Relation.h

//-----------------------------------------------------------------------------
// $Id$
//-----------------------------------------------------------------------------
// Filename : Relation.h
// Section  : TRG CDC
// Owner    : Yoshihito Iwasaki
// Email    : yoshihito.iwasaki@kek.jp
//-----------------------------------------------------------------------------
// Description : A class to represent relations between TrackBase and MC track
//-----------------------------------------------------------------------------
// $Log$
//-----------------------------------------------------------------------------
 
#ifndef TRGCDCRelation_FLAG_
#define TRGCDCRelation_FLAG_
 
#include <map>
 
#ifdef TRGCDC_SHORT_NAMES
#define TCRelation TRGCDCRelation
#endif
 
#include "trg/cdc/TrackBase.h"
 
namespace Belle2 {
//class G4Track;
  class MCParticle;
//class TRGCDCTrackBase;
 
  class TRGCDCRelation {
 
  public:
 
    TRGCDCRelation(const TRGCDCTrackBase& track,
                   const std::map<unsigned, unsigned>& relation);
 
    virtual ~TRGCDCRelation();
 
  public:// Selectors
 
    const TRGCDCTrackBase& track(void) const;
 
    unsigned nContributors(void) const;
 
//  G4Track contributor(unsigned i = 0) const;
    unsigned contributor(unsigned i = 0) const;
 
    const MCParticle& mcParticle(unsigned i = 0) const;
 
    float purity(unsigned i = 0) const;
 
    float purity3D(unsigned trkID) const;
 
    float efficiency3D(unsigned trkID, std::map<unsigned, unsigned>& numTSsParticle) const;
 
    void dump(const std::string& message = std::string(""),
              const std::string& prefix = std::string("")) const;
 
  private:
 
    const TRGCDCTrackBase& _track;
 
    const std::map<unsigned, unsigned> _relations;
 
    mutable const std::pair<unsigned, unsigned>** _pairs;
  };
 
//-----------------------------------------------------------------------------
 
  inline
  const TRGCDCTrackBase&
  TRGCDCRelation::track(void) const
  {
    return _track;
  }
 
  inline
  unsigned
  TRGCDCRelation::nContributors(void) const
  {
    return _relations.size();
  }
 
} // namespace Belle2
#endif /* TRGCDCRelation_FLAG_ */