update from pr task : sjena

[u/mrichter/AliRoot.git] / PWGDQ / dielectron / macrosLMEE / Config_shin_pPb.C

void InitHistograms(AliDielectron *die, Int_t cutDefinition);
void InitCF(AliDielectron* die, Int_t cutDefinition);

void SetupTrackCuts(AliDielectron *die, Int_t cutDefinition);

AliESDtrackCuts *SetupESDtrackCuts(Int_t cutDefinition);
AliDielectronPID *SetPIDcuts(Int_t cutDefinition);


//start systematics
TString names= ("TPCTOFPID+phiv");


	Bool_t kRot = 0;
	Bool_t kMix = 1;

TObjArray *arrNames=names.Tokenize(";");
const Int_t nDie=arrNames->GetEntriesFast();

AliDielectron* Config_shin_pPb(Int_t cutDefinition=1)
{
  //
  // Setup the instance of AliDielectron
  //
  
  // create the actual framework object
  TString name=Form("%02d",cutDefinition);
  if (cutDefinition<arrNames->GetEntriesFast()){
    name=arrNames->At(cutDefinition)->GetName();
  }
  AliDielectron *die =
    new AliDielectron(Form("%s",name.Data()),
                      Form("Track cuts: %s",name.Data()));
  
  
  if(kRot){
    AliDielectronTrackRotator *rot = new AliDielectronTrackRotator;
    rot->SetConeAnglePhi(TMath::Pi());
    rot->SetIterations(10);
    die->SetTrackRotator(rot);
  }//kRot
  
  
  if(kMix){
    AliDielectronMixingHandler *mix = new AliDielectronMixingHandler;
    mix->SetMixType(AliDielectronMixingHandler::kAll);
    mix->AddVariable(AliDielectronVarManager::kZvPrim,"-10., -7.5, -5., -2.5 , 0., 2.5, 5., 7.5 , 10.");
    mix->SetDepth(10);
    die->SetMixingHandler(mix);
  }//kMix
  
  

  SetupTrackCuts(die,cutDefinition);
  SetupPairCuts(die,cutDefinition);
  
  // histogram setup
  // only if an AliDielectronHistos object is attached to the
  // dielectron framework histograms will be filled
  InitHistograms(die,cutDefinition);
  //  InitCF(die,cutDefinition);
  

  return die;

}

//______________________________________________________________________________________
void SetupTrackCuts(AliDielectron *die, Int_t cutDefinition)
{
  //
  // Setup the track cuts
  //
	//options
        die->SetPreFilterAllSigns();
	//pairing with TLorentzVector
	die->SetUseKF(kFALSE);

	AliDielectronTrackCuts *TrackCuts = new AliDielectronTrackCuts("StandardCuts","StandardCut");
	//refit
	TrackCuts->SetRequireTPCRefit(kTRUE);
	TrackCuts->SetRequireITSRefit(kTRUE);
	//SPD require
	TrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD,AliESDtrackCuts::kFirst);
	die->GetTrackFilter().AddCuts(TrackCuts);
		
	AliDielectronVarCuts *varCuts = new AliDielectronVarCuts("varCuts","varCuts");
	varCuts->AddCut(AliDielectronVarManager::kPt,0.2,100.);
	varCuts->AddCut(AliDielectronVarManager::kEta,-0.8,0.8);
	varCuts->AddCut(AliDielectronVarManager::kImpactParZ,-3.,3.);
	varCuts->AddCut(AliDielectronVarManager::kImpactParXY,-1.,1.);
	varCuts->AddCut(AliDielectronVarManager::kKinkIndex0,0.);

	//TPC track
	varCuts->AddCut(AliDielectronVarManager::kNclsTPC,80.,500.);
	varCuts->AddCut(AliDielectronVarManager::kNFclsTPCr,100.,500.); // or NFclsTPCr?
	varCuts->AddCut(AliDielectronVarManager::kTPCchi2Cl,0.,4.);
	varCuts->AddCut(AliDielectronVarManager::kNFclsTPCfCross,0.5,500.);

	//ITS track
	varCuts->AddCut(AliDielectronVarManager::kNclsITS,3.,7.);
	
	die->GetTrackFilter().AddCuts(varCuts);
	
	// PID	
	//	if(cutDefinition > 0){
	AliDielectronPID *pid = new AliDielectronPID("pid","pid"); 	
	pid->AddCut(AliDielectronPID::kTPC,AliPID::kElectron,-1.5,3.,0.2,100.,kFALSE);
	pid->AddCut(AliDielectronPID::kTPC,AliPID::kPion,-100.,3.,0.2,100.,kTRUE);
	pid->AddCut(AliDielectronPID::kTOF,AliPID::kElectron,-3.,3.,0.4, 5., kFALSE);	  
	die->GetTrackFilter().AddCuts(pid);
	//	}
	
}
//----------------------------------Pair Cut-------------------------------------------
void SetupPairCuts(AliDielectron *die, Int_t cutDefinition)
{
  
  //V0 cuts
  AliDielectronTrackCuts *noconv=new AliDielectronTrackCuts("noConv","conversion tagging");
  noconv->SetV0DaughterCut(AliPID::kElectron,kTRUE);
  die->GetTrackFilter().AddCuts(noconv);
  
    
  //Pre filter
  AliDielectronVarCuts *PhiV = new AliDielectronVarCuts("PhiV","PhiV");//mass and Phiv together
  PhiV->AddCut(AliDielectronVarManager::kM, 0. , 0.1);
  PhiV->AddCut(AliDielectronVarManager::kPhivPair, 2.5 , 3.2);
  die->GetPairPreFilter().AddCuts(PhiV);

    

}

