Gabriele Cosmo
165 lines
3.9 KiB
C++
165 lines
3.9 KiB
C++
#include "G4AssemblyCreator.hh"
|
|
#include "G4Assembly.hh"
|
|
extern void HeaderSchemaInit (Registry & reg);
|
|
|
|
G4AssemblyCreator G4AssemblyCreator::ci;
|
|
|
|
G4AssemblyCreator::G4AssemblyCreator()
|
|
{
|
|
G4GeometryTable::RegisterObject(this);
|
|
}
|
|
|
|
G4AssemblyCreator::G4AssemblyCreator(G4String fileName, G4String readerName)
|
|
{
|
|
// Name of STEP file to read in, existance of file should be checked
|
|
//here also...
|
|
STEPfileName = fileName;
|
|
|
|
// define type of STEP file reader to use
|
|
if(readerName == "NIST")
|
|
StepReader = new G4NISTStepReader();
|
|
//else...
|
|
|
|
}
|
|
|
|
G4AssemblyCreator::~G4AssemblyCreator()
|
|
{
|
|
delete StepReader;
|
|
}
|
|
|
|
|
|
void G4AssemblyCreator::ReadStepFile()
|
|
{
|
|
StepReader->ReadSTEPFile(STEPfileName);
|
|
instanceManager = StepReader->GetInstanceManager();
|
|
}
|
|
|
|
void G4AssemblyCreator::CreateG4Geometry(STEPentity& Ent)
|
|
{
|
|
// Advanced_Brep_Shape_Representation are created into
|
|
// Context_Dependent_Shape_Representation and
|
|
// Shape_Definition_Representation
|
|
G4int AdvancedBrepShapes = instanceManager.EntityKeywordCount
|
|
("Advanced_Brep_Shape_Representation");
|
|
|
|
|
|
G4int ConDepShapes = instanceManager.EntityKeywordCount
|
|
("Context_Dependent_Shape_Representation");
|
|
|
|
G4int ShapeDefReps = instanceManager.EntityKeywordCount
|
|
("Shape_Definition_Representation");
|
|
|
|
G4int instanceCount = instanceManager.InstanceCount();
|
|
|
|
STEPentity* ent=0;
|
|
index = 0;
|
|
int tmpindex;
|
|
|
|
void *tmp = 0;
|
|
G4PlacedSolidVector* psv = new G4PlacedSolidVector();
|
|
G4PlacedSolid* ps = new G4PlacedSolid();
|
|
G4int a;
|
|
|
|
// L. Broglia
|
|
// Total assumption !
|
|
|
|
if(ConDepShapes>0)
|
|
{
|
|
G4cout<<"\n\n Creating the Context_Dependent_Shape_Representation"<<endl;
|
|
index = 0;
|
|
|
|
for( a=0; a< ConDepShapes; a++)
|
|
{
|
|
G4cout<<"loop "<<a+1<<" of "<<ConDepShapes<<endl;
|
|
|
|
// Be careful, tmpindex not correspond to STEPfile_id !
|
|
tmpindex =
|
|
instanceManager.GetIndex("Context_Dependent_Shape_Representation",
|
|
index );
|
|
|
|
ent = instanceManager.GetSTEPentity(tmpindex);
|
|
|
|
if(ent!= ENTITY_NULL)
|
|
{
|
|
tmp =G4GeometryTable::CreateObject(*ent);
|
|
|
|
G4PlacedSolidVector* tmpV = (G4PlacedSolidVector*)tmp;
|
|
G4int entr = tmpV->entries();
|
|
|
|
for(G4int b=0; b<entr; b++)
|
|
{
|
|
ps = tmpV->at(b);
|
|
psv->append(ps);
|
|
}
|
|
|
|
index = ent->STEPfile_id ;
|
|
G4cout<<" Context_Dependent_Shape_Representation find in index "
|
|
<<index<<endl;
|
|
}
|
|
|
|
// Set index to the true value
|
|
index = tmpindex + 1;
|
|
ent=0;
|
|
}
|
|
|
|
}
|
|
else
|
|
{
|
|
G4cout<<"\n Creating the Shape_Definition_Representation"<<endl;
|
|
for(a=0; a< ShapeDefReps ; a++)
|
|
{
|
|
G4cout<<"loop "<<a+1<<" of "<<ShapeDefReps<<endl;
|
|
|
|
// Be careful, tmpindex not correspond to STEPfile_id !
|
|
tmpindex = instanceManager.GetIndex("Shape_Definition_Representation",
|
|
index );
|
|
|
|
ent = instanceManager.GetSTEPentity(tmpindex);
|
|
|
|
if(ent!= ENTITY_NULL)
|
|
{
|
|
tmp = G4GeometryTable::CreateObject(*ent);
|
|
|
|
G4PlacedSolidVector* tmpV = (G4PlacedSolidVector*)tmp;
|
|
G4int entr = tmpV->entries();
|
|
|
|
for(G4int b=0; b<entr; b++)
|
|
{
|
|
ps = tmpV->at(b);
|
|
psv->append(ps);
|
|
}
|
|
|
|
index = ent->STEPfile_id ;
|
|
G4cout<<" Shape_Definition_Representation find in index "<<index<<endl;
|
|
}
|
|
|
|
// Set index to the true value
|
|
index = tmpindex + 1;
|
|
ent=0;
|
|
}
|
|
}
|
|
|
|
createdObject = psv;
|
|
}
|
|
|
|
|
|
void G4AssemblyCreator::CreateSTEPGeometry(void* G4obj)
|
|
{
|
|
Registry reg(&::HeaderSchemaInit);
|
|
SdaiCONFIG_CONTROL_DESIGNInit (reg);
|
|
|
|
//G4Placement* plc = (G4Placement*)G4obj;
|
|
//G4String name("Axis2Placement3d");
|
|
// G4FCylindricalSurface *fCyl = (G4FCylindricalSurface *)G4obj;
|
|
// G4String name("Cylindrical_Surface");
|
|
G4ToroidalSurface *tor = (G4ToroidalSurface *)G4obj;
|
|
G4String name("Toroidal_Surface");
|
|
{
|
|
void *tmp =G4GeometryTable::CreateSTEPObject(G4obj, name);
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|