AliDJetTTreeReader.cxx

Go to the documentation of this file.

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

#include <TFile.h>
#include <TH1D.h>
#include <TH2D.h>
#include <TChain.h>
#include <Riostream.h>

#include <AliAnalysisTaskDmesonJets.h>

#include "AliDJetTTreeReader.h"

ClassImp(AliDJetTTreeReader);

AliDJetTTreeReader::AliDJetTTreeReader():
  AliDJetVReader(),
  fInputFileNames(),
  fTreeName(),
  fDBranchName(),
  fJetBranchName(),
  fmassmin(0),
  fmassmax(0),
  fmasswidth(0)
{
}

AliDJetTTreeReader::AliDJetTTreeReader(const AliDJetTTreeReader &source):
    AliDJetVReader(source),
  fInputFileNames(source.fInputFileNames),
  fTreeName(source.fTreeName),
  fDBranchName(source.fDBranchName),
  fJetBranchName(source.fJetBranchName),
  fmassmin(source.fmassmin),
  fmassmax(source.fmassmax),
  fmasswidth(source.fmasswidth)
{
}

AliDJetTTreeReader::~AliDJetTTreeReader()
{
}

TChain* AliDJetTTreeReader::GenerateChain()
{
  Printf("Generating the chain...");
  TChain* chain = new TChain(fTreeName);
  for (std::string fname : fInputFileNames) {
    chain->Add(fname.c_str());
  }
  return chain;
}

Bool_t AliDJetTTreeReader::ExtractInputMassPlotEffScale()
{
  std::cout << "Extracting input mass plot: " << fpTmin << " to " << fpTmax << std::endl;

  TTree *tree = GenerateChain();
  if (!tree) {
    std::cout << "Error in setting the tree/branch names! Exiting..." << std::endl;
    return kFALSE;
  }

  AliAnalysisTaskDmesonJets::AliD0InfoSummary *brD = 0;
  AliAnalysisTaskDmesonJets::AliJetInfoSummary *brJet = 0;
  tree->SetBranchAddress(fDBranchName,&brD);
  tree->SetBranchAddress(fJetBranchName,&brJet);

  TString hname = TString::Format("invMass_JetPt_%.0f_%.0f", fpTmin, fpTmax);
  fMassPlot = new TH1D(hname,hname, (fmassmax-fmassmin) / fmasswidth / fMassRebin, fmassmin, fmassmax);
  fMassPlot->Sumw2();

  for (int k = 0; k < tree->GetEntries(); k++) {
    tree->GetEntry(k);
    if (brJet->fEta < -0.5 || brJet->fEta >= 0.5) continue;
    if (brJet->fPt < fpTmin || brJet->fPt >= fpTmax) continue;
    int j = 0;
    while (brD->fPt >= fDbinpTedges[j + 1] && j < fnDbins) j++;
    if (brD->fPt < fDbinpTedges[j] || brD->fPt >= fDbinpTedges[j + 1]) continue;
    fMassPlot->Fill(brD->fInvMass, 1. / fDEffValues[j]);
  }

  return kTRUE;
}

Bool_t AliDJetTTreeReader::ExtractInputMassPlotSideband()
{
  std::cout << "Extracting input mass plot: " << fpTmin << " to " << fpTmax << std::endl;

  TTree *tree = GenerateChain();
  if (!tree) {
    std::cout << "Error in setting the tree/branch names! Exiting..." << std::endl;
    return kFALSE;
  }

  AliAnalysisTaskDmesonJets::AliD0InfoSummary *brD = 0;
  AliAnalysisTaskDmesonJets::AliJetInfoSummary *brJet = 0;
  tree->SetBranchAddress(fDBranchName,&brD);
  tree->SetBranchAddress(fJetBranchName,&brJet);

  TString hname = TString::Format("invMass_DPt_%.0f_%.0f", fpTmin, fpTmax);
  fMassPlot = new TH1D(hname,hname, (fmassmax-fmassmin) / fmasswidth / fMassRebin, fmassmin, fmassmax);
  fMassPlot->Sumw2();

  if (fnJetPtbins > 1) {
    fMassVsJetPtPlot = new TH2D("hInvMassJetPt", "hInvMassJetPt", (fmassmax-fmassmin) / fmasswidth, fmassmin, fmassmax, fnJetPtbins, fJetPtBinEdges);
    fMassVsJetPtPlot->Sumw2();
  }

  if (fnJetzbins > 1) {
    fMassVsJetzPlot = new TH2D("hInvMassJetz", "hInvMassJetz", (fmassmax-fmassmin) / fmasswidth, fmassmin, fmassmax, fnJetzbins, fJetzBinEdges);
    fMassVsJetzPlot->Sumw2();
  }

  if (fpTmin < fDbinpTedges[0] || fpTmax > fDbinpTedges[fnDbins]) {
    AliErrorStream() << "Efficiency bin edges " << fDbinpTedges[0] << ", " << fDbinpTedges[fnDbins] <<
        " do not include the current bin " << fpTmin << ", " << fpTmax << std::endl;
    throw;
  }

  for (int k = 0; k < tree->GetEntries(); k++) {
    tree->GetEntry(k);
    if (brJet->fEta < -0.5 || brJet->fEta >= 0.5) continue;
    if (fnJetPtbins > 0 && (brJet->fPt < fJetPtBinEdges[0] || brJet->fPt >= fJetPtBinEdges[fnJetPtbins])) continue;
    if (brD->fPt < fpTmin || brD->fPt >= fpTmax) continue;
    Double_t w = 1.0;
    if (fEfficiencyWeightSB) {
      int j = 0;
      while (brD->fPt >= fDbinpTedges[j + 1] && j < fnDbins) j++;
      w /= fDEffValues[j];
    }
    fMassPlot->Fill(brD->fInvMass, w);
    if (fMassVsJetPtPlot) fMassVsJetPtPlot->Fill(brD->fInvMass, brJet->fPt, w);
    if (fMassVsJetzPlot) fMassVsJetzPlot->Fill(brD->fInvMass, brJet->fZ, w);
  }

  return kTRUE;
}