TGeoFromDddService Class Reference

#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 &)=delete
	TGeoFromDddService (const TGeoFromDddService &)=delete

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_

Detailed Description

Description: [one line class summary]

Usage: <usage>

Definition at line 46 of file TGeoFromDddService.h.

Constructor & Destructor Documentation

TGeoFromDddService() [1/2]

TGeoFromDddService::TGeoFromDddService	(	const edm::ParameterSet &	pset,
		edm::ActivityRegistry &	ar
	)

Definition at line 58 of file TGeoFromDddService.cc.

    : m_level(pset.getUntrackedParameter<int>("level", 10)),
      m_verbose(pset.getUntrackedParameter<bool>("verbose", false)),
      m_eventSetup(nullptr),
      m_geoManager(nullptr) {
  ar.watchPostBeginRun(this, &TGeoFromDddService::postBeginRun);
  ar.watchPostEndRun(this, &TGeoFromDddService::postEndRun);
}

References postBeginRun(), postEndRun(), edm::ActivityRegistry::watchPostBeginRun(), and edm::ActivityRegistry::watchPostEndRun().

~TGeoFromDddService()

TGeoFromDddService::~TGeoFromDddService ( )

virtual

Definition at line 67 of file TGeoFromDddService.cc.

                                        {
  if (m_geoManager) {
    delete m_geoManager;
  }
}

References m_geoManager.

TGeoFromDddService() [2/2]

TGeoFromDddService::TGeoFromDddService ( const TGeoFromDddService &  )

privatedelete

Member Function Documentation

createManager()

TGeoManager * TGeoFromDddService::createManager ( int  level )

private

