// (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"

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

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

public:
	CStep03App () : 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 () {
	}


	// - AsdkStep03._myCreate command (do not rename)
	static void AsdkStep03_myCreate(void)
	{
		// Add your code for command AsdkStep03._myCreate 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.") ;
	}

	// - AsdkStep03._mySetLayer command (do not rename)
	static void AsdkStep03_mySetLayer(void)
	{
		// Add your code for command AsdkStep03._mySetLayer here
		// TODO: Implement the command
		
		// 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 () ;
	}
} ;

//-----------------------------------------------------------------------------
IMPLEMENT_ARX_ENTRYPOINT(CStep03App)

ACED_ARXCOMMAND_ENTRY_AUTO(CStep03App, AsdkStep03, _myCreate, myCreate, ACRX_CMD_TRANSPARENT, NULL)
ACED_ARXCOMMAND_ENTRY_AUTO(CStep03App, AsdkStep03, _mySetLayer, mySetLayer, ACRX_CMD_TRANSPARENT, NULL)