TrackerTreeGenerator Class Reference

#include <Alignment/TrackerAlignment/plugins/TrackerTreeGenerator.cc>

Inheritance diagram for TrackerTreeGenerator:

Public Member Functions
	TrackerTreeGenerator (const edm::ParameterSet &)
	~TrackerTreeGenerator ()=default
Public Member Functions inherited from edm::one::EDAnalyzer< edm::one::SharedResources >
	EDAnalyzer ()=default
Public Member Functions inherited from edm::one::EDAnalyzerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
	EDAnalyzerBase ()
ModuleDescription const &	moduleDescription () const
virtual	~EDAnalyzerBase ()
Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const
	EDConsumerBase ()
	EDConsumerBase (EDConsumerBase const &)=delete
	EDConsumerBase (EDConsumerBase &&)=default
ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const
void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFromEvent () const
void	labelsForToken (EDGetToken iToken, Labels &oLabels) const
void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
EDConsumerBase const &	operator= (EDConsumerBase const &)=delete
EDConsumerBase &	operator= (EDConsumerBase &&)=default
bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
virtual	~EDConsumerBase () noexcept(false)

Private Member Functions
virtual void	analyze (const edm::Event &, const edm::EventSetup &) override
virtual void	beginJob () override
virtual void	endJob () override

Private Attributes
edm::ParameterSet	config_
const bool	createEntryForDoubleSidedModule_
std::vector< TrackerTreeVariables >	vTkTreeVar_

Additional Inherited Members
Public Types inherited from edm::one::EDAnalyzerBase
typedef EDAnalyzerBase	ModuleType
Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels
Static Public Member Functions inherited from edm::one::EDAnalyzerBase
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &descriptions)
Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)
EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)
ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...
template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()
void	consumesMany (const TypeToGet &id)
template<BranchType B>
void	consumesMany (const TypeToGet &id)
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Description: <one line="" class="" summary>=""> Implementation: <Notes on="" implementation>="">

Definition at line 66 of file TrackerTreeGenerator.cc.

Constructor & Destructor Documentation

TrackerTreeGenerator::TrackerTreeGenerator ( const edm::ParameterSet &  config )

explicit

Definition at line 94 of file TrackerTreeGenerator.cc.

References TFileService::kSharedResource.

                                                                       :
  createEntryForDoubleSidedModule_(config.getParameter<bool>("createEntryForDoubleSidedModule")),
  config_(config)
{
  usesResource(TFileService::kSharedResource);
}

TrackerTreeGenerator::~TrackerTreeGenerator ( )

default

Member Function Documentation

