acrxentrypoint.cpp

vs2003+arxobject2006相信不用说

// (C) Copyright 2002-2005 by Autodesk, Inc. 
//
// Permission to use, copy, modify, and distribute this software in
// object code form for any purpose and without fee is hereby granted, 
// provided that the above copyright notice appears in all copies and 
// that both that copyright notice and the limited warranty and
// restricted rights notice below appear in all supporting 
// documentation.
//
// AUTODESK PROVIDES THIS PROGRAM "AS IS" AND WITH ALL FAULTS. 
// AUTODESK SPECIFICALLY DISCLAIMS ANY IMPLIED WARRANTY OF
// MERCHANTABILITY OR FITNESS FOR A PARTICULAR USE.  AUTODESK, INC. 
// DOES NOT WARRANT THAT THE OPERATION OF THE PROGRAM WILL BE
// UNINTERRUPTED OR ERROR FREE.
//
// Use, duplication, or disclosure by the U.S. Government is subject to 
// restrictions set forth in FAR 52.227-19 (Commercial Computer
// Software - Restricted Rights) and DFAR 252.227-7013(c)(1)(ii)
// (Rights in Technical Data and Computer Software), as applicable.
//

//-----------------------------------------------------------------------------
//----- acrxEntryPoint.h
//-----------------------------------------------------------------------------
#include "StdAfx.h"
#include "resource.h"
#include "utilities.h"
#include "EmployeeDetails\AsdkEmployeeDetails.h" 

//-----------------------------------------------------------------------------
#define szRDS _RXST("Asdk")

//-----------------------------------------------------------------------------
//----- ObjectARX EntryPoint
class CStep05App : public AcRxArxApp {

public:
	CStep05App () : AcRxArxApp () {}

	virtual AcRx::AppRetCode On_kInitAppMsg (void *pkt) {
		// TODO: Load dependencies here

		// You *must* call On_kInitAppMsg here
		AcRx::AppRetCode retCode =AcRxArxApp::On_kInitAppMsg (pkt) ;
		
		// TODO: Add your initialization code here

		return (retCode) ;
	}

	virtual AcRx::AppRetCode On_kUnloadAppMsg (void *pkt) {
		// TODO: Add your code here

		// You *must* call On_kUnloadAppMsg here
		AcRx::AppRetCode retCode =AcRxArxApp::On_kUnloadAppMsg (pkt) ;

		// TODO: Unload dependencies here

		return (retCode) ;
	}

	virtual void RegisterServerComponents () {
	}


	// - AsdkStep05._Create command (do not rename)
	static void AsdkStep05_Create(void)
	{
		// Add your code for command AsdkStep05._Create here
		// TODO: Implement the command
		
		// Create a new layer named "USER"
		// createLayer returns the object ID of the newly created layer
		AcDbObjectId layerId ;
		if ( createLayer ("USER", layerId) != Acad::eOk ) {
			acutPrintf ("\nERROR: Couldn't create layer record.") ;
			return ;
		}
		// This is not always needed, but a call to 'applyCurDwgLayerTableChanges()'
		// will synchronize the newly created layer table change with the 
		// rest of the current DWG database.
		applyCurDwgLayerTableChanges () ;

		acutPrintf ("\nLayer USER successfully created.") ;

		// Create a new block definition named "EMPLOYEE"
		if ( createBlockRecord ("EMPLOYEE") != Acad::eOk )
			acutPrintf ("\nERROR: Couldn't create block record.") ;
		else
			acutPrintf ("\nBlock EMPLOYEE successfully created.") ;
	}

