Updated macros for D0 analysis (Massimo)

diff --git a/PWGLF/RESONANCES/macros/mini/AddAnalysisTaskD0.C b/PWGLF/RESONANCES/macros/mini/AddAnalysisTaskD0.C
index 1bb4aa9f156d89aa5e4ddb5e477103ebbbd35483..b9eb0d954b50db1dc4199fdee6dbc280981e8c22 100644 (file)
--- a/PWGLF/RESONANCES/macros/mini/AddAnalysisTaskD0.C
+++ b/PWGLF/RESONANCES/macros/mini/AddAnalysisTaskD0.C
@@ -30,6 +30,10 @@ AliRsnMiniAnalysisTask * AddAnalysisTaskD0
    Int_t       NTPCcluster = 70,
    Double_t    minpt = 0.15,
    TString     triggerMask = AliVEvent::kMB,
+   Short_t     maxSisters = 2,
+   Bool_t      checkP = kTRUE,
+   Bool_t      minDCAcutFixed = kFALSE,
+   Bool_t      maxDCAcutFixed = kFALSE,
    Int_t       nmix = 5,
    Double_t    minYlab =  -0.5,
    Double_t    maxYlab =  0.5,
@@ -63,6 +67,11 @@ AliRsnMiniAnalysisTask * AddAnalysisTaskD0
    }
 
    task->SelectCollisionCandidates(trigger);
+   task->SetMaxNDaughters(maxSisters);
+   task->SetCheckMomentumConservation(checkP);
+   
+   ::Info("AddAnalysisTaskD0", Form("Maximum numbers of daughters allowed (-1 means cut not applied): %i",maxSisters));
+   ::Info("AddAnalysisTaskD0", Form("Are we checking the momentum conservation?: %s", checkP? "yes" : "no"));
 
 
    if (isPP) 
@@ -147,24 +156,25 @@ AliRsnMiniAnalysisTask * AddAnalysisTaskD0
    cutsPairY->ShowCuts();
    cutsPairY->PrintSetInfo();
    
-   /*AliRsnCutSet *cutsPairDCAp = new AliRsnCutSet("pairCutsDCAp", AliRsnTarget::kMother);
-   cutsPairDCAp->AddCut(cutDCAproduct);
-   cutsPairDCAp->UseMonitor(kTRUE);
-   cutsPairDCAp->SetCutScheme("setPairD0_DCAp");
-   cutsPairDCAp->ShowCuts();
-   cutsPairDCAp->PrintSetInfo();*/
+   AliRsnCutSet *cutsPair = new AliRsnCutSet("pairCuts", AliRsnTarget::kMother);
+   cutsPair->AddCut(cutY);
+   cutsPair->AddCut(cutDCAproduct);
+   cutsPair->UseMonitor(kTRUE);
+   cutsPair->SetCutScheme(Form("%s&%s", cutY->GetName(), cutDCAproduct->GetName()));
+   cutsPair->ShowCuts();
+   cutsPair->PrintSetInfo();
   
-   
+ 
    
    //
    // -- CONFIG ANALYSIS --------------------------------------------------------------------------
    gROOT->LoadMacro("$ALICE_ROOT/PWGLF/RESONANCES/macros/mini/ConfigD0.C");
 
    if (isMC) {
-       Printf("========================== MC analysis - PID cuts not used");
+       Printf("========================== MC analysis - PID cuts used");
    } else 
      Printf("========================== DATA analysis - PID cuts used");
-   if (!ConfigD0(task, isPP, isMC, nsigmaTPCPi, nsigmaTPCKa, nsigmaTOFPi, nsigmaTOFKa, aodFilterBit, trackDCAcutMax, trackDCAcutMin, NTPCcluster, minpt, "", cutsPairY)) return 0x0;
+   if (!ConfigD0(task, isPP, isMC, nsigmaTPCPi, nsigmaTPCKa, nsigmaTOFPi, nsigmaTOFKa, aodFilterBit, trackDCAcutMax, trackDCAcutMin, NTPCcluster, minpt, maxSisters, checkP,  minDCAcutFixed, maxDCAcutFixed, "", cutsPairY, cutsPair)) return 0x0;
    
    //
    // -- CONTAINERS --------------------------------------------------------------------------------

