StDijetFilter.h

#ifndef STAR_StDijetFilter
#define STAR_StDijetFilter

#include <vector>
#include <string>
#include "StMCFilter.h"

class StGenParticleMaster;
class StGenParticle;

class JetFourVec
{

 private:
  float px; 
  float py; 
  float pz; 
  float en; 
  int code; 
 
 public:
  JetFourVec(); 
  JetFourVec(JetFourVec*); 
  JetFourVec(StGenParticle*); 
  ~JetFourVec(){} 

  JetFourVec operator +(JetFourVec); 
  bool operator ==(JetFourVec); 

  void setCode(int x){code = x;}; 
  void setPx(float x){px = x;} 
  void setPy(float x){py = x;} 
  void setPz(float x){pz = x;} 
  void setEn(float x){en = x;} 
  void setPxPyPzEn(float,float,float,float); 
  void setPtEtaPhiM(float,float,float,float); 
  void setPtEtaPhiE(float,float,float,float); 

  int getCode(){return code;} 
  float getPx(){return px;} 
  float getPy(){return py;} 
  float getPz(){return pz;} 
  float getEn(){return en;} 

  float Pt(); 
  float Eta(); 
  float Phi(); 
  float Theta(); 
  float M(); 
  float P(); 
};


class StDijetFilter : public StMCFilter
{
 private:

  float mRBTOW; 
  float mRcone; 
  float mSeed; 
  float mAssoc; 
  float mSplitfraction; 
  bool mAddmidpoints; 
  bool mStablemidpoints; 
  bool mSplitmerge; 
  float mParticleEtaRange; 
  float mJetEtaHigh; 
  float mJetEtaLow; 
  float mDPhi; 
  float mDEta; 
  float mPtLow; 
  float mPtHigh; 
  float mMinJetPt; 

  float mRecohadron; 
  float mRecolepton; 
  int *nEvents;
  double mVertex[3]; 

  float dR(StGenParticle*,StGenParticle*) const; 
  float dR(StGenParticle*,JetFourVec*) const; 
  float dR(JetFourVec*,StGenParticle*) const; 
  float dR(JetFourVec*,JetFourVec*) const; 

  JetFourVec* combineTracks(std::vector<JetFourVec*>) const; 
  JetFourVec* recoJet(std::vector<JetFourVec*>, double*) const; 
  std::vector<JetFourVec*> EtOrderedList(std::vector<JetFourVec*>) const; 
  std::vector< std::vector<JetFourVec*> > EtOrderedList(std::vector< std::vector<JetFourVec*> >) const; 
  std::vector< std::vector<JetFourVec*> > RemoveDuplicates(std::vector< std::vector<JetFourVec*> >) const; 

  float overlapEnergy(std::vector<JetFourVec*>,std::vector<JetFourVec*>) const;
  std::vector< std::vector<JetFourVec*> > doSplitMerge(std::vector< std::vector<JetFourVec*> >) const; 

  std::vector<JetFourVec*> merge(std::vector<JetFourVec*>,std::vector<JetFourVec*>) const; 
  void split(std::vector<JetFourVec*> &v1,std::vector<JetFourVec*> &v2) const; 

  std::vector<JetFourVec*> addMidPoints(std::vector<JetFourVec*>) const; 

  void readConfig(); 

 public:
  StDijetFilter(); 
  virtual ~StDijetFilter();

  int RejectEG(const StGenParticleMaster &ptl) const;

  int RejectGT(const StGenParticleMaster &ptl) const;

  int RejectGE(const StGenParticleMaster &ptl) const;

  void parseConfig(std::string, float); 
};

#endif