	// - AsdkStep05._SetLayer command (do not rename)
	static void AsdkStep05_SetLayer(void)
	{
		// Add your code for command AsdkStep05._SetLayer here
		// Iterate through Model Space to find every instance of the EMPLOYEE block
		// When found, change its layer to "USER"
		Acad::ErrorStatus es ;
		AcDbBlockTable *pBlockTbl ;
		AcDbBlockTableRecord *pMS ;

		// Get the block table
		if ( (es =acdbHostApplicationServices ()->workingDatabase ()->getBlockTable (pBlockTbl, AcDb::kForRead)) != Acad::eOk ) {
			acutPrintf ("\nCouldn't open the block table!") ;
			return ;
		}
		// Get the Model Space record and open it for read.
		if ( (es =pBlockTbl->getAt (ACDB_MODEL_SPACE, pMS, AcDb::kForWrite)) != Acad::eOk ) {
			acutPrintf ("\nCouldn't get Model Space! Drawing corrupt.\n") ;
			pBlockTbl->close () ;
			return ;
		}
		pBlockTbl->close () ;

		// Declare the appropriate iterator type
		// Get the iterator from the object to be iterated through

		// In this case, the Model Space block table record will provide the iterator
		// start at the beginning of the record and skip deleted entities
		AcDbBlockTableRecordIterator *pBtrIter ;
		if ( (es =pMS->newIterator (pBtrIter) ) != Acad::eOk ) {
			acutPrintf ("\nCouldn't create Model Space iterator.") ;
			pMS->close () ;
			return ;
		}

		char *blockName ;
		AcDbEntity *pEnt ;
		AcDbBlockTableRecord *pCurEntBlock ;
		AcDbObjectId blockId ;

		for ( pBtrIter->start (); !pBtrIter->done () ; pBtrIter->step () ) {
			// First open each entity for read, just to check its class
			// if it's what we want, we can upgrade open later
			// Don't bother with erased entities
			if ( (es =pBtrIter->getEntity (pEnt, AcDb::kForRead)) != Acad::eOk ) {
				acutPrintf ("\nCouldn't open entity.");
				continue ;
			}
			if ( pEnt->isA() != AcDbBlockReference::desc () ) {
				pEnt->close () ;
				continue ;
			}
			// Get the insert's block table record and compare its name
			// to make sure we've got the right block.  If so, set the layer
			blockId =(AcDbBlockReference::cast (pEnt))->blockTableRecord () ;
			if ( acdbOpenObject ((AcDbObject *&)pCurEntBlock, blockId, AcDb::kForRead) == Acad::eOk ) {
				pCurEntBlock->getName(blockName); 
				if ( strcmp (blockName, "EMPLOYEE") == 0 ) {
					if ( pEnt->upgradeOpen () == Acad::eOk )
						// setLayer also has an overload that takes a layer ID
						// but to avoid global variables we specify the layer name
						pEnt->setLayer ("USER") ;
				}
				pCurEntBlock->close () ;
				acdbFree (blockName) ;
			}
			pEnt->close () ;
		}

		// delete, rather than close, the iterator object
		delete pBtrIter ;
		pMS->close () ;
	}

	// - AsdkStep05._AddDetail command (do not rename)
	static void AsdkStep05_AddDetail(void)
	{
		// Add your code for command AsdkStep05._AddDetail here
		// Prompt the user for the employee details
		ads_name ename ;
		ads_point pt ;
		// Get the data from the user
		if ( acedEntSel ("Select employee: ", ename, pt) != RTNORM )
			return ;
		// Do a quick check
		// a more comprehensive check could include 
		// whether we already have the detail object on this candidate
		AcDbObjectId idO ;
		if ( acdbGetObjectId (idO, ename) != Acad::eOk )
			return ;
		AcDbObject *pO ;
		if ( acdbOpenAcDbObject (pO, idO, AcDb::kForWrite) != Acad::eOk )
			return ;
		if ( !pO->isKindOf (AcDbBlockReference::desc ()) ) {
			acutPrintf ("\nThis is not a block reference.") ;
			pO->close () ;
			return ;
		}
		// Get user input
		int id, cubeNumber ;
		char strFirstName [133] ;
		char strLastName [133] ;
		if (   acedGetInt ("Enter employee ID: ", &id) != RTNORM
			|| acedGetInt ("Enter cube number: ", &cubeNumber) != RTNORM
			|| acedGetString (0, "Enter employee first name: ", strFirstName) != RTNORM
			|| acedGetString (0, "Enter employee last name: ", strLastName) != RTNORM
		) {
			pO->close () ;
			return ;
		}
		// Get the extension dictionary
		if ( (idO =pO->extensionDictionary ()) == AcDbObjectId::kNull ) {
			if ( pO->createExtensionDictionary () != Acad::eOk ) {
				pO->close () ;
				acutPrintf ("\nFailed to create ext. dictionary.") ;
				return ;
			}
			idO =pO->extensionDictionary () ;
		}
		// We do not need the block reference object anymore.
		pO->close () ;
		// Make sure you open erased extension dictionaries
		// you may need to unerase them
		AcDbDictionary *pExtDict ;
		if ( acdbOpenAcDbObject ((AcDbObject *&)pExtDict, idO, AcDb::kForWrite, Adesk::kTrue) != Acad::eOk ) {
			acutPrintf ("\nFailed to open ext. dictionary.") ;
			return ;
		}
		// Unerase the ext. dictionary if it was erased
		if ( pExtDict->isErased () )
			pExtDict->erase (Adesk::kFalse) ;
		// See if our dictionary is already there
		AcDbDictionary *pEmployeeDict ;
		if ( pExtDict->getAt ("ASDK_EMPLOYEE_DICTIONARY", idO) == Acad::eKeyNotFound ) {
			// Create it if not
			pEmployeeDict =new AcDbDictionary ;
			Acad::ErrorStatus es ;
			if ( (es =pExtDict->setAt ("ASDK_EMPLOYEE_DICTIONARY", pEmployeeDict, idO)) != Acad::eOk ) {
				pExtDict->close () ;
				acutPrintf ("\nFailed to create the 'Employee' dictionary.") ;
				return ;
			}
		} else {
			// Open our dictionary for write if it is already there
			if ( acdbOpenAcDbObject (pO, idO, AcDb::kForWrite) != Acad::eOk ) {
				pExtDict->close () ;