StEEmcSmdCluster.h

#ifndef __StEEmcSmdCluster_h__
#define __StEEmcSmdCluster_h__

#include <cassert>
#include <TObject.h>
#include "StEEmcPool/StEEmcA2EMaker/StEEmcStrip.h"
#include "StEEmcPool/StEEmcA2EMaker/StEEmcTower.h"

#include "StEEmcBaseCluster.h"

class TH1F;

class StEmcCluster;

class StEEmcSmdCluster : public StEEmcBaseCluster {
public:
  StEEmcSmdCluster();
  StEEmcSmdCluster ( const StEEmcSmdCluster &c );
  virtual ~StEEmcSmdCluster(){ /* nada */ };

  void add(const StEEmcStrip &strip, Float_t weight=1.0 ); // Add an SMD strip to this cluster with the specified weight
  void add(const StEEmcStripVec_t &strips );

  Float_t energy()const {return mEnergy;} // Return the energy of this cluster
  Float_t mean()const {return mMean;} // Return the mean strip number of this cluster
  Float_t sigma()const {return mSigma;} // Return the sigma -- sqrt(variance) -- of the cluster

  void mean(Float_t x){ mMean=x; } // Set the mean of the cluster
  void sigma(Float_t s){ mSigma=s; } // Set the width of the cluster
  //void energy( Float_t e){ mEnergy=e; } // Set the energy of the cluster

  Float_t energy( Int_t nmax, Option_t *opts="mean" ) const; // Return the energy of this cluster summed over +/- nmax strips about either the mean (opts=="mean") or seed (opts=="seed") smd strip.
  Float_t sigma( Int_t nmax, Option_t *opts="mean" ) const; // As above, but returns sigma instead of energy

  Int_t size()const {return (Int_t)mSize;} // Return the size (number of strips) of the cluster

  StEEmcTowerVec_t &towers() {return mMatchedTowers;} // Return list of towers matching this cluster
  const StEEmcTowerVec_t &towers() const {return mMatchedTowers;} // Return list of towers matching this cluster
  Int_t numberOfMatchedTowers() const {return (Int_t)mMatchedTowers.size();} // Return number of towers matching this cluster
  StEEmcTower &tower(Int_t t) {return mMatchedTowers[t];} // Return a specific tower matching this SMD cluster
  const StEEmcTower &tower(Int_t t) const {return mMatchedTowers[t];} // Return a specific tower matching this SMD cluster

  Bool_t operator<( const StEEmcSmdCluster &other ) const { return this->energy() < other.energy(); }
  Bool_t operator>( const StEEmcSmdCluster &other ) const { return this->energy() > other.energy(); }

  Int_t numberOfStrips()const{ return (Int_t)mStrips.size(); } // Returns the number of SMD strips in the cluster

  StEEmcStrip &strip(Int_t s){ return mStrips[s]; }
  const StEEmcStrip &strip(Int_t s)const{ return mStrips[s]; }

  Float_t weight(Int_t s)const{ return mWeights[s]; } 

  StEEmcStrip &seed(){ assert(mStrips.size()>0); return mStrips[0]; } 
  const StEEmcStrip &seed() const { assert(mStrips.size()>0); return mStrips[0]; } 

  Int_t plane() const { return mPlane; } // Return the plane of the cluster
  Int_t sector() const { return mSector; } // Return the sector of the cluster
  void sector( Int_t s ) { mSector=s; } // Set the sector
  void plane( Int_t p ){ mPlane = p; } // Set the plane

  StEmcCluster *stemc();
  void stemc( StEmcCluster *c ){ mEmcCluster = c; }

  void print(Option_t *opts="") const;
  void printLine(Bool_t endline=false) const;

  Int_t next(Int_t direct) const { if(direct>0) return mRight; else if(direct<0) return mLeft; else return -999; }

  void copy( TH1F *h ) const;

protected:
  
  StEEmcStripVec_t mStrips;         
  std::vector<Float_t> mWeights;    

  Int_t mSize;  

  Float_t mSumXW;
  Float_t mSumX2W;

  Float_t mMean;
  Float_t mSigma;
  Int_t   mPlane;
  Int_t   mSector;

  StEEmcTowerVec_t mMatchedTowers;  

  Int_t mLeft;
  Int_t mRight;

  StEmcCluster *mEmcCluster;

  ClassDef(StEEmcSmdCluster,1);
};
//inline Bool_t           StEEmcSmdCluster::operator>(StEEmcSmdCluster &other){ return (energy() > other.energy()); }


inline Bool_t inner( const StEEmcSmdCluster &me, const StEEmcSmdCluster &you) { return me.mean()<you.mean(); } 
inline Bool_t outer( const StEEmcSmdCluster &me, const StEEmcSmdCluster &you) { return me.mean()>you.mean(); }
inline Bool_t Energy( const StEEmcSmdCluster &me, const StEEmcSmdCluster &you) { return me.energy()>you.energy(); }
inline Bool_t Key( const StEEmcSmdCluster &me, const StEEmcSmdCluster &you) { return me.key()<you.key(); }

typedef std::vector<StEEmcSmdCluster> StEEmcSmdClusterVec_t;

std::ostream& operator<<(std::ostream &out, const StEEmcSmdCluster &c );

#endif