CaloGeometry.cc

Go to the documentation of this file.

#include "Geometry/CaloGeometry/interface/CaloCellGeometry.h"
#include "Geometry/CaloGeometry/interface/CaloSubdetectorGeometry.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"

const std::vector<DetId> CaloGeometry::k_emptyVec ( 0 ) ;

CaloGeometry::CaloGeometry() :
   m_geos ( kLength, nullptr )
{
}

unsigned int 
CaloGeometry::makeIndex( DetId::Detector det    , 
             int             subdet ,
             bool&           ok       ) const 
{
   const unsigned int idet ( det ) ;

   ok = ( kMinDet <= idet   &&
      kMaxDet >= idet   &&
      0       <  subdet &&
      kMaxSub >= subdet    ) ;
   
   return ( ( det - kMinDet )*kMaxSub + subdet - 1 ) ;
}

void 
CaloGeometry::setSubdetGeometry( DetId::Detector                det    , 
                 int                            subdet , 
                 const CaloSubdetectorGeometry* geom     ) 
{
   bool ok ;
   const unsigned int index = makeIndex( det, subdet, ok ) ;
   if( ok ) m_geos[index] = geom ;

//   std::cout<<"Detector="<<(int)det<<", subset="<<subdet<<", index="<<index
//      <<", size="<<m_geos.size()<<std::endl;

   assert( ok ) ;
}

const CaloSubdetectorGeometry* 
CaloGeometry::getSubdetectorGeometry( const DetId& id ) const 
{
   bool ok ;

   const unsigned int index ( makeIndex( id.det(),
                     id.subdetId(),
                     ok             ) ) ;
   return ( ok ? m_geos[ index ] : nullptr ) ;
}

const CaloSubdetectorGeometry* 
CaloGeometry::getSubdetectorGeometry( DetId::Detector det    , 
                      int             subdet  ) const 
{
   bool ok ;

   const unsigned int index ( makeIndex( det,
                     subdet,
                     ok             ) ) ;
   return ( ok ? m_geos[ index ] : nullptr ) ;
}

static const GlobalPoint notFound(0,0,0);

const GlobalPoint& 
CaloGeometry::getPosition( const DetId& id ) const 
{
   const CaloSubdetectorGeometry* geom=getSubdetectorGeometry( id ) ;
   const CaloCellGeometry* cell ( ( nullptr == geom ? nullptr : geom->getGeometry( id ) ) ) ;
   return ( nullptr == cell ?  notFound : cell->getPosition() ) ;
}

const CaloCellGeometry* 
CaloGeometry::getGeometry( const DetId& id ) const 
{
   const CaloSubdetectorGeometry* geom ( getSubdetectorGeometry( id ) ) ;
   const CaloCellGeometry* cell ( nullptr == geom ? nullptr : geom->getGeometry( id ) ) ;
   return cell ;
}

bool 
CaloGeometry::present( const DetId& id ) const 
{
   const CaloSubdetectorGeometry* geom ( getSubdetectorGeometry( id ) ) ;
   return ( nullptr == geom ? false : geom->present( id ) ) ;
}

std::vector<DetId> CaloGeometry::getValidDetIds() const
{
   std::vector<DetId> returnValue ;
   returnValue.reserve( kLength ) ;

   bool doneHcal ( false ) ;
   for( unsigned int i ( 0 ) ; i != m_geos.size() ; ++i ) 
   {     
      if( nullptr != m_geos[i] )
      {
     const std::vector< DetId >& aVec = m_geos[i]->getValidDetIds();     
     if( aVec.empty() ) {
       edm::LogWarning("EmptyDetIdList") << "Valid det id list at index " << i << " is empty!" << std::endl;
     }
     const bool isHcal ( !aVec.empty() && DetId::Hcal == aVec.front().det() ) ;
     if( !doneHcal ||
         !isHcal      )
     {
        returnValue.insert( returnValue.end(), aVec.begin(), aVec.end() ) ;
        if( !doneHcal &&
        isHcal        ) doneHcal = true ;
     }
      }
   }
   return returnValue ;
}

const std::vector<DetId>&
CaloGeometry::getValidDetIds( DetId::Detector det    , 
                  int             subdet  ) const 
{
   bool ok ;

   const unsigned int index ( makeIndex( det,
                     subdet,
                     ok             ) ) ;

   return ( ok && ( nullptr != m_geos[ index ] ) ?
        m_geos[ index ]->getValidDetIds( det, subdet ) :
        k_emptyVec ) ;
}