/**************************************************************************
 * Copyright(c) 1998-2009, 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.                  *
 **************************************************************************/

/////////////////////////////////////////////////////////////
//
// AliAnalysisTaskSE for D0 candidates invariant mass histogram
// and comparison with the MC truth and cut variables distributions.
//
// Authors: A.Dainese, andrea.dainese@lnl.infn.it
// Chiara Bianchin, chiara.bianchin@pd.infn.it (invariant mass)
// Carmelo Di Giglio, carmelo.digiglio@ba.infn.it (like sign)
/////////////////////////////////////////////////////////////

#include <Riostream.h>
#include <TClonesArray.h>
#include <TCanvas.h>
#include <TNtuple.h>
#include <TList.h>
#include <TH1F.h>
#include <TH2F.h>
#include <TDatabasePDG.h>

#include "AliAnalysisManager.h"
#include "AliESDtrack.h"
#include "AliAODHandler.h"
#include "AliAODEvent.h"
#include "AliAODVertex.h"
#include "AliAODTrack.h"
#include "AliAODMCHeader.h"
#include "AliAODMCParticle.h"
#include "AliAODRecoDecayHF2Prong.h"
#include "AliAODRecoCascadeHF.h"
#include "AliAnalysisVertexingHF.h"
#include "AliAnalysisTaskSE.h"
#include "AliAnalysisTaskSED0Mass.h"


ClassImp(AliAnalysisTaskSED0Mass)


//________________________________________________________________________
AliAnalysisTaskSED0Mass::AliAnalysisTaskSED0Mass():
AliAnalysisTaskSE(),
fOutputPPR(0),
fOutputmycuts(0),
fNentries(0),
fDistr(0), 
fChecks(0),
fVHFPPR(0),
fVHFmycuts(0),
fArray(0),
fReadMC(0),
fLsNormalization(1.)

{
  // Default constructor
  for(Int_t i=0;i<5;i++) {fTotPosPairs[i]=0; fTotNegPairs[i]=0;}
}

//________________________________________________________________________
AliAnalysisTaskSED0Mass::AliAnalysisTaskSED0Mass(const char *name):
AliAnalysisTaskSE(name),
fOutputPPR(0), 
fOutputmycuts(0), 
fNentries(0),
fDistr(0),
fChecks(0),
fVHFPPR(0),
fVHFmycuts(0),
fArray(0),
fReadMC(0),
fLsNormalization(1.)

{
  // Default constructor
  for(Int_t i=0;i<5;i++) {fTotPosPairs[i]=0; fTotNegPairs[i]=0;}

  // Output slot #1 writes into a TList container
  DefineOutput(1,TList::Class());  //My private output
  // Output slot #2 writes into a TList container
  DefineOutput(2,TList::Class());  //My private output
  // Output slot #3 writes into a TH1F container
  DefineOutput(3,TH1F::Class());  //My private output
  // Output slot #4 writes into a TList container
  DefineOutput(4,TList::Class());  //My private output
  // Output slot #5 writes into a TList container
  DefineOutput(5,TList::Class());  //My private output
}

//________________________________________________________________________
AliAnalysisTaskSED0Mass::~AliAnalysisTaskSED0Mass()
{
   if (fOutputPPR) {
    delete fOutputPPR;
    fOutputPPR = 0;
  }
  if (fVHFPPR) {
    delete fVHFPPR;
    fVHFPPR = 0;
  }
  if (fOutputmycuts) {
    delete fOutputmycuts;
    fOutputmycuts = 0;
  }

  if (fDistr) {
    delete fDistr;
    fDistr = 0;
  }

  if (fChecks) {
    delete fChecks;
    fChecks = 0;
  }

  if (fVHFmycuts) {
    delete fVHFmycuts;
    fVHFmycuts = 0;
  }
  if (fNentries){
    delete fNentries;
    fNentries = 0;
  }


}  

//________________________________________________________________________
void AliAnalysisTaskSED0Mass::Init()
{
  // Initialization

  if(fDebug > 1) printf("AnalysisTaskSED0Mass::Init() \n");

  gROOT->LoadMacro("$ALICE_ROOT/PWG3/vertexingHF/ConfigVertexingHF.C");

  // 2 sets of dedidcated cuts -- defined in UserExec
  fVHFPPR = (AliAnalysisVertexingHF*)gROOT->ProcessLine("ConfigVertexingHF()");
  fVHFmycuts = (AliAnalysisVertexingHF*)gROOT->ProcessLine("ConfigVertexingHF()");  
  
  return;
}

