TrackerGeometry.cc

Go to the documentation of this file.

#include <typeinfo>

#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "Geometry/CommonDetUnit/interface/GeomDetUnit.h"
#include "Geometry/CommonDetUnit/interface/GeomDetType.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "DataFormats/SiStripDetId/interface/StripSubdetector.h"
#include "DataFormats/SiPixelDetId/interface/PixelSubdetector.h"
#include "DataFormats/DetId/interface/DetId.h"
#include "FWCore/Utilities/interface/Exception.h"

#include <algorithm>
#include <iostream>
#include <map>


#include "Geometry/TrackerNumberingBuilder/interface/GeometricDet.h"

namespace {
GeomDetEnumerators::SubDetector
geometricDetToGeomDet(GeometricDet::GDEnumType gdenum) {
  // provide a map between the GeometricDet enumerators and the GeomDet enumerators of the possible tracker subdetectors
  if(gdenum == GeometricDet::GDEnumType::PixelBarrel ) return GeomDetEnumerators::SubDetector::PixelBarrel;
  if(gdenum == GeometricDet::GDEnumType::PixelEndCap) return GeomDetEnumerators::SubDetector::PixelEndcap;
  if(gdenum == GeometricDet::GDEnumType::TIB) return GeomDetEnumerators::SubDetector::TIB;
  if(gdenum == GeometricDet::GDEnumType::TID) return GeomDetEnumerators::SubDetector::TID;
  if(gdenum == GeometricDet::GDEnumType::TOB) return GeomDetEnumerators::SubDetector::TOB;
  if(gdenum == GeometricDet::GDEnumType::TEC) return GeomDetEnumerators::SubDetector::TEC;
  if(gdenum == GeometricDet::GDEnumType::PixelPhase1Barrel) return GeomDetEnumerators::SubDetector::P1PXB;
  if(gdenum == GeometricDet::GDEnumType::PixelPhase1EndCap) return GeomDetEnumerators::SubDetector::P1PXEC;
  if(gdenum == GeometricDet::GDEnumType::PixelPhase2EndCap) return GeomDetEnumerators::SubDetector::P2PXEC;
  if(gdenum == GeometricDet::GDEnumType::OTPhase2Barrel) return GeomDetEnumerators::SubDetector::P2OTB;
  if(gdenum == GeometricDet::GDEnumType::OTPhase2EndCap) return GeomDetEnumerators::SubDetector::P2OTEC;
  return GeomDetEnumerators::SubDetector::invalidDet;
}
class DetIdComparator {
public:
  bool operator()(GeometricDet const* gd1, GeometricDet const * gd2) const {
    uint32_t det1 = gd1->geographicalId();
    uint32_t det2 = gd2->geographicalId();
    return det1 < det2;
  }                                      
  
};
}

enum class TrackerGeometry::ModuleType {
  UNKNOWN, 
  PXB, 
  PXF, 
  IB1, 
  IB2, 
  OB1, 
  OB2, 
  W1A, 
  W2A, 
  W3A, 
  W1B, 
  W2B, 
  W3B, 
  W4, 
  W5, 
  W6, 
  W7, 
  Ph1PXB, 
  Ph1PXF, 
  Ph2PXB, 
  Ph2PXF, 
  Ph2PSP, 
  Ph2PSS, 
  Ph2SS
};
 

TrackerGeometry::TrackerGeometry(GeometricDet const* gd) :  theTrackerDet(gd)
{
  for(unsigned int i=0;i<6;++i) {
    theSubDetTypeMap[i] = GeomDetEnumerators::invalidDet;
    theNumberOfLayers[i] = 0;
  }
  GeometricDet::ConstGeometricDetContainer subdetgd = gd->components();
  
  LogDebug("BuildingSubDetTypeMap") << "GeometriDet and GeomDetEnumerators enumerator values of the subdetectors";
  for(unsigned int i=0;i<subdetgd.size();++i) {
    assert(subdetgd[i]->geographicalId().subdetId()>0 && subdetgd[i]->geographicalId().subdetId()<7);
    theSubDetTypeMap[subdetgd[i]->geographicalId().subdetId()-1]= geometricDetToGeomDet(subdetgd[i]->type());
    theNumberOfLayers[subdetgd[i]->geographicalId().subdetId()-1]= subdetgd[i]->components().size();
    LogTrace("BuildingSubDetTypeMap") << "subdet " << i 
                      << " Geometric Det type " << subdetgd[i]->type()
                      << " Geom Det type " << theSubDetTypeMap[subdetgd[i]->geographicalId().subdetId()-1]
                      << " detid " <<  subdetgd[i]->geographicalId()
                      << " subdetid " <<  subdetgd[i]->geographicalId().subdetId()
                      << " number of layers " << subdetgd[i]->components().size();
  }
  LogDebug("SubDetTypeMapContent") << "Content of theSubDetTypeMap";
  for(unsigned int i=1;i<7;++i) {
    LogTrace("SubDetTypeMapContent") << " detid subdet "<< i << " Geom Det type " << geomDetSubDetector(i); 
  }
  LogDebug("NumberOfLayers") << "Content of theNumberOfLayers";
  for(unsigned int i=1;i<7;++i) {
    LogTrace("NumberOfLayers") << " detid subdet "<< i << " number of layers " << numberOfLayers(i); 
  }
  std::vector<const GeometricDet*> deepcomp;
  gd->deepComponents(deepcomp);
   
  sort(deepcomp.begin(), deepcomp.end(), DetIdComparator());

  LogDebug("ThicknessAndType") << " Total Number of Detectors " << deepcomp.size() ;
  LogDebug("ThicknessAndType") << "Dump of sensors names and bounds";
  for(auto det : deepcomp) {
    fillTestMap(det); 
    LogDebug("ThicknessAndType") << det->geographicalId() << " " << det->name().fullname() << " " << det->bounds()->thickness();
  }
  LogDebug("DetTypeList") << " Content of DetTypetList : size " << theDetTypetList.size();
  for (auto iVal : theDetTypetList) {
    LogDebug("DetTypeList") << " DetId " <<  std::get<0>(iVal) << " Type " << std::get<1>(iVal)<< " Thickness " << std::get<2>(iVal);
  }  
}