//______________________________________________________________________________________
void InitHistograms(AliDielectron *die, Int_t cutDefinition)
{
  //
  // Initialise the histograms
  //
  
  //Setup histogram classes
  AliDielectronHistos *histos=
    new AliDielectronHistos(die->GetName(),
                            die->GetTitle());
  


  //Initialise histogram classes
  histos->SetReservedWords("Track;Pair");
  
  //Event class
  histos->AddClass("Event");
  

  //Track classes
  //to fill also track info from 2nd event loop until 2
  for (Int_t i=0; i<2; ++i){
    histos->AddClass(Form("Track_%s",AliDielectron::TrackClassName(i)));
  }
  
  //Pair classes
  // to fill also mixed event histograms loop until 10

  for (Int_t i=0; i<3; ++i){
    histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(i)));

  }

  if(kMix){
    histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(3))); //ME ++
    histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(4)));//ME -+
    histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(6)));//ME +-
    histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(7))); // ME --
  }
  
  if(kRot)histos->AddClass(Form("Pair_%s",AliDielectron::PairClassName(10)));//Rot
  
  
  //add histograms to event class
  histos->UserHistogram("Event","nEvents","Number of processed events after cuts;Number events",1,0,1,AliDielectronVarManager::kNevents);
  histos->UserHistogram("Event","ZVertex","ZVertex;ZVertex/cm",480,-12.,12.,AliDielectronVarManager::kZvPrim);
  histos->UserHistogram("Event","nESDTracks","ESD tracks;ESD tracks;Number events",100,0,200,AliDielectronVarManager::kTracks);
  histos->UserHistogram("Event","Nacc","Number of accepted tracks;Number events",100,0,200,AliDielectronVarManager::kNacc);
  histos->UserHistogram("Event","NVtxContrib","Number of Vertex Contributor;NVtx;Number events",100,0,100,AliDielectronVarManager::kNVtxContrib);

  
  //add histograms to Track classes
  //histos->UserHistogram("Track","P","P;P [GeV];#tracks",500,0.,10.,AliDielectronVarManager::kPIn);
  histos->UserHistogram("Track","Pt","Pt;Pt [GeV];#tracks",500,0.,10.,AliDielectronVarManager::kPt);
  histos->UserHistogram("Track","Eta","Eta; Eta ;#tracks",100,-1.,1.,AliDielectronVarManager::kEta);
  histos->UserHistogram("Track","Phi","Phi; Phi ;#tracks",640,0.,6.4,AliDielectronVarManager::kPhi);


  histos->UserHistogram("Track","Eta_phi","Eta vs Phi;Eta;Phi",500,-1.,1.,320,0.,6.4,AliDielectronVarManager::kEta,AliDielectronVarManager::kPhi);
  histos->UserHistogram("Track","Eta_pt","Eta vs Pt;Eta;Pt",100,-1.,1.,500,0.,10.,AliDielectronVarManager::kEta,AliDielectronVarManager::kPt);
  histos->UserHistogram("Track","Phi_pt","Phi vs Pt;Phi;Pt",640,0,6.4.,500,0.,10.,AliDielectronVarManager::kPhi,AliDielectronVarManager::kPt);