diff --git a/PWGLF/RESONANCES/macros/mini/ConfigD0.C b/PWGLF/RESONANCES/macros/mini/ConfigD0.C
index 768f5a155fe6c3e7e09088ae2d9b5f51d31c7974..640c0d80d132b7a88011a4f8754a43a3d0e8e031 100644 (file)
--- a/PWGLF/RESONANCES/macros/mini/ConfigD0.C
+++ b/PWGLF/RESONANCES/macros/mini/ConfigD0.C
@@ -21,9 +21,14 @@ Bool_t ConfigD0
    Float_t                trackDCAcutMax = 7.0,
    Float_t                trackDCAcutMin = 0.0,
    Int_t                  NTPCcluster = 70,
-   Double_t                minpt = 0.15, 
+   Double_t                minpt = 0.15,
+   Short_t                        maxSisters = 2,
+   Bool_t                  checkP = kTRUE,
+   Bool_t                  minDCAcutFixed = kFALSE,
+   Bool_t                  maxDCAcutFixed = kFALSE,
    const char            *suffix,
-   AliRsnCutSet           *cutsPairY
+   AliRsnCutSet           *cutsPairY,
+   AliRsnCutSet           *cutsPair
 )
 {
    // manage suffix
@@ -35,7 +40,7 @@ Bool_t ConfigD0
    
    TString s2 = ""; s2+=trackDCAcutMin; s2+="*(0.0015+0.0050/pt^1.01)";
 
-   const char *formula2 = s2;
+   const char *formulaMin = s2;
    
    
    // 
@@ -48,10 +53,11 @@ Bool_t ConfigD0
    cutPi->SetTOFPionPIDCut(nsigmaTOFPi);
    AliRsnCutTrackQuality *cutQuality = (AliRsnCutTrackQuality*) cutPi->CutQuality();
    cutQuality->SetCheckOnlyFilterBit(kFALSE);
-   cutQuality->SetAODTestFilterBit(aodFilterBit);               
-   cutQuality->SetDCARPtFormula(formula);
-   //cutQuality->SetDCARPtFormula2(formula2);
-   cutQuality->SetDCARmin(trackDCAcutMin);
+   cutQuality->SetAODTestFilterBit(aodFilterBit);
+   if(maxDCAcutFixed)cutQuality->SetDCARmax(trackDCAcutMax);            
+   else cutQuality->SetDCARPtFormula(formula);
+   if(minDCAcutFixed) cutQuality->SetDCARmin(trackDCAcutMin);
+   else cutQuality->SetDCARPtFormulaMin(formulaMin); 
    cutQuality->SetTPCminNClusters(NTPCcluster);
    cutQuality->SetPtRange(minpt,1E20);
    cutQuality->SetEtaRange(-0.8, 0.8);
@@ -80,10 +86,11 @@ Bool_t ConfigD0
    cutK->SetTOFKaonPIDCut(nsigmaTOFKa);        
    AliRsnCutTrackQuality *cutQuality = (AliRsnCutTrackQuality*) cutK->CutQuality();
    cutQuality->SetCheckOnlyFilterBit(kFALSE);
-   cutQuality->SetAODTestFilterBit(aodFilterBit);               
-   cutQuality->SetDCARPtFormula(formula);
-   //cutQuality->SetDCARPtFormula2(formula2);
-   cutQuality->SetDCARmin(trackDCAcutMin);
+   cutQuality->SetAODTestFilterBit(aodFilterBit);
+   if(maxDCAcutFixed)cutQuality->SetDCARmax(trackDCAcutMax);            
+   else cutQuality->SetDCARPtFormula(formula);
+   if(minDCAcutFixed) cutQuality->SetDCARmin(trackDCAcutMin);
+   else cutQuality->SetDCARPtFormulaMin(formulaMin);
    cutQuality->SetTPCminNClusters(NTPCcluster);
    cutQuality->SetPtRange(minpt,1E20);
    cutQuality->SetEtaRange(-0.8, 0.8);
@@ -117,6 +124,7 @@ Bool_t ConfigD0
   /* second daughter pt */ Int_t daug2ptID  = task->CreateValue(AliRsnMiniValue::kSecondDaughterPt, kFALSE);
   /* first daughter dca */ Int_t daug1dcaID = task->CreateValue(AliRsnMiniValue::kFirstDaughterDCA, kFALSE);
   /* second daughter dca*/ Int_t daug2dcaID = task->CreateValue(AliRsnMiniValue::kSecondDaughterDCA, kFALSE);
+  /* number of Sisters  */ Int_t nsistID    = task->CreateValue(AliRsnMiniValue::kNSisters, kFALSE);
    
    //
    // -- Create all needed outputs -----------------------------------------------------------------
@@ -127,19 +135,19 @@ Bool_t ConfigD0
    // [1] = mixing
    // [2] = like ++
    // [3] = like --
-   Bool_t   use     [14] = { 1       ,  1       ,  1       ,  1       ,  1       ,  1        ,  1        ,  1        ,  1       ,  1       ,  isMC    ,   isMC    ,  isMC    ,   isMC   };
-   Bool_t   useIM   [14] = { 1       ,  1       ,  1       ,  1       ,  1       ,  1        ,  1        ,  1        ,  1       ,  1       ,  1       ,   1       ,  0       ,   0      };
-   TString  name    [14] = {"Unlike1", "Unlike2", "Mixing1", "Mixing2", "RotateK1", "RotateP1", "RotateK2", "RotateP2", "LikePP" , "LikeMM" , "Trues1" ,  "Trues2" , "Res1"   ,  "Res2"  };
-   TString  comp    [14] = {"PAIR"   , "PAIR"   , "MIX"    , "MIX"    , "ROTATE1" , "ROTATE2" , "ROTATE1" , "ROTATE2" , "PAIR"   , "PAIR"   , "TRUE"   ,  "TRUE"   , "TRUE"   ,  "TRUE"  };
-   TString  output  [14] = {"SPARSE" , "SPARSE" , "SPARSE" , "SPARSE" , "SPARSE"  , "SPARSE"  , "SPARSE"  , "SPARSE"  , "SPARSE" , "SPARSE" , "SPARSE" ,  "SPARSE" , "SPARSE" ,  "SPARSE"  };
-   Char_t   charge1 [14] = {'+'      , '-'      , '+'      , '-'      , '+'      , '+'       , '-'       , '-'       , '+'      , '-'      , '+'      ,  '-'      , '+'      ,  '-'     };
-   Char_t   charge2 [14] = {'-'      , '+'      , '-'      , '+'      , '-'      , '-'       , '+'       , '+'       , '+'      , '-'      , '-'      ,  '+'      , '-'      ,  '+'     };
-   Int_t    cutID1  [14] = { iCutK   ,  iCutK   ,  iCutK   ,  iCutK   ,  iCutK    ,  iCutK    ,  iCutK    ,  iCutK    ,  iCutK   ,  iCutK   ,  iCutK   ,   iCutK   ,  iCutK   ,   iCutK  };
-   Int_t    cutID2  [14] = { iCutPi  ,  iCutPi  ,  iCutPi  ,  iCutPi  ,  iCutPi   ,  iCutPi   ,  iCutPi   ,  iCutPi   ,  iCutPi  ,  iCutPi  ,  iCutPi  ,   iCutPi  ,  iCutPi  ,   iCutPi };
-   Int_t    ipdg    [14] = { 421     , -421     ,  421     , -421     ,  421     ,  421      , -421      , -421      ,  421     , -421     ,  421     , -421      ,  421     ,  -421    };
-   Double_t mass    [14] = { 1.86486 ,  1.86486 ,  1.86486 ,  1.86486 ,  1.86486  ,  1.86486  ,  1.86486  ,  1.86486  ,  1.86486 ,  1.86486 ,  1.86486 ,  1.86486  ,  1.86486 ,  1.86486 };
-   
-   for (Int_t i = 0; i < 14; i++) {
+   Bool_t   use     [10] = { 1       ,  1       ,  1       ,  1       ,  1       ,  1        ,  1        ,  1        ,  1       ,  1       };
+   Bool_t   useIM   [10] = { 1       ,  1       ,  1       ,  1       ,  1       ,  1        ,  1        ,  1        ,  1       ,  1       };
+   TString  name    [10] = {"Unlike1", "Unlike2", "Mixing1", "Mixing2", "RotateK1", "RotateP1", "RotateK2", "RotateP2", "LikePP" , "LikeMM" };
+   TString  comp    [10] = {"PAIR"   , "PAIR"   , "MIX"    , "MIX"    , "ROTATE1" , "ROTATE2" , "ROTATE1" , "ROTATE2" , "PAIR"   , "PAIR"   };
+   TString  output  [10] = {"SPARSE" , "SPARSE" , "SPARSE" , "SPARSE" , "SPARSE"  , "SPARSE"  , "SPARSE"  , "SPARSE"  , "SPARSE" , "SPARSE" };
+   Char_t   charge1 [10] = {'+'      , '-'      , '+'      , '-'      , '+'      , '+'       , '-'       , '-'       , '+'      , '-'      };
+   Char_t   charge2 [10] = {'-'      , '+'      , '-'      , '+'      , '-'      , '-'       , '+'       , '+'       , '+'      , '-'      };
+   Int_t    cutID1  [10] = { iCutK   ,  iCutK   ,  iCutK   ,  iCutK   ,  iCutK    ,  iCutK    ,  iCutK    ,  iCutK    ,  iCutK   ,  iCutK   };
+   Int_t    cutID2  [10] = { iCutPi  ,  iCutPi  ,  iCutPi  ,  iCutPi  ,  iCutPi   ,  iCutPi   ,  iCutPi   ,  iCutPi   ,  iCutPi  ,  iCutPi  };
+   Int_t    ipdg    [10] = { 421     , -421     ,  421     , -421     ,  421     ,  421      , -421      , -421      ,  421     , -421     };
+   Double_t mass    [10] = { 1.86486 ,  1.86486 ,  1.86486 ,  1.86486 ,  1.86486  ,  1.86486  ,  1.86486  ,  1.86486  ,  1.86486 ,  1.86486 };
+   
+   for (Int_t i = 0; i < 10; i++) {
       if (!use[i]) continue;
       
       // create output
@@ -154,38 +162,30 @@ Bool_t ConfigD0
       out->SetMotherPDG(ipdg[i]);
       out->SetMotherMass(mass[i]);
       // pair cuts
-      out->SetPairCuts(cutsPairY);
+      out->SetPairCuts(cutsPair);
 
       // axis X: invmass (or resolution)
-      if (useIM[i]) 
+      //if (useIM[i]) 
          out->AddAxis(imID, 800, 1.4, 2.2);
-      else
-         out->AddAxis(resID, 200, -0.02, 0.02);
+      //else
+      //   out->AddAxis(resID, 200, -0.02, 0.02);
       // axis Y: transverse momentum
       out->AddAxis(ptID, 100, 0.0, 10.0);
       
       // axiz Z: rapidity
       //out->AddAxis(yID, 100, -1, 1);
       
-      // more axis: daughter's dca product
-      out->AddAxis(dcapID, 100, -0.001, 0);
-      
+      // more axis: daughter's dca product and more
+      //out->AddAxis(dcapID, 100, -0.001, 0.001);      
       //out->AddAxis(daug1ptID, 150, 0.0, 15.0);
-     
       //out->AddAxis(daug2ptID, 150, 0.0, 15.0);
-      
       //out->AddAxis(daug1dcaID, 200, -1.0, 1.0);
-      
-      //out->AddAxis(daug2dcaID, 200, -1.0, 1.0);
-      
+      //out->AddAxis(daug2dcaID, 200, -1.0, 1.0);    
       
       if (!isPP) out->AddAxis(centID, 100, 0.0, 100.0);
    }
    
    
-   /*AddMonitorOutput_PairY(cutsPairY->GetMonitorOutput());
-   AddMonitorOutput_PairDCAProduct(cutsPairDCAp->GetMonitorOutput());*/
-   
    AddMonitorOutput_PionEta(cutSetPi->GetMonitorOutput());
    AddMonitorOutput_PionY(cutSetPi->GetMonitorOutput());
    AddMonitorOutput_PionMinPt(cutSetPi->GetMonitorOutput());
@@ -202,14 +202,116 @@ Bool_t ConfigD0
    AddMonitorOutput_KaonTOF_PIDCut(cutSetK->GetMonitorOutput());
    AddMonitorOutput_KaonNTPC(cutSetK->GetMonitorOutput());
    
+   if (isMC) {
+   
+   // TRUE RECONSTRUCTED PAIRS
+   
+   TString mode = "SPARSE";
+   
+   // create output
+   AliRsnMiniOutput *out = task->CreateOutput("D0_True1", mode.Data(), "TRUE");
+   // selection settings
+   out->SetCutID(0, iCutK);
+   out->SetCutID(1, iCutPi);
+   out->SetDaughter(0, AliRsnDaughter::kKaon);
+   out->SetDaughter(1, AliRsnDaughter::kPion);
+   out->SetCharge(0, '-');
+   out->SetCharge(1, '+');
+   out->SetMotherPDG(421);
+   out->SetMotherMass(1.86486);
+   // pair cuts
+   out->SetPairCuts(cutsPair);
+   out->SetMaxNSisters(maxSisters);
+   out->SetCheckMomentumConservation(checkP);
+   // binnings
+   out->AddAxis(imID, 3000, 0., 3.);
+   out->AddAxis(ptID, 100, 0.0, 10.0);
+   //out->AddAxis(yID, 100, -1, 1);
+   //out->AddAxis(dcapID, 100, -0.001, 0.001);
+   //out->AddAxis(nsistID, 10, 0, 5);
 
+   if (!isPP) out->AddAxis(centID, 100, 0.0, 100.0);
+   
+   // create output
+   AliRsnMiniOutput *out = task->CreateOutput("D0_True2", mode.Data(), "TRUE");
+   // selection settings
+   out->SetCharge(0, '+');
+   out->SetCharge(1, '-');
+   out->SetDaughter(0, AliRsnDaughter::kKaon);
+   out->SetDaughter(1, AliRsnDaughter::kPion);
+   out->SetCharge(0, '+');
+   out->SetCharge(1, '-');
+   out->SetMotherPDG(-421);
+   out->SetMotherMass(1.86486);
+   // pair cuts
+   out->SetPairCuts(cutsPair);
+   out->SetMaxNSisters(maxSisters);
+   out->SetCheckMomentumConservation(checkP);
+   // binnings
+   out->AddAxis(imID,3000, 0., 3.);
+   out->AddAxis(ptID, 100, 0.0, 10.0);
+   //out->AddAxis(yID, 100, -1, 1);
+   //out->AddAxis(dcapID, 100, -0.001, 0.001);
+   //out->AddAxis(nsistID, 10, 0, 5);
+
+   if (!isPP) out->AddAxis(centID, 100, 0.0, 100.0);
    
-   if (isMC) {
+   
+   // INVARIANT RESOLUTION
    
    TString mode = "SPARSE";
    
    // create output
-   AliRsnMiniOutput *out = task->CreateOutput(Form("D0_TrueMC%s", suffix), mode.Data(), "MOTHER");
+   AliRsnMiniOutput *out = task->CreateOutput("D0_Res1", mode.Data(), "TRUE");
+   // selection settings
+   out->SetDaughter(0, AliRsnDaughter::kKaon);
+   out->SetDaughter(1, AliRsnDaughter::kPion);
+   out->SetCharge(0, '-');
+   out->SetCharge(1, '+');
+   out->SetMotherPDG(421);
+   out->SetMotherMass(1.86486);
+   // pair cuts
+   out->SetPairCuts(cutsPair);
+   out->SetMaxNSisters(maxSisters);
+   out->SetCheckMomentumConservation(checkP);
+   // binnings
+   out->AddAxis(resID, 1000, -0.1, 0.1);
+   out->AddAxis(ptID, 100, 0.0, 10.0);
+   //out->AddAxis(yID, 100, -1, 1);
+   //out->AddAxis(dcapID, 100, -0.001, 0.001);
+   //out->AddAxis(nsistID, 10, 0, 5);
+
+   if (!isPP) out->AddAxis(centID, 100, 0.0, 100.0);
+   
+   // create output
+   AliRsnMiniOutput *out = task->CreateOutput("D0_Res2", mode.Data(), "TRUE");
+   // selection settings
+   out->SetDaughter(0, AliRsnDaughter::kKaon);
+   out->SetDaughter(1, AliRsnDaughter::kPion);
+   out->SetCharge(0, '+');
+   out->SetCharge(1, '-');
+   out->SetMotherPDG(-421);
+   out->SetMotherMass(1.86486);
+   // pair cuts
+   out->SetPairCuts(cutsPair);
+   out->SetMaxNSisters(maxSisters);
+   out->SetCheckMomentumConservation(checkP);
+   // binnings
+   out->AddAxis(resID, 1000, -0.1, 0.1);
+   out->AddAxis(ptID, 100, 0.0, 10.0);
+   //out->AddAxis(yID, 100, -1, 1);
+   //out->AddAxis(dcapID, 100, -0.001, 0.001);
+   //out->AddAxis(nsistID, 10, 0, 5);
+
+   if (!isPP) out->AddAxis(centID, 100, 0.0, 100.0);
+   
+   
+   // GENERATED MOTHERS
+   
+   TString mode = "SPARSE";
+   
+   // create output
+   AliRsnMiniOutput *out = task->CreateOutput("D0_TrueMC1", mode.Data(), "MOTHER");
    // selection settings
    out->SetDaughter(0, AliRsnDaughter::kKaon);
    out->SetDaughter(1, AliRsnDaughter::kPion);
@@ -218,14 +320,14 @@ Bool_t ConfigD0
    // pair cuts
    out->SetPairCuts(cutsPairY);
    // binnings
-   out->AddAxis(imID, 800, 1.4, 2.2);
+   out->AddAxis(imID, 3000, 0.0, 3.0);
    out->AddAxis(ptID, 100, 0.0, 10.0);
    //out->AddAxis(yID, 100, -1, 1);
 
    if (!isPP) out->AddAxis(centID, 100, 0.0, 100.0);
    
    // create output
-   AliRsnMiniOutput *out = task->CreateOutput(Form("D0bar_TrueMC%s", suffix), mode.Data(), "MOTHER");
+   AliRsnMiniOutput *out = task->CreateOutput("D0_TrueMC2", mode.Data(), "MOTHER");
    // selection settings
    out->SetDaughter(0, AliRsnDaughter::kKaon);
    out->SetDaughter(1, AliRsnDaughter::kPion);
@@ -234,7 +336,7 @@ Bool_t ConfigD0
    // pair cuts
    out->SetPairCuts(cutsPairY);
    // binnings
-   out->AddAxis(imID, 800, 1.4, 2.2);
+   out->AddAxis(imID, 3000, 0, 3.0);
    out->AddAxis(ptID, 100, 0.0, 10.0);
    //out->AddAxis(yID, 100, -1, 1);
 
@@ -243,43 +345,10 @@ Bool_t ConfigD0
    
    }
 
-   return kTRUE;
-}
 
-/*void AddMonitorOutput_PairY(TObjArray *mon=0,TString opt="",AliRsnLoopPair *pairy=0)
-{
-
-   // Pair Y
-   AliRsnValuePair *axisPairY = new AliRsnValuePair("pair_y", AliRsnValuePair::kY);
-   axisPairY->SetBins(-1.0,1.0,0.001);
-
-   // output: 2D histogram
-   AliRsnListOutput *outMonitorPairY = new AliRsnListOutput("Pair_Y", AliRsnListOutput::kHistoDefault);
-   outMonitorPairY->AddValue(axisPairY);
-
-   // add outputs to loop
-   if (mon) mon->Add(outMonitorPairY);
-   if (pairy) pairy->AddOutput(outMonitorPairY);
-  
+   return kTRUE;
 }
 
-void AddMonitorOutput_PairDCAProduct(TObjArray *mon=0,TString opt="",AliRsnLoopPair*pairdcaproduct=0)
-{
-
-   // Pair DCA Product
-   AliRsnValuePair *axisPairDCAProduct = new AliRsnValuePair("pair_dcaproduct", AliRsnValuePair::kDCAproduct);
-   axisPairDCAProduct->SetBins(-1.0,1.0,0.001);
-
-   // output: 2D histogram
-   AliRsnListOutput *outMonitorPairDCAProduct = new AliRsnListOutput("Pair_DCAProduct", AliRsnListOutput::kHistoDefault);
-   outMonitorPairDCAProduct->AddValue(axisPairDCAProduct);
-
-   // add outputs to loop
-   if (mon) mon->Add(outMonitorPairDCAProduct);
-   if (pairdcaproduct) pairdcaproduct->AddOutput(outMonitorPairDCAProduct);
-  
-}*/
-
 
 void AddMonitorOutput_PionEta(TObjArray *mon=0,TString opt="",AliRsnLoopDaughter *peta=0)
 {