void TrackerTreeGenerator::analyze ( const edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

overrideprivatevirtual

Definition at line 108 of file TrackerTreeGenerator.cc.

References PV3DBase< T, PVType, FrameType >::barePhi(), TrackerTreeVariables::blade, TrackerGeomBuilderFromGeometricDet::build(), createEntryForDoubleSidedModule_, MuonCkfTrajectoryBuilder_cfi::deltaPhi, TrackerGeometry::detIds(), TrackerGeometry::detUnitIds(), dPhi(), PFRecoTauDiscriminationAgainstElectronDeadECAL_cfi::dR, PV3DBase< T, PVType, FrameType >::eta(), edm::EventSetup::get(), TrackerTreeVariables::half, TrackerGeometry::idToDet(), TrackerTreeVariables::isDoubleSide, TrackerTreeVariables::isRPhi, TrackerTreeVariables::isStereo, TrackerTreeVariables::layer, TrackerTreeVariables::module, TrackerTreeVariables::nStrips, TrackerTreeVariables::outerInner, TrackerTreeVariables::panel, PV3DBase< T, PVType, FrameType >::perp(), TrackerTreeVariables::petal, PixelSubdetector::PixelBarrel, PixelSubdetector::PixelEndcap, TrackerTreeVariables::posEta, TrackerTreeVariables::posPhi, TrackerTreeVariables::posR, TrackerTreeVariables::posX, TrackerTreeVariables::posY, TrackerTreeVariables::posZ, edm::ESHandle< T >::product(), TrackerTreeVariables::rawId, TrackerTreeVariables::ring, TrackerTreeVariables::rod, TrackerTreeVariables::side, TrackerTreeVariables::subdetId, GeomDet::surface(), StripSubdetector::TEC, StripSubdetector::TIB, StripSubdetector::TID, StripSubdetector::TOB, Surface::toGlobal(), TrackerTreeVariables::uDirection, TrackerTreeVariables::vDirection, vTkTreeVar_, TrackerTreeVariables::wDirection, PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

{
  //iSetup.get<TrackerDigiGeometryRecord>().get(tkGeom);
  // now try to take directly the ideal geometry independent of used geometry in Global Tag
  edm::ESHandle<GeometricDet> geometricDet;
  iSetup.get<IdealGeometryRecord>().get(geometricDet);

  //Retrieve tracker topology from geometry
  edm::ESHandle<TrackerTopology> tTopoHandle;
  iSetup.get<TrackerTopologyRcd>().get(tTopoHandle);
  const TrackerTopology* const tTopo = tTopoHandle.product();

  edm::ESHandle<PTrackerParameters> ptp;
  iSetup.get<PTrackerParametersRcd>().get(ptp);
  TrackerGeomBuilderFromGeometricDet trackerBuilder;
  const TrackerGeometry* tkGeom = trackerBuilder.build(&(*geometricDet), *ptp, tTopo);
  AlignableTracker alignableTracker{tkGeom, tTopo};
  const auto& ns = alignableTracker.trackerNameSpace();

  edm::LogInfo("TrackerTreeGenerator")
    << "@SUB=TrackerTreeGenerator::analyze"
    << "There are " << tkGeom->detIds().size()
    << " dets and "<<tkGeom->detUnitIds().size()
    <<" detUnits in the Geometry Record";

  if (createEntryForDoubleSidedModule_) {
    edm::LogInfo("TrackerTreeGenerator")
      << "@SUB=TrackerTreeGenerator::analyze"
      << "Create entry for each module AND one entry for virtual "
      << "double-sided module in addition";
  } else {
    edm::LogInfo("TrackerTreeGenerator")
      << "@SUB=TrackerTreeGenerator::analyze"
      << "Create one entry for each physical module, do NOT create additional "
      << "entry for virtual double-sided module";
  }

  for (const auto& detId: tkGeom->detIds()){

    const GeomDet& geomDet = *tkGeom->idToDet(detId);
    const Surface& surface = geomDet.surface();

    TrackerTreeVariables tkTreeVar;
    const auto rawId = detId.rawId();
    tkTreeVar.rawId    = rawId;
    tkTreeVar.subdetId = detId.subdetId();

    switch (tkTreeVar.subdetId) {
    case PixelSubdetector::PixelBarrel:
      tkTreeVar.layer  = tTopo->pxbLayer(detId);
      tkTreeVar.half   = ns.tpb().halfBarrelNumber(rawId);
      tkTreeVar.rod    = tTopo->pxbLadder(detId);     // ... so, ladder is not per halfBarrel-Layer, but per barrel-layer!
      tkTreeVar.module = tTopo->pxbModule(detId);
      break;
    case PixelSubdetector::PixelEndcap:
      tkTreeVar.layer  = tTopo->pxfDisk(detId);
      tkTreeVar.side   = tTopo->pxfSide(detId);
      tkTreeVar.half   = ns.tpe().halfCylinderNumber(rawId);
      tkTreeVar.blade  = tTopo->pxfBlade(detId);
      tkTreeVar.panel  = tTopo->pxfPanel(detId);
      tkTreeVar.module = tTopo->pxfModule(detId);
      break;
    case StripSubdetector::TIB:
      tkTreeVar.layer        = tTopo->tibLayer(detId);
      tkTreeVar.side         = tTopo->tibStringInfo(detId)[0];
      tkTreeVar.half         = ns.tib().halfShellNumber(rawId);
      tkTreeVar.rod          = tTopo->tibStringInfo(detId)[2];
      tkTreeVar.outerInner   = tTopo->tibStringInfo(detId)[1];
      tkTreeVar.module       = tTopo->tibModule(detId);
      tkTreeVar.isDoubleSide = tTopo->tibIsDoubleSide(detId);
      tkTreeVar.isRPhi       = tTopo->tibIsRPhi(detId);
      tkTreeVar.isStereo     = tTopo->tibIsStereo(detId);
      break;
    case StripSubdetector::TID:
      tkTreeVar.layer        = tTopo->tidWheel(detId);
      tkTreeVar.side         = tTopo->tidSide(detId);
      tkTreeVar.ring         = tTopo->tidRing(detId);
      tkTreeVar.outerInner   = tTopo->tidModuleInfo(detId)[0];
      tkTreeVar.module       = tTopo->tidModuleInfo(detId)[1];
      tkTreeVar.isDoubleSide = tTopo->tidIsDoubleSide(detId);
      tkTreeVar.isRPhi       = tTopo->tidIsRPhi(detId);
      tkTreeVar.isStereo     = tTopo->tidIsStereo(detId);
      break;
    case StripSubdetector::TOB:
      tkTreeVar.layer        = tTopo->tobLayer(detId);
      tkTreeVar.side         = tTopo->tobRodInfo(detId)[0];
      tkTreeVar.rod          = tTopo->tobRodInfo(detId)[1];
      tkTreeVar.module       = tTopo->tobModule(detId);
      tkTreeVar.isDoubleSide = tTopo->tobIsDoubleSide(detId);
      tkTreeVar.isRPhi       = tTopo->tobIsRPhi(detId);
      tkTreeVar.isStereo     = tTopo->tobIsStereo(detId);
      break;
    case StripSubdetector::TEC:
      tkTreeVar.layer        = tTopo->tecWheel(detId);
      tkTreeVar.side         = tTopo->tecSide(detId);
      tkTreeVar.ring         = tTopo->tecRing(detId);
      tkTreeVar.petal        = tTopo->tecPetalInfo(detId)[1];
      tkTreeVar.outerInner   = tTopo->tecPetalInfo(detId)[0];
      tkTreeVar.module       = tTopo->tecModule(detId);
      tkTreeVar.isDoubleSide = tTopo->tecIsDoubleSide(detId);
      tkTreeVar.isRPhi       = tTopo->tecIsRPhi(detId);
      tkTreeVar.isStereo     = tTopo->tecIsStereo(detId);
      break;
    }


    LocalPoint lPModule(0.,0.,0.),
               lUDirection(1.,0.,0.),
               lVDirection(0.,1.,0.),
               lWDirection(0.,0.,1.);
    GlobalPoint gPModule    = surface.toGlobal(lPModule),
                gUDirection = surface.toGlobal(lUDirection),
                gVDirection = surface.toGlobal(lVDirection),
                gWDirection = surface.toGlobal(lWDirection);
    double dR(999.), dPhi(999.), dZ(999.);
    switch (tkTreeVar.subdetId) {
    case PixelSubdetector::PixelBarrel:
    case StripSubdetector::TIB:
    case StripSubdetector::TOB:
      dR   = gWDirection.perp() - gPModule.perp();
      dPhi = deltaPhi(gUDirection.barePhi(),gPModule.barePhi());
      dZ   = gVDirection.z() - gPModule.z();
      tkTreeVar.uDirection = dPhi>0. ? 1 : -1;
      tkTreeVar.vDirection = dZ>0.   ? 1 : -1;
      tkTreeVar.wDirection = dR>0.   ? 1 : -1;
      break;
    case PixelSubdetector::PixelEndcap:
      dR   = gUDirection.perp() - gPModule.perp();
      dPhi = deltaPhi(gVDirection.barePhi(),gPModule.barePhi());
      dZ   = gWDirection.z() - gPModule.z();
      tkTreeVar.uDirection = dR>0.   ? 1 : -1;
      tkTreeVar.vDirection = dPhi>0. ? 1 : -1;
      tkTreeVar.wDirection = dZ>0.   ? 1 : -1;
      break;
    case StripSubdetector::TID:
    case StripSubdetector::TEC:
      dR = gVDirection.perp() - gPModule.perp();
      dPhi = deltaPhi(gUDirection.barePhi(),gPModule.barePhi());
      dZ = gWDirection.z() - gPModule.z();
      tkTreeVar.uDirection = dPhi>0. ? 1 : -1;
      tkTreeVar.vDirection = dR>0.   ? 1 : -1;
      tkTreeVar.wDirection = dZ>0.   ? 1 : -1;
      break;
    }
    tkTreeVar.posR         = gPModule.perp();
    tkTreeVar.posPhi       = gPModule.barePhi();     // = gPModule.barePhi().degrees();
    tkTreeVar.posEta       = gPModule.eta();
    tkTreeVar.posX         = gPModule.x();
    tkTreeVar.posY         = gPModule.y();
    tkTreeVar.posZ         = gPModule.z();


    if(auto stripGeomDetUnit = dynamic_cast<const StripGeomDetUnit*>(&geomDet)){  //is it a single physical module?
      switch (tkTreeVar.subdetId) {
      case StripSubdetector::TIB:
      case StripSubdetector::TOB:
      case StripSubdetector::TID:
      case StripSubdetector::TEC:
        auto& topol =
          dynamic_cast<const StripTopology&>(stripGeomDetUnit->specificTopology());
        tkTreeVar.nStrips = topol.nstrips();
        break;
      }
    }


    if(!createEntryForDoubleSidedModule_){
      // do so only for individual modules and not also one entry for the combined doubleSided Module
      if (tkTreeVar.isDoubleSide) continue;
    }
    vTkTreeVar_.push_back(tkTreeVar);
  }

}

void TrackerTreeGenerator::beginJob ( void  )

overrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 286 of file TrackerTreeGenerator.cc.

{
}

void TrackerTreeGenerator::endJob ( void  )

overrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 293 of file TrackerTreeGenerator.cc.

References DEFINE_FWK_MODULE, F(), fileService, TFileDirectory::make(), TFileService::mkdir(), python.rootplot.argparse::module, module::module(), RecoTauValidation_cfi::posX, RecoTauValidation_cfi::posY, relativeConstraints::ring, and vTkTreeVar_.

Referenced by o2olib.O2ORunMgr::executeJob().

                             {
  UInt_t rawId(999), subdetId(999), layer(999), side(999), half(999), rod(999),
    ring(999), petal(999), blade(999), panel(999), outerInner(999),
    module(999), nStrips(999);
  Bool_t isDoubleSide(false), isRPhi(false), isStereo(false);
  Int_t uDirection(999), vDirection(999), wDirection(999);
  Float_t posR(999.F), posPhi(999.F), posEta(999.F),
    posX(999.F), posY(999.F), posZ(999.F);

  edm::Service<TFileService> fileService;
  TFileDirectory treeDir = fileService->mkdir("TrackerTree");
  auto trackerTree{treeDir.make<TTree>("TrackerTree",
                                       "IDs of all modules (ideal geometry)")};
  trackerTree->Branch("RawId", &rawId, "RawId/i");
  trackerTree->Branch("SubdetId", &subdetId, "SubdetId/i");
  trackerTree->Branch("Layer", &layer, "Layer/i");                // Barrel: Layer, Forward: Disk
  trackerTree->Branch("Side", &side, "Side/i");                   // Rod/Ring in +z or -z
  trackerTree->Branch("Half", &half, "Half/i");                   // PXB: HalfBarrel, PXF: HalfCylinder, TIB: HalfShell
  trackerTree->Branch("Rod", &rod, "Rod/i");                      // Barrel (Ladder or String or Rod)
  trackerTree->Branch("Ring", &ring, "Ring/i");                   // Forward
  trackerTree->Branch("Petal", &petal, "Petal/i");                // TEC
  trackerTree->Branch("Blade", &blade, "Blade/i");                // PXF
  trackerTree->Branch("Panel", &panel, "Panel/i");                // PXF
  trackerTree->Branch("OuterInner", &outerInner, "OuterInner/i"); // front/back String,Ring,Petal
  trackerTree->Branch("Module", &module, "Module/i");             // Module ID
  trackerTree->Branch("NStrips", &nStrips, "NStrips/i");
  trackerTree->Branch("IsDoubleSide", &isDoubleSide, "IsDoubleSide/O");
  trackerTree->Branch("IsRPhi", &isRPhi, "IsRPhi/O");
  trackerTree->Branch("IsStereo", &isStereo, "IsStereo/O");
  trackerTree->Branch("UDirection", &uDirection, "UDirection/I");
  trackerTree->Branch("VDirection", &vDirection, "VDirection/I");
  trackerTree->Branch("WDirection", &wDirection, "WDirection/I");
  trackerTree->Branch("PosR", &posR, "PosR/F");
  trackerTree->Branch("PosPhi", &posPhi, "PosPhi/F");
  trackerTree->Branch("PosEta", &posEta, "PosEta/F");
  trackerTree->Branch("PosX", &posX, "PosX/F");
  trackerTree->Branch("PosY", &posY, "PosY/F");
  trackerTree->Branch("PosZ", &posZ, "PosZ/F");

  for(const auto& iTree: vTkTreeVar_){
    rawId        = iTree.rawId;
    subdetId     = iTree.subdetId;
    layer        = iTree.layer;
    side         = iTree.side;
    half         = iTree.half;
    rod          = iTree.rod;
    ring         = iTree.ring;
    petal        = iTree.petal;
    blade        = iTree.blade;
    panel        = iTree.panel;
    outerInner   = iTree.outerInner;
    module       = iTree.module;
    nStrips      = iTree.nStrips;
    isDoubleSide = iTree.isDoubleSide;
    isRPhi       = iTree.isRPhi;
    isStereo     = iTree.isStereo;
    uDirection   = iTree.uDirection;
    vDirection   = iTree.vDirection;
    wDirection   = iTree.wDirection;
    posR         = iTree.posR;
    posPhi       = iTree.posPhi;
    posEta       = iTree.posEta;
    posX         = iTree.posX;
    posY         = iTree.posY;
    posZ         = iTree.posZ;

    trackerTree->Fill();
  }
  edm::LogInfo("TrackerTreeGenerator")
    << "@SUB=TrackerTreeGenerator::endJob"
    << "TrackerTree contains "<< vTkTreeVar_.size() <<" entries overall";
}

Member Data Documentation

edm::ParameterSet TrackerTreeGenerator::config_

private

Definition at line 80 of file TrackerTreeGenerator.cc.

const bool TrackerTreeGenerator::createEntryForDoubleSidedModule_

private

Definition at line 78 of file TrackerTreeGenerator.cc.

Referenced by analyze().

std::vector<TrackerTreeVariables> TrackerTreeGenerator::vTkTreeVar_

private

Definition at line 79 of file TrackerTreeGenerator.cc.

Referenced by analyze(), and endJob().