[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEcuts.cxx

/**************************************************************************
* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
*                                                                        *
* Author: The ALICE Off-line Project.                                    *
* Contributors are mentioned in the code where appropriate.              *
*                                                                        *
* Permission to use, copy, modify and distribute this software and its   *
* documentation strictly for non-commercial purposes is hereby granted   *
* without fee, provided that the above copyright notice appears in all   *
* copies and that both the copyright notice and this permission notice   *
* appear in the supporting documentation. The authors make no claims     *
* about the suitability of this software for any purpose. It is          *
* provided "as is" without express or implied warranty.                  *
**************************************************************************/
/************************************************************************
 *                                                                      *
 * Cut menagement class implemented by the                              *
 * ALICE Heavy Flavour Electron Group                                   *
 *                                                                      *
 * Authors:                                                             *
 *   Raphaelle Bailhache <R.Bailhache@gsi.de>                           *
 *   Markus Fasel <M.Fasel@gsi.de>                                      *
 *   Markus Heide <mheide@uni-muenster.de>                              *
 *   Matus Kalisky <m.kalisky@uni-muenster.de>                          *
 *                                                                      *
 ************************************************************************/
#include <TClass.h>
#include <TList.h>
#include <TObjArray.h>
#include <TString.h>

#include "AliCFAcceptanceCuts.h"
#include "AliCFCutBase.h"
#include "AliCFManager.h"
#include "AliCFParticleGenCuts.h"
#include "AliCFTrackIsPrimaryCuts.h"
#include "AliCFTrackKineCuts.h"
#include "AliCFTrackQualityCuts.h"
#include "AliESDtrack.h"
#include "AliMCParticle.h"

#include "AliHFEcuts.h"

ClassImp(AliHFEcuts)

//__________________________________________________________________
AliHFEcuts::AliHFEcuts():
  fRequirements(0),
  fMinClustersTPC(0),
  fMinTrackletsTRD(0),
  fCutITSPixel(0),
  fMaxChi2clusterTPC(0.),
  fMinClusterRatioTPC(0.),
  fSigmaToVtx(0.),
  fHistQA(0x0),
  fCutList(0x0),
  fDebugLevel(0)
{
  //
  // Default Constructor
  //
  memset(fProdVtx, 0, sizeof(Double_t) * 4);
  memset(fDCAtoVtx, 0, sizeof(Double_t) * 2);
  memset(fPtRange, 0, sizeof(Double_t) * 2);
  fCutList = new TObjArray();
  fCutList->SetOwner();
}

//__________________________________________________________________
AliHFEcuts::AliHFEcuts(const AliHFEcuts &c):
  TObject(c),
  fRequirements(c.fRequirements),
  fMinClustersTPC(c.fMinClustersTPC),
  fMinTrackletsTRD(c.fMinTrackletsTRD),
  fCutITSPixel(c.fCutITSPixel),
  fMaxChi2clusterTPC(c.fMaxChi2clusterTPC),
  fMinClusterRatioTPC(c.fMinClusterRatioTPC),
  fSigmaToVtx(c.fSigmaToVtx),
  fHistQA(0x0),
  fCutList(0x0),
  fDebugLevel(0)
{
  //
  // Copy Constructor
  //
  memcpy(fProdVtx, c.fProdVtx, sizeof(Double_t) * 4);
  memcpy(fDCAtoVtx, c.fDCAtoVtx, sizeof(Double_t) * 4);
  memcpy(fPtRange, c.fPtRange, sizeof(Double_t) *2);
  fCutList = dynamic_cast<TObjArray *>(c.fCutList->Clone());
  fCutList->SetOwner();
}

//__________________________________________________________________
AliHFEcuts::~AliHFEcuts(){
  //
  // Destruktor
  //
  if(fCutList){
    fCutList->Delete();
    delete fCutList;
  }
  fCutList = 0x0;
  if(fHistQA) fHistQA->Clear();
  delete fHistQA;
}

//__________________________________________________________________
void AliHFEcuts::Initialize(AliCFManager *cfm){

  // Call all the setters for the cuts
  SetParticleGenCutList();
  SetAcceptanceCutList();
  SetRecKineTPCCutList();
  SetRecKineITSCutList();
  SetRecPrimaryCutList();
  SetHFElectronITSCuts();
  SetHFElectronTPCCuts();
  SetHFElectronTRDCuts();

  
  // Connect the cuts
  cfm->SetParticleCutsList(kStepMCGenerated, dynamic_cast<TObjArray *>(fCutList->FindObject("fPartGenCuts")));
  cfm->SetParticleCutsList(kStepMCInAcceptance, dynamic_cast<TObjArray *>(fCutList->FindObject("fPartAccCuts")));
  cfm->SetParticleCutsList(kStepRecKineTPC, dynamic_cast<TObjArray *>(fCutList->FindObject("fPartRecKineTPCCuts")));
  cfm->SetParticleCutsList(kStepRecKineITS, dynamic_cast<TObjArray *>(fCutList->FindObject("fPartRecKineITSCuts")));
  cfm->SetParticleCutsList(kStepRecPrim, dynamic_cast<TObjArray *>(fCutList->FindObject("fPartPrimCuts")));
  cfm->SetParticleCutsList(kStepHFEcutsITS, dynamic_cast<TObjArray *>(fCutList->FindObject("fPartHFECutsITS")));
  cfm->SetParticleCutsList(kStepHFEcutsTPC, dynamic_cast<TObjArray *>(fCutList->FindObject("fPartHFECutsTPC")));
  cfm->SetParticleCutsList(kStepHFEcutsTRD, dynamic_cast<TObjArray *>(fCutList->FindObject("fPartHFECutsTRD")));

}

//__________________________________________________________________
void AliHFEcuts::Initialize(){
  // Call all the setters for the cuts
  SetParticleGenCutList();
  SetAcceptanceCutList();
  SetRecKineTPCCutList();
  SetRecKineITSCutList();
  SetRecPrimaryCutList();
  SetHFElectronITSCuts();
  SetHFElectronTPCCuts();
  SetHFElectronTRDCuts();

}

//__________________________________________________________________
void AliHFEcuts::SetParticleGenCutList(){
  //
  // Initialize Particle Cuts for Monte Carlo Tracks
  // Production Vertex: < 1cm (Beampipe)
  // Particle Species: Electrons
  // Eta: < 0.9 (TRD-TOF acceptance)
  //
  AliCFParticleGenCuts *genCuts = new AliCFParticleGenCuts("fCutsGenMC", "Particle Generation Cuts");
  genCuts->SetRequireIsCharged();
  if(IsRequirePrimary()) genCuts->SetRequireIsPrimary();
  if(IsRequireProdVertex()){
    genCuts->SetProdVtxRangeX(fProdVtx[0], fProdVtx[1]);
    genCuts->SetProdVtxRangeY(fProdVtx[2], fProdVtx[3]);
  }
  genCuts->SetRequirePdgCode(11/*, kTRUE*/);
  
  AliCFTrackKineCuts *kineMCcuts = new AliCFTrackKineCuts("fCutsKineMC","MC Kine Cuts");
  kineMCcuts->SetPtRange(fPtRange[0], fPtRange[1]);
  kineMCcuts->SetEtaRange(-0.9, 0.9);

  if(IsInDebugMode()){
    genCuts->SetQAOn(fHistQA);
    kineMCcuts->SetQAOn(fHistQA);
  }

  TObjArray *mcCuts = new TObjArray;
  mcCuts->SetName("fPartGenCuts");
  mcCuts->AddLast(genCuts);
  mcCuts->AddLast(kineMCcuts);
  fCutList->AddLast(mcCuts);
}

//__________________________________________________________________
void AliHFEcuts::SetAcceptanceCutList(){
  //
  // Initialize Particle (Monte Carlo) acceptance cuts
  // Min. Required Hist for Detectors:
  //          ITS [3]
  //          TPC [2]
  //          TRD [12]
  //          TOF [0]
  //
  AliCFAcceptanceCuts *accCuts = new AliCFAcceptanceCuts("fCutsAccMC", "MC Acceptance Cuts");
  accCuts->SetMinNHitITS(3);
  accCuts->SetMinNHitTPC(2);
  accCuts->SetMinNHitTRD(12);
  if(IsInDebugMode()) accCuts->SetQAOn(fHistQA);
  
  TObjArray *partAccCuts = new TObjArray();
  partAccCuts->SetName("fPartAccCuts");
  partAccCuts->AddLast(accCuts);
  fCutList->AddLast(partAccCuts);
}

//__________________________________________________________________
void AliHFEcuts::SetRecKineTPCCutList(){
  //
  // Track Kinematics and Quality cuts (Based on the Standard cuts from PWG0)
  //
  // Variances:
  //  y: 2
  //  z: 2
  //  sin(phi): 0.5
  //  tan(theta): 0.5
  //  1/pt: 2
  // Min. Number of Clusters:
  //  TPC: 50
  // RefitRequired:
  //  TPC
  // Chi2 per TPC cluster: 3.5
  //
  // Kinematics:
  //  Momentum Range: 100MeV - 20GeV
  //  Eta: < 0.9 (TRD-TOF acceptance)
  //
  AliCFTrackQualityCuts *trackQuality = new AliCFTrackQualityCuts("fCutsQualityRec","REC Track Quality Cuts");
  trackQuality->SetMinNClusterTPC(fMinClustersTPC);
  trackQuality->SetMaxChi2PerClusterTPC(fMaxChi2clusterTPC);
  trackQuality->SetStatus(AliESDtrack::kTPCrefit);
  trackQuality->SetMaxCovDiagonalElements(2., 2., 0.5, 0.5, 2); 

  AliCFTrackKineCuts *kineCuts = new AliCFTrackKineCuts("fCutsKineRec", "REC Kine Cuts");
  kineCuts->SetPtRange(fPtRange[0], fPtRange[1]);
  kineCuts->SetEtaRange(-0.9, 0.9);
  
  if(IsInDebugMode()){
    trackQuality->SetQAOn(fHistQA);
    kineCuts->SetQAOn(fHistQA);
  }
  
  TObjArray *recCuts = new TObjArray;
  recCuts->SetName("fPartRecKineTPCCuts");
  recCuts->AddLast(trackQuality);
  recCuts->AddLast(kineCuts);
  fCutList->AddLast(recCuts);
}

//__________________________________________________________________
void AliHFEcuts::SetRecKineITSCutList(){
  //
  // Track Kinematics and Quality cuts (Based on the Standard cuts from PWG0)
  //
  // RefitRequired:
  //  ITS
  //
  AliCFTrackQualityCuts *trackQuality = new AliCFTrackQualityCuts("fCutsQualityRec","REC Track Quality Cuts");
  trackQuality->SetStatus(AliESDtrack::kITSrefit);
  
  if(IsInDebugMode()){
    trackQuality->SetQAOn(fHistQA);
  }
  
  TObjArray *recCuts = new TObjArray;
  recCuts->SetName("fPartRecKineITSCuts");
  recCuts->AddLast(trackQuality);
  fCutList->AddLast(recCuts);
}

//__________________________________________________________________
void AliHFEcuts::SetRecPrimaryCutList(){
  //
  // Primary cuts (based on standard cuts from PWG0):
  //  DCA to Vertex: 
  //    XY: 3. cm
  //    Z:  10. cm
  //  No Kink daughters
  //
  AliCFTrackIsPrimaryCuts *primaryCut = new AliCFTrackIsPrimaryCuts("fCutsPrimaryCuts", "REC Primary Cuts");
  if(IsRequireDCAToVertex()){
    primaryCut->SetDCAToVertex2D(kTRUE);
    primaryCut->SetMaxDCAToVertexXY(fDCAtoVtx[0]);
    primaryCut->SetMaxDCAToVertexZ(fDCAtoVtx[1]);
  }
  if(IsRequireSigmaToVertex()){
    primaryCut->SetRequireSigmaToVertex(kTRUE);
    primaryCut->SetMaxNSigmaToVertex(fSigmaToVtx);
  }
  primaryCut->SetAcceptKinkDaughters(kFALSE);
  if(IsInDebugMode()) primaryCut->SetQAOn(fHistQA);
  
  TObjArray *primCuts = new TObjArray;
  primCuts->SetName("fPartPrimCuts");
  primCuts->AddLast(primaryCut);
  fCutList->AddLast(primCuts);
}

//__________________________________________________________________
void AliHFEcuts::SetHFElectronITSCuts(){
  //
  // Special Cuts introduced by the HFElectron Group: ITS
  //
  AliHFEextraCuts *hfecuts = new AliHFEextraCuts("fCutsHFElectronGroupPixels","Extra cuts from the HFE group");
  if(IsRequireITSpixel()){
    hfecuts->SetRequireITSpixel(AliHFEextraCuts::ITSPixel_t(fCutITSPixel));
  }
  if(IsRequireDCAToVertex()){
    hfecuts->SetMaxImpactParamR(fDCAtoVtx[0]);
    hfecuts->SetMaxImpactParamZ(fDCAtoVtx[1]);
  }
  
  if(IsInDebugMode()) hfecuts->SetQAOn(fHistQA);
  
  TObjArray *hfeCuts = new TObjArray;
  hfeCuts->SetName("fPartHFECutsITS");
  hfeCuts->AddLast(hfecuts);
  fCutList->AddLast(hfeCuts);
}

//__________________________________________________________________
void AliHFEcuts::SetHFElectronTPCCuts(){
  //
  // Special Cuts introduced by the HFElectron Group: TPC
  //
  AliHFEextraCuts *hfecuts = new AliHFEextraCuts("fCutsHFElectronGroupTPC","Extra cuts from the HFE group");
  if(fMinClusterRatioTPC > 0.) hfecuts->SetClusterRatioTPC(fMinClusterRatioTPC);
  if(IsInDebugMode()) hfecuts->SetQAOn(fHistQA);
  
  TObjArray *hfeCuts = new TObjArray;
  hfeCuts->SetName("fPartHFECutsTPC");
  hfeCuts->AddLast(hfecuts);
  fCutList->AddLast(hfeCuts);
}

//__________________________________________________________________
void AliHFEcuts::SetHFElectronTRDCuts(){
  //
  // Special Cuts introduced by the HFElectron Group: TRD
  //
  AliHFEextraCuts *hfecuts = new AliHFEextraCuts("fCutsHFElectronGroupTRD","Extra cuts from the HFE group");
  if(fMinTrackletsTRD > 0.) hfecuts->SetMinTrackletsTRD(fMinTrackletsTRD);
  if(IsInDebugMode()) hfecuts->SetQAOn(fHistQA);
  
  TObjArray *hfeCuts = new TObjArray;
  hfeCuts->SetName("fPartHFECutsTRD");
  hfeCuts->AddLast(hfecuts);
  fCutList->AddLast(hfeCuts);
}

//__________________________________________________________________
void AliHFEcuts::SetDebugMode(){ 
  //
  // Switch on QA
  //
  SetBit(kDebugMode, kTRUE); 
  fHistQA = new TList;
  fHistQA->SetName("CutQAhistograms");
  fHistQA->SetOwner(kFALSE);
}

//__________________________________________________________________
Bool_t AliHFEcuts::CheckParticleCuts(CutStep_t step, TObject *o){
  //
  // Checks the cuts without using the correction framework manager
  // 
  TString stepnames[kNcutSteps] = {"fPartGenCuts", "fPartAccCuts", "fPartRecCuts", "fPartPrimCuts", "fPartHFECuts"};
  TObjArray *cuts = dynamic_cast<TObjArray *>(fCutList->FindObject(stepnames[step].Data()));
  if(!cuts) return kTRUE;
  TIterator *it = cuts->MakeIterator();
  AliCFCutBase *mycut;
  Bool_t status = kTRUE;
  while((mycut = dynamic_cast<AliCFCutBase *>(it->Next()))){
    status &= mycut->IsSelected(o);
  }
  delete it;
  return status;
}
Commit	Line	Data
809a4336	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15	/************************************************************************
	16	* *
	17	* Cut menagement class implemented by the *
	18	* ALICE Heavy Flavour Electron Group *
	19	* *
	20	* Authors: *
dbe3abbe	21	* Raphaelle Bailhache <R.Bailhache@gsi.de> *
809a4336	22	* Markus Fasel <M.Fasel@gsi.de> *
	23	* Markus Heide <mheide@uni-muenster.de> *
	24	* Matus Kalisky <m.kalisky@uni-muenster.de> *
	25	* *
	26	************************************************************************/
	27	#include <TClass.h>
	28	#include <TList.h>
	29	#include <TObjArray.h>
	30	#include <TString.h>
	31
	32	#include "AliCFAcceptanceCuts.h"
	33	#include "AliCFCutBase.h"
	34	#include "AliCFManager.h"
	35	#include "AliCFParticleGenCuts.h"
	36	#include "AliCFTrackIsPrimaryCuts.h"
	37	#include "AliCFTrackKineCuts.h"
	38	#include "AliCFTrackQualityCuts.h"
	39	#include "AliESDtrack.h"
	40	#include "AliMCParticle.h"
	41
	42	#include "AliHFEcuts.h"
	43
	44	ClassImp(AliHFEcuts)
	45
809a4336	46	//__________________________________________________________________
	47	AliHFEcuts::AliHFEcuts():
	48	fRequirements(0),
	49	fMinClustersTPC(0),
	50	fMinTrackletsTRD(0),
	51	fCutITSPixel(0),
	52	fMaxChi2clusterTPC(0.),
	53	fMinClusterRatioTPC(0.),
	54	fSigmaToVtx(0.),
809a4336	55	fHistQA(0x0),
dbe3abbe	56	fCutList(0x0),
dbe3abbe	57	fDebugLevel(0)
809a4336	58	{
	59	//
	60	// Default Constructor
	61	//
	62	memset(fProdVtx, 0, sizeof(Double_t) * 4);
	63	memset(fDCAtoVtx, 0, sizeof(Double_t) * 2);
	64	memset(fPtRange, 0, sizeof(Double_t) * 2);
	65	fCutList = new TObjArray();
	66	fCutList->SetOwner();
	67	}
	68
	69	//__________________________________________________________________
	70	AliHFEcuts::AliHFEcuts(const AliHFEcuts &c):
	71	TObject(c),
	72	fRequirements(c.fRequirements),
	73	fMinClustersTPC(c.fMinClustersTPC),
	74	fMinTrackletsTRD(c.fMinTrackletsTRD),
	75	fCutITSPixel(c.fCutITSPixel),
	76	fMaxChi2clusterTPC(c.fMaxChi2clusterTPC),
	77	fMinClusterRatioTPC(c.fMinClusterRatioTPC),
	78	fSigmaToVtx(c.fSigmaToVtx),
809a4336	79	fHistQA(0x0),
dbe3abbe	80	fCutList(0x0),
dbe3abbe	81	fDebugLevel(0)
809a4336	82	{
	83	//
	84	// Copy Constructor
	85	//
	86	memcpy(fProdVtx, c.fProdVtx, sizeof(Double_t) * 4);
	87	memcpy(fDCAtoVtx, c.fDCAtoVtx, sizeof(Double_t) * 4);
	88	memcpy(fPtRange, c.fPtRange, sizeof(Double_t) *2);
	89	fCutList = dynamic_cast<TObjArray *>(c.fCutList->Clone());
	90	fCutList->SetOwner();
	91	}
	92
	93	//__________________________________________________________________
	94	AliHFEcuts::~AliHFEcuts(){
	95	//
	96	// Destruktor
	97	//
	98	if(fCutList){
	99	fCutList->Delete();
	100	delete fCutList;
	101	}
	102	fCutList = 0x0;
	103	if(fHistQA) fHistQA->Clear();
	104	delete fHistQA;
	105	}
	106
	107	//__________________________________________________________________
	108	void AliHFEcuts::Initialize(AliCFManager *cfm){
dbe3abbe	109
809a4336	110	// Call all the setters for the cuts
	111	SetParticleGenCutList();
	112	SetAcceptanceCutList();
dbe3abbe	113	SetRecKineTPCCutList();
dbe3abbe	114	SetRecKineITSCutList();
809a4336	115	SetRecPrimaryCutList();
dbe3abbe	116	SetHFElectronITSCuts();
	117	SetHFElectronTPCCuts();
	118	SetHFElectronTRDCuts();
	119
809a4336	120
	121	// Connect the cuts
	122	cfm->SetParticleCutsList(kStepMCGenerated, dynamic_cast<TObjArray *>(fCutList->FindObject("fPartGenCuts")));
	123	cfm->SetParticleCutsList(kStepMCInAcceptance, dynamic_cast<TObjArray *>(fCutList->FindObject("fPartAccCuts")));
dbe3abbe	124	cfm->SetParticleCutsList(kStepRecKineTPC, dynamic_cast<TObjArray *>(fCutList->FindObject("fPartRecKineTPCCuts")));
dbe3abbe	125	cfm->SetParticleCutsList(kStepRecKineITS, dynamic_cast<TObjArray *>(fCutList->FindObject("fPartRecKineITSCuts")));
809a4336	126	cfm->SetParticleCutsList(kStepRecPrim, dynamic_cast<TObjArray *>(fCutList->FindObject("fPartPrimCuts")));
dbe3abbe	127	cfm->SetParticleCutsList(kStepHFEcutsITS, dynamic_cast<TObjArray *>(fCutList->FindObject("fPartHFECutsITS")));
	128	cfm->SetParticleCutsList(kStepHFEcutsTPC, dynamic_cast<TObjArray *>(fCutList->FindObject("fPartHFECutsTPC")));
	129	cfm->SetParticleCutsList(kStepHFEcutsTRD, dynamic_cast<TObjArray *>(fCutList->FindObject("fPartHFECutsTRD")));
	130
809a4336	131	}
	132
	133	//__________________________________________________________________
	134	void AliHFEcuts::Initialize(){
	135	// Call all the setters for the cuts
	136	SetParticleGenCutList();
	137	SetAcceptanceCutList();
dbe3abbe	138	SetRecKineTPCCutList();
dbe3abbe	139	SetRecKineITSCutList();
809a4336	140	SetRecPrimaryCutList();
dbe3abbe	141	SetHFElectronITSCuts();
	142	SetHFElectronTPCCuts();
	143	SetHFElectronTRDCuts();
	144
809a4336	145	}
	146
	147	//__________________________________________________________________
	148	void AliHFEcuts::SetParticleGenCutList(){
	149	//
	150	// Initialize Particle Cuts for Monte Carlo Tracks
	151	// Production Vertex: < 1cm (Beampipe)
	152	// Particle Species: Electrons
	153	// Eta: < 0.9 (TRD-TOF acceptance)
	154	//
	155	AliCFParticleGenCuts *genCuts = new AliCFParticleGenCuts("fCutsGenMC", "Particle Generation Cuts");
	156	genCuts->SetRequireIsCharged();
	157	if(IsRequirePrimary()) genCuts->SetRequireIsPrimary();
	158	if(IsRequireProdVertex()){
	159	genCuts->SetProdVtxRangeX(fProdVtx[0], fProdVtx[1]);
	160	genCuts->SetProdVtxRangeY(fProdVtx[2], fProdVtx[3]);
	161	}
	162	genCuts->SetRequirePdgCode(11/, kTRUE/);
	163
	164	AliCFTrackKineCuts *kineMCcuts = new AliCFTrackKineCuts("fCutsKineMC","MC Kine Cuts");
	165	kineMCcuts->SetPtRange(fPtRange[0], fPtRange[1]);
	166	kineMCcuts->SetEtaRange(-0.9, 0.9);
	167
	168	if(IsInDebugMode()){
	169	genCuts->SetQAOn(fHistQA);
	170	kineMCcuts->SetQAOn(fHistQA);
	171	}
	172
	173	TObjArray *mcCuts = new TObjArray;
	174	mcCuts->SetName("fPartGenCuts");
	175	mcCuts->AddLast(genCuts);
	176	mcCuts->AddLast(kineMCcuts);
	177	fCutList->AddLast(mcCuts);
	178	}
	179
	180	//__________________________________________________________________
	181	void AliHFEcuts::SetAcceptanceCutList(){
	182	//
	183	// Initialize Particle (Monte Carlo) acceptance cuts
	184	// Min. Required Hist for Detectors:
	185	// ITS [3]
	186	// TPC [2]
	187	// TRD [12]
	188	// TOF [0]
	189	//
	190	AliCFAcceptanceCuts *accCuts = new AliCFAcceptanceCuts("fCutsAccMC", "MC Acceptance Cuts");
	191	accCuts->SetMinNHitITS(3);
	192	accCuts->SetMinNHitTPC(2);
	193	accCuts->SetMinNHitTRD(12);
	194	if(IsInDebugMode()) accCuts->SetQAOn(fHistQA);
	195
75d81601	196	TObjArray *partAccCuts = new TObjArray();
	197	partAccCuts->SetName("fPartAccCuts");
	198	partAccCuts->AddLast(accCuts);
	199	fCutList->AddLast(partAccCuts);
809a4336	200	}
	201
	202	//__________________________________________________________________
dbe3abbe	203	void AliHFEcuts::SetRecKineTPCCutList(){
809a4336	204	//
	205	// Track Kinematics and Quality cuts (Based on the Standard cuts from PWG0)
	206	//
	207	// Variances:
	208	// y: 2
	209	// z: 2
	210	// sin(phi): 0.5
	211	// tan(theta): 0.5
	212	// 1/pt: 2
	213	// Min. Number of Clusters:
	214	// TPC: 50
	215	// RefitRequired:
809a4336	216	// TPC
	217	// Chi2 per TPC cluster: 3.5
	218	//
	219	// Kinematics:
	220	// Momentum Range: 100MeV - 20GeV
	221	// Eta: < 0.9 (TRD-TOF acceptance)
	222	//
	223	AliCFTrackQualityCuts *trackQuality = new AliCFTrackQualityCuts("fCutsQualityRec","REC Track Quality Cuts");
	224	trackQuality->SetMinNClusterTPC(fMinClustersTPC);
	225	trackQuality->SetMaxChi2PerClusterTPC(fMaxChi2clusterTPC);
dbe3abbe	226	trackQuality->SetStatus(AliESDtrack::kTPCrefit);
809a4336	227	trackQuality->SetMaxCovDiagonalElements(2., 2., 0.5, 0.5, 2);
	228
	229	AliCFTrackKineCuts *kineCuts = new AliCFTrackKineCuts("fCutsKineRec", "REC Kine Cuts");
	230	kineCuts->SetPtRange(fPtRange[0], fPtRange[1]);
	231	kineCuts->SetEtaRange(-0.9, 0.9);
	232
	233	if(IsInDebugMode()){
	234	trackQuality->SetQAOn(fHistQA);
	235	kineCuts->SetQAOn(fHistQA);
	236	}
	237
	238	TObjArray *recCuts = new TObjArray;
dbe3abbe	239	recCuts->SetName("fPartRecKineTPCCuts");
809a4336	240	recCuts->AddLast(trackQuality);
	241	recCuts->AddLast(kineCuts);
	242	fCutList->AddLast(recCuts);
	243	}
	244
dbe3abbe	245	//__________________________________________________________________
	246	void AliHFEcuts::SetRecKineITSCutList(){
	247	//
	248	// Track Kinematics and Quality cuts (Based on the Standard cuts from PWG0)
	249	//
	250	// RefitRequired:
	251	// ITS
	252	//
	253	AliCFTrackQualityCuts *trackQuality = new AliCFTrackQualityCuts("fCutsQualityRec","REC Track Quality Cuts");
	254	trackQuality->SetStatus(AliESDtrack::kITSrefit);
	255
	256	if(IsInDebugMode()){
	257	trackQuality->SetQAOn(fHistQA);
	258	}
	259
	260	TObjArray *recCuts = new TObjArray;
	261	recCuts->SetName("fPartRecKineITSCuts");
	262	recCuts->AddLast(trackQuality);
	263	fCutList->AddLast(recCuts);
	264	}
	265
809a4336	266	//__________________________________________________________________
	267	void AliHFEcuts::SetRecPrimaryCutList(){
	268	//
	269	// Primary cuts (based on standard cuts from PWG0):
	270	// DCA to Vertex:
	271	// XY: 3. cm
	272	// Z: 10. cm
	273	// No Kink daughters
	274	//
	275	AliCFTrackIsPrimaryCuts *primaryCut = new AliCFTrackIsPrimaryCuts("fCutsPrimaryCuts", "REC Primary Cuts");
	276	if(IsRequireDCAToVertex()){
	277	primaryCut->SetDCAToVertex2D(kTRUE);
	278	primaryCut->SetMaxDCAToVertexXY(fDCAtoVtx[0]);
	279	primaryCut->SetMaxDCAToVertexZ(fDCAtoVtx[1]);
	280	}
	281	if(IsRequireSigmaToVertex()){
	282	primaryCut->SetRequireSigmaToVertex(kTRUE);
	283	primaryCut->SetMaxNSigmaToVertex(fSigmaToVtx);
	284	}
	285	primaryCut->SetAcceptKinkDaughters(kFALSE);
	286	if(IsInDebugMode()) primaryCut->SetQAOn(fHistQA);
	287
	288	TObjArray *primCuts = new TObjArray;
	289	primCuts->SetName("fPartPrimCuts");
	290	primCuts->AddLast(primaryCut);
	291	fCutList->AddLast(primCuts);
	292	}
	293
	294	//__________________________________________________________________
dbe3abbe	295	void AliHFEcuts::SetHFElectronITSCuts(){
809a4336	296	//
dbe3abbe	297	// Special Cuts introduced by the HFElectron Group: ITS
809a4336	298	//
dbe3abbe	299	AliHFEextraCuts *hfecuts = new AliHFEextraCuts("fCutsHFElectronGroupPixels","Extra cuts from the HFE group");
809a4336	300	if(IsRequireITSpixel()){
	301	hfecuts->SetRequireITSpixel(AliHFEextraCuts::ITSPixel_t(fCutITSPixel));
	302	}
dbe3abbe	303	if(IsRequireDCAToVertex()){
ad75027f	304	hfecuts->SetMaxImpactParamR(fDCAtoVtx[0]);
ad75027f	305	hfecuts->SetMaxImpactParamZ(fDCAtoVtx[1]);
dbe3abbe	306	}
dbe3abbe	307
	308	if(IsInDebugMode()) hfecuts->SetQAOn(fHistQA);
	309
	310	TObjArray *hfeCuts = new TObjArray;
	311	hfeCuts->SetName("fPartHFECutsITS");
	312	hfeCuts->AddLast(hfecuts);
	313	fCutList->AddLast(hfeCuts);
	314	}
	315
	316	//__________________________________________________________________
	317	void AliHFEcuts::SetHFElectronTPCCuts(){
	318	//
	319	// Special Cuts introduced by the HFElectron Group: TPC
	320	//
	321	AliHFEextraCuts *hfecuts = new AliHFEextraCuts("fCutsHFElectronGroupTPC","Extra cuts from the HFE group");
809a4336	322	if(fMinClusterRatioTPC > 0.) hfecuts->SetClusterRatioTPC(fMinClusterRatioTPC);
	323	if(IsInDebugMode()) hfecuts->SetQAOn(fHistQA);
	324
	325	TObjArray *hfeCuts = new TObjArray;
dbe3abbe	326	hfeCuts->SetName("fPartHFECutsTPC");
	327	hfeCuts->AddLast(hfecuts);
	328	fCutList->AddLast(hfeCuts);
	329	}
	330
	331	//__________________________________________________________________
	332	void AliHFEcuts::SetHFElectronTRDCuts(){
	333	//
	334	// Special Cuts introduced by the HFElectron Group: TRD
	335	//
	336	AliHFEextraCuts *hfecuts = new AliHFEextraCuts("fCutsHFElectronGroupTRD","Extra cuts from the HFE group");
	337	if(fMinTrackletsTRD > 0.) hfecuts->SetMinTrackletsTRD(fMinTrackletsTRD);
	338	if(IsInDebugMode()) hfecuts->SetQAOn(fHistQA);
	339
	340	TObjArray *hfeCuts = new TObjArray;
	341	hfeCuts->SetName("fPartHFECutsTRD");
809a4336	342	hfeCuts->AddLast(hfecuts);
	343	fCutList->AddLast(hfeCuts);
	344	}
	345
	346	//__________________________________________________________________
	347	void AliHFEcuts::SetDebugMode(){
	348	//
	349	// Switch on QA
	350	//
	351	SetBit(kDebugMode, kTRUE);
	352	fHistQA = new TList;
	353	fHistQA->SetName("CutQAhistograms");
	354	fHistQA->SetOwner(kFALSE);
	355	}
	356
	357	//__________________________________________________________________
	358	Bool_t AliHFEcuts::CheckParticleCuts(CutStep_t step, TObject *o){
	359	//
	360	// Checks the cuts without using the correction framework manager
	361	//
	362	TString stepnames[kNcutSteps] = {"fPartGenCuts", "fPartAccCuts", "fPartRecCuts", "fPartPrimCuts", "fPartHFECuts"};
	363	TObjArray cuts = dynamic_cast<TObjArray >(fCutList->FindObject(stepnames[step].Data()));
	364	if(!cuts) return kTRUE;
	365	TIterator *it = cuts->MakeIterator();
	366	AliCFCutBase *mycut;
	367	Bool_t status = kTRUE;
	368	while((mycut = dynamic_cast<AliCFCutBase *>(it->Next()))){
	369	status &= mycut->IsSelected(o);
	370	}
	371	delete it;
	372	return status;
	373	}