StFmsPi0Filter.cxx

// @(#)root/eg:$Id: StFmsPi0Filter.cxx,v 1.4 2011/01/07 16:57:18 jwebb Exp $
// Author: Victor Perev  17/03/2009

//______________________________________________________________________________
#include "stdlib.h"
#include <stdio.h>
#include "math.h"
#include "StFmsPi0Filter.h"
#include "StGenParticle.h"


static StFmsPi0Filter fmsFilter;



StFmsPi0Filter::StFmsPi0Filter(): StMCFilter("fmsfilt")
{
  mEtaMin=2.8; mEtaMax=4.1; mPtMin=1.0;
  cout << "StFmsPi0Filter::StFmsPi0Filter   Setting Default parameters"<<endl;
  cout << "StFmsPi0Filter::StFmsPi0Filter   EtaMin="<<mEtaMin<<endl;
  cout << "StFmsPi0Filter::StFmsPi0Filter   EtaMax="<<mEtaMax<<endl;
  cout << "StFmsPi0Filter::StFmsPi0Filter   PtMin= "<<mPtMin<<endl;
};

//______________________________________________________________________________
int StFmsPi0Filter::RejectGT(const StGenParticleMaster &ptl) const
{
  //ptl.Print("************** In RejectGT ************** ");
  const StGenParticle *tk=0;
  int n = ptl.Size();
  int ntk=0;
  for (int i=0;i<n;i++) {
    tk = ptl(i); if (!tk)       continue;
    if (tk->GetPdgCode()==111 &&
        tk->Eta() > mEtaMin   &&
        tk->Eta() < mEtaMax   &&
        tk->Pt()  > mPtMin    ){
      double p[4];
      tk->Momentum(p);
#ifdef __DEBUG__
      printf("PI0!!! %3d St=%3d Pid=%5d Gid=%3d P=%7.2f %7.2f %7.2f %7.2f pt=%7.2f eta=%7.2f\n",
             i,tk->GetStatusCode(),tk->GetPdgCode(),tk->GetGeaCode(),
             p[0],p[1],p[2],p[3],tk->Pt(),tk->Eta()
             );
#endif
      ntk++;
    }
  }  
  if (ntk==0) {return 1;}

  printf("Accepted/Total EG=%10d/%10d  GT=%10d/%10d  GE=%10d/%10d\n",
           GetNTotEG()-1-GetNRejEG(),GetNTotEG(),
           GetNTotGT()-GetNRejGT(),GetNTotGT(),
           GetNTotGE()-GetNRejGE(),GetNTotGE());
  return 0;
}
//______________________________________________________________________________
int StFmsPi0Filter::RejectEG(const StGenParticleMaster &ptl) const
{
  //ptl.Print("************** In RejectEG ************** ");
  return 0;
}
//______________________________________________________________________________
int StFmsPi0Filter::RejectGE(const StGenParticleMaster &ptl) const
{
  //ptl.Print("************** In RejectGE ************** ");
  return 0;
}
//______________________________________________________________________________
void StFmsPi0Filter::ChangeConfig(string attr, float val)
{
  if (attr == "EtaMin"){
      cout << "StFmsPi0Filter::ChangeConfig()   Setting EtaMin to " << val << endl;
      mEtaMin = val;
  } else if (attr == "EtaMax"){
      cout << "StFmsPi0Filter::ChangeConfig()   Setting EtaMax to " << val << endl;
      mEtaMax = val;
  } else if (attr == "PtMin"){
      cout << "StFmsPi0Filter::ChangeConfig()   Setting PtMin to " << val << endl;
      mPtMin = val;
  } else {
      cout << "StFmsPi0Filter::ChangeConfig()   Unknown parameter " << attr << endl;
  }
}