ClusterImpl.h

Go to the documentation of this file.

#ifndef IMPL_CLUSTERIMPL_H
#define IMPL_CLUSTERIMPL_H 1


#include "EVENT/Cluster.h"
#include "AccessChecked.h"
#include <map>
#include <bitset>

#define NERRPOS 6
#define NERRDIR 3

#define NSHAPE_OLD 6

namespace IMPL {


  class ClusterImpl : public EVENT::Cluster, public AccessChecked {
    
  public: 
    
    ClusterImpl() ;
    
    virtual ~ClusterImpl() ; 

    virtual int id() const { return simpleUID() ; }

    virtual int getType() const ;

    virtual float getEnergy() const ;

    virtual float getEnergyError() const ;

    virtual const float* getPosition() const ;

    virtual const EVENT::FloatVec & getPositionError() const ;

    virtual float getITheta() const ;

    virtual float getIPhi() const ;

    virtual const EVENT::FloatVec & getDirectionError() const ;

    virtual const EVENT::FloatVec & getShape() const ;

//     /** Type hypotheses: 3 Parameters: compatible with EM, HAD, muon cluster
//      */
//     virtual const EVENT::FloatVec & getParticleType() const ;


    virtual const EVENT::ParticleIDVec & getParticleIDs() const ;


    virtual const EVENT::ClusterVec & getClusters() const ;


    virtual const EVENT::CalorimeterHitVec & getCalorimeterHits() const ;

    virtual const EVENT::FloatVec & getHitContributions() const ;


    virtual const EVENT::FloatVec & getSubdetectorEnergies() const ;

    //setters 
    void setTypeBit( int index , bool val=true) ;
    void setEnergy(float energy ) ;
    void setEnergyError(float energyError ) ;
    void setPosition(const float* position) ;
    void setPositionError(const EVENT::FloatVec &errpos) ;
    void setPositionError(const float* errpos) ;
    void setITheta(float theta) ;
    void setIPhi(float phi) ;
    void setDirectionError(const EVENT::FloatVec &errdir) ;
    void setDirectionError(const float* errdir) ;
    //    void setShape(const float* shape) ;
    void setShape(const EVENT::FloatVec &shape) ;

    void addParticleID( EVENT::ParticleID*  pid ) ;

//     void setEMWeight(float emWeight ) ;
//     void setHADWeight(float hadWeight ) ;
//     void setMuonWeight(float muonWeight ) ;

    void addCluster(EVENT::Cluster* cluster) ;
    void addHit(EVENT::CalorimeterHit* hit  , float contribution) ;


    EVENT::FloatVec& subdetectorEnergies() ;

  protected:
    void setType(int type ) ; 

    std::bitset<32> _type{0} ;
    float _energy{0} ;
    float _energyError{0} ;
    float _position[3]={0,0,0} ;
    EVENT::FloatVec _errpos ;
    float _theta{0};
    float _phi{0} ;
    EVENT::FloatVec _errdir ;
    EVENT::FloatVec _shape{} ;
    EVENT::ParticleIDVec _pid{} ;
    EVENT::ClusterVec _clusters{} ;
    EVENT::CalorimeterHitVec _hits{} ;
    EVENT::FloatVec _weights{} ;
    EVENT::FloatVec _subdetectorEnergies{} ;

}; // class

} // namespace IMPL
#endif /* ifndef IMPL_CLUSTERIMLP_H */