//________________________________________________________________________
void AliAnalysisTaskSED0Mass::UserCreateOutputObjects()
{

  // Create the output container
  //
  if(fDebug > 1) printf("AnalysisTaskSED0Mass::UserCreateOutputObjects() \n");

  // Several histograms are more conveniently managed in a TList
  fOutputPPR = new TList();
  fOutputPPR->SetOwner();
  fOutputPPR->SetName("listPPR");

  fOutputmycuts = new TList();
  fOutputmycuts->SetOwner();
  fOutputmycuts->SetName("listloose");

  fDistr = new TList();
  fDistr->SetOwner();
  fDistr->SetName("distributionslist");

  fChecks = new TList();
  fChecks->SetOwner();
  fChecks->SetName("checkHistograms");

  const Int_t nhist=5;

  TString nameMass=" ",nameSgn27=" ",nameSgn=" ", nameBkg=" ", nameRfl=" ",nameMassNocutsS =" ",nameMassNocutsB =" ", namedistr=" ";

  for(Int_t i=0;i<nhist;i++){

    nameMass="histMass_";
    nameMass+=i+1;
    nameSgn27="histSgn27_";
    nameSgn27+=i+1;
    nameSgn="histSgn_";
    nameSgn+=i+1;
    nameBkg="histBkg_";
    nameBkg+=i+1;
    nameRfl="histRfl_";
    nameRfl+=i+1;
    nameMassNocutsS="hMassS_";
    nameMassNocutsS+=i+1;
    nameMassNocutsB="hMassB_";
    nameMassNocutsB+=i+1;

    //histograms of cut variable distributions

    //  pT
    namedistr="hptpiS_";
    namedistr+=i+1;
    TH1F *hptpiS = new TH1F(namedistr.Data(), "P_{T} distribution (pions);p_{T} [GeV/c]",200,0.,8.);

    namedistr="hptKS_";
    namedistr+=i+1;
    TH1F *hptKS = new TH1F(namedistr.Data(), "P_{T} distribution (kaons);p_{T} [GeV/c]",200,0.,8.);

    namedistr="hptB_";
    namedistr+=i+1;
    TH1F *hptB = new TH1F(namedistr.Data(), "P_{T} distribution;p_{T} [GeV/c]",200,0.,8.);

    //  pT no mass cut
    namedistr="hptpiSnoMcut_";
    namedistr+=i+1;
    TH1F *hptpiSnoMcut = new TH1F(namedistr.Data(), "P_{T} distribution (pions);p_{T} [GeV/c]",200,0.,8.);

    namedistr="hptKSnoMcut_";
    namedistr+=i+1;
    TH1F *hptKSnoMcut = new TH1F(namedistr.Data(), "P_{T} distribution (kaons);p_{T} [GeV/c]",200,0.,8.);

    namedistr="hptB1prongnoMcut_";
    namedistr+=i+1;
    TH1F *hptB1pnoMcut = new TH1F(namedistr.Data(), "P_{T} distribution;p_{T} [GeV/c]",200,0.,8.);

    namedistr="hptB2prongsnoMcut_";
    namedistr+=i+1;
    TH1F *hptB2pnoMcut = new TH1F(namedistr.Data(), "P_{T} distribution;p_{T} [GeV/c]",200,0.,8.);

    
    //  dca
    namedistr="hdcaS_";
    namedistr+=i+1;
    TH1F *hdcaS = new TH1F(namedistr.Data(), "DCA distribution;dca [cm]",200,0.,0.1);
    namedistr="hdcaB_";
    namedistr+=i+1;
    TH1F *hdcaB = new TH1F(namedistr.Data(), "DCA distribution;dca [cm]",200,0.,0.1);

    //  costhetastar
    namedistr="hcosthetastarS_";
    namedistr+=i+1;
    TH1F *hcosthetastarS = new TH1F(namedistr.Data(), "cos#theta* distribution;cos#theta*",200,-1.,1.);
    namedistr="hcosthetastarB_";
    namedistr+=i+1;
    TH1F *hcosthetastarB = new TH1F(namedistr.Data(), "cos#theta* distribution;cos#theta*",200,-1.,1.);

    // impact parameter
    namedistr="hd0piS_";
    namedistr+=i+1;
    TH1F *hd0piS = new TH1F(namedistr.Data(), "Impact parameter distribution (pions);d0(#pi) [cm]",200,-0.1,0.1);

    namedistr="hd0KS_";
    namedistr+=i+1;
    TH1F *hd0KS = new TH1F(namedistr.Data(), "Impact parameter distribution (kaons);d0(K) [cm]",200,-0.1,0.1);
    namedistr="hd0B_";
    namedistr+=i+1;
    TH1F *hd0B = new TH1F(namedistr.Data(), "Impact parameter distribution;d0 [cm]",200,-0.1,0.1);

    namedistr="hd0d0S_";
    namedistr+=i+1;
    TH1F *hd0d0S = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution;d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);
    namedistr="hd0d0B_";
    namedistr+=i+1;
    TH1F *hd0d0B = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution;d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);

    //  costhetapoint
    namedistr="hcosthetapointS_";
    namedistr+=i+1;
    TH1F *hcosthetapointS = new TH1F(namedistr.Data(), "cos#theta_{Point} distribution;cos#theta_{Point}",200,0,1.);
    namedistr="hcosthetapointB_";
    namedistr+=i+1;
    TH1F *hcosthetapointB = new TH1F(namedistr.Data(), "cos#theta_{Point} distribution;cos#theta_{Point}",200,0,1.);

    namedistr="hcosthpointd0d0S_";
    namedistr+=i+1;
    TH2F *hcosthpointd0d0S= new TH2F(namedistr.Data(),"Correlation cos#theta_{Point}-d_{0}#timesd_{0};cos#theta_{Point};d_{0}#timesd_{0} [cm^{2}]",200,0,1.,200,-0.001,0.001);
    namedistr="hcosthpointd0d0B_";
    namedistr+=i+1;
    TH2F *hcosthpointd0d0B= new TH2F(namedistr.Data(),"Correlation cos#theta_{Point}-d_{0}#timesd_{0};cos#theta_{Point};d_{0}#timesd_{0} [cm^{2}]",200,0,1.,200,-0.001,0.001);

    fDistr->Add(hptpiS);
    fDistr->Add(hptKS);
    fDistr->Add(hptB);

    fDistr->Add(hptpiSnoMcut);
    fDistr->Add(hptKSnoMcut);
    fDistr->Add(hptB1pnoMcut);
    fDistr->Add(hptB2pnoMcut);

    fDistr->Add(hdcaS);
    fDistr->Add(hdcaB);

    fDistr->Add(hd0piS);
    fDistr->Add(hd0KS);
    fDistr->Add(hd0B);

    fDistr->Add(hd0d0S);
    fDistr->Add(hd0d0B);

    fDistr->Add(hcosthetastarS);
    fDistr->Add(hcosthetastarB);

    fDistr->Add(hcosthetapointS);
    fDistr->Add(hcosthetapointB);

    fDistr->Add(hcosthpointd0d0S);
    fDistr->Add(hcosthpointd0d0B);


    //histograms of invariant mass distributions

    TH1F* tmpMt = new TH1F(nameMass.Data(),"D^{0} invariant mass; M [GeV]; Entries",200,1.765,1.965);
    TH1F *tmpMl=(TH1F*)tmpMt->Clone();
    tmpMt->Sumw2();
    tmpMl->Sumw2();

    //to compare with AliAnalysisTaskCharmFraction
    TH1F* tmpS27t = new TH1F(nameSgn27.Data(),"D^{0} invariant mass in M(D^{0}) +/- 27 MeV - MC; M [GeV]; Entries",200,1.765,1.965);
    TH1F *tmpS27l=(TH1F*)tmpS27t->Clone();
    tmpS27t->Sumw2();
    tmpS27l->Sumw2();
 
    //distribution w/o cuts
    //   TH1F* tmpMS = new TH1F(nameMassNocutsS.Data(),"D^{0} invariant mass; M [GeV]; Entries",300,0.7,3.);
    TH1F* tmpMS = new TH1F(nameMassNocutsS.Data(),"D^{0} invariant mass; M [GeV]; Entries",300,1.56484,2.16484); //range (MD0-300MeV, mD0 + 300MeV)
    TH1F *tmpMB=(TH1F*)tmpMS->Clone();
    tmpMB->SetName(nameMassNocutsB.Data());
    tmpMS->Sumw2();
    tmpMB->Sumw2();

    //MC signal and background
    TH1F* tmpSt = new TH1F(nameSgn.Data(), "D^{0} invariant mass - MC; M [GeV]; Entries",200,1.765,1.965);
    TH1F *tmpSl=(TH1F*)tmpSt->Clone();
    tmpSt->Sumw2();
    tmpSl->Sumw2();

    TH1F* tmpBt = new TH1F(nameBkg.Data(), "Background invariant mass - MC; M [GeV]; Entries",200,1.765,1.965);
    TH1F *tmpBl=(TH1F*)tmpBt->Clone();
    tmpBt->Sumw2();
    tmpBl->Sumw2();

    //Reflection: histo filled with D0Mass which pass the cut (also) as D0bar and with D0bar which pass (also) the cut as D0
    TH1F* tmpRt = new TH1F(nameRfl.Data(), "Reflected signal invariant mass - MC; M [GeV]; Entries",200,1.765,1.965);
    TH1F *tmpRl=(TH1F*)tmpRt->Clone();
    tmpRt->Sumw2();
    tmpRl->Sumw2();
  //  printf("Created histograms %s\t%s\t%s\t%s\n",tmpM->GetName(),tmpS->GetName(),tmpB->GetName(),tmpR->GetName());

    fOutputPPR->Add(tmpMt);
    fOutputPPR->Add(tmpSt);
    fOutputPPR->Add(tmpS27t);
    fOutputPPR->Add(tmpBt);
    fOutputPPR->Add(tmpRt);

    fOutputmycuts->Add(tmpMl);
    fOutputmycuts->Add(tmpSl);
    fOutputmycuts->Add(tmpS27l);
    fOutputmycuts->Add(tmpBl);
    fOutputmycuts->Add(tmpRl);
 
    fDistr->Add(tmpMS);
    fDistr->Add(tmpMB);


  }

  //histograms for vertex checking
  TString checkname="hptGoodTr";

  TH1F* hptGoodTr=new TH1F(checkname.Data(),"Pt distribution of 'good' tracks;p_{t}[GeV];Number",200,0.,8.);
  hptGoodTr->SetTitleOffset(1.3,"Y");
  checkname="hdistrGoodTr";

  TH1F* hdistrGoodTr=new TH1F(checkname.Data(),"Distribution of number of good tracks per event;no.good-tracks/ev;Entries",31,0,31);
  hdistrGoodTr->SetTitleOffset(1.3,"Y");

  //conta gli eventi con vertice buoni e almeno due tracce utilizzabili 
  fChecks->Add(hptGoodTr);
  fChecks->Add(hdistrGoodTr);

  fNentries=new TH1F("nentriesD0", "nentriesD0->Integral(1,2) = number of AODs *** nentriesD0->Integral(3,4) = number of candidates selected with cuts *** nentriesD0->Integral(5,6) = number of D0 selected with cuts *** nentriesD0->Integral(7,8) = events with good vertex", 5,0.,5.);

  fNentries->GetXaxis()->SetBinLabel(1,"nEventsAnal");
  fNentries->GetXaxis()->SetBinLabel(2,"nCandidatesSelected");
  fNentries->GetXaxis()->SetBinLabel(3,"nD0Selected");
  fNentries->GetXaxis()->SetBinLabel(4,"nEventsGoodVtx");
  fNentries->GetXaxis()->SetBinLabel(5,"nEventsGoodVtx+>2tracks");
  fNentries->GetXaxis()->SetNdivisions(1,kFALSE);

  return;
}

