//*-- Author : Dan Magestro
//*-- Created: 06/09/01
//*-- Last modified: 07/11/01

//_HADES_CLASS_DESCRIPTION 
/////////////////////////////////////////////////////////////////////////
//
// HCutParticle - cuts on HParticles
//
// 

//
/////////////////////////////////////////////////////////////////////////
using namespace std;
#include <TRandom.h>
#include <time.h>
#include <iostream> 
#include <iomanip>
#include <math.h>
#include <TMath.h>
#include "phyanadef.h"
#include "hcutparticle.h"
#include "hcut.h"
#include "hparticle.h"
#include "hades.h"
#include "hiterator.h"
#include "hcategory.h"
#include "hlinearcategory.h"
#include "hphysicsconstants.h"
#include "hrecevent.h"
#include "hlocation.h"

 HCutParticle::HCutParticle() : HCut() {
  // Default constructor
  fPartCat = 0;
  iterPart=0;

  reset();
}

 HCutParticle::HCutParticle(const Text_t *name, const Text_t *title)
         : HCut(name, title) {
  fPartCat = 0;
  iterPart = 0;

  reset();
}

 void HCutParticle::setStandardCuts() {
// Not implemented yet
  return;
  }

 void HCutParticle::reset() {
// Disables all cuts by resetting to kFALSE
  kFlagCharge = kFALSE;
  kFlagP = kFALSE;
  kFlagPt = kFALSE;
  kFlagTheta = kFALSE;
  kFlagPhi = kFALSE;
  kFlagRapidity = kFALSE;
  return;
  }

 void HCutParticle::printCutList() {
// Prints a table of enabled cuts to stdio
  printf("n--------------------------------------------n");
  printf("               HCutParticle cuts              n");
  printf("               -----------------              n");

  if(kFlagCharge)
     printf("fCharge = %5i   n",fCharge);
  if(kFlagP)
     printf("fMinP          = %5.1f   fMaxP          = %5.1fn",fMinP,fMaxP);
  if(kFlagPt)
     printf("fMinPt         = %5.1f   fMaxPt         = %5.1fn",fMinPt,fMaxPt);
  if(kFlagRapidity)
     printf("fMinRapidity   = %5.1f   fMaxRapidity   = %5.1fn",fMinRapidity,fMaxRapidity);
  if(kFlagTheta)
     printf("fMinTheta      = %5.1f   fMaxTheta      = %5.1fn",fMinTheta,fMaxTheta);
  if(kFlagPhi)
     printf("fMinPhi        = %5.1f   fMaxPhi        = %5.1fn",fMinPhi,fMaxPhi);

  printf("--------------------------------------------n\n");
  return;
  }

 Int_t HCutParticle::check(HParticle *part) {
// Checks one HParticle and returns 1 if particle passes cuts
  Float_t rad2deg = 57.29578;
  fNCheck++;

  if( kFlagCharge && (part->getCharge() != fCharge) ) {
    if(fVerbose) printf("Particle not accepted (Charge cut).n");
    fnCharge++;
    return kFALSE;
  }

  if( kFlagP && ( part->P() > fMaxP || part->P() < fMinP) ) {
    if(fVerbose) printf("Particle not accepted (P cut).n");
    fnP++;
    return kFALSE;
  }

  if( kFlagPt && ( part->Pt() > fMaxP || part->Pt() < fMinP ) ) {
    if(fVerbose) printf("Particle not accepted (Pt cut).n");
    fnPt++;
    return kFALSE;
  }

//  if( kFlagCMRapidity && () ) {
//    if(fVerbose) printf("Particle not accepted (CMRapidity cut).n");
//    fnCMRapidity++;
//    return kFALSE;
//  }

  if( kFlagTheta && ( part->Theta()*rad2deg > fMaxTheta || part->Theta()*rad2deg < fMinTheta ) ) {
    if(fVerbose) printf("Particle not accepted (Theta cut).n");
    fnTheta++;
    return kFALSE;
  }

  if( kFlagPhi && ( part->Phi()*rad2deg > fMaxPhi || part->Phi()*rad2deg < fMinPhi ) ) {
    if(fVerbose) printf("Particle not accepted (Phi cut).n");
    fnPhi++;
    return kFALSE;
  }

  fNGood++;
  return kTRUE;
}

 TObjArray* HCutParticle::getAcceptedParticles(TObjArray *partList) {
// Not implemented yet
  partList = 0;
  return partList;
}

ClassImp(HCutParticle)