StEEmcPool/StEEmcAssociationMaker/StEEmcAssociationMaker.h

#ifndef STAR_StEEmcAssociationMaker_HH
#define STAR_StEEmcAssociationMaker_HH

#define NDETECTORS 4

#ifndef StMaker_H
#include "StMaker.h"
#include "StMcEvent/StMcCalorimeterHit.hh"
#include "TMatrix.h"
#endif

#include <map>
#include <utility>
#if !defined(ST_NO_NAMESPACES)
using std::multimap;
using std::pair;
#endif

class StEvent;
class StEmcRawHit;
class StEmcCluster;
class StEmcPoint;
class StMcTrack;

class StEEmcAssociation
{
  private:
    StMcTrack        *mTrack;
  public:
                     StEEmcAssociation(StMcTrack *t);
    virtual          ~StEEmcAssociation();
    const StMcTrack* getTrack() const           { return mTrack; }   
    ClassDef(StEEmcAssociation, 1)
};

class StEEmcClusterAssociation:public StEEmcAssociation
{
  private:
    StEmcCluster     *mCluster;
    float            mFTrack;
    float            mFEmc;
  public:
                     StEEmcClusterAssociation(StMcTrack*, StEmcCluster*, float,float);
    virtual          ~StEEmcClusterAssociation();
    const StEmcCluster* getCluster() const      { return mCluster; }  
    float            getFractionTrack() const   { return mFTrack; }   
    float            getFractionCluster() const { return mFEmc; }     
    ClassDef(StEEmcClusterAssociation, 1)
};
class StEEmcPointAssociation:public StEEmcAssociation
{
  private:
    StEmcPoint       *mPoint;
    int              mAssocType;
  public:
                     StEEmcPointAssociation(StMcTrack*, StEmcPoint*, int);
    virtual          ~StEEmcPointAssociation();
    const StEmcPoint*getPoint() const           { return mPoint; }      
    int              getAssociation() const     { return mAssocType; }  
    int              getAssociation(int det)    { return (det>0 && det<=NDETECTORS) ? ((mAssocType&(1<<(det-1)))!=0): 0; } 
    ClassDef(StEEmcPointAssociation, 1)
};

// define basic multimap structures

#ifndef __CINT__
struct compEEmcTrack   { bool operator()(const StMcTrack* a, const StMcTrack* b) const {return a<b; } };
struct compEEmcHit     { bool operator()(const StEmcRawHit* a, const StEmcRawHit* b) const {return a<b; }};
struct compEEmcCluster { bool operator()(const StEmcCluster* a, const StEmcCluster* b) const {return a<b; }};
struct compEEmcPoint   { bool operator()(const StEmcPoint* a, const StEmcPoint* b) const {return a<b; }};
// define maps and iterators
typedef  multimap<const StMcTrack*, const StEEmcClusterAssociation*, compEEmcTrack>      multiEEmcTrackCluster;
typedef  multimap<const StEmcCluster*, const StEEmcClusterAssociation*, compEEmcCluster> multiEEmcClusterTrack;
typedef  multimap<const StMcTrack*, const StEEmcPointAssociation*, compEEmcTrack>        multiEEmcTrackPoint;
typedef  multimap<const StEmcPoint*, const StEEmcPointAssociation*, compEEmcPoint>       multiEEmcPointTrack;
//defining iterators
typedef  multiEEmcTrackCluster::iterator      multiEEmcTrackClusterIter; 
typedef  multiEEmcClusterTrack::iterator      multiEEmcClusterTrackIter; 
typedef  multiEEmcTrackPoint::iterator        multiEEmcTrackPointIter;   
typedef  multiEEmcPointTrack::iterator        multiEEmcPointTrackIter;   
typedef  multiEEmcTrackCluster::value_type    multiEEmcTrackClusterValue; 
typedef  multiEEmcClusterTrack::value_type    multiEEmcClusterTrackValue; 
typedef  multiEEmcTrackPoint::value_type      multiEEmcTrackPointValue;   
typedef  multiEEmcPointTrack::value_type      multiEEmcPointTrackValue;   
#else
class multiEEmcTrackCluster;
class multiEEmcClusterTrack;
class multiEEmcTrackPoint;
class multiEEmcPointTrack;
class multiEEmcTrackClusterIter;
class multiEEmcClusterTrackIter;
class multiEEmcTrackPointIter;
class multiEEmcPointTrackIter;
class multiEEmcTrackClusterValue;
class multiEEmcClusterTrackValue;
class multiEEmcTrackPointValue;
class multiEEmcPointTrackValue;
#endif


class StEEmcAssociationMaker : public StMaker
{
  public:
                                StEEmcAssociationMaker(const char* name="EEmcAssoc");
         virtual                ~StEEmcAssociationMaker();
                 Int_t          Init();
                 Int_t          Make();
                 Int_t          Finish();
                 void           Clear(const char*);
                 
                 void           printMaps();
         
         multiEEmcTrackCluster*  getTrackClusterMap(Int_t i)      { if(i>0 && i<=NDETECTORS) return mTrackCluster[i-1]; else return NULL; } 
         multiEEmcTrackCluster*  getTrackClusterMap(const char*);  
         multiEEmcClusterTrack*  getClusterTrackMap(Int_t i)      { if(i>0 && i<=NDETECTORS) return mClusterTrack[i-1]; else return NULL; } 
         multiEEmcClusterTrack*  getClusterTrackMap(const char*);  
         multiEEmcTrackPoint*    getTrackPointMap()               { return mTrackPoint;} 
         multiEEmcPointTrack*    getPointTrackMap()               { return mPointTrack;} 
                 Int_t          getDetNum(const char*);          
                 
                 void           setPrint(Bool_t a)               {mPrint = a;}; 
     void          printHits(StEvent*); 
 
         
         
  protected: 
                 Bool_t         mPrint;
                 
         multiEEmcTrackCluster*  mTrackCluster[NDETECTORS];
         multiEEmcClusterTrack*  mClusterTrack[NDETECTORS];
         multiEEmcTrackPoint*    mTrackPoint;
         multiEEmcPointTrack*    mPointTrack;
   ClassDef(StEEmcAssociationMaker, 1)
};

#endif

/*******************************************************************************/
//
// $Id: StEEmcAssociationMaker.h,v 1.1.1.1 2005/05/31 18:54:14 wzhang Exp $
// $Log: StEEmcAssociationMaker.h,v $
// Revision 1.1.1.1  2005/05/31 18:54:14  wzhang
// First version
//
//

---