Allow for differerent selection flag than kMB, take into account filterbit in the...

[u/mrichter/AliRoot.git] / PWG4 / macros / ConfigJetCorrel.C

AliJetCorrelSelector* ConfigJetCorrel(){

  ///////////////////////////////////
  // set correlation input parameters
  ///////////////////////////////////
  // set generic selections:
  Bool_t kGenQA = kFALSE;      // generate QA histos
  UInt_t kDPhiNumBins = 30;    // number of bins in DeltaPhi
  UInt_t kDEtaNumBins = 16;    // number of bins in DeltaEta
  Float_t kPoutBinWidth = 0.2; // number of bins in Pout = max(assocBins)/kPoutBinWidth
  Bool_t kUseAliKF = kFALSE;   // use AliKF or TLorentzVector for parent reconstruction
  UInt_t poolDepth = 5;
  UInt_t correlTypes[] = {0};  // 0=dihadron, 1=pi0-hadron, 2=photon-hadron
  Float_t centrBins[] = {1,500};
  Float_t zVertBins[] = {-5,5};
  Float_t triggBins[] = {6,8,10,20};
  Float_t assocBins[] = {2,10};
  TString sTrigg[] = {"ALL"}; // selects events where one of the strings is matched; "ALL"=no cut
  //TString sTrigg[] = {"CINT1B-"};
  // set track selections:
  Bool_t itsRefit = kTRUE;
  Bool_t tpcRefit = kTRUE;
  Bool_t trdRefit = kTRUE;         // used only for electron tracks
  Float_t maxEta = 0.8;
  UInt_t minNClusTPC = 70;
  Float_t maxTPCChi2 = 4.0;        // max track Chi2 per TPC cluster
  Bool_t rejectKinkChild = kTRUE;  // reject track comming from a kink
  Float_t trkPairCut = 0.;         // track pair proximity cut (dist at TPC entrance)
  // code that applies next 3 cuts (NClusITS,ITSChi2,NsigVtx) currently commented out
  UInt_t minNClusITS = 0;
  Float_t maxITSChi2 = 35;        // max track Chi2 per ITS cluster
  Float_t maxNsigVtx = 35;        // max dist to primary vertex (sigma)
  Float_t maxTrkVtx  = 2.4;       // max dist to primary vertex (absolute) - temporarily instead of sigma

  //////////////////////////////////
  // load them into selector object:
  //////////////////////////////////
  AliJetCorrelSelector* selector = new AliJetCorrelSelector();
  selector->SetQA(kGenQA);
  selector->SetUseAliKF(kUseAliKF);
  selector->SetDPhiNumBins(kDPhiNumBins);
  selector->SetDEtaNumBins(kDEtaNumBins);
  selector->SetPoutBinWidth(kPoutBinWidth);
  selector->SetPoolDepth(poolDepth);
  selector->SetCorrelTypes(sizeof(correlTypes)/sizeof(UInt_t),correlTypes);
  selector->SetBinningCentr(sizeof(centrBins)/sizeof(Float_t),centrBins);
  selector->SetBinningZvert(sizeof(zVertBins)/sizeof(Float_t),zVertBins);
  selector->SetBinningTrigg(sizeof(triggBins)/sizeof(Float_t),triggBins);
  selector->SetBinningAssoc(sizeof(assocBins)/sizeof(Float_t),assocBins);
  selector->SetTriggers(sizeof(sTrigg)/sizeof(TString),sTrigg);
  selector->SetITSRefit(itsRefit);
  selector->SetTPCRefit(tpcRefit);
  selector->SetTRDRefit(trdRefit);
  selector->SetMaxEta(maxEta);
  selector->SetMinNClusITS(minNClusITS);
  selector->SetMinNClusTPC(minNClusTPC);
  selector->SetMaxITSChi2(maxITSChi2);
  selector->SetMaxTPCChi2(maxTPCChi2);
  selector->SetMaxNsigmaVtx(maxNsigVtx);
  selector->SetMaxTrkVtx(maxTrkVtx);
  selector->SetRejectKinkChild(rejectKinkChild);
  selector->SetTrkProximityCut(trkPairCut);
  selector->Show();

  return selector;
}