CellIDEncoder.h

#ifndef CellIDEncoder_h
#define CellIDEncoder_h 1

#include "EVENT/LCCollection.h"
#include "UTIL/BitField64.h"
#include "IMPL/LCFlagImpl.h"

#include <string>

//using namespace lcio ;

//forward declarations for template specializations 
namespace EVENT{
  class SimCalorimeterHit ;
  class RawCalorimeterHit ;
  class CalorimeterHit ;
  class TrackerData ;
  class TrackerPulse ;
  class TrackerRawData ;
  class SimTrackerHit ;
  class TrackerHit ;
  class TrackerHitPlane ;
  class TrackerHitZCylinder ;
}

namespace UTIL{


  template <class T>
  int CellIDEncoder_cellID1Bit()  { return -1 ; } 

  template<> int CellIDEncoder_cellID1Bit<EVENT::SimCalorimeterHit>() ;

  template<> int CellIDEncoder_cellID1Bit<EVENT::RawCalorimeterHit>() ;

  template<> int CellIDEncoder_cellID1Bit<EVENT::CalorimeterHit>() ; 

  template<> int CellIDEncoder_cellID1Bit<EVENT::TrackerData>() ;

  template<> int CellIDEncoder_cellID1Bit<EVENT::TrackerPulse>() ;

  template<> int CellIDEncoder_cellID1Bit<EVENT::TrackerRawData>() ;

  template<> int CellIDEncoder_cellID1Bit<EVENT::TrackerHit>() ;

  template<> int CellIDEncoder_cellID1Bit<EVENT::TrackerHitPlane>() ;

  template<> int CellIDEncoder_cellID1Bit<EVENT::TrackerHitZCylinder>() ;

  template<> int CellIDEncoder_cellID1Bit<EVENT::SimTrackerHit>() ;


  template <class T>
  void CellIDEncoder_setCellID(T* hit, int low, int high)  {  

    hit->setCellID0( low ) ;
    hit->setCellID1( high ) ;
  } 
 
//  /** Specialization for SimTrackerHits that have only one cellID */
//  template<> 
//  void CellIDEncoder_setCellID<IMPL::SimTrackerHitImpl>( IMPL::SimTrackerHitImpl* hit, 
//                                                       int low, int high);
//


  template <class T> 
  class CellIDEncoder : public BitField64 {

  public:  
    
    CellIDEncoder( const std::string& cellIDEncoding ,  EVENT::LCCollection* col) :

      BitField64( cellIDEncoding  ),

      _col( col ) {
      
      _col->parameters().setValue( EVENT::LCIO::CellIDEncoding , cellIDEncoding ) ;

        setCellIDFlag() ;
    }

    inline void setCellID( T* hit) {

      CellIDEncoder_setCellID( hit , lowWord() , highWord() ) ;
    }

    void setCellIDFlag() {
      
      int bit = CellIDEncoder_cellID1Bit<typename T::lcobject_type>() ;
      
      if(  bit >= 0 ) {
        
        IMPL::LCFlagImpl f( _col->getFlag() ) ;
        
        if( highestBit() > 31 ) {
          
          //      std::cout << " setting bit " <<  bit << std::endl ;
          f.setBit( bit ) ;
          
        } else {
          //      std::cout << " unsetting bit " <<  bit << std::endl ;
          f.unsetBit( bit ) ;
        }
        
        _col->setFlag( f.getFlag()  ) ;
      }
    }
    
  protected:
    
    //    int cellID1Bit() { return -1 ; }

 
    EVENT::LCCollection* _col ;
    //    T* _oldHit ;
  } ; 
  



} // namespace
#endif