Definition at line 133 of file TGeoFromDddService.cc.

                                                        {
  using namespace edm;
 
  ESTransientHandle<DDCompactView> viewH;
  m_eventSetup->get<IdealGeometryRecord>().get(viewH);
 
  if (!viewH.isValid()) {
    return nullptr;
  }
 
  TGeoManager* geo_mgr = new TGeoManager("cmsGeo", "CMS Detector");
  // NOTE: the default constructor does not create the identity matrix
  if (gGeoIdentity == nullptr) {
    gGeoIdentity = new TGeoIdentity("Identity");
  }
 
  std::cout << "about to initialize the DDCompactView walker" << std::endl;
  auto walker = math::GraphWalker<DDLogicalPart, DDPosData*>(viewH->graph(), viewH->root());
  auto 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 == nullptr) {
    return nullptr;
  }
 
  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 nullptr;
  }
 
  do {
    auto 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 = (nullptr != nameToVolume_[info.first.name().fullname()]);
    TGeoVolume* child = createVolume(info.first.name().fullname(), info.first.solid(), info.first.material());
    if (nullptr != child && info.second != nullptr) {
      parentStack.back()->AddNode(
          child, info.second->copyno(), createPlacement(info.second->rotation(), info.second->translation()));
      child->SetLineColor(kBlue);
    } else {
      if (info.second == nullptr) {
        break;
      }
    }
    if (nullptr == child || childAlreadyExists || level == int(parentStack.size())) {
      if (nullptr != 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;
}

References gather_cfg::cout, dd4hep::createPlacement(), createVolume(), edm::EventSetup::get(), get, DDCompactView::graph(), mps_fire::i, info(), edm::ESHandleBase::isValid(), personalPlayback::level, m_eventSetup, m_verbose, DDSolidShapesName::name(), nameToMaterial_, nameToMedium_, nameToShape_, nameToVolume_, and DDCompactView::root().

Referenced by getGeoManager().

createMaterial()

TGeoMaterial * TGeoFromDddService::createMaterial ( const DDMaterial &  iMaterial )

private

Definition at line 466 of file TGeoFromDddService.cc.

                                                                            {
  std::string mat_name = iMaterial.name().fullname();
  TGeoMaterial* mat = nameToMaterial_[mat_name];
 
  if (mat == nullptr) {
    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;
}

References DDMaterial::a(), DDMaterial::constituent(), DDMaterial::density(), DDName::fullname(), g, mps_fire::i, GeneratorMix_cff::mix, DDBase< N, C >::name(), nameToMaterial_, DDMaterial::noOfConstituents(), AlCaHLTBitMon_QueryRunRegistry::string, and DDMaterial::z().

Referenced by createVolume().

createShape()

TGeoShape * TGeoFromDddService::createShape ( const std::string &  iName, const DDSolid &  iSolid  )

private

Definition at line 237 of file TGeoFromDddService.cc.

                                                                                        {
  TGeoShape* rSolid = nameToShape_[iName];
  if (rSolid == nullptr) {
    const std::vector<double>& params = iSolid.parameters();
    //      std::cout <<"  shape "<<iSolid<<std::endl;
    switch (iSolid.shape()) {
      case DDSolidShape::ddbox:
        rSolid = new TGeoBBox(iName.c_str(), params[0] / cm, params[1] / cm, params[2] / cm);
        break;
      case DDSolidShape::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 DDSolidShape::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[3] / deg + params[4] / deg);
        break;
      case DDSolidShape::ddcuttubs:
        //Order in params is  zhalf,rIn,rOut,startPhi,deltaPhi
        rSolid = new TGeoCtub(iName.c_str(),
                              params[1] / cm,
                              params[2] / cm,
                              params[0] / cm,
                              params[3] / deg,
                              params[3] / deg + params[4] / deg,
                              params[5],
                              params[6],
                              params[7],
                              params[8],
                              params[9],
                              params[10]);
        break;
      case DDSolidShape::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 DDSolidShape::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 DDSolidShape::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 DDSolidShape::ddextrudedpolygon: {
        DDExtrudedPolygon extrPgon(iSolid);
        std::vector<double> x = extrPgon.xVec();
        std::transform(x.begin(), x.end(), x.begin(), [](double d) { return d / cm; });
        std::vector<double> y = extrPgon.yVec();
        std::transform(y.begin(), y.end(), y.begin(), [](double d) { return d / cm; });
        std::vector<double> z = extrPgon.zVec();
        std::vector<double> zx = extrPgon.zxVec();
        std::vector<double> zy = extrPgon.zyVec();
        std::vector<double> zscale = extrPgon.zscaleVec();
 
        TGeoXtru* mySolid = new TGeoXtru(z.size());
        mySolid->DefinePolygon(x.size(), &(*x.begin()), &(*y.begin()));
        for (size_t i = 0; i < params[0]; ++i) {
          mySolid->DefineSection(i, z[i] / cm, zx[i] / cm, zy[i] / cm, zscale[i]);
        }
 
        rSolid = mySolid;
      } break;
      case DDSolidShape::ddpseudotrap: {
        //implementation taken from SimG4Core/Geometry/src/DDG4SolidConverter.cc
        const 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::unique_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::unique_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 DDSolidShape::ddsubtraction: {
        DDBooleanSolid boolSolid(iSolid);
        if (!boolSolid) {
          throw cms::Exception("GeomConvert") << "conversion to DDBooleanSolid failed";
        }
 
        std::unique_ptr<TGeoShape> left(createShape(boolSolid.solidA().name().fullname(), boolSolid.solidA()));
        std::unique_ptr<TGeoShape> right(createShape(boolSolid.solidB().name().fullname(), boolSolid.solidB()));
        if (nullptr != left.get() && nullptr != 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 DDSolidShape::ddunion: {
        DDBooleanSolid boolSolid(iSolid);
        if (!boolSolid) {
          throw cms::Exception("GeomConvert") << "conversion to DDBooleanSolid failed";
        }
 
        std::unique_ptr<TGeoShape> left(createShape(boolSolid.solidA().name().fullname(), boolSolid.solidA()));
        std::unique_ptr<TGeoShape> right(createShape(boolSolid.solidB().name().fullname(), boolSolid.solidB()));
        //DEBUGGING
        //break;
        if (nullptr != left.get() && nullptr != 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 DDSolidShape::ddintersection: {
        DDBooleanSolid boolSolid(iSolid);
        if (!boolSolid) {
          throw cms::Exception("GeomConvert") << "conversion to DDBooleanSolid failed";
        }
 
        std::unique_ptr<TGeoShape> left(createShape(boolSolid.solidA().name().fullname(), boolSolid.solidA()));
        std::unique_ptr<TGeoShape> right(createShape(boolSolid.solidB().name().fullname(), boolSolid.solidB()));
        if (nullptr != left.get() && nullptr != 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 == nullptr) {
    std::cerr << "COULD NOT MAKE " << iName << std::endl;
  }
  return rSolid;
}

References cms::cuda::assert(), EcnaPython_AdcPeg12_S1_10_R170298_1_0_150_Dee0::cerr, filterCSVwithJSON::copy, dd4hep::createPlacement(), ztail::d, ddbox, ddcons, ddcuttubs, ddextrudedpolygon, ddintersection, ddpolycone_rrz, ddpolyhedra_rrz, ddpseudotrap, ddsubtraction, ddtrap, ddtubs, ddunion, dumpMFGeometry_cfg::delta, Exception, DDName::fullname(), mps_fire::i, DDRotation::matrix(), DDBase< N, C >::name(), nameToShape_, DDSolid::parameters(), CalibrationSummaryClient_cfi::params, DiDispStaMuonMonitor_cfi::pt, alignCSCRings::r, DDBooleanSolid::rotation(), DDSolid::shape(), DDBooleanSolid::solidA(), DDBooleanSolid::solidB(), mathSSE::sqrt(), groupFilesInBlocks::temp, HcalDetIdTransform::transform(), DDBooleanSolid::translation(), x, DDExtrudedPolygon::xVec(), y, DDExtrudedPolygon::yVec(), z, DDExtrudedPolygon::zscaleVec(), DDExtrudedPolygon::zVec(), DDExtrudedPolygon::zxVec(), and DDExtrudedPolygon::zyVec().

Referenced by createVolume().

createVolume()

TGeoVolume * TGeoFromDddService::createVolume ( const std::string &  iName, const DDSolid &  iSolid, const DDMaterial &  iMaterial  )

private

Definition at line 445 of file TGeoFromDddService.cc.

                                                                          {
  TGeoVolume* v = nameToVolume_[iName];
  if (v == nullptr) {
    TGeoShape* solid = createShape(iSolid.name().fullname(), iSolid);
    std::string mat_name = iMaterial.name().fullname();
    TGeoMedium* geo_med = nameToMedium_[mat_name];
    if (geo_med == nullptr) {
      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;
}

References createMaterial(), createShape(), DDName::fullname(), DDBase< N, C >::name(), nameToMedium_, nameToVolume_, AlCaHLTBitMon_QueryRunRegistry::string, and findQualityFiles::v.

Referenced by createManager().

getGeoManager()

TGeoManager * TGeoFromDddService::getGeoManager

(

)

Definition at line 97 of file TGeoFromDddService.cc.

                                               {
  if (m_geoManager == nullptr) {
    if (m_eventSetup == nullptr)
      edm::LogError("TGeoFromDddService") << "getGeoManager() -- EventSetup not present.\n";
    else {
      m_geoManager = createManager(m_level);
      if (m_geoManager == nullptr)
        edm::LogError("TGeoFromDddService") << "getGeoManager() -- creation failed.\n";
    }
  }
  gGeoManager = m_geoManager;
  return m_geoManager;
}

References createManager(), m_eventSetup, m_geoManager, and m_level.

operator=()

const TGeoFromDddService& TGeoFromDddService::operator= ( const TGeoFromDddService &  )

privatedelete

postBeginRun()

void TGeoFromDddService::postBeginRun	(	const edm::Run &	,
		const edm::EventSetup &	es
	)

Definition at line 77 of file TGeoFromDddService.cc.

                                                                            {
  printf("TGeoFromDddService::postBeginRun\n");
 
  m_eventSetup = &es;
}

References m_eventSetup.

Referenced by TGeoFromDddService().

postEndRun()

void TGeoFromDddService::postEndRun	(	const edm::Run &	,
		const edm::EventSetup &
	)

Definition at line 83 of file TGeoFromDddService.cc.

                                                                       {
  printf("TGeoFromDddService::postEndRun\n");
 
  // Construction of geometry fails miserably on second attempt ...
  /*
   if (m_geoManager)
   {
      delete m_geoManager;
      m_geoManager = 0;
   }
   */
  m_eventSetup = nullptr;
}

References m_eventSetup.

Referenced by TGeoFromDddService().

Member Data Documentation

m_eventSetup

const edm::EventSetup* TGeoFromDddService::m_eventSetup

private

Definition at line 76 of file TGeoFromDddService.h.

Referenced by createManager(), getGeoManager(), postBeginRun(), and postEndRun().

m_geoManager

TGeoManager* TGeoFromDddService::m_geoManager

private

Definition at line 77 of file TGeoFromDddService.h.

Referenced by getGeoManager(), and ~TGeoFromDddService().

m_level

int TGeoFromDddService::m_level

private

Definition at line 74 of file TGeoFromDddService.h.

Referenced by getGeoManager().

m_verbose

bool TGeoFromDddService::m_verbose

private

Definition at line 75 of file TGeoFromDddService.h.

Referenced by createManager().

nameToMaterial_

std::map<std::string, TGeoMaterial*> TGeoFromDddService::nameToMaterial_

private

Definition at line 81 of file TGeoFromDddService.h.

Referenced by createManager(), and createMaterial().

nameToMedium_

std::map<std::string, TGeoMedium*> TGeoFromDddService::nameToMedium_

private

Definition at line 82 of file TGeoFromDddService.h.

Referenced by createManager(), and createVolume().

nameToShape_

std::map<std::string, TGeoShape*> TGeoFromDddService::nameToShape_

private

Definition at line 79 of file TGeoFromDddService.h.

Referenced by createManager(), and createShape().

nameToVolume_

std::map<std::string, TGeoVolume*> TGeoFromDddService::nameToVolume_

private

Definition at line 80 of file TGeoFromDddService.h.

Referenced by createManager(), and createVolume().