CMSSW_4_4_3_patch1/src/Geometry/CaloGeometry/src/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"

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

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

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 ] : 0 ) ;
}

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 ] : 0 ) ;
}

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

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

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

bool 
CaloGeometry::present( const DetId& id ) const 
{
   const CaloSubdetectorGeometry* geom ( getSubdetectorGeometry( id ) ) ;
   return ( 0 == 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( 0    != m_geos[i] )
      {
         const std::vector< DetId >& aVec ( m_geos[i]->getValidDetIds() ) ;
         const bool isHcal ( 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 && ( 0 != m_geos[ index ] ) ?
            m_geos[ index ]->getValidDetIds( det, subdet ) :
            k_emptyVec ) ;
}