New class for D* in new CF (Alessandro)
authordainese <dainese@f7af4fe6-9843-0410-8265-dc069ae4e863>
Sun, 30 Jan 2011 21:18:18 +0000 (21:18 +0000)
committerdainese <dainese@f7af4fe6-9843-0410-8265-dc069ae4e863>
Sun, 30 Jan 2011 21:18:18 +0000 (21:18 +0000)
PWG3/CMakelibPWG3vertexingHF.pkg
PWG3/PWG3vertexingHFLinkDef.h
PWG3/vertexingHF/AliCFVertexingHFCascade.cxx [new file with mode: 0644]
PWG3/vertexingHF/AliCFVertexingHFCascade.h [new file with mode: 0644]

index db2eb03..d48507a 100644 (file)
@@ -25,7 +25,7 @@
 # SHLIBS - Shared Libraries and objects for linking (Executables only)           #
 #--------------------------------------------------------------------------------#
 
-set ( SRCS    vertexingHF/AliAODHFUtil.cxx vertexingHF/AliAODRecoDecayHF.cxx vertexingHF/AliAODRecoDecayHF2Prong.cxx  vertexingHF/AliAODRecoDecayHF3Prong.cxx vertexingHF/AliAODRecoDecayHF4Prong.cxx vertexingHF/AliAODRecoCascadeHF.cxx vertexingHF/AliAODPidHF.cxx vertexingHF/AliRDHFCuts.cxx vertexingHF/AliHFSystErr.cxx vertexingHF/AliRDHFCutsD0toKpi.cxx vertexingHF/AliRDHFCutsJpsitoee.cxx vertexingHF/AliRDHFCutsDplustoKpipi.cxx vertexingHF/AliRDHFCutsDstoKKpi.cxx vertexingHF/AliRDHFCutsLctopKpi.cxx vertexingHF/AliRDHFCutsLctoV0.cxx vertexingHF/AliRDHFCutsD0toKpipipi.cxx vertexingHF/AliRDHFCutsDStartoKpipi.cxx vertexingHF/AliAnalysisVertexingHF.cxx vertexingHF/AliAnalysisTaskSEVertexingHF.cxx vertexingHF/AliAnalysisTaskMEVertexingHF.cxx vertexingHF/AliAnalysisTaskSESelectHF.cxx vertexingHF/AliAnalysisTaskSECompareHF.cxx vertexingHF/AliAnalysisTaskSEDplus.cxx vertexingHF/AliAnalysisTaskSEDs.cxx vertexingHF/AliAnalysisTaskSELambdac.cxx vertexingHF/AliAnalysisTaskSED0Mass.cxx vertexingHF/AliAnalysisTaskSECharmFraction.cxx vertexingHF/AliCFVertexingHF.cxx vertexingHF/AliCFVertexingHF2Prong.cxx vertexingHF/AliCFVertexingHF3Prong.cxx vertexingHF/AliCFTaskVertexingHF.cxx vertexingHF/AliCFHeavyFlavourTaskMultiVarMultiStep.cxx vertexingHF/AliCFTaskForDStarAnalysis.cxx vertexingHF/AliAnalysisTaskSEDStarJets.cxx vertexingHF/AliAnalysisTaskSEDStarSpectra.cxx vertexingHF/AliAnalysisTaskSESignificance.cxx vertexingHF/AliAnalysisTaskSEHFQA.cxx vertexingHF/AliMultiDimVector.cxx vertexingHF/AliSignificanceCalculator.cxx vertexingHF/AliHFMassFitter.cxx vertexingHF/AliHFPtSpectrum.cxx vertexingHF/AliNormalizationCounter.cxx vertexingHF/AliAnalysisTaskSEMonitNorm.cxx vertexingHF/AliAnalysisTaskSEBkgLikeSignJPSI.cxx vertexingHF/AliAnalysisTaskSEBkgLikeSignD0.cxx vertexingHF/AliAnalysisTaskSEJPSItoEle.cxx vertexingHF/AliAnalysisBtoJPSItoEle.cxx vertexingHF/AliAnalysisTaskSEBtoJPSItoEle.cxx vertexingHF/AliBtoJPSItoEleCDFfitFCN.cxx vertexingHF/AliBtoJPSItoEleCDFfitHandler.cxx)
+set ( SRCS    vertexingHF/AliAODHFUtil.cxx vertexingHF/AliAODRecoDecayHF.cxx vertexingHF/AliAODRecoDecayHF2Prong.cxx  vertexingHF/AliAODRecoDecayHF3Prong.cxx vertexingHF/AliAODRecoDecayHF4Prong.cxx vertexingHF/AliAODRecoCascadeHF.cxx vertexingHF/AliAODPidHF.cxx vertexingHF/AliRDHFCuts.cxx vertexingHF/AliHFSystErr.cxx vertexingHF/AliRDHFCutsD0toKpi.cxx vertexingHF/AliRDHFCutsJpsitoee.cxx vertexingHF/AliRDHFCutsDplustoKpipi.cxx vertexingHF/AliRDHFCutsDstoKKpi.cxx vertexingHF/AliRDHFCutsLctopKpi.cxx vertexingHF/AliRDHFCutsLctoV0.cxx vertexingHF/AliRDHFCutsD0toKpipipi.cxx vertexingHF/AliRDHFCutsDStartoKpipi.cxx vertexingHF/AliAnalysisVertexingHF.cxx vertexingHF/AliAnalysisTaskSEVertexingHF.cxx vertexingHF/AliAnalysisTaskMEVertexingHF.cxx vertexingHF/AliAnalysisTaskSESelectHF.cxx vertexingHF/AliAnalysisTaskSECompareHF.cxx vertexingHF/AliAnalysisTaskSEDplus.cxx vertexingHF/AliAnalysisTaskSEDs.cxx vertexingHF/AliAnalysisTaskSELambdac.cxx vertexingHF/AliAnalysisTaskSED0Mass.cxx vertexingHF/AliAnalysisTaskSECharmFraction.cxx vertexingHF/AliCFVertexingHF.cxx vertexingHF/AliCFVertexingHF2Prong.cxx vertexingHF/AliCFVertexingHF3Prong.cxx vertexingHF/AliCFVertexingHFCascade.cxx vertexingHF/AliCFTaskVertexingHF.cxx vertexingHF/AliCFHeavyFlavourTaskMultiVarMultiStep.cxx vertexingHF/AliCFTaskForDStarAnalysis.cxx vertexingHF/AliAnalysisTaskSEDStarJets.cxx vertexingHF/AliAnalysisTaskSEDStarSpectra.cxx vertexingHF/AliAnalysisTaskSESignificance.cxx vertexingHF/AliAnalysisTaskSEHFQA.cxx vertexingHF/AliMultiDimVector.cxx vertexingHF/AliSignificanceCalculator.cxx vertexingHF/AliHFMassFitter.cxx vertexingHF/AliHFPtSpectrum.cxx vertexingHF/AliNormalizationCounter.cxx vertexingHF/AliAnalysisTaskSEMonitNorm.cxx vertexingHF/AliAnalysisTaskSEBkgLikeSignJPSI.cxx vertexingHF/AliAnalysisTaskSEBkgLikeSignD0.cxx vertexingHF/AliAnalysisTaskSEJPSItoEle.cxx vertexingHF/AliAnalysisBtoJPSItoEle.cxx vertexingHF/AliAnalysisTaskSEBtoJPSItoEle.cxx vertexingHF/AliBtoJPSItoEleCDFfitFCN.cxx vertexingHF/AliBtoJPSItoEleCDFfitHandler.cxx)
 
 string ( REPLACE ".cxx" ".h" HDRS "${SRCS}" )
 