TrackerGeometry::~TrackerGeometry() {
    for (auto d : theDets) delete const_cast<GeomDet*>(d);
    for (auto d : theDetTypes) delete const_cast<GeomDetType*>(d);
}

void TrackerGeometry::finalize() {
    theDetTypes.shrink_to_fit();  // owns the DetTypes
    theDetUnits.shrink_to_fit();  // they're all also into 'theDets', so we assume 'theDets' owns them
    theDets.shrink_to_fit();     // owns *ONLY* the GeomDet * corresponding to GluedDets.
    theDetUnitIds.shrink_to_fit();
    theDetIds.shrink_to_fit();
  
    thePXBDets.shrink_to_fit(); // not owned: they're also in 'theDets'
    thePXFDets.shrink_to_fit(); // not owned: they're also in 'theDets'
    theTIBDets.shrink_to_fit(); // not owned: they're also in 'theDets'
    theTIDDets.shrink_to_fit(); // not owned: they're also in 'theDets'
    theTOBDets.shrink_to_fit(); // not owned: they're also in 'theDets'
    theTECDets.shrink_to_fit(); // not owned: they're also in 'theDets'
}


void TrackerGeometry::addType(GeomDetType const * p) {
  theDetTypes.push_back(p);  // add to vector
}

void TrackerGeometry::addDetUnit(GeomDetUnit const * p) {
  // set index
  const_cast<GeomDetUnit *>(p)->setIndex(theDetUnits.size());
  theDetUnits.push_back(p);  // add to vector
  theMapUnit.insert(std::make_pair(p->geographicalId().rawId(),p));
}

void TrackerGeometry::addDetUnitId(DetId p){
  theDetUnitIds.push_back(p);
}

void TrackerGeometry::addDet(GeomDet const * p) {
  // set index
  const_cast<GeomDet *>(p)->setGdetIndex(theDets.size());
  theDets.push_back(p);  // add to vector
  theMap.insert(std::make_pair(p->geographicalId().rawId(),p));
  DetId id(p->geographicalId());
  switch(id.subdetId()){
  case PixelSubdetector::PixelBarrel:
    thePXBDets.push_back(p);
    break;
  case PixelSubdetector::PixelEndcap:
    thePXFDets.push_back(p);
    break;
  case StripSubdetector::TIB:
    theTIBDets.push_back(p);
    break;
  case StripSubdetector::TID:
    theTIDDets.push_back(p);
    break;
  case StripSubdetector::TOB:
    theTOBDets.push_back(p);
    break;
  case StripSubdetector::TEC:
    theTECDets.push_back(p);
    break;
  default:
    edm::LogError("TrackerGeometry")<<"ERROR - I was expecting a Tracker Subdetector, I got a "<<id.subdetId();
  }


}

void TrackerGeometry::addDetId(DetId p){
  theDetIds.push_back(p);
}


const TrackerGeometry::DetContainer&
TrackerGeometry::detsPXB() const
{
  return thePXBDets;
}

const TrackerGeometry::DetContainer&
TrackerGeometry::detsPXF() const
{
  return thePXFDets;
}

const TrackerGeometry::DetContainer&
TrackerGeometry::detsTIB() const
{
  return theTIBDets;
}

const TrackerGeometry::DetContainer&
TrackerGeometry::detsTID() const
{
  return theTIDDets;
}

const TrackerGeometry::DetContainer&
TrackerGeometry::detsTOB() const
{
  return theTOBDets;
}

const TrackerGeometry::DetContainer&
TrackerGeometry::detsTEC() const
{
  return theTECDets;
}

