using System ;
using Autodesk.AutoCAD.Runtime ;
using Autodesk.AutoCAD.ApplicationServices;
using Autodesk.AutoCAD.DatabaseServices;
using Autodesk.AutoCAD.Geometry;
using Autodesk.AutoCAD.Colors;
using Autodesk.AutoCAD.EditorInput;
[assembly: CommandClass( typeof (ClassLibrary.Class))]
namespace ClassLibrary
{
/// <summary>
/// Summary description for Class.
/// </summary>
public class Class
{
public Class()
{
//
// TODO: Add constructor logic here
//
}
// This function returns the ObjectId for the BlockTableRecord called "EmployeeBlock",
// creating it if necessary. The block contains three entities - circle, text
// and ellipse.
public ObjectId CreateEmployeeDefinition()
{
ObjectId newBtrId = new ObjectId(); // The return value for this function
Database db = HostApplicationServices.WorkingDatabase; // save some space
Transaction trans = db.TransactionManager.StartTransaction(); // begin the transaction
Editor ed = Autodesk.AutoCAD.ApplicationServices.Application.DocumentManager.MdiActiveDocument.Editor;
try
{
// Now, drill into the database and obtain a reference to the BlockTable
BlockTable bt = (BlockTable)trans.GetObject(db.BlockTableId, OpenMode.ForWrite);
if ((bt.Has( " EmployeeBlock " )))
{
newBtrId = bt[ " EmployeeBlock " ];
}
else
{
Point3d center = new Point3d( 10 , 10 , 0 ); // convenient declaration
// Declare and define the entities we want to add:
// Circle:
Circle circle = new Circle(center, Vector3d.ZAxis, 2 );
// Text:
// MText text = new MText();
// text.Contents = "Earnest Shackleton";
// text.Location = center;
// Attribute Definition
AttributeDefinition text = new AttributeDefinition(center, " NoName " , " Name: " , " Enter Name " , db.Textstyle);
text.ColorIndex = 2 ;
// Ellipse:
Ellipse ellipse = new Ellipse(center, Vector3d.ZAxis, new Vector3d( 3 , 0 , 0 ), 0.5 , 0 , 0 );
// Next, create a layer with the helper function, and assign
// the layer to our entities.
ObjectId empId = CreateLayer();
text.LayerId = empId;
circle.LayerId = empId;
ellipse.LayerId = empId;
// Set the color for each entity irrespective of the layer's color.
text.ColorIndex = 2 ;
circle.ColorIndex = 1 ;
ellipse.ColorIndex = 3 ;
// Create a new block definition called EmployeeBlock
BlockTableRecord newBtr = new BlockTableRecord();
newBtr.Name = " EmployeeBlock " ;
newBtrId = bt.Add(newBtr); // Add the block, and set the id as the return value of our function
trans.AddNewlyCreatedDBObject(newBtr, true ); // Let the transaction know about any object/entity you add to the database!
newBtr.AppendEntity(circle); // Append our entities
newBtr.AppendEntity(text);
newBtr.AppendEntity(ellipse);
trans.AddNewlyCreatedDBObject(circle, true ); // Again, let the transaction know about our newly added entities.
trans.AddNewlyCreatedDBObject(text, true );
trans.AddNewlyCreatedDBObject(ellipse, true );
}
trans.Commit(); // All done, no errors? Go ahead and commit!
}
catch
{
ed.WriteMessage( " Error Creating Employee Block " );
}
finally
{
trans.Dispose();
}
// CreateDivision(); // Create the Employee Division dictionaries.
return newBtrId;
}
// This function creates a new BlockReference to the "EmployeeBlock" object,
// and adds it to ModelSpace.
public ObjectId CreateEmployee( string name, string division, double salary, Point3d pos)
{
Database db = HostApplicationServices.WorkingDatabase;
Transaction trans = db.TransactionManager.StartTransaction();
try
{
BlockTable bt = (BlockTable)(trans.GetObject(db.BlockTableId, OpenMode.ForWrite));
BlockTableRecord btr = (BlockTableRecord)trans.GetObject(bt[BlockTableRecord.ModelSpace], OpenMode.ForWrite);
// Create the block referenceuse the return from CreateEmployeeDefinition directly!
BlockReference br = new BlockReference(pos, CreateEmployeeDefinition());
// btr.AppendEntity(br); // Add the reference to ModelSpace
// trans.AddNewlyCreatedDBObject(br, true); // Let the transaction know about it
AttributeReference attRef = new AttributeReference();
// Iterate the employee block and find the attribute definition
BlockTableRecord empBtr = (BlockTableRecord)trans.GetObject(bt[ " EmployeeBlock " ], OpenMode.ForRead);
foreach (ObjectId id in empBtr)
{
Entity ent = (Entity)trans.GetObject(id, OpenMode.ForRead, false );
// Use it to open the current object!
if (ent is AttributeDefinition) // We use .NET's RTTI to establish type.
{
// Set the properties from the attribute definition on our attribute reference
AttributeDefinition attDef = ((AttributeDefinition)(ent));
attRef.SetPropertiesFrom(attDef);
attRef.Position = new Point3d(attDef.Position.X + br.Position.X, attDef.Position.Y + br.Position.Y, attDef.Position.Z + br.Position.Z);
attRef.Height = attDef.Height;
attRef.Rotation = attDef.Rotation;
attRef.Tag = attDef.Tag;
attRef.TextString = name;
}
}
// Add the reference to ModelSpace
btr.AppendEntity(br);
// Add the attribute reference to the block reference
br.AttributeCollection.AppendAttribute(attRef);
// let the transaction know
trans.AddNewlyCreatedDBObject(attRef, true );
trans.AddNewlyCreatedDBObject(br, true );
// Create the custom per-employee data
Xrecord xRec = new Xrecord();
// We want to add 'Name', 'Salary' and 'Division' information. Here is how:
xRec.Data = new ResultBuffer(
new TypedValue(( int )DxfCode.Text, name),
new TypedValue(( int )DxfCode.Real, salary),
new TypedValue(( int )DxfCode.Text, division));
// Next, we need to add this data to the 'Extension Dictionary' of the employee.
br.CreateExtensionDictionary();
DBDictionary brExtDict = (DBDictionary)trans.GetObject(br.ExtensionDictionary, OpenMode.ForWrite, false );
brExtDict.SetAt( " EmployeeData " , xRec); // Set our XRecord in the dictionary at 'EmployeeData'.
trans.AddNewlyCreatedDBObject(xRec, true );
trans.Commit();
return br.ObjectId;
}
finally
{
trans.Dispose();
}
}
// This function returns the objectId for the "EmployeeLayer", creating it if necessary.
public ObjectId CreateLayer()
{
ObjectId layerId;
Database db = HostApplicationServices.WorkingDatabase;
Transaction trans = db.TransactionManager.StartTransaction();
// Get the layer table first
LayerTable lt = (LayerTable)trans.GetObject(db.LayerTableId, OpenMode.ForWrite);
// Check if EmployeeLayer exists
if (lt.Has( " EmployeeLayer " ))
{
layerId = lt[ " EmployeeLayer " ];
}
else
{
// If not, create the layer here.
LayerTableRecord ltr = new LayerTableRecord();
ltr.Name = " EmployeeLayer " ; // Set the layer name
ltr.Color = Color.FromColorIndex(ColorMethod.ByAci, 2 );
layerId = lt.Add(ltr);
trans.AddNewlyCreatedDBObject(ltr, true );
}
trans.Commit();
trans.Dispose();
return layerId;
}
public ObjectId CreateDivision( string division, string manager)
{
Database db = HostApplicationServices.WorkingDatabase;
Transaction trans = db.TransactionManager.StartTransaction();
try
{
// First, get the NOD
DBDictionary NOD = (DBDictionary)trans.GetObject(db.NamedObjectsDictionaryId, OpenMode.ForWrite);
// Define a corporate level dictionary
DBDictionary acmeDict;
try
{
// Just throw if it doesn't existdo nothing else
acmeDict = (DBDictionary)trans.GetObject(NOD.GetAt( " ACME_DIVISION " ), OpenMode.ForRead);
}
catch
{
// Doesn't exist, so create one, and set it in the NOD?
acmeDict = new DBDictionary();
NOD.SetAt( " ACME_DIVISION " , acmeDict);
trans.AddNewlyCreatedDBObject(acmeDict, true );
}
// Now get the division we want from acmeDict
DBDictionary divDict;
try
{
divDict = (DBDictionary)trans.GetObject(acmeDict.GetAt(division), OpenMode.ForWrite);
}
catch
{
divDict = new DBDictionary();
// Division doesn't exist, create one
acmeDict.UpgradeOpen();
acmeDict.SetAt(division, divDict);
trans.AddNewlyCreatedDBObject(divDict, true );
}
// Now get the manager info from the division
// We need to add the name of the division supervisor. We'll do this with another XRecord.
Xrecord mgrXRec;
try
{
mgrXRec = (Xrecord)trans.GetObject(divDict.GetAt( " Department Manager " ), OpenMode.ForWrite);
}
catch
{
mgrXRec = new Xrecord();
mgrXRec.Data = new ResultBuffer( new TypedValue(( int )DxfCode.Text, manager));
divDict.SetAt( " Department Manager " , mgrXRec);
trans.AddNewlyCreatedDBObject(mgrXRec, true );
}
trans.Commit();
// Return the department manager XRecord
return mgrXRec.ObjectId;
}
finally
{
trans.Dispose();
}
}
[CommandMethod( " EMPLOYEECOUNT " )]
public void EmployeeCount()
{
Database db = HostApplicationServices.WorkingDatabase;
Transaction trans = db.TransactionManager.StartTransaction(); // Start the transaction.
int nEmployeeCount = 0 ;
try
{
// First, get at the BlockTable, and the ModelSpace BlockTableRecord
BlockTable bt = (BlockTable)trans.GetObject(db.BlockTableId, OpenMode.ForRead);
BlockTableRecord btr = (BlockTableRecord)trans.GetObject(bt[BlockTableRecord.ModelSpace], OpenMode.ForRead);
// Now, we need to be able to print to the commandline. Here is an object which will help us:
Editor ed = Application.DocumentManager.MdiActiveDocument.Editor;
// Now, here is the fun part. This is where we iterate through ModelSpace:
foreach (ObjectId id in btr)
{
Entity ent = (Entity)trans.GetObject(id, OpenMode.ForRead, false ); // Use it to open the current object!
if (ent.GetType() == typeof (BlockReference)) // We use .NET's RTTI to establish type.
{
nEmployeeCount += 1 ;
}
}
ed.WriteMessage( " Employees Found: " + nEmployeeCount.ToString());
trans.Commit();
}
finally
{
trans.Dispose();
}
}
// We want a command which will go through and list all the relevant employee data.
public static void ListEmployee(ObjectId employeeId, ref string [] saEmployeeList)
{
int nEmployeeDataCount = 0 ;
Database db = HostApplicationServices.WorkingDatabase;
Transaction trans = db.TransactionManager.StartTransaction(); // Start the transaction
try
{
Entity ent = (Entity)trans.GetObject(employeeId, OpenMode.ForRead, false ); // Use it to open the current object!
if (ent.GetType() == typeof (BlockReference)) // We use .NET's RTTI to establish type.
{
// Not all BlockReferences will have our employee data, so we must make sure we can handle failure
bool bHasOurDict = true ;
Xrecord EmployeeXRec = null ;
try
{
BlockReference br = (BlockReference)ent;
DBDictionary extDict = (DBDictionary)trans.GetObject(br.ExtensionDictionary, OpenMode.ForRead, false );
EmployeeXRec = (Xrecord)trans.GetObject(extDict.GetAt( " EmployeeData " ), OpenMode.ForRead, false );
}
catch
{
bHasOurDict = false ; // Something bad happenedour dictionary and/or XRecord is not accessible
}
if (bHasOurDict) // If obtaining the Extension Dictionary, and our XRecord is successful
{
// allocate memory for the list
saEmployeeList = new String[ 4 ];
TypedValue resBuf = EmployeeXRec.Data.AsArray()[ 0 ];
saEmployeeList.SetValue( string .Format( " {0}
" , resBuf.Value), nEmployeeDataCount);
nEmployeeDataCount += 1 ;
resBuf = EmployeeXRec.Data.AsArray()[ 1 ];
saEmployeeList.SetValue( string .Format( " {0}
" , resBuf.Value), nEmployeeDataCount);
nEmployeeDataCount += 1 ;
resBuf = EmployeeXRec.Data.AsArray()[ 2 ];
string str = ( string )resBuf.Value;
saEmployeeList.SetValue( string .Format( " {0}
" , resBuf.Value), nEmployeeDataCount);
nEmployeeDataCount += 1 ;
DBDictionary NOD = (DBDictionary)trans.GetObject(db.NamedObjectsDictionaryId, OpenMode.ForRead, false );
DBDictionary acmeDict = (DBDictionary)trans.GetObject(NOD.GetAt( " ACME_DIVISION " ), OpenMode.ForRead);
DBDictionary salesDict = (DBDictionary)trans.GetObject(acmeDict.GetAt(( string )EmployeeXRec.Data.AsArray()[ 2 ].Value), OpenMode.ForRead);
Xrecord salesXRec = (Xrecord)trans.GetObject(salesDict.GetAt( " Department Manager " ), OpenMode.ForRead);
resBuf = salesXRec.Data.AsArray()[ 0 ];
saEmployeeList.SetValue( string .Format( " {0}
" , resBuf.Value), nEmployeeDataCount);
nEmployeeDataCount += 1 ;
}
}
trans.Commit();
}
finally
{
trans.Dispose();
}
}
[CommandMethod( " PRINTOUTEMPLOYEE " )]
public static void PrintoutEmployee()
{
Editor ed = Application.DocumentManager.MdiActiveDocument.Editor;
Database db = HostApplicationServices.WorkingDatabase;
Transaction trans = db.TransactionManager.StartTransaction();
try
{
BlockTable bt = (BlockTable)trans.GetObject(HostApplicationServices.WorkingDatabase.BlockTableId, OpenMode.ForRead);
BlockTableRecord btr = (BlockTableRecord)trans.GetObject(bt[BlockTableRecord.ModelSpace], OpenMode.ForRead);
foreach (ObjectId id in btr)
{
Entity ent = (Entity)trans.GetObject(id, OpenMode.ForRead, false );
if (ent is BlockReference)
{
string [] saEmployeeList = null ;
ListEmployee(id, ref saEmployeeList);
if ((saEmployeeList.Length == 4 ))
{
ed.WriteMessage( " Employee Name: {0} " , saEmployeeList[ 0 ]);
ed.WriteMessage( " Employee Salary: {0} " , saEmployeeList[ 1 ]);
ed.WriteMessage( " Employee Division: {0} " , saEmployeeList[ 2 ]);
ed.WriteMessage( " Division Manager: {0} " , saEmployeeList[ 3 ]);
}
}
}
}
finally
{
}
}
[CommandMethod( " CREATE " )]
public void CreateEmployee()
{
Database db = HostApplicationServices.WorkingDatabase;
Editor ed = Application.DocumentManager.MdiActiveDocument.Editor;
Transaction trans = db.TransactionManager.StartTransaction();
try
{
string empName = " Earnest Shackleton " ;
string divName = " Sales " ;
double salary = new double ();
salary = 10000 ;
Point3d position = new Point3d( 0 , 0 , 0 );
bool gotPosition = new bool ();
// boolean to check if a point has been entered
gotPosition = false ;
// Prompts for each employee detail
PromptStringOptions prName = new PromptStringOptions( " Enter Employee Name < " + empName + " > " );
PromptStringOptions prDiv = new PromptStringOptions( " Enter Employee Division < " + divName + " > " );
PromptDoubleOptions prSal = new PromptDoubleOptions( " Enter Employee Salary < " + salary + " > " );
PromptPointOptions prPos = new PromptPointOptions( " Enter Employee Position or " );
// Add keywords when prompting for position
prPos.Keywords.Add( " Name " );
prPos.Keywords.Add( " Division " );
prPos.Keywords.Add( " Salary " );
// Set conditions for prompting
prPos.AllowNone = false ; // Do not allow null values
// prompt results
PromptResult prNameRes;
PromptResult prDivRes;
PromptDoubleResult prSalRes;
PromptPointResult prPosRes;
// Loop to get employee details. Exit the loop when positon is entered
while ( ! gotPosition)
{
// Prompt for position
prPosRes = ed.GetPoint(prPos);
// Got a point
if (prPosRes.Status == PromptStatus.OK)
{
gotPosition = true ;
position = prPosRes.Value;
}
else if (prPosRes.Status == PromptStatus.Keyword) // Got a keyword
{
// Name keyword entered
if (prPosRes.StringResult == " Name " )
{
// Get employee name
prName.AllowSpaces = true ;
prNameRes = ed.GetString(prName);
if (prNameRes.Status != PromptStatus.OK)
{
return ;
}
// we got the employee name successfully
if (prNameRes.StringResult != "" )
{
empName = prNameRes.StringResult;
}
}
// Division keyword entered
if (prPosRes.StringResult == " Division " )
{
// Get employee division
prDiv.AllowSpaces = true ;
prDivRes = ed.GetString(prDiv);
if (prDivRes.Status != PromptStatus.OK)
{
return ;
}
if (prDivRes.StringResult != "" )
{
divName = prDivRes.StringResult;
}
} // Division
// Salary keyword entered
if (prPosRes.StringResult == " Salary " )
{
// Get employee salary
prSal.AllowNegative = false ;
prSal.AllowNone = true ;
prSal.AllowZero = false ;
prSalRes = ed.GetDouble(prSal);
if (prSalRes.Status != PromptStatus.OK & prSalRes.Status != PromptStatus.None)
{
return ;
}
if (prSalRes.Status != PromptStatus.None)
{
salary = prSalRes.Value;
}
} // Salary
}
else
{
// Error in getting a point
ed.WriteMessage( " ***Error in getting a point, exiting!!*** " + "
" );
return ;
} // If got a point
}
// Create the Employee
CreateEmployee(empName, divName, salary, position);
string manager = "" ;
// Now create the division
// Pass an empty string for manager to check if it already exists
Xrecord depMgrXRec;
ObjectId xRecId;
xRecId = CreateDivision(divName, manager);
// Open the department manager XRecord
depMgrXRec = (Xrecord)trans.GetObject(xRecId, OpenMode.ForRead);
TypedValue[] typedVal = depMgrXRec.Data.AsArray();
foreach (TypedValue val in typedVal)
{
string str;
str = ( string )val.Value;
if (str == "" )
{
// Manager was not set, now set it
// Prompt for manager name first
ed.WriteMessage( "
" );
PromptStringOptions prManagerName = new PromptStringOptions( " No manager set for the division! Enter Manager Name " );
prManagerName.AllowSpaces = true ;
PromptResult prManagerNameRes = ed.GetString(prManagerName);
if (prManagerNameRes.Status != PromptStatus.OK)
{
return ;
}
// Set a manager name
depMgrXRec.Data = new ResultBuffer( new TypedValue(( int )DxfCode.Text, prManagerNameRes.StringResult));
}
}
trans.Commit();
}
finally
{
trans.Dispose();
}
}
[CommandMethod( " LISTEMPLOYEES " )]
public void List()
{
Editor ed = Application.DocumentManager.MdiActiveDocument.Editor;
try
{
PromptSelectionOptions Opts = new PromptSelectionOptions();
TypedValue[] filList = new TypedValue[ 1 ];
// Build a filter list so that only block references are selected
filList[ 0 ] = new TypedValue(( int )DxfCode.Start, " INSERT " );
SelectionFilter filter = new SelectionFilter(filList);
PromptSelectionResult res = ed.GetSelection(Opts, filter);
// Do nothing if selection is unsuccessful
if (res.Status != PromptStatus.OK)
return ;
Autodesk.AutoCAD.EditorInput.SelectionSet SS = res.Value;
ObjectId[] idArray;
idArray = SS.GetObjectIds();
string [] saEmployeeList = new string [ 4 ];
// collect all employee details in saEmployeeList array
foreach (ObjectId employeeId in idArray)
{
ListEmployee(employeeId, ref saEmployeeList);
// Print employee details to the command line
foreach ( string employeeDetail in saEmployeeList)
{
ed.WriteMessage(employeeDetail);
}
// separator
ed.WriteMessage( " ---------------------- " + "
" );
}
}
finally
{
}
}
[CommandMethod( " Test " )]
public void Test()
{
CreateDivision( " Sales " , " Randolph P. Brokwell " );
}
}
}