index 3a13a41..20b631f 100644 (file)
@@ -35,6 +35,7 @@
 #pragma link C++ class AliCFVertexingHF+;
 #pragma link C++ class AliCFVertexingHF2Prong+;
 #pragma link C++ class AliCFVertexingHF3Prong+;
+#pragma link C++ class AliCFVertexingHFCascade+;
 #pragma link C++ class AliCFTaskVertexingHF+;
 #pragma link C++ class AliCFHeavyFlavourTaskMultiVarMultiStep+;
 #pragma link C++ class AliCFTaskForDStarAnalysis+;
diff --git a/PWG3/vertexingHF/AliCFVertexingHFCascade.cxx b/PWG3/vertexingHF/AliCFVertexingHFCascade.cxx
new file mode 100644 (file)
index 0000000..7d71513
--- /dev/null
@@ -0,0 +1,389 @@
+/**************************************************************************
+ * Copyright(c) 1998-2011, 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.                  *
+ **************************************************************************/
+
+//-----------------------------------------------------------------------
+// Class for HF corrections as a function of many variables and steps
+// For D* and other cascades
+// 
+// Author : A.Grelli a.grelli@uu.nl  UTECHT
+//-----------------------------------------------------------------------
+
+#include "AliAODRecoDecayHF2Prong.h"
+#include "AliAODMCParticle.h"
+#include "AliAODEvent.h"
+#include "TClonesArray.h"
+#include "AliCFVertexingHF.h"
+#include "AliESDtrack.h"
+#include "TDatabasePDG.h"
+#include "AliAODRecoCascadeHF.h"
+#include "AliCFVertexingHFCascade.h"
+#include "AliCFContainer.h"
+
+ClassImp(AliCFVertexingHFCascade)
+
+
+//_________________________________________
+  AliCFVertexingHFCascade::AliCFVertexingHFCascade(TClonesArray *mcArray, UShort_t originDselection):
+  AliCFVertexingHF(mcArray, originDselection)
+{      
+  // standard constructor
+
+  SetNProngs(3);
+}
+
+
+//_____________________________________
+AliCFVertexingHFCascade& AliCFVertexingHFCascade::operator=(const AliCFVertexingHFCascade& c)
+{
+  // operator =
+  if  (this != &c) {
+
+    AliCFVertexingHF::operator=(c);
+   
+  }
+  return *this;
+}
+
+//__________________________________________
+Bool_t AliCFVertexingHFCascade::SetRecoCandidateParam(AliAODRecoDecayHF *recoCand)
+{
+  // set the AliAODRecoDecay candidate
+  
+  Bool_t bSignAssoc = kFALSE;
+  fRecoCandidate = recoCand;
+  AliAODRecoCascadeHF* dstarD0pi = (AliAODRecoCascadeHF*)recoCand;
+
+  if (!fRecoCandidate) {
+    AliError("fRecoCandidate not found, problem in assignement\n");
+    return bSignAssoc;
+  }
+  
+  if ( fRecoCandidate->GetPrimaryVtx()) AliDebug(3,"fReco Candidate has a pointer to PrimVtx\n");
+  //if (recoCand->GetPrimaryVtx()) printf("Reco Cand has a pointer to PrimVtx\n");
+  
+  //Int_t pdgCand = 413;
+
+  Int_t pdgDgDStartoD0pi[2]={421,211};
+  Int_t pdgDgD0toKpi[2]={321,211};
+
+  Int_t nentries = fmcArray->GetEntriesFast();
+
+  AliDebug(3,Form("nentries = %d\n", nentries));
+  Int_t mcLabel =  dstarD0pi->MatchToMC(413,421,pdgDgDStartoD0pi,pdgDgD0toKpi,fmcArray); 
+  
+  if (mcLabel < 0) return bSignAssoc;
+  SetMCLabel(mcLabel);
+  fmcPartCandidate = dynamic_cast<AliAODMCParticle*>(fmcArray->At(fmcLabel)); 
+
+  if (!fmcPartCandidate){
+    AliDebug(3,"No part candidate");
+    return bSignAssoc;
+  }
+  bSignAssoc = kTRUE;
+  return bSignAssoc;
+}
+
+//______________________________________________
+Bool_t AliCFVertexingHFCascade::GetGeneratedValuesFromMCParticle(Double_t* vectorMC) 
+{
+  // 
+  // collecting all the necessary info (pt, y, cosThetaStar, ptPi, ptKa, cT) from MC particle
+  //
+       
+       Bool_t bGenValues = kFALSE;
+       
+       
+       Int_t daughter0ds = fmcPartCandidate->GetDaughter(0);
+       Int_t daughter1ds = fmcPartCandidate->GetDaughter(1);
+
+        //the D0
+       AliAODMCParticle* mcPartDaughterD0 = dynamic_cast<AliAODMCParticle*>(fmcArray->At(daughter0ds));
+       AliAODMCParticle* mcPartDaughterPis = dynamic_cast<AliAODMCParticle*>(fmcArray->At(daughter1ds));
+
+       Double_t vtx1[3] = {0,0,0};   // primary vertex         
+       Double_t vtx2daughter0[3] = {0,0,0};   // secondary vertex from daughter 0
+       Double_t vtx2daughter1[3] = {0,0,0};   // secondary vertex from daughter 1
+       fmcPartCandidate->XvYvZv(vtx1);  // cm
+
+       //D0 daughters
+       Int_t daughter0 = mcPartDaughterD0->GetDaughter(0);
+       Int_t daughter1 = mcPartDaughterD0->GetDaughter(1);
+
+       AliAODMCParticle* mcPartDaughter0 = dynamic_cast<AliAODMCParticle*>(fmcArray->At(daughter0)); //D0
+       AliAODMCParticle* mcPartDaughter1 = dynamic_cast<AliAODMCParticle*>(fmcArray->At(daughter1)); //pis
+
+       // getting vertex from daughters
+       mcPartDaughter0->XvYvZv(vtx2daughter0);  // cm
+       mcPartDaughter1->XvYvZv(vtx2daughter1);  //cm
+       if (TMath::Abs(vtx2daughter0[0] - vtx2daughter1[0])>1E-5 || TMath::Abs(vtx2daughter0[1]- vtx2daughter1[1])>1E-5 || TMath::Abs(vtx2daughter0[2] - vtx2daughter1[2])>1E-5) {
+         AliError("Daughters have different secondary vertex, skipping the track");
+         return bGenValues;
+       }
+       
+       Int_t nprongs = 2;
+       Short_t charge = 0;
+       // always instantiate the AliAODRecoDecay with the positive daughter first, the negative second
+       AliAODMCParticle* positiveDaugh = mcPartDaughter0;
+       AliAODMCParticle* negativeDaugh = mcPartDaughter1;
+       if (mcPartDaughter0->GetPdgCode()<0 && mcPartDaughter1->GetPdgCode()>0){
+               // inverting in case the positive daughter is the second one
+               positiveDaugh = mcPartDaughter1;
+               negativeDaugh = mcPartDaughter0;
+       }
+       // getting the momentum from the daughters
+       Double_t px[2] = {positiveDaugh->Px(), negativeDaugh->Px()};            
+       Double_t py[2] = {positiveDaugh->Py(), negativeDaugh->Py()};            
+       Double_t pz[2] = {positiveDaugh->Pz(), negativeDaugh->Pz()};
+       
+       Double_t d0[2] = {0.,0.};               
+       
+       AliAODRecoDecayHF* decay = new AliAODRecoDecayHF(vtx1,vtx2daughter0,nprongs,charge,px,py,pz,d0);
+       
+       Double_t cosThetaStar = 0.;
+       Double_t cosThetaStarD0 = 0.;
+       Double_t cosThetaStarD0bar = 0.;
+       cosThetaStarD0 = decay->CosThetaStar(1,421,211,321);
+       cosThetaStarD0bar = decay->CosThetaStar(0,421,321,211);
+       if (mcPartDaughterD0->GetPdgCode() == 421){  // D0
+         AliDebug(3, Form("D0, with pdgprong0 = %d, pdgprong1 = %d",mcPartDaughter0->GetPdgCode(),mcPartDaughter1->GetPdgCode()));
+         cosThetaStar = cosThetaStarD0;
+       }
+       else if (mcPartDaughterD0->GetPdgCode() == -421){  // D0bar{
+         AliDebug(3, Form("D0bar, with pdgprong0 = %d, pdgprong1 = %d",mcPartDaughter0->GetPdgCode(),mcPartDaughter1->GetPdgCode()));
+         cosThetaStar = cosThetaStarD0bar;
+       }
+       else{
+         AliWarning("There are problems!! particle was expected to be either a D0 or a D0bar, check...");
+         return vectorMC;
+       }
+       if (cosThetaStar < -1 || cosThetaStar > 1) {
+         AliWarning("Invalid value for cosine Theta star, returning");
+         return bGenValues;
+       }
+       
+       Double_t vectorD0[2] ={0.,0.};
+
+       // evaluate the correct cascade
+       if (!EvaluateIfD0toKpi(mcPartDaughterD0,vectorD0)) {
+         AliDebug(2, "Error! the D0 MC doesn't have correct daughters!!");
+         return bGenValues;  
+       }       
+
+       //ct
+       Double_t cT = decay->Ct(421);
+       // get the pT of the daughters
+       Double_t pTD0 = 0.;
+       Double_t pTpi = 0.;
+       
+       if (TMath::Abs(fmcPartCandidate->GetPdgCode()) == 413) {
+               pTD0 = mcPartDaughterD0->Pt();
+               pTpi = mcPartDaughterPis->Pt();
+       }
+
+       
+       vectorMC[0] = fmcPartCandidate->Pt();
+       vectorMC[1] = fmcPartCandidate->Y() ;
+       vectorMC[2] = cosThetaStar ;
+       vectorMC[3] = vectorD0[0]; 
+       vectorMC[4] = vectorD0[1];
+       vectorMC[5] = cT*1.E4 ;  // in micron
+       vectorMC[6] = 0.;   // dummy value, meaningless in MC
+       vectorMC[7] = 0.;   // dummy value, meaningless in MC, in micron
+       vectorMC[8] = 0.;   // dummy value, meaningless in MC, in micron
+       vectorMC[9] = -100000.; // dummy value, meaningless in MC, in micron^2
+       vectorMC[10] = 1.01;    // dummy value, meaningless in MC
+       vectorMC[11] = fmcPartCandidate->Phi(); 
+       vectorMC[12] = fzMCVertex;    // z of reconstructed of primary vertex
+       
+
+       delete decay;
+       bGenValues = kTRUE;
+       return bGenValues;
+}
+//____________________________________________
+Bool_t AliCFVertexingHFCascade::GetRecoValuesFromCandidate(Double_t *vectorReco) const
+{ 
+  // read the variables for the container
+
+  Bool_t bFillRecoValues=kFALSE;
+  
+  //Get the D* and the D0 from D*
+  AliAODRecoCascadeHF* dstarD0pi = (AliAODRecoCascadeHF*)fRecoCandidate;
+  AliAODRecoDecayHF2Prong* d0toKpi = (AliAODRecoDecayHF2Prong*)dstarD0pi->Get2Prong();
+  
+
+  if (dstarD0pi->GetPrimaryVtx())printf("dstarD0pi has primary vtx\n");
+  if (fRecoCandidate->GetPrimaryVtx())printf("fRecoCandidateDstar has primary vtx\n");
+
+  Double_t pt =  dstarD0pi->Pt();
+  Double_t rapidity =  dstarD0pi->YDstar();
+  Double_t invMass=0.;
+  Double_t cosThetaStar = 9999.;
+  Double_t pTpi = 0.;
+  Double_t pTK = 0.;
+  Double_t dca = d0toKpi->GetDCA();
+  Double_t d0pi = 0.;
+  Double_t d0K = 0.;
+  Double_t d0xd0 = d0toKpi->Prodd0d0();
+  Double_t cosPointingAngle = d0toKpi->CosPointingAngle();
+  Double_t phi = dstarD0pi->Phi();
+
+  Int_t pdgCode = fmcPartCandidate->GetPdgCode();
+  if (pdgCode > 0){
+    cosThetaStar = d0toKpi->CosThetaStarD0();
+    pTpi = d0toKpi->PtProng(0);
+    pTK = d0toKpi->PtProng(1);
+    d0pi = d0toKpi->Getd0Prong(0);
+    d0K = d0toKpi->Getd0Prong(1);
+    invMass=d0toKpi->InvMassD0();
+  }
+  else {
+    cosThetaStar = d0toKpi->CosThetaStarD0bar();
+    pTpi = d0toKpi->PtProng(1);
+    pTK = d0toKpi->PtProng(0);
+    d0pi = d0toKpi->Getd0Prong(1);
+    d0K = d0toKpi->Getd0Prong(0);
+    invMass= d0toKpi->InvMassD0bar();
+  }
+  
+  Double_t cT = d0toKpi->CtD0();
+  
+  vectorReco[0] = pt;
+  vectorReco[1] = rapidity;
+  vectorReco[2] = cosThetaStar;
+  vectorReco[3] = pTpi;
+  vectorReco[4] = pTK;
+  vectorReco[5] = cT*1.E4;  // in micron
+  vectorReco[6] = dca*1.E4;  // in micron
+  vectorReco[7] = d0pi*1.E4;  // in micron
+  vectorReco[8] = d0K*1.E4;  // in micron
+  vectorReco[9] = d0xd0*1.E8;  // in micron^2
+  vectorReco[10] = cosPointingAngle;  // in micron
+  vectorReco[11] = phi;  
+  vectorReco[12] = fzPrimVertex;    // z of reconstructed of primary vertex
+
+  bFillRecoValues = kTRUE;
+
+  return bFillRecoValues;
+}
+
+
+//_____________________________________________________________
+Bool_t AliCFVertexingHFCascade::CheckMCChannelDecay() const
+{ 
+  // check the required decay channel
+
+  Bool_t checkCD = kFALSE;
+
+  
+  Int_t daughter0 = fmcPartCandidate->GetDaughter(0);
+  Int_t daughter1 = fmcPartCandidate->GetDaughter(1);
+  AliAODMCParticle* mcPartDaughter0 = dynamic_cast<AliAODMCParticle*>(fmcArray->At(daughter0));
+  AliAODMCParticle* mcPartDaughter1 = dynamic_cast<AliAODMCParticle*>(fmcArray->At(daughter1));
+
+  if (!mcPartDaughter0 || !mcPartDaughter1) {
+    AliDebug (2,"Problems in the MC Daughters\n");
+    return checkCD;
+  }
+
+  if (!(TMath::Abs(mcPartDaughter0->GetPdgCode())==421 &&
+       TMath::Abs(mcPartDaughter1->GetPdgCode())==211) && 
+      !(TMath::Abs(mcPartDaughter0->GetPdgCode())==211 &&
+       TMath::Abs(mcPartDaughter1->GetPdgCode())==421)) {
+    AliDebug(2, "The D0 MC doesn't come from a Kpi decay, skipping!!");
+    return checkCD;  
+  }
+  
+  Double_t vectorD0[2] ={0.,0.};
+
+  // D* is a cascade ...evaluate the correct cascade
+  if (!EvaluateIfD0toKpi(mcPartDaughter0,vectorD0)) {
+    AliDebug(2, "Error! the D0 MC doesn't have correct daughters!!");
+    return checkCD;  
+  }
+   
+  checkCD = kTRUE;
+  return checkCD;
+  
+}
+
+//__________________________________________
+Bool_t AliCFVertexingHFCascade::EvaluateIfD0toKpi(AliAODMCParticle* neutralDaugh, Double_t* vectorD0)const
+{  
+  //
+  // chack wether D0 is decaing into kpi
+  //
+  
+  Bool_t isHadronic = kFALSE;
+  
+  Int_t daughterD00 = neutralDaugh->GetDaughter(0);
+  Int_t daughterD01 = neutralDaugh->GetDaughter(1);
+  
+  AliDebug(2, Form("daughter0 = %d and daughter1 = %d",daughterD00,daughterD01));
+  if (daughterD00 == 0 || daughterD01 == 0) {
+    AliDebug(2, "Error! the D0 MC doesn't have correct daughters!!");
+    return isHadronic;  
+  }
+  
+  if (TMath::Abs(daughterD01 - daughterD00) != 1) { // should be everytime true - see PDGdatabooklet
+    AliDebug(2, "The D0 MC doesn't come from a 2-prong decay, skipping!!");
+    return isHadronic;  
+  }
+  
+  AliAODMCParticle* mcPartDaughterD00 = dynamic_cast<AliAODMCParticle*>(fmcArray->At(daughterD00));
+  AliAODMCParticle* mcPartDaughterD01 = dynamic_cast<AliAODMCParticle*>(fmcArray->At(daughterD01));
+  if (!mcPartDaughterD00 || !mcPartDaughterD01) {
+    AliWarning("D0 MC analysis: At least one Daughter Particle not found in tree, skipping"); 
+    return isHadronic;  
+  }
+  
+  if (!(TMath::Abs(mcPartDaughterD00->GetPdgCode())==321 &&
+       TMath::Abs(mcPartDaughterD01->GetPdgCode())==211) && 
+      !(TMath::Abs(mcPartDaughterD00->GetPdgCode())==211 &&
+       TMath::Abs(mcPartDaughterD01->GetPdgCode())==321)) {
+    AliDebug(2, "The D0 MC doesn't come from a Kpi decay, skipping!!");
+    return isHadronic;  
+  }
+  
+  Double_t pTD0pi = 0;
+  Double_t pTD0K = 0;
+  
+  
+  if (TMath::Abs(mcPartDaughterD00->GetPdgCode()) == 211) {
+    pTD0pi = mcPartDaughterD00->Pt();
+    pTD0K = mcPartDaughterD01->Pt();
+  }
+  else {
+    pTD0pi = mcPartDaughterD01->Pt();
+    pTD0K  = mcPartDaughterD00->Pt();
+  }
+  
+  isHadronic = kTRUE;
+  
+  vectorD0[0] = pTD0pi;
+  vectorD0[1] = pTD0K;
+  return isHadronic;
+
+}
+
+
+
+
diff --git a/PWG3/vertexingHF/AliCFVertexingHFCascade.h b/PWG3/vertexingHF/AliCFVertexingHFCascade.h
new file mode 100644 (file)
index 0000000..58fb74f
--- /dev/null
@@ -0,0 +1,65 @@
+#ifndef ALICFVERTEXINGHFCASCADE_H
+#define ALICFVERTEXINGHFCASCADE_H
+
+/**************************************************************************
+ * Copyright(c) 1998-2011, 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.                  *
+ **************************************************************************/
+
+//-----------------------------------------------------------------------
+// Class for HF corrections as a function of many variables and steps
+// For D* and other cascades
+// 
+// Author : A.GRELLI - a.grelli@uu.nl UTRECHT
+//-----------------------------------------------------------------------
+
+
+#include "AliCFVertexingHF.h"
+#include "AliAODRecoDecayHF.h"
+#include "AliAODRecoCascadeHF.h"
+#include "AliAODRecoDecayHF2Prong.h"
+
+class AliAODMCParticle;
+class TClonesArray;
+class AliCFVertexingHF;
+class AliESDtrack;
+class TDatabasePDG;
+
+class AliCFVertexingHFCascade : public AliCFVertexingHF{
+ public:
+               
+  AliCFVertexingHFCascade(){};
+  AliCFVertexingHFCascade(TClonesArray *mcArray, UShort_t originDselection);
+       
+  //virtual ~AliCFVertexingHFCascade(){};
+  
+  
+  Bool_t GetGeneratedValuesFromMCParticle(Double_t* /*vectorMC*/);
+  Bool_t GetRecoValuesFromCandidate(Double_t* /*vectorReco*/ ) const;
+  Bool_t CheckMCChannelDecay()const;
+  
+  Bool_t SetRecoCandidateParam(AliAODRecoDecayHF *recoCand);
+  Bool_t EvaluateIfD0toKpi(AliAODMCParticle* neutralDaugh, Double_t* VectorD0)const;
+
+ protected:
+  
+  
+ private:      
+  AliCFVertexingHFCascade(const AliCFVertexingHFCascade& c);
+  AliCFVertexingHFCascade& operator= (const AliCFVertexingHFCascade& other);
+  
+  ClassDef(AliCFVertexingHFCascade, 1); // CF class for D* and other cascades
+  
+};
+
+#endif