StPmdDiscriminatorMaker.h

/******************************************************************
 *
 * $Id: StPmdDiscriminatorMaker.h,v 1.4 2004/10/30 00:08:06 subhasis Exp $
 *
 * Author: Subhasis Chattopadhyay
 *
 ******************************************************************
 *
 * Description: This routine describes methods for photon/hadron
 * discrimination. 
 ******************************************************************
 *
 * $Log: StPmdDiscriminatorMaker.h,v $
 * Revision 1.4  2004/10/30 00:08:06  subhasis
 * TranFlag added and set to 0 in ctor
 *
 * Revision 1.3  2003/09/10 19:47:27  perev
 * ansi corrs
 *
 * Revision 1.2  2003/05/29 13:12:51  subhasis
 * several changes to include NN
 *
 ******************************************************************/

#ifndef STAR_StPmdDiscriminatorMaker
#define STAR_StPmdDiscriminatorMaker
#ifndef StMaker_H
#include "StMaker.h"
#endif

//For CC5 compatibility
#include <vector>
#if !defined(ST_NO_NAMESPACES)
using std::vector;
#endif

#ifdef ST_NO_TEMPLATE_DEF_ARGS
// Syntax currently required by Solaris compiler
#define StVector(T) vector<T, allocator<T> >
#else
#define StVector(T) vector<T>
#endif
// *************************
#include <TH2.h>
#include <TH1.h>
#include <TCanvas.h>

class StPmdCollection;
class StPhmdCollection;
class StPmdDetector;
class StPmdCluster;
class StPhmdClusterCollection;
class StPhmdCluster;
class StNNCluster;

//typedef StVector(StPhmdCluster*) StPmdCl;
typedef StVector(StNNCluster*) StPmdCl;

class StPmdDiscriminatorMaker: public StMaker{

  private:
  Int_t m_photonlike;
          Float_t mEdepThreshold;
          Float_t mDeltaEta;
          Float_t mDeltaPhi;
          Int_t mApplyFlagNN;
          Int_t mTrainFlag;
  protected:
  TH1F *mEdepPmd;    
  TH1F *mEtaPmd;    
  TH1F *mPhiPmd;    
  TH1F *mEtadiff;    
  TH1F *mPhidiff;    
  TH1F *mMCPID;    
  TH2F *mEtaPhi;    
  TH1F *mClusterPID;    
  TH1F *mClusterEdepPID;    
  TH2F *mEdepVsPID; 
  TH1F *mcpvmatched; 
  
   public:
  StPmdDiscriminatorMaker(const char *name="StPmdDiscriminator"); 
  ~StPmdDiscriminatorMaker();

  virtual Int_t Init();  
  virtual Int_t Make();
  virtual Int_t Finish();
  void getEdepThreshold(Float_t); 
  void SetDeltaEta(Float_t);      
  void SetDeltaPhi(Float_t);      
  void SetApplyFlag(Int_t);      
  void SetTrainFlag(Int_t);      
  void bookHistograms();          
  void fillHistograms(StPmdDetector*, StPmdDetector*);
  void Matching(StPmdDetector*, StPmdDetector*);
  Int_t PrepareInputforNN(StPmdDetector*,StPmdDetector*,StPhmdClusterCollection*,StPhmdClusterCollection*);
  //void Matching();
  void fillStEvent(StPmdCollection*,StPhmdCollection*);
  void  Browse(TBrowser* b);
//NN hist outs (made public to be accessed from StPmdNeunet
  TH1F *mNNoutput; 

  ClassDef(StPmdDiscriminatorMaker,0) 
};

inline void StPmdDiscriminatorMaker::getEdepThreshold(Float_t de){mEdepThreshold=de;}
inline void StPmdDiscriminatorMaker::SetDeltaEta(Float_t deta){mDeltaEta=deta;}
inline void StPmdDiscriminatorMaker::SetDeltaPhi(Float_t dphi){mDeltaPhi=dphi;}
inline void StPmdDiscriminatorMaker::SetApplyFlag(Int_t flag){mApplyFlagNN=flag;}
inline void StPmdDiscriminatorMaker::SetTrainFlag(Int_t flag){mTrainFlag=flag;}
#endif