const TrackerGeomDet * 
TrackerGeometry::idToDetUnit(DetId s)const
{
  mapIdToDetUnit::const_iterator p=theMapUnit.find(s.rawId());
  if (p != theMapUnit.end())
    return static_cast<const TrackerGeomDet *>(p->second);
  edm::LogError("TrackerGeometry")<<"Invalid DetID: no GeomDetUnit associated";
  return nullptr;
}

const TrackerGeomDet* 
TrackerGeometry::idToDet(DetId s)const
{
  mapIdToDet::const_iterator p=theMap.find(s.rawId());
  if (p != theMap.end())
    return static_cast<const TrackerGeomDet *>(p->second);
  edm::LogError("TrackerGeometry")<<"Invalid DetID: no GeomDet associated";
  return nullptr;
}

const GeomDetEnumerators::SubDetector 
TrackerGeometry::geomDetSubDetector(int subdet) const {
  if(subdet>=1 && subdet<=6) {
    return theSubDetTypeMap[subdet-1];
  } else {
    throw cms::Exception("WrongTrackerSubDet") << "Subdetector " << subdet;
  }
}

unsigned int
TrackerGeometry::numberOfLayers(int subdet) const {
  if(subdet>=1 && subdet<=6) {
    return theNumberOfLayers[subdet-1];
  } else {
    throw cms::Exception("WrongTrackerSubDet") << "Subdetector " << subdet;
  }
}

bool
TrackerGeometry::isThere(GeomDetEnumerators::SubDetector subdet) const {
  for(unsigned int i=1;i<7;++i) {
    if(subdet == geomDetSubDetector(i)) return true;
  }
  return false;
}

void TrackerGeometry::fillTestMap(const GeometricDet* gd) {
    
  std::string temp = gd->name();
  std::string name = temp.substr(temp.find(":")+1); 
  DetId detid = gd->geographicalId();
  float thickness = gd->bounds()->thickness();
  std::string nameTag;  
  TrackerGeometry::ModuleType mtype = moduleType(name);
  if (theDetTypetList.size() == 0) {
    theDetTypetList.push_back({std::make_tuple(detid, mtype, thickness)});
  } else {
    auto  & t = (*(theDetTypetList.end()-1));
    if (std::get<1>(t) != mtype) theDetTypetList.push_back({std::make_tuple(detid, mtype, thickness)});
    else {
      if  ( detid > std::get<0>(t) ) std::get<0>(t) = detid;
    }
  }
}

TrackerGeometry::ModuleType TrackerGeometry::getDetectorType(DetId detid) const {
  for (auto iVal : theDetTypetList) {
    DetId detid_max = std::get<0>(iVal);
    TrackerGeometry::ModuleType mtype =  std::get<1>(iVal);     
    if (detid.rawId() <=  detid_max.rawId()) return mtype;
  }
  return TrackerGeometry::ModuleType::UNKNOWN;
}
float TrackerGeometry::getDetectorThickness(DetId detid) const {
  for (auto iVal : theDetTypetList) {
    DetId detid_max = std::get<0>(iVal);
    if (detid.rawId() <=  detid_max.rawId()) 
      return std::get<2>(iVal);
  }
  return -1.0;
}

TrackerGeometry::ModuleType TrackerGeometry::moduleType(const std::string& name) const {
  if ( name.find("PixelBarrel") != std::string::npos) return ModuleType::Ph1PXB;
  else if (name.find("PixelForward") != std::string::npos) return ModuleType::Ph1PXF;
  else if ( name.find("TIB") != std::string::npos) {
    if ( name.find("0") != std::string::npos) return ModuleType::IB1;
    else return ModuleType::IB2;
  } else if ( name.find("TOB") != std::string::npos) {
    if ( name.find("0") != std::string::npos) return ModuleType::OB1;
    else return ModuleType::OB2;
  } else if ( name.find("TID") != std::string::npos) {
    if ( name.find("0") != std::string::npos) return ModuleType::W1A; 
    else if ( name.find("1") != std::string::npos) return ModuleType::W2A;
    else if ( name.find("2") != std::string::npos) return ModuleType::W3A;
  } else if ( name.find("TEC") != std::string::npos) { 
    if ( name.find("0") != std::string::npos) return ModuleType::W1B;
    else if ( name.find("1") != std::string::npos) return ModuleType::W2B;
    else if ( name.find("2") != std::string::npos) return ModuleType::W3B;
    else if ( name.find("3") != std::string::npos) return ModuleType::W4;
    else if ( name.find("4") != std::string::npos) return ModuleType::W5;
    else if ( name.find("5") != std::string::npos) return ModuleType::W6;
    else if ( name.find("6") != std::string::npos) return ModuleType::W7;
  } else if ( name.find("BModule") != std::string::npos || name.find("EModule") != std::string::npos ) { 
    if (name.find("PSMacroPixel") != std::string::npos) return ModuleType::Ph2PSP;
    else if (name.find("PSStrip") != std::string::npos) return ModuleType::Ph2PSS;
    else if (name.find("2S") != std::string::npos) return ModuleType::Ph2SS;
  }
  return ModuleType::UNKNOWN;  
}