|
|
|
#include <Fireworks/Geometry/interface/TGeoFromDddService.h>
Public Member Functions | |
| TGeoManager * | getGeoManager () |
| void | postBeginRun (const edm::Run &, const edm::EventSetup &) |
| void | postEndRun (const edm::Run &, const edm::EventSetup &) |
| TGeoFromDddService (const edm::ParameterSet &, edm::ActivityRegistry &) | |
| virtual | ~TGeoFromDddService () |
Private Member Functions | |
| TGeoManager * | createManager (int level) |
| TGeoMaterial * | createMaterial (const DDMaterial &iMaterial) |
| TGeoShape * | createShape (const std::string &iName, const DDSolid &iSolid) |
| TGeoVolume * | createVolume (const std::string &iName, const DDSolid &iSolid, const DDMaterial &iMaterial) |
| const TGeoFromDddService & | operator= (const TGeoFromDddService &) |
| TGeoFromDddService (const TGeoFromDddService &) | |
Private Attributes | |
| const edm::EventSetup * | m_eventSetup |
| TGeoManager * | m_geoManager |
| int | m_level |
| bool | m_verbose |
| std::map< std::string, TGeoMaterial * > | nameToMaterial_ |
| std::map< std::string, TGeoMedium * > | nameToMedium_ |
| std::map< std::string, TGeoShape * > | nameToShape_ |
| std::map< std::string, TGeoVolume * > | nameToVolume_ |
Description: [one line class summary]
Usage: <usage>
Definition at line 49 of file TGeoFromDddService.h.
| TGeoFromDddService::TGeoFromDddService | ( | const edm::ParameterSet & | pset, |
| edm::ActivityRegistry & | ar | ||
| ) |
Definition at line 58 of file TGeoFromDddService.cc.
References postBeginRun(), postEndRun(), edm::ActivityRegistry::watchPostBeginRun(), and edm::ActivityRegistry::watchPostEndRun().
: m_level (pset.getUntrackedParameter<int> ("level", 10)), m_verbose (pset.getUntrackedParameter<bool>("verbose",false)), m_eventSetup (0), m_geoManager (0) { ar.watchPostBeginRun(this, &TGeoFromDddService::postBeginRun); ar.watchPostEndRun (this, &TGeoFromDddService::postEndRun); }
| TGeoFromDddService::~TGeoFromDddService | ( | ) | [virtual] |
Definition at line 68 of file TGeoFromDddService.cc.
References m_geoManager.
{
if (m_geoManager)
{
delete m_geoManager;
}
}
| TGeoFromDddService::TGeoFromDddService | ( | const TGeoFromDddService & | ) | [private] |
| TGeoManager * TGeoFromDddService::createManager | ( | int | level | ) | [private] |
Definition at line 151 of file TGeoFromDddService.cc.
References gather_cfg::cout, createVolume(), edm::EventSetup::get(), i, info, m_eventSetup, m_verbose, mergeVDriftHistosByStation::name, nameToMaterial_, nameToMedium_, nameToShape_, and nameToVolume_.
Referenced by getGeoManager().
{
using namespace edm;
ESTransientHandle<DDCompactView> viewH;
m_eventSetup->get<IdealGeometryRecord>().get(viewH);
if ( ! viewH.isValid() )
{
return 0;
}
TGeoManager *geo_mgr = new TGeoManager("cmsGeo","CMS Detector");
// NOTE: the default constructor does not create the identity matrix
if (gGeoIdentity == 0)
{
gGeoIdentity = new TGeoIdentity("Identity");
}
std::cout << "about to initialize the DDCompactView walker" << std::endl;
DDCompactView::walker_type walker(viewH->graph());
DDCompactView::walker_type::value_type info = walker.current();
// The top most item is actually the volume holding both the
// geometry AND the magnetic field volumes!
walker.firstChild();
TGeoVolume *top = createVolume(info.first.name().fullname(),
info.first.solid(),
info.first.material());
if (top == 0)
{
return 0;
}
geo_mgr->SetTopVolume(top);
// ROOT chokes unless colors are assigned
top->SetVisibility(kFALSE);
top->SetLineColor(kBlue);
std::vector<TGeoVolume*> parentStack;
parentStack.push_back(top);
if( ! walker.firstChild() ) {
return 0;
}
do
{
DDCompactView::walker_type::value_type info = walker.current();
if (m_verbose)
{
for(unsigned int i=0; i<parentStack.size();++i) {
std::cout <<" ";
}
std::cout << info.first.name()<<" "<<info.second->copyno_<<" "
<< DDSolidShapesName::name(info.first.solid().shape())<<std::endl;
}
bool childAlreadyExists = (0 != nameToVolume_[info.first.name().fullname()]);
TGeoVolume *child = createVolume(info.first.name().fullname(),
info.first.solid(),
info.first.material());
if (0!=child && info.second != 0)
{
parentStack.back()->AddNode(child,
info.second->copyno_,
createPlacement(info.second->rotation(),
info.second->translation()));
child->SetLineColor(kBlue);
}
else
{
if ( info.second == 0 ) {
break;
}
}
if (0 == child || childAlreadyExists || level == int(parentStack.size()))
{
if (0!=child)
{
child->SetLineColor(kRed);
}
//stop descending
if ( ! walker.nextSibling())
{
while (walker.parent())
{
parentStack.pop_back();
if (walker.nextSibling()) {
break;
}
}
}
}
else
{
if (walker.firstChild())
{
parentStack.push_back(child);
}
else
{
if ( ! walker.nextSibling())
{
while (walker.parent())
{
parentStack.pop_back();
if (walker.nextSibling()) {
break;
}
}
}
}
}
} while ( ! parentStack.empty());
geo_mgr->CloseGeometry();
geo_mgr->DefaultColors();
nameToShape_.clear();
nameToVolume_.clear();
nameToMaterial_.clear();
nameToMedium_.clear();
return geo_mgr;
}
| TGeoMaterial * TGeoFromDddService::createMaterial | ( | const DDMaterial & | iMaterial | ) | [private] |
Definition at line 544 of file TGeoFromDddService.cc.
References DDMaterial::a(), DDMaterial::constituent(), DDMaterial::density(), DDName::fullname(), g, i, reco_application_tbsim_DetSim-Digi_cfg::mix, DDBase< N, C >::name(), nameToMaterial_, DDMaterial::noOfConstituents(), and DDMaterial::z().
Referenced by createVolume().
{
std::string mat_name = iMaterial.name().fullname();
TGeoMaterial *mat = nameToMaterial_[mat_name];
if (mat == 0)
{
if (iMaterial.noOfConstituents() > 0)
{
TGeoMixture *mix = new TGeoMixture(mat_name.c_str(),
iMaterial.noOfConstituents(),
iMaterial.density()*cm3/g);
for (int i = 0; i < iMaterial.noOfConstituents(); ++i)
{
mix->AddElement(createMaterial(iMaterial.constituent(i).first),
iMaterial.constituent(i).second);
}
mat = mix;
}
else
{
mat = new TGeoMaterial(mat_name.c_str(),
iMaterial.a()*mole/g, iMaterial.z(),
iMaterial.density()*cm3/g);
}
nameToMaterial_[mat_name] = mat;
}
return mat;
}
| TGeoShape * TGeoFromDddService::createShape | ( | const std::string & | iName, |
| const DDSolid & | iSolid | ||
| ) | [private] |
Definition at line 283 of file TGeoFromDddService.cc.
References DDPseudoTrap::atMinusZ(), benchmark_cfg::cerr, filterCSVwithJSON::copy, ddbox, ddcons, ddintersection, ddpolycone_rrz, ddpolyhedra_rrz, ddpseudotrap, ddsubtraction, ddtrap, ddtubs, ddunion, delta, Exception, DDName::fullname(), h, DDPseudoTrap::halfZ(), DDRotation::matrix(), DDName::name(), DDBase< N, C >::name(), nameToShape_, DDSolid::parameters(), DDPseudoTrap::radius(), DDBooleanSolid::rotation(), DDSolid::shape(), DDBooleanSolid::solidA(), DDBooleanSolid::solidB(), mathSSE::sqrt(), cond::rpcobtemp::temp, DDBooleanSolid::translation(), x, DDPseudoTrap::x1(), DDPseudoTrap::x2(), DDPseudoTrap::y1(), and DDPseudoTrap::y2().
Referenced by createVolume().
{
TGeoShape* rSolid= nameToShape_[iName];
if (rSolid == 0)
{
const std::vector<double>& params = iSolid.parameters();
// std::cout <<" shape "<<iSolid<<std::endl;
switch(iSolid.shape())
{
case ddbox:
rSolid = new TGeoBBox(
iName.c_str(),
params[0]/cm,
params[1]/cm,
params[2]/cm);
break;
case ddcons:
rSolid = new TGeoConeSeg(
iName.c_str(),
params[0]/cm,
params[1]/cm,
params[2]/cm,
params[3]/cm,
params[4]/cm,
params[5]/deg,
params[6]/deg+params[5]/deg
);
break;
case ddtubs:
//Order in params is zhalf,rIn,rOut,startPhi,deltaPhi
rSolid= new TGeoTubeSeg(
iName.c_str(),
params[1]/cm,
params[2]/cm,
params[0]/cm,
params[3]/deg,
params[4]/deg);
break;
case ddtrap:
rSolid =new TGeoTrap(
iName.c_str(),
params[0]/cm, //dz
params[1]/deg, //theta
params[2]/deg, //phi
params[3]/cm, //dy1
params[4]/cm, //dx1
params[5]/cm, //dx2
params[6]/deg, //alpha1
params[7]/cm, //dy2
params[8]/cm, //dx3
params[9]/cm, //dx4
params[10]/deg);//alpha2
break;
case ddpolycone_rrz:
rSolid = new TGeoPcon(
iName.c_str(),
params[0]/deg,
params[1]/deg,
(params.size()-2)/3) ;
{
std::vector<double> temp(params.size()+1);
temp.reserve(params.size()+1);
temp[0]=params[0]/deg;
temp[1]=params[1]/deg;
temp[2]=(params.size()-2)/3;
std::copy(params.begin()+2,params.end(),temp.begin()+3);
for(std::vector<double>::iterator it=temp.begin()+3;
it != temp.end();
++it) {
*it /=cm;
}
rSolid->SetDimensions(&(*(temp.begin())));
}
break;
case ddpolyhedra_rrz:
rSolid = new TGeoPgon(
iName.c_str(),
params[1]/deg,
params[2]/deg,
static_cast<int>(params[0]),
(params.size()-3)/3);
{
std::vector<double> temp(params.size()+1);
temp[0]=params[1]/deg;
temp[1]=params[2]/deg;
temp[2]=params[0];
temp[3]=(params.size()-3)/3;
std::copy(params.begin()+3,params.end(),temp.begin()+4);
for(std::vector<double>::iterator it=temp.begin()+4;
it != temp.end();
++it) {
*it /=cm;
}
rSolid->SetDimensions(&(*(temp.begin())));
}
break;
case ddpseudotrap:
{
//implementation taken from SimG4Core/Geometry/src/DDG4SolidConverter.cc
static DDRotationMatrix s_rot(ROOT::Math::RotationX(90.*deg));
DDPseudoTrap pt(iSolid);
assert(pt.radius() < 0);
double x=0;
double r = fabs(pt.radius());
if( pt.atMinusZ()) {
x=pt.x1();
} else {
x=pt.x2();
}
double openingAngle = 2.0*asin(x/r);
double h=pt.y1()<pt.y2()? pt.y2() :pt.y1();
h+=h/20.;
double displacement=0;
double startPhi = 0;
double delta = sqrt((r+x)*(r-x));
if(pt.atMinusZ()) {
displacement=-pt.halfZ() - delta;
startPhi = 270.-openingAngle/deg/2.0;
}else {
displacement = pt.halfZ() + delta;
startPhi = 90. - openingAngle/deg/2.;
}
std::auto_ptr<TGeoShape> trap( new TGeoTrd2(pt.name().name().c_str(),
pt.x1()/cm,
pt.x2()/cm,
pt.y1()/cm,
pt.y2()/cm,
pt.halfZ()/cm) );
std::auto_ptr<TGeoShape> tubs( new TGeoTubeSeg(pt.name().name().c_str(),
0.,
r/cm,
h/cm,
startPhi,
openingAngle) );
TGeoSubtraction* sub = new TGeoSubtraction(trap.release(),
tubs.release(),
createPlacement(s_rot,
DDTranslation(0.,
0.,
displacement)));
rSolid = new TGeoCompositeShape(iName.c_str(),
sub);
break;
}
case ddsubtraction:
{
DDBooleanSolid boolSolid(iSolid);
if(!boolSolid) {
throw cms::Exception("GeomConvert") <<"conversion to DDBooleanSolid failed";
}
std::auto_ptr<TGeoShape> left( createShape(boolSolid.solidA().name().fullname(),
boolSolid.solidA()) );
std::auto_ptr<TGeoShape> right( createShape(boolSolid.solidB().name().fullname(),
boolSolid.solidB()));
if( 0 != left.get() &&
0 != right.get() ) {
TGeoSubtraction* sub = new TGeoSubtraction(left.release(),right.release(),
gGeoIdentity,
createPlacement(
*(boolSolid.rotation().matrix()),
boolSolid.translation()));
rSolid = new TGeoCompositeShape(iName.c_str(),
sub);
}
break;
}
case ddunion:
{
DDBooleanSolid boolSolid(iSolid);
if(!boolSolid) {
throw cms::Exception("GeomConvert") <<"conversion to DDBooleanSolid failed";
}
std::auto_ptr<TGeoShape> left( createShape(boolSolid.solidA().name().fullname(),
boolSolid.solidA()) );
std::auto_ptr<TGeoShape> right( createShape(boolSolid.solidB().name().fullname(),
boolSolid.solidB()));
//DEBUGGING
//break;
if( 0 != left.get() &&
0 != right.get() ) {
TGeoUnion* boolS = new TGeoUnion(left.release(),right.release(),
gGeoIdentity,
createPlacement(
*(boolSolid.rotation().matrix()),
boolSolid.translation()));
rSolid = new TGeoCompositeShape(iName.c_str(),
boolS);
}
break;
}
case ddintersection:
{
DDBooleanSolid boolSolid(iSolid);
if(!boolSolid) {
throw cms::Exception("GeomConvert") <<"conversion to DDBooleanSolid failed";
}
std::auto_ptr<TGeoShape> left( createShape(boolSolid.solidA().name().fullname(),
boolSolid.solidA()) );
std::auto_ptr<TGeoShape> right( createShape(boolSolid.solidB().name().fullname(),
boolSolid.solidB()));
if( 0 != left.get() &&
0 != right.get() ) {
TGeoIntersection* boolS = new TGeoIntersection(left.release(),
right.release(),
gGeoIdentity,
createPlacement(
*(boolSolid.rotation().matrix()),
boolSolid.translation()));
rSolid = new TGeoCompositeShape(iName.c_str(),
boolS);
}
break;
}
default:
break;
}
nameToShape_[iName]=rSolid;
}
if (rSolid == 0)
{
std::cerr <<"COULD NOT MAKE "<<iName<<std::endl;
}
return rSolid;
}
| TGeoVolume * TGeoFromDddService::createVolume | ( | const std::string & | iName, |
| const DDSolid & | iSolid, | ||
| const DDMaterial & | iMaterial | ||
| ) | [private] |
Definition at line 515 of file TGeoFromDddService.cc.
References createMaterial(), createShape(), DDName::fullname(), DDBase< N, C >::name(), nameToMedium_, nameToVolume_, and v.
Referenced by createManager().
{
TGeoVolume* v=nameToVolume_[iName];
if (v == 0)
{
TGeoShape* solid = createShape(iSolid.name().fullname(),
iSolid);
std::string mat_name = iMaterial.name().fullname();
TGeoMedium *geo_med = nameToMedium_[mat_name];
if (geo_med == 0)
{
TGeoMaterial *geo_mat = createMaterial(iMaterial);
geo_med = new TGeoMedium(mat_name.c_str(), 0, geo_mat);
nameToMedium_[mat_name] = geo_med;
}
if (solid)
{
v = new TGeoVolume(iName.c_str(),
solid,
geo_med);
}
nameToVolume_[iName] = v;
}
return v;
}
| TGeoManager * TGeoFromDddService::getGeoManager | ( | ) |
Definition at line 103 of file TGeoFromDddService.cc.
References createManager(), m_eventSetup, m_geoManager, and m_level.
{
if (m_geoManager == 0)
{
if (m_eventSetup == 0)
edm::LogError("TGeoFromDddService") << "getGeoManager() -- EventSetup not present.\n";
else
{
m_geoManager = createManager(m_level);
if (m_geoManager == 0)
edm::LogError("TGeoFromDddService") << "getGeoManager() -- creation failed.\n";
}
}
gGeoManager = m_geoManager;
return m_geoManager;
}
| const TGeoFromDddService& TGeoFromDddService::operator= | ( | const TGeoFromDddService & | ) | [private] |
| void TGeoFromDddService::postBeginRun | ( | const edm::Run & | , |
| const edm::EventSetup & | es | ||
| ) |
Definition at line 81 of file TGeoFromDddService.cc.
References m_eventSetup.
Referenced by TGeoFromDddService().
{
printf("TGeoFromDddService::postBeginRun\n");
m_eventSetup = &es;
}
| void TGeoFromDddService::postEndRun | ( | const edm::Run & | , |
| const edm::EventSetup & | |||
| ) |
Definition at line 88 of file TGeoFromDddService.cc.
References m_eventSetup.
Referenced by TGeoFromDddService().
{
printf("TGeoFromDddService::postEndRun\n");
// Construction of geometry fails miserably on second attempt ...
/*
if (m_geoManager)
{
delete m_geoManager;
m_geoManager = 0;
}
*/
m_eventSetup = 0;
}
const edm::EventSetup* TGeoFromDddService::m_eventSetup [private] |
Definition at line 84 of file TGeoFromDddService.h.
Referenced by createManager(), getGeoManager(), postBeginRun(), and postEndRun().
TGeoManager* TGeoFromDddService::m_geoManager [private] |
Definition at line 85 of file TGeoFromDddService.h.
Referenced by getGeoManager(), and ~TGeoFromDddService().
int TGeoFromDddService::m_level [private] |
Definition at line 82 of file TGeoFromDddService.h.
Referenced by getGeoManager().
bool TGeoFromDddService::m_verbose [private] |
Definition at line 83 of file TGeoFromDddService.h.
Referenced by createManager().
std::map<std::string, TGeoMaterial*> TGeoFromDddService::nameToMaterial_ [private] |
Definition at line 89 of file TGeoFromDddService.h.
Referenced by createManager(), and createMaterial().
std::map<std::string, TGeoMedium*> TGeoFromDddService::nameToMedium_ [private] |
Definition at line 90 of file TGeoFromDddService.h.
Referenced by createManager(), and createVolume().
std::map<std::string, TGeoShape*> TGeoFromDddService::nameToShape_ [private] |
Definition at line 87 of file TGeoFromDddService.h.
Referenced by createManager(), and createShape().
std::map<std::string, TGeoVolume*> TGeoFromDddService::nameToVolume_ [private] |
Definition at line 88 of file TGeoFromDddService.h.
Referenced by createManager(), and createVolume().
1.7.3