PIDHandler.h

#ifndef PIDHandler_h
#define PIDHandler_h 1

#include "UTIL/CollectionParameterMap.h"

#include "EVENT/LCCollection.h"
#include "Exceptions.h"

#include "IMPL/ParticleIDImpl.h"
#include "IMPL/ReconstructedParticleImpl.h"

#include <lcio.h>

#include <string>

using namespace lcio ;


namespace UTIL{
  
  
  class PIDHandler {

    typedef CollectionParameterMap CPM ;
    typedef std::map< int, StringVec > PNM ;
    typedef  std::map< CPM::map_type::mapped_type , CPM::map_type::key_type > CPMINV ;


  public:  
    
    PIDHandler( const LCCollection* col ) ;
    
    PIDHandler( LCCollection* col ) ;


    ~PIDHandler() ;

    int addAlgorithm( const std::string& algoName, const StringVec& parameterNames ) ;
    
    int getAlgorithmID( const std::string& algoName ) ;
    
    const std::string& getAlgorithmName(  int algoID ) ;

    int getParameterIndex( int algorithmID, const std::string& pName ) ;
      
    const ParticleID& getParticleID( LCObject* particle , int algorithmID ) ;
    

    ParticleIDVec getParticleIDs( LCObject* p , int id ) ;


    void setParticleIDUsed( IMPL::ReconstructedParticleImpl* particle , int algorithmID  ) ;


    const StringVec&  getParameterNames( int algorithmID  ) ;
    
    const IntVec& getAlgorithmIDs() ;


    void setParticleID( LCObject* p ,
                        int userType, 
                        int PDG,
                        float likelihood,
                        int algorithmID, 
                        const FloatVec& params
                        ) ;

  protected:

    int nextID() { return ++_maxID ; }
    void init( const LCCollection* col ) ;

    PIDHandler();
 
    LCCollection* _col ;
    CPM _cpm ;
    int _type ; 
    int _maxID ;
    PNM _pNames ; 
    CPMINV _cpmInv ;
    IntVec _ids ;

  } ; 
  

  class UnknownAlgorithm : public Exception {
    
  protected:
    UnknownAlgorithm() {  /*no_op*/ ; } 
  public: 
    virtual ~UnknownAlgorithm() throw() { /*no_op*/; } 
    
    UnknownAlgorithm( std::string text ){
      message = "lcio::UnknownAlgorithm: " + text ;
    }
  }; 
 



} // namespace
#endif