//________________________________________________________________________
void AliAnalysisTaskSED0Mass::UserExec(Option_t */*option*/)
{
  // Execute analysis for current event:
  // heavy flavor candidates association to MC truth
  //cout<<"I'm in UserExec"<<endl;
  AliAODEvent *aod = dynamic_cast<AliAODEvent*> (InputEvent());

  TString bname;
  if(fArray==0){ //D0 candidates
    // load D0->Kpi candidates
    //cout<<"D0 candidates"<<endl;
    bname="D0toKpi";
  } else { //LikeSign candidates
    // load like sign candidates
    //cout<<"LS candidates"<<endl;
    bname="LikeSign2Prong";
  }

  TClonesArray *inputArray=0;
 
  if(!aod && AODEvent() && IsStandardAOD()) {
    // In case there is an AOD handler writing a standard AOD, use the AOD 
    // event in memory rather than the input (ESD) event.    
    aod = dynamic_cast<AliAODEvent*> (AODEvent());
    // in this case the braches in the deltaAOD (AliAOD.VertexingHF.root)
    // have to taken from the AOD event hold by the AliAODExtension
    AliAODHandler* aodHandler = (AliAODHandler*) 
      ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());

    if(aodHandler->GetExtensions()) {
      AliAODExtension *ext = (AliAODExtension*)aodHandler->GetExtensions()->FindObject("AliAOD.VertexingHF.root");
      AliAODEvent* aodFromExt = ext->GetAOD();
      inputArray=(TClonesArray*)aodFromExt->GetList()->FindObject(bname.Data());
    }
  } else {
    inputArray=(TClonesArray*)aod->GetList()->FindObject(bname.Data());
  }


  if(!inputArray) {
    printf("AliAnalysisTaskSED0Mass::UserExec: input branch not found!\n");
    return;
  }
  
  // AOD primary vertex
  AliAODVertex *vtx1 = (AliAODVertex*)aod->GetPrimaryVertex();

  Bool_t isGoodVtx=kFALSE;

   //vtx1->Print();
  TString primTitle = vtx1->GetTitle();
  if(primTitle.Contains("VertexerTracks") && vtx1->GetNContributors()>0) {
    isGoodVtx=kTRUE;
    fNentries->Fill(3);
  }

  //cout<<"Start checks"<<endl;
  Int_t ntracks=0,isGoodTrack=0;

  if(aod) ntracks=aod->GetNTracks();
  //cout<<"ntracks = "<<ntracks<<endl;
  //cout<<"Before loop"<<endl;
  //loop on tracks in the event
  for (Int_t k=0;k<ntracks;k++){
    AliAODTrack* track=aod->GetTrack(k);
    //cout<<"in loop"<<endl;
    //check clusters of the tracks
    Int_t nclsTot=0,nclsSPD=0;
    
    for(Int_t l=0;l<6;l++) {
	if(TESTBIT(track->GetITSClusterMap(),l)) {
	  nclsTot++; if(l<2) nclsSPD++;
	}
    }
    
    if (track->Pt()>0.3 &&
	track->GetStatus()&AliESDtrack::kTPCrefit &&
	track->GetStatus()&AliESDtrack::kITSrefit &&
	nclsTot>3 &&
	nclsSPD>0) {//fill hist good tracks
      //cout<<"in if"<<endl;
      ((TH1F*)fChecks->FindObject("hptGoodTr"))->Fill(track->Pt());
      isGoodTrack++;
    }
    //cout<<"isGoodTrack = "<<isGoodTrack<<endl;
    ((TH1F*)fChecks->FindObject("hdistrGoodTr"))->Fill(isGoodTrack);
  }
  //number of events with good vertex and at least 2 good tracks
  if (isGoodTrack>=2 && isGoodVtx) fNentries->Fill(4);
  
  TClonesArray *mcArray = 0;
  AliAODMCHeader *mcHeader = 0;

  if(fReadMC) {
    // load MC particles
    mcArray = (TClonesArray*)aod->GetList()->FindObject(AliAODMCParticle::StdBranchName());
    if(!mcArray) {
      printf("AliAnalysisTaskSED0Mass::UserExec: MC particles branch not found!\n");
      return;
    }
  
    // load MC header
    mcHeader = (AliAODMCHeader*)aod->GetList()->FindObject(AliAODMCHeader::StdBranchName());
    if(!mcHeader) {
      printf("AliAnalysisTaskSED0Mass::UserExec: MC header branch not found!\n");
      return;
    }
  }
  
  //printf("VERTEX Z %f %f\n",vtx1->GetZ(),mcHeader->GetVtxZ());

  //histogram filled with 1 for every AOD
  fNentries->Fill(0);
  PostData(3,fNentries);
  //cout<<"First PostData"<<endl;

  // loop over candidates
  Int_t nInD0toKpi = inputArray->GetEntriesFast();
  if(fDebug>1) printf("Number of D0->Kpi: %d\n",nInD0toKpi);

  for (Int_t iD0toKpi = 0; iD0toKpi < nInD0toKpi; iD0toKpi++) {
    //Int_t nPosPairs=0, nNegPairs=0;
    //cout<<"inside the loop"<<endl;
    AliAODRecoDecayHF2Prong *d = (AliAODRecoDecayHF2Prong*)inputArray->UncheckedAt(iD0toKpi);
    Bool_t unsetvtx=kFALSE;
    if(!d->GetOwnPrimaryVtx()) {
      d->SetOwnPrimaryVtx(vtx1); // needed to compute all variables
      unsetvtx=kTRUE;
    }
  
    //check reco daughter in acceptance
    Double_t eta0=d->EtaProng(0);
    Double_t eta1=d->EtaProng(1);
    Bool_t prongsinacc=kFALSE;
    if (TMath::Abs(eta0) < 0.9 && TMath::Abs(eta1) < 0.9) prongsinacc=kTRUE;

    //add distr here
    UInt_t pdgs[2];
    
    Double_t mPDG=TDatabasePDG::Instance()->GetParticle(421)->Mass();
    pdgs[0]=211;
    pdgs[1]=321;
    Double_t minvD0 = d->InvMassD0();
    pdgs[1]=211;
    pdgs[0]=321;
    Double_t minvD0bar = d->InvMassD0bar();
    //cout<<"inside mass cut"<<endl;

    Int_t pdgDgD0toKpi[2]={321,211};
    Int_t lab=-9999;
    if(fReadMC) lab=d->MatchToMC(421,mcArray,2,pdgDgD0toKpi); //return MC particle label if the array corresponds to a D0, -1 if not (cf. AliAODRecoDecay.cxx)
    Double_t pt = d->Pt(); //mother pt
 
    if(lab>=0 && fReadMC){ //signal

      //check pdg of the prongs
      AliAODTrack *prong0=(AliAODTrack*)d->GetDaughter(0);
      AliAODTrack *prong1=(AliAODTrack*)d->GetDaughter(1);
      Int_t labprong[2];
      labprong[0]=prong0->GetLabel();
      labprong[1]=prong1->GetLabel();
      AliAODMCParticle *mcprong=0;
      Int_t pdgProng[2]={0,0};
      for (Int_t iprong=0;iprong<2;iprong++){
	if(labprong[iprong]>=0)  mcprong= (AliAODMCParticle*)mcArray->At(labprong[iprong]);
	pdgProng[iprong]=mcprong->GetPdgCode();
      }

 
      //no mass cut ditributions: ptbis
	
      if(pt>0. && pt<=1.) {
	if (TMath::Abs(pdgProng[0]) == 211 && TMath::Abs(pdgProng[1]) == 321){
	  ((TH1F*)fDistr->FindObject("hptpiSnoMcut_1"))->Fill(d->PtProng(0));
	  ((TH1F*)fDistr->FindObject("hptKSnoMcut_1"))->Fill(d->PtProng(1));
	}else {
	  if (TMath::Abs(pdgProng[0]) == 321 && TMath::Abs(pdgProng[1]) == 211){
	    ((TH1F*)fDistr->FindObject("hptKSnoMcut_1"))->Fill(d->PtProng(0));
	    ((TH1F*)fDistr->FindObject("hptpiSnoMcut_1"))->Fill(d->PtProng(1));
	  }
	}
      }
      if(pt>1. && pt<=2.) {
	if (TMath::Abs(pdgProng[0]) == 211 && TMath::Abs(pdgProng[1]) == 321){
	  ((TH1F*)fDistr->FindObject("hptpiSnoMcut_2"))->Fill(d->PtProng(0));
	  ((TH1F*)fDistr->FindObject("hptKSnoMcut_2"))->Fill(d->PtProng(1));
	}else {
	  if (TMath::Abs(pdgProng[0]) == 321 && TMath::Abs(pdgProng[1]) == 211){
	    ((TH1F*)fDistr->FindObject("hptKSnoMcut_2"))->Fill(d->PtProng(0));
	    ((TH1F*)fDistr->FindObject("hptpiSnoMcut_2"))->Fill(d->PtProng(1));
	  }
	}
      }
      if(pt>2. && pt<=3.) {
	if (TMath::Abs(pdgProng[0]) == 211 && TMath::Abs(pdgProng[1]) == 321){
	  ((TH1F*)fDistr->FindObject("hptpiSnoMcut_3"))->Fill(d->PtProng(0));
	  ((TH1F*)fDistr->FindObject("hptKSnoMcut_3"))->Fill(d->PtProng(1));
	}else {
	  if (TMath::Abs(pdgProng[0]) == 321 && TMath::Abs(pdgProng[1]) == 211){
	    ((TH1F*)fDistr->FindObject("hptKSnoMcut_3"))->Fill(d->PtProng(0));
	    ((TH1F*)fDistr->FindObject("hptpiSnoMcut_3"))->Fill(d->PtProng(1));
	  }
	}
      }
      if(pt>3. && pt<=5.) {

	if (TMath::Abs(pdgProng[0]) == 211 && TMath::Abs(pdgProng[1]) == 321){
	  ((TH1F*)fDistr->FindObject("hptpiSnoMcut_4"))->Fill(d->PtProng(0));
	  ((TH1F*)fDistr->FindObject("hptKSnoMcut_4"))->Fill(d->PtProng(1));
	}else {
	  if (TMath::Abs(pdgProng[0]) == 321 && TMath::Abs(pdgProng[1]) == 211){
	    ((TH1F*)fDistr->FindObject("hptKSnoMcut_4"))->Fill(d->PtProng(0));
	    ((TH1F*)fDistr->FindObject("hptpiSnoMcut_4"))->Fill(d->PtProng(1));
	  }
	}
      }
      if(pt>5.)           {

	if (TMath::Abs(pdgProng[0]) == 211 && TMath::Abs(pdgProng[1]) == 321){
	  ((TH1F*)fDistr->FindObject("hptpiSnoMcut_5"))->Fill(d->PtProng(0));
	  ((TH1F*)fDistr->FindObject("hptKSnoMcut_5"))->Fill(d->PtProng(1));
	}else {
	  if (TMath::Abs(pdgProng[0]) == 321 && TMath::Abs(pdgProng[1]) == 211){
	    ((TH1F*)fDistr->FindObject("hptKSnoMcut_5"))->Fill(d->PtProng(0));
	    ((TH1F*)fDistr->FindObject("hptpiSnoMcut_5"))->Fill(d->PtProng(1));
	  }
	}
      }

      if (((AliAODMCParticle*)mcArray->At(lab))->GetPdgCode() == 421){//D0
		  	 
	//no mass cut ditributions: mass, costhetastar
	
	if(pt>0. && pt<=1.) {
	  ((TH1F*)fDistr->FindObject("hMassS_1"))->Fill(minvD0);
	}
	if(pt>1. && pt<=2.) {
	  ((TH1F*)fDistr->FindObject("hMassS_2"))->Fill(minvD0);

	}
	if(pt>2. && pt<=3.) {
	  ((TH1F*)fDistr->FindObject("hMassS_3"))->Fill(minvD0);

	}
	if(pt>3. && pt<=5.) {
	  ((TH1F*)fDistr->FindObject("hMassS_4"))->Fill(minvD0);

	}
	if(pt>5.)           {
	  ((TH1F*)fDistr->FindObject("hMassS_5"))->Fill(minvD0);

	}

      }
      else { //D0bar

	//no mass cut ditributions: mass

	if(pt>0. && pt<=1.) {
	  ((TH1F*)fDistr->FindObject("hMassS_1"))->Fill(minvD0bar);

	}
	if(pt>1. && pt<=2.) {
	  ((TH1F*)fDistr->FindObject("hMassS_2"))->Fill(minvD0bar);

	}
	if(pt>2. && pt<=3.) {
	  ((TH1F*)fDistr->FindObject("hMassS_3"))->Fill(minvD0bar);

	}
	if(pt>3. && pt<=5.) {
	  ((TH1F*)fDistr->FindObject("hMassS_4"))->Fill(minvD0bar);

	}
	if(pt>5.)           {
	  ((TH1F*)fDistr->FindObject("hMassS_5"))->Fill(minvD0bar);
	}

      }

      //apply cut on invariant mass on the pair
      if(TMath::Abs(minvD0-mPDG)<0.03 || TMath::Abs(minvD0bar-mPDG)<0.03){

	if(fArray==1) cout<<"LS signal: ERROR"<<endl;
	for (Int_t iprong=0; iprong<2; iprong++){
	  AliAODTrack *prong=(AliAODTrack*)d->GetDaughter(iprong);
	  labprong[iprong]=prong->GetLabel();

	  //cout<<"prong name = "<<prong->GetName()<<" label = "<<prong->GetLabel()<<endl;
	  if(labprong[iprong]>=0)  mcprong= (AliAODMCParticle*)mcArray->At(labprong[iprong]);
	  Int_t pdgprong=mcprong->GetPdgCode();
	  if(TMath::Abs(pdgprong)==211) {
	    //cout<<"pi"<<endl;
	    if(pt>0. && pt<=1.){
	      ((TH1F*)fDistr->FindObject("hptpiS_1"))->Fill(d->PtProng(iprong));
	      ((TH1F*)fDistr->FindObject("hd0piS_1"))->Fill(d->Getd0Prong(iprong));
	    }
	    if(pt>1. && pt<=2.){
	      ((TH1F*)fDistr->FindObject("hptpiS_2"))->Fill(d->PtProng(iprong));
	      ((TH1F*)fDistr->FindObject("hd0piS_2"))->Fill(d->Getd0Prong(iprong));
	    }

	    if(pt>2. && pt<=3.){
	      ((TH1F*)fDistr->FindObject("hptpiS_3"))->Fill(d->PtProng(iprong));
	      ((TH1F*)fDistr->FindObject("hd0piS_3"))->Fill(d->Getd0Prong(iprong));

	    }
	    if(pt>3. && pt<=5.){
	      ((TH1F*)fDistr->FindObject("hptpiS_4"))->Fill(d->PtProng(iprong));
	      ((TH1F*)fDistr->FindObject("hd0piS_4"))->Fill(d->Getd0Prong(iprong));

	    }
	    if(pt>5.)          {
	      ((TH1F*)fDistr->FindObject("hptpiS_5"))->Fill(d->PtProng(iprong));
	      ((TH1F*)fDistr->FindObject("hd0piS_5"))->Fill(d->Getd0Prong(iprong));

	    }


	}

	if(TMath::Abs(pdgprong)==321) {
	    //cout<<"kappa"<<endl;
	    if(pt>0. && pt<=1.){
	      ((TH1F*)fDistr->FindObject("hptKS_1"))->Fill(d->PtProng(iprong));
	      ((TH1F*)fDistr->FindObject("hd0KS_1"))->Fill(d->Getd0Prong(iprong));
	    }
	    if(pt>1. && pt<=2.){
	      ((TH1F*)fDistr->FindObject("hptKS_2"))->Fill(d->PtProng(iprong));
	      ((TH1F*)fDistr->FindObject("hd0KS_2"))->Fill(d->Getd0Prong(iprong));
	    }
	    if(pt>2. && pt<=3.){
	      ((TH1F*)fDistr->FindObject("hptKS_3"))->Fill(d->PtProng(iprong));
	      ((TH1F*)fDistr->FindObject("hd0KS_3"))->Fill(d->Getd0Prong(iprong));
	    }
	    if(pt>3. && pt<=5.){
	      ((TH1F*)fDistr->FindObject("hptKS_4"))->Fill(d->PtProng(iprong));
	      ((TH1F*)fDistr->FindObject("hd0KS_4"))->Fill(d->Getd0Prong(iprong));

	    }
	    if(pt>5.)          {
	      ((TH1F*)fDistr->FindObject("hptKS_5"))->Fill(d->PtProng(iprong));
	      ((TH1F*)fDistr->FindObject("hd0KS_5"))->Fill(d->Getd0Prong(iprong));

	    }

	  }

	  if(pt>0. && pt<=1.){
	    ((TH1F*)fDistr->FindObject("hdcaS_1"))->Fill(d->GetDCA());

	    if (((AliAODMCParticle*)mcArray->At(lab))->GetPdgCode() == 421)
	      ((TH1F*)fDistr->FindObject("hcosthetastarS_1"))->Fill(d->CosThetaStarD0());
	    else ((TH1F*)fDistr->FindObject("hcosthetastarS_1"))->Fill(d->CosThetaStarD0bar());

	  }
	  if(pt>1. && pt<=2.){
	    ((TH1F*)fDistr->FindObject("hdcaS_2"))->Fill(d->GetDCA());

	    if (((AliAODMCParticle*)mcArray->At(lab))->GetPdgCode() == 421)
	      ((TH1F*)fDistr->FindObject("hcosthetastarS_2"))->Fill(d->CosThetaStarD0());
	    else ((TH1F*)fDistr->FindObject("hcosthetastarS_2"))->Fill(d->CosThetaStarD0bar());

	  }
	  if(pt>2. && pt<=3.){
	    ((TH1F*)fDistr->FindObject("hdcaS_3"))->Fill(d->GetDCA());

	    if (((AliAODMCParticle*)mcArray->At(lab))->GetPdgCode() == 421)
	      ((TH1F*)fDistr->FindObject("hcosthetastarS_3"))->Fill(d->CosThetaStarD0());
	    else ((TH1F*)fDistr->FindObject("hcosthetastarS_3"))->Fill(d->CosThetaStarD0bar());

	  }
	  if(pt>3. && pt<=5.){
	    ((TH1F*)fDistr->FindObject("hdcaS_4"))->Fill(d->GetDCA());

	    if (((AliAODMCParticle*)mcArray->At(lab))->GetPdgCode() == 421)
	      ((TH1F*)fDistr->FindObject("hcosthetastarS_4"))->Fill(d->CosThetaStarD0());
	    else ((TH1F*)fDistr->FindObject("hcosthetastarS_4"))->Fill(d->CosThetaStarD0bar());

	  }
	  if(pt>5.)          {
	    ((TH1F*)fDistr->FindObject("hdcaS_5"))->Fill(d->GetDCA());

	    if (((AliAODMCParticle*)mcArray->At(lab))->GetPdgCode() == 421)
	      ((TH1F*)fDistr->FindObject("hcosthetastarS_5"))->Fill(d->CosThetaStarD0());
	    else ((TH1F*)fDistr->FindObject("hcosthetastarS_5"))->Fill(d->CosThetaStarD0bar());

	  }
	  
	}

	if(pt>0. && pt<=1.){
	  ((TH1F*)fDistr->FindObject("hd0d0S_1"))->Fill(d->Prodd0d0());
	  ((TH1F*)fDistr->FindObject("hcosthetapointS_1"))->Fill(d->CosPointingAngle());
	  ((TH1F*)fDistr->FindObject("hcosthpointd0d0S_1"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
	}
	if(pt>1. && pt<=2.){
	  ((TH1F*)fDistr->FindObject("hd0d0S_2"))->Fill(d->Prodd0d0());
	  ((TH1F*)fDistr->FindObject("hcosthetapointS_2"))->Fill(d->CosPointingAngle());
	  ((TH1F*)fDistr->FindObject("hcosthpointd0d0S_2"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
	}
	if(pt>2. && pt<=3.){
	  ((TH1F*)fDistr->FindObject("hd0d0S_3"))->Fill(d->Prodd0d0());
	  ((TH1F*)fDistr->FindObject("hcosthetapointS_3"))->Fill(d->CosPointingAngle());
	  ((TH1F*)fDistr->FindObject("hcosthpointd0d0S_3"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
	}
	if(pt>3. && pt<=5.){
	  ((TH1F*)fDistr->FindObject("hd0d0S_4"))->Fill(d->Prodd0d0());
	  ((TH1F*)fDistr->FindObject("hcosthetapointS_4"))->Fill(d->CosPointingAngle());
	  ((TH1F*)fDistr->FindObject("hcosthpointd0d0S_4"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
	}
	if(pt>5.)          {
	  ((TH1F*)fDistr->FindObject("hd0d0S_5"))->Fill(d->Prodd0d0());
	  ((TH1F*)fDistr->FindObject("hcosthetapointS_5"))->Fill(d->CosPointingAngle());
	  ((TH1F*)fDistr->FindObject("hcosthpointd0d0S_5"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
	}

      } //invmass cut

    } else{ //Background or LS
      //cout<<"is background"<<endl;
      pt = d->Pt();

      //no mass cut distributions: mass, ptbis
      if(pt>0. && pt<=1.) {
	((TH1F*)fDistr->FindObject("hMassB_1"))->Fill(minvD0);
	((TH1F*)fDistr->FindObject("hMassB_1"))->Fill(minvD0bar);

	((TH1F*)fDistr->FindObject("hptB1prongnoMcut_1"))->Fill(d->PtProng(0));
	((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_1"))->Fill(d->PtProng(0));
	((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_1"))->Fill(d->PtProng(1));

      }
      if(pt>1. && pt<=2.) {
	((TH1F*)fDistr->FindObject("hMassB_2"))->Fill(minvD0);
	((TH1F*)fDistr->FindObject("hMassB_2"))->Fill(minvD0bar);

	((TH1F*)fDistr->FindObject("hptB1prongnoMcut_2"))->Fill(d->PtProng(0));
	((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_2"))->Fill(d->PtProng(0));
	((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_2"))->Fill(d->PtProng(1));

      }
      if(pt>2. && pt<=3.) {
	((TH1F*)fDistr->FindObject("hMassB_3"))->Fill(minvD0);
	((TH1F*)fDistr->FindObject("hMassB_3"))->Fill(minvD0bar);

	((TH1F*)fDistr->FindObject("hptB1prongnoMcut_3"))->Fill(d->PtProng(0));
	((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_3"))->Fill(d->PtProng(0));
	((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_3"))->Fill(d->PtProng(1));

      }
      if(pt>3. && pt<=5.) {
	((TH1F*)fDistr->FindObject("hMassB_4"))->Fill(minvD0);
	((TH1F*)fDistr->FindObject("hMassB_4"))->Fill(minvD0bar);

	((TH1F*)fDistr->FindObject("hptB1prongnoMcut_4"))->Fill(d->PtProng(0));
	((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_4"))->Fill(d->PtProng(0));
	((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_4"))->Fill(d->PtProng(1));
 
      }
      if(pt>5.)           { 
	((TH1F*)fDistr->FindObject("hMassB_5"))->Fill(minvD0);
	((TH1F*)fDistr->FindObject("hMassB_5"))->Fill(minvD0bar);

	((TH1F*)fDistr->FindObject("hptB1prongnoMcut_5"))->Fill(d->PtProng(0));
	((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_5"))->Fill(d->PtProng(0));
	((TH1F*)fDistr->FindObject("hptB2prongsnoMcut_5"))->Fill(d->PtProng(1));
 
      }

      //apply cut on invariant mass on the pair
      if(TMath::Abs(minvD0-mPDG)<0.03 || TMath::Abs(minvD0bar-mPDG)<0.03){


	AliAODTrack *prong=(AliAODTrack*)d->GetDaughter(0);
	if(!prong) cout<<"No daughter found";
	else{
	  if(prong->Charge()==1) {fTotPosPairs[4]++;} else {fTotNegPairs[4]++;}
	}
	if(pt>0. && pt<=1.){
	  //normalise pt distr to half afterwards
	  ((TH1F*)fDistr->FindObject("hptB_1"))->Fill(d->PtProng(0));((TH1F*)fDistr->FindObject("hptB_1"))->Fill(d->PtProng(1));
	  ((TH1F*)fDistr->FindObject("hd0B_1"))->Fill(d->Getd0Prong(0));
	  ((TH1F*)fDistr->FindObject("hdcaB_1"))->Fill(d->GetDCA());
	  ((TH1F*)fDistr->FindObject("hcosthetastarB_1"))->Fill(d->CosThetaStarD0());
	  ((TH1F*)fDistr->FindObject("hcosthetastarB_1"))->Fill(d->CosThetaStarD0bar());	
	  ((TH1F*)fDistr->FindObject("hd0d0B_1"))->Fill(d->Prodd0d0());
	  ((TH1F*)fDistr->FindObject("hcosthetapointB_1"))->Fill(d->CosPointingAngle());
	  ((TH1F*)fDistr->FindObject("hcosthpointd0d0B_1"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
	}
	if(pt>1. && pt<=2.){
	  ((TH1F*)fDistr->FindObject("hptB_2"))->Fill(d->PtProng(0));((TH1F*)fDistr->FindObject("hptB_2"))->Fill(d->PtProng(1));
	  ((TH1F*)fDistr->FindObject("hd0B_2"))->Fill(d->Getd0Prong(0));
	  ((TH1F*)fDistr->FindObject("hdcaB_2"))->Fill(d->GetDCA());
	  ((TH1F*)fDistr->FindObject("hcosthetastarB_2"))->Fill(d->CosThetaStarD0());
	  ((TH1F*)fDistr->FindObject("hcosthetastarB_2"))->Fill(d->CosThetaStarD0bar());	
	  ((TH1F*)fDistr->FindObject("hd0d0B_2"))->Fill(d->Prodd0d0());
	  ((TH1F*)fDistr->FindObject("hcosthetapointB_2"))->Fill(d->CosPointingAngle());
	  ((TH1F*)fDistr->FindObject("hcosthpointd0d0B_2"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
	}
	if(pt>2. && pt<=3.){
	  ((TH1F*)fDistr->FindObject("hptB_3"))->Fill(d->PtProng(0));((TH1F*)fDistr->FindObject("hptB_3"))->Fill(d->PtProng(1));
	  ((TH1F*)fDistr->FindObject("hd0B_3"))->Fill(d->Getd0Prong(0));
	  ((TH1F*)fDistr->FindObject("hdcaB_3"))->Fill(d->GetDCA());
	  ((TH1F*)fDistr->FindObject("hcosthetastarB_3"))->Fill(d->CosThetaStarD0());
	  ((TH1F*)fDistr->FindObject("hcosthetastarB_3"))->Fill(d->CosThetaStarD0bar());	
	  ((TH1F*)fDistr->FindObject("hd0d0B_3"))->Fill(d->Prodd0d0());
	  ((TH1F*)fDistr->FindObject("hcosthetapointB_3"))->Fill(d->CosPointingAngle());
	  ((TH1F*)fDistr->FindObject("hcosthpointd0d0B_3"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
	}
	if(pt>3. && pt<=5.){
	  ((TH1F*)fDistr->FindObject("hptB_4"))->Fill(d->PtProng(0));((TH1F*)fDistr->FindObject("hptB_4"))->Fill(d->PtProng(1));
	  ((TH1F*)fDistr->FindObject("hd0B_4"))->Fill(d->Getd0Prong(0));
	  ((TH1F*)fDistr->FindObject("hdcaB_4"))->Fill(d->GetDCA());
	  ((TH1F*)fDistr->FindObject("hcosthetastarB_4"))->Fill(d->CosThetaStarD0());
	  ((TH1F*)fDistr->FindObject("hcosthetastarB_4"))->Fill(d->CosThetaStarD0bar());	
	  ((TH1F*)fDistr->FindObject("hd0d0B_4"))->Fill(d->Prodd0d0());
	  ((TH1F*)fDistr->FindObject("hcosthetapointB_4"))->Fill(d->CosPointingAngle());
	  ((TH1F*)fDistr->FindObject("hcosthpointd0d0B_4"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
	}
	if(pt>5.)         {
	  ((TH1F*)fDistr->FindObject("hptB_5"))->Fill(d->PtProng(0));((TH1F*)fDistr->FindObject("hptB_5"))->Fill(d->PtProng(1));
	  ((TH1F*)fDistr->FindObject("hd0B_5"))->Fill(d->Getd0Prong(0));
	  ((TH1F*)fDistr->FindObject("hdcaB_5"))->Fill(d->GetDCA());
	  ((TH1F*)fDistr->FindObject("hcosthetastarB_5"))->Fill(d->CosThetaStarD0());
	  ((TH1F*)fDistr->FindObject("hcosthetastarB_5"))->Fill(d->CosThetaStarD0bar());	
	  ((TH1F*)fDistr->FindObject("hd0d0B_5"))->Fill(d->Prodd0d0());
	  ((TH1F*)fDistr->FindObject("hcosthetapointB_5"))->Fill(d->CosPointingAngle());
	  ((TH1F*)fDistr->FindObject("hcosthpointd0d0B_5"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
	}       


      }// end if inv mass cut
    }//end if background

  
  
     //cuts order
//       printf("    |M-MD0| [GeV]    < %f\n",fD0toKpiCuts[0]);
//     printf("    dca    [cm]  < %f\n",fD0toKpiCuts[1]);
//     printf("    cosThetaStar     < %f\n",fD0toKpiCuts[2]);
//     printf("    pTK     [GeV/c]    > %f\n",fD0toKpiCuts[3]);
//     printf("    pTpi    [GeV/c]    > %f\n",fD0toKpiCuts[4]);
//     printf("    |d0K|  [cm]  < %f\n",fD0toKpiCuts[5]);
//     printf("    |d0pi| [cm]  < %f\n",fD0toKpiCuts[6]);
//     printf("    d0d0  [cm^2] < %f\n",fD0toKpiCuts[7]);
//     printf("    cosThetaPoint    > %f\n",fD0toKpiCuts[8]);
  
    Int_t ptbin=0;
    Bool_t isInRange=kFALSE;
    
    //cout<<"P_t = "<<pt<<endl;
    if (pt>0. && pt<=1.) {
      ptbin=1;
      isInRange=kTRUE;
      /*
      //test d0 cut
      fVHFPPR->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.1,-0.0002,0.5);
      fVHFmycuts->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.1,-0.00025,0.7);
      */
      //original
      fVHFPPR->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.05,-0.0002,0.5);
      fVHFmycuts->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.05,-0.00025,0.7);
      
//       fVHFPPR->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.05,-0.0002,0.7);
//       fVHFmycuts->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,1,1,-0.00015,0.5);
      //printf("I'm in the bin %d\n",ptbin);
      
    }
    
    if(pt>1. && pt<=3.) {
      if(pt>1. && pt<=2.) ptbin=2;  
      if(pt>2. && pt<=3.) ptbin=3;  
      isInRange=kTRUE;
      /*
      //test d0 cut
      fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.1,-0.0002,0.6);
      fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,1,0.1,-0.00025,0.8);
      */
      //original
      fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.0002,0.6);
      fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,1,1,-0.00025,0.8);
      
      //printf("I'm in the bin %d\n",ptbin);
    }
 
    if(pt>3. && pt<=5.){
      ptbin=4;  
      isInRange=kTRUE;
      /*
      //test d0 cut
      fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.1,-0.0001,0.8);
      fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.1,-0.00015,0.8);
      */
      //original
      fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.0001,0.8);
      fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.00015,0.8);
      
      //printf("I'm in the bin %d\n",ptbin);
    }
    if(pt>5.&& pt<=10.){ //additional upper limit to compare with Correction Framework
      ptbin=5;
      isInRange=kTRUE;
      /*
      //test d0 cut
      fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.1,-0.00005,0.8);
      fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.1,-0.00015,0.9);
      */
      //original
      fVHFPPR->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.00005,0.8);
      fVHFmycuts->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.00015,0.9);
      
    }//if(pt>5)    if (pt>0. && pt<=1.) {
    //printf("I'm in the bin %d\n",ptbin);
    //old
    //fVHF->SetD0toKpiCuts(0.7,0.03,0.8,0.06,0.06,0.05,0.05,-0.0002,0.6); //2.p-p vertex reconstructed    
    if(prongsinacc && isInRange){
      FillHists(ptbin,d,mcArray,fVHFPPR,fOutputPPR);
      FillHists(ptbin,d,mcArray,fVHFmycuts,fOutputmycuts);
    }
    if(unsetvtx) d->UnsetOwnPrimaryVtx();
  } //end for prongs
  

     
  
   
  // Post the data
  PostData(1,fOutputPPR);
  PostData(2,fOutputmycuts);
  PostData(4,fDistr);
  PostData(5,fChecks);
  //cout<<"Other PostData"<<endl;
  return;
}
//____________________________________________________________________________*
void AliAnalysisTaskSED0Mass::FillHists(Int_t ptbin, AliAODRecoDecayHF2Prong *part, TClonesArray *arrMC, AliAnalysisVertexingHF *vhf, TList *listout){
  //
  // function used in UserExec:
  //


  Double_t mPDG=TDatabasePDG::Instance()->GetParticle(421)->Mass();

  Int_t okD0=0,okD0bar=0;
  //cout<<"inside FillHist"<<endl;
  if(part->SelectD0(vhf->GetD0toKpiCuts(),okD0,okD0bar)) {//selected
    Double_t invmassD0 = part->InvMassD0(), invmassD0bar = part->InvMassD0bar();
    //printf("SELECTED\n");


    AliAODTrack *prong=(AliAODTrack*)part->GetDaughter(0);
    if(!prong) cout<<"No daughter found";
    else{
      if(prong->Charge()==1) {fTotPosPairs[ptbin-1]++;} else {fTotNegPairs[ptbin-1]++;}
    }

    TString fillthis="";
    Int_t pdgDgD0toKpi[2]={321,211};
    Int_t labD0=-1;
    if (fReadMC) labD0 = part->MatchToMC(421,arrMC,2,pdgDgD0toKpi); //return MC particle label if the array corresponds to a D0, -1 if not (cf. AliAODRecoDecay.cxx)

    //count candidates selected by cuts
    fNentries->Fill(1);
    //count true D0 selected by cuts
    if (fReadMC && labD0>=0) fNentries->Fill(2);
    PostData(3,fNentries);

    if (okD0==1) {
      fillthis="histMass_";
      fillthis+=ptbin;
      //cout<<"Filling "<<fillthis<<endl;

      //printf("Fill mass with D0");
      ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
 
      if(labD0>=0) {
	if(fArray==1) cout<<"LS signal ERROR"<<endl;

	AliAODMCParticle *partD0 = (AliAODMCParticle*)arrMC->At(labD0);
	Int_t pdgD0 = partD0->GetPdgCode();
	//cout<<"pdg = "<<pdgD0<<endl;
	if (pdgD0==421){ //D0
	  //cout<<"Fill S with D0"<<endl;
	  fillthis="histSgn_";
	  fillthis+=ptbin;
	  ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
	  if(TMath::Abs(invmassD0 - mPDG) < 0.027){
	    fillthis="histSgn27_";
	    fillthis+=ptbin;
	    ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
	  }
	} else{ //it was a D0bar
	  fillthis="histRfl_";
	  fillthis+=ptbin;
	  ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
	}
      } else {//background
	fillthis="histBkg_";
	fillthis+=ptbin;
	((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
      }
      
    }
    if (okD0bar==1) {
      fillthis="histMass_";
      fillthis+=ptbin;
      //printf("Fill mass with D0bar");
      ((TH1F*)listout->FindObject(fillthis))->Fill(invmassD0bar);
 
      if(labD0>=0) {
	if(fArray==1) cout<<"LS signal ERROR"<<endl;
	AliAODMCParticle *partD0 = (AliAODMCParticle*)arrMC->At(labD0);
	Int_t pdgD0 = partD0->GetPdgCode();
	//cout<<" pdg = "<<pdgD0<<endl;
	if (pdgD0==-421){ //D0bar
	  fillthis="histSgn_";
	  fillthis+=ptbin;
	  ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
	  if (TMath::Abs(invmassD0bar - mPDG) < 0.027){
	    fillthis="histSgn27_";
	    fillthis+=ptbin;
	    ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
	  }

	  
	} else{
	  fillthis="histRfl_";
	  fillthis+=ptbin;
	  ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
	}
      } else {//background or LS
	fillthis="histBkg_";
	fillthis+=ptbin;
	((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
      }
    }


  } //else cout<<"NOT SELECTED"<<endl;

}
//________________________________________________________________________
void AliAnalysisTaskSED0Mass::Terminate(Option_t */*option*/)
{
  // Terminate analysis
  //
  if(fDebug > 1) printf("AnalysisTaskSED0Mass: Terminate() \n");


  fOutputPPR = dynamic_cast<TList*> (GetOutputData(1));
  if (!fOutputPPR) {     
    printf("ERROR: fOutputthight not available\n");
    return;
  }
  fOutputmycuts = dynamic_cast<TList*> (GetOutputData(2));
  if (!fOutputmycuts) {     
    printf("ERROR: fOutputmycuts not available\n");
    return;
  }
  fDistr = dynamic_cast<TList*> (GetOutputData(4));
  if (!fDistr) {
    printf("ERROR: fDistr not available\n");
    return;
  }

  fChecks = dynamic_cast<TList*> (GetOutputData(5));
  if (!fChecks) {
    printf("ERROR: fChecks not available\n");
    return;
  }

  if(fArray==1){
    for(Int_t ipt=0;ipt<4;ipt++){
      fLsNormalization = 2.*TMath::Sqrt(fTotPosPairs[ipt]*fTotNegPairs[ipt]); 


      if(fLsNormalization>0) {
	
	TString massName="histMass_";
	massName+=ipt+1;
	((TH1F*)fOutputPPR->FindObject(massName))->Scale((1/fLsNormalization)*((TH1F*)fOutputPPR->FindObject(massName))->GetEntries());
	((TH1F*)fOutputmycuts->FindObject(massName))->Scale((1/fLsNormalization)*((TH1F*)fOutputmycuts->FindObject(massName))->GetEntries());
      }
    }

    fLsNormalization = 2.*TMath::Sqrt(fTotPosPairs[4]*fTotNegPairs[4]);

    if(fLsNormalization>0) {

      TString nameDistr="hptB";
      ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
      nameDistr="hdcaB";
      ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
      nameDistr="hcosthetastarB";
      ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
      nameDistr="hd0B";
      ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
      nameDistr="hd0d0B";
      ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
      nameDistr="hcosthetapointB";
      ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
      nameDistr="hcosthpointd0d0B";
      ((TH2F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH2F*)fDistr->FindObject(nameDistr))->GetEntries());

    }
  }

  TString cvname;

  if (fArray==0){
    cvname="D0invmass";
  } else cvname="LSinvmass";

  TCanvas *cMass=new TCanvas(cvname,cvname);
  cMass->cd();
  ((TH1F*)fOutputPPR->FindObject("histMass_4"))->Draw();

  TCanvas* cStat=new TCanvas("cstat","Stat");
  cStat->cd();
  cStat->SetGridy();
  fNentries->Draw("htext0");

  return;
}