[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / UserTasks / AliAnalysisTaskEmcalJetMass.cxx

//
// Jet mass analysis task.
//
// Author: M.Verweij

#include <TClonesArray.h>
#include <TH1F.h>
#include <TH2F.h>
#include <TH3F.h>
#include <THnSparse.h>
#include <TList.h>
#include <TLorentzVector.h>
#include <TProfile.h>
#include <TChain.h>
#include <TSystem.h>
#include <TFile.h>
#include <TKey.h>

#include "AliVCluster.h"
#include "AliVTrack.h"
#include "AliEmcalJet.h"
#include "AliRhoParameter.h"
#include "AliLog.h"
#include "AliEmcalParticle.h"
#include "AliMCEvent.h"
#include "AliGenPythiaEventHeader.h"
#include "AliAODMCHeader.h"
#include "AliMCEvent.h"
#include "AliAnalysisManager.h"
#include "AliJetContainer.h"

#include "AliAODEvent.h"

#include "AliAnalysisTaskEmcalJetMass.h"

ClassImp(AliAnalysisTaskEmcalJetMass)

//________________________________________________________________________
AliAnalysisTaskEmcalJetMass::AliAnalysisTaskEmcalJetMass() : 
  AliAnalysisTaskEmcalJet("AliAnalysisTaskEmcalJetMass", kTRUE),
  fContainerBase(0),
  fMinFractionShared(0),
  fh2PtJet1VsLeadPtAllSel(0),
  fh2PtJet1VsLeadPtTagged(0),
  fh2PtVsMassJet1All(0),
  fh2PtVsMassJet1Tagged(0),
  fpPtVsMassJet1All(0),
  fpPtVsMassJet1Tagged(0),
  fh2MassVsAreaJet1All(0),
  fh2MassVsAreaJet1Tagged(0),
  fh2MassVsNConstJet1All(0),
  fh2MassVsNConstJet1Tagged(0),
  fh2EtMassOverEtRSq(0)
{
  // Default constructor.

  fh2PtJet1VsLeadPtAllSel      = new TH2F*[fNcentBins];
  fh2PtJet1VsLeadPtTagged      = new TH2F*[fNcentBins];
  fh2PtVsMassJet1All           = new TH2F*[fNcentBins];
  fh2PtVsMassJet1Tagged        = new TH2F*[fNcentBins];
  fpPtVsMassJet1All            = new TProfile*[fNcentBins];
  fpPtVsMassJet1Tagged         = new TProfile*[fNcentBins];
  fh2MassVsAreaJet1All         = new TH2F*[fNcentBins];
  fh2MassVsAreaJet1Tagged      = new TH2F*[fNcentBins];
  fh2MassVsNConstJet1All       = new TH2F*[fNcentBins];
  fh2MassVsNConstJet1Tagged    = new TH2F*[fNcentBins];
  fh2EtMassOverEtRSq           = new TH2F*[fNcentBins];

  for (Int_t i = 0; i < fNcentBins; i++) {
    fh2PtJet1VsLeadPtAllSel[i]     = 0;
    fh2PtJet1VsLeadPtTagged[i]     = 0;
    fh2PtVsMassJet1All[i]          = 0;
    fh2PtVsMassJet1Tagged[i]       = 0;
    fpPtVsMassJet1All[i]           = 0;
    fpPtVsMassJet1Tagged[i]        = 0;
    fh2MassVsAreaJet1All[i]        = 0;
    fh2MassVsAreaJet1Tagged[i]     = 0;
    fh2MassVsNConstJet1All[i]      = 0;
    fh2MassVsNConstJet1Tagged[i]   = 0;
    fh2EtMassOverEtRSq[i]          = 0;
  }

  SetMakeGeneralHistograms(kTRUE);
  
}

//________________________________________________________________________
AliAnalysisTaskEmcalJetMass::AliAnalysisTaskEmcalJetMass(const char *name) : 
  AliAnalysisTaskEmcalJet(name, kTRUE),  
  fContainerBase(0),
  fMinFractionShared(0),
  fh2PtJet1VsLeadPtAllSel(0),
  fh2PtJet1VsLeadPtTagged(0),
  fh2PtVsMassJet1All(0),
  fh2PtVsMassJet1Tagged(0),
  fpPtVsMassJet1All(0),
  fpPtVsMassJet1Tagged(0),
  fh2MassVsAreaJet1All(0),
  fh2MassVsAreaJet1Tagged(0),
  fh2MassVsNConstJet1All(0),
  fh2MassVsNConstJet1Tagged(0),
  fh2EtMassOverEtRSq(0)
{
  // Standard constructor.

  fh2PtJet1VsLeadPtAllSel      = new TH2F*[fNcentBins];
  fh2PtJet1VsLeadPtTagged      = new TH2F*[fNcentBins];
  fh2PtVsMassJet1All           = new TH2F*[fNcentBins];
  fh2PtVsMassJet1Tagged        = new TH2F*[fNcentBins];
  fpPtVsMassJet1All            = new TProfile*[fNcentBins];
  fpPtVsMassJet1Tagged         = new TProfile*[fNcentBins];
  fh2MassVsAreaJet1All         = new TH2F*[fNcentBins];
  fh2MassVsAreaJet1Tagged      = new TH2F*[fNcentBins];
  fh2MassVsNConstJet1All       = new TH2F*[fNcentBins];
  fh2MassVsNConstJet1Tagged    = new TH2F*[fNcentBins];
  fh2EtMassOverEtRSq           = new TH2F*[fNcentBins];

  for (Int_t i = 0; i < fNcentBins; i++) {
    fh2PtJet1VsLeadPtAllSel[i]     = 0;
    fh2PtJet1VsLeadPtTagged[i]     = 0;
    fh2PtVsMassJet1All[i]          = 0;
    fh2PtVsMassJet1Tagged[i]       = 0;
    fpPtVsMassJet1All[i]           = 0;
    fpPtVsMassJet1Tagged[i]        = 0;
    fh2MassVsAreaJet1All[i]        = 0;
    fh2MassVsAreaJet1Tagged[i]     = 0;
    fh2MassVsNConstJet1All[i]      = 0;
    fh2MassVsNConstJet1Tagged[i]   = 0;
    fh2EtMassOverEtRSq[i]          = 0;
  }

  SetMakeGeneralHistograms(kTRUE);
}

//________________________________________________________________________
AliAnalysisTaskEmcalJetMass::~AliAnalysisTaskEmcalJetMass()
{
  // Destructor.
}

//________________________________________________________________________
void AliAnalysisTaskEmcalJetMass::UserCreateOutputObjects()
{
  // Create user output.

  AliAnalysisTaskEmcalJet::UserCreateOutputObjects();

  Bool_t oldStatus = TH1::AddDirectoryStatus();
  TH1::AddDirectory(kFALSE);

  const Int_t nBinsPt  = 250;
  const Double_t minPt = -50.;
  const Double_t maxPt = 200.;

  const Int_t nBinsArea = 100;
  const Double_t minArea = 0.;
  const Double_t maxArea = 1.;

  const Int_t nBinsNConst = 100;
  const Double_t minNConst = 0.;
  const Double_t maxNConst = 500.;

  TString histName = "";
  TString histTitle = "";
  for (Int_t i = 0; i < fNcentBins; i++) {
    histName = TString::Format("fh2PtJet1VsLeadPtAllSel_%d",i);
    histTitle = TString::Format("%s;#it{p}_{T,jet1};#it{p}_{T,lead trk}",histName.Data());
    fh2PtJet1VsLeadPtAllSel[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,20,0.,20.);
    fOutput->Add(fh2PtJet1VsLeadPtAllSel[i]);

    histName = TString::Format("fh2PtJet1VsLeadPtTagged_%d",i);
    histTitle = TString::Format("%s;#it{p}_{T,jet1};#it{p}_{T,lead trk}",histName.Data());
    fh2PtJet1VsLeadPtTagged[i] =  new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,20,0.,20.);
    fOutput->Add(fh2PtJet1VsLeadPtTagged[i]);

    histName = TString::Format("fh2PtVsMassJet1All_%d",i);
    histTitle = TString::Format("%s;#it{p}_{T,jet1};#it{M}_{jet1}",histName.Data());
    fh2PtVsMassJet1All[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,nBinsPt,minPt,maxPt);
    fOutput->Add(fh2PtVsMassJet1All[i]);

    histName = TString::Format("fh2PtVsMassJet1Tagged_%d",i);
    histTitle = TString::Format("%s;#it{p}_{T,jet1};#it{M}_{jet1}",histName.Data());
    fh2PtVsMassJet1Tagged[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,nBinsPt,minPt,maxPt);
    fOutput->Add(fh2PtVsMassJet1Tagged[i]);

    histName = TString::Format("fpPtVsMassJet1All_%d",i);
    histTitle = TString::Format("%s;#it{p}_{T,jet1};Avg #it{M}_{jet1}",histName.Data());
    fpPtVsMassJet1All[i] = new TProfile(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt);
    fOutput->Add(fpPtVsMassJet1All[i]);

    histName = TString::Format("fpPtVsMassJet1Tagged_%d",i);
    histTitle = TString::Format("%s;#it{p}_{T,jet1};Avg #it{M}_{jet1}",histName.Data());
    fpPtVsMassJet1Tagged[i] = new TProfile(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt);
    fOutput->Add(fpPtVsMassJet1Tagged[i]);

    histName = TString::Format("fh2MassVsAreaJet1All_%d",i);
    histTitle = TString::Format("%s;#it{M}_{jet1};#it{A}",histName.Data());
    fh2MassVsAreaJet1All[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,nBinsArea,minArea,maxArea);
    fOutput->Add(fh2MassVsAreaJet1All[i]);

    histName = TString::Format("fh2MassVsAreaJet1Tagged_%d",i);
    histTitle = TString::Format("%s;#it{M}_{jet1};#it{A}",histName.Data());
    fh2MassVsAreaJet1Tagged[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,nBinsArea,minArea,maxArea);
    fOutput->Add(fh2MassVsAreaJet1Tagged[i]);

    histName = TString::Format("fh2MassVsNConstJet1All_%d",i);
    histTitle = TString::Format("%s;#it{M}_{jet1};#it{N}_{constituents}",histName.Data());
    fh2MassVsNConstJet1All[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,nBinsNConst,minNConst,maxNConst);
    fOutput->Add(fh2MassVsNConstJet1All[i]);

    histName = TString::Format("fh2MassVsNConstJet1Tagged_%d",i);
    histTitle = TString::Format("%s;#it{M}_{jet1};#it{N}_{constituents}",histName.Data());
    fh2MassVsNConstJet1Tagged[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,nBinsNConst,minNConst,maxNConst);
    fOutput->Add(fh2MassVsNConstJet1Tagged[i]);

    histName = TString::Format("fh2EtMassOverEtRSq_%d",i);
    histTitle = TString::Format("%s;#it{E}_{T};(#it{M}/(#it{E}_{T}#it{R}))^{2}",histName.Data());
    fh2EtMassOverEtRSq[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,100,0.,1.);
    fOutput->Add(fh2EtMassOverEtRSq[i]);
  }

  // =========== Switch on Sumw2 for all histos ===========
  for (Int_t i=0; i<fOutput->GetEntries(); ++i) {
    TH1 *h1 = dynamic_cast<TH1*>(fOutput->At(i));
    if (h1){
      h1->Sumw2();
      continue;
    }
    THnSparse *hn = dynamic_cast<THnSparse*>(fOutput->At(i));
    if(hn)hn->Sumw2();
  }

  TH1::AddDirectory(oldStatus);

  PostData(1, fOutput); // Post data for ALL output slots > 0 here.
}

//________________________________________________________________________
Bool_t AliAnalysisTaskEmcalJetMass::Run()
{
  // Run analysis code here, if needed. It will be executed before FillHistograms().

  return kTRUE;
}

//________________________________________________________________________
Bool_t AliAnalysisTaskEmcalJetMass::FillHistograms()
{
  // Fill histograms.

  AliEmcalJet* jet1 = NULL;

  AliJetContainer *jetCont = GetJetContainer(fContainerBase);
  if(jetCont) {
    jetCont->ResetCurrentID();
    while((jet1 = jetCont->GetNextAcceptJet())) {
      Double_t fraction = jetCont->GetFractionSharedPt(jet1);
      if(fMinFractionShared>0. && fraction<fMinFractionShared) continue;

      Double_t ptJet1 = jet1->Pt() - GetRhoVal(fContainerBase)*jet1->Area();//jetCont->GetJetPtCorr(jetCont->GetCurrentID());//
      fh2PtJet1VsLeadPtAllSel[fCentBin]->Fill(ptJet1,jet1->MaxTrackPt());
      fh2PtVsMassJet1All[fCentBin]->Fill(ptJet1,jet1->M());
      fpPtVsMassJet1All[fCentBin]->Fill(ptJet1,jet1->M());
      fh2MassVsAreaJet1All[fCentBin]->Fill(jet1->M(),jet1->Area());
      fh2MassVsNConstJet1All[fCentBin]->Fill(jet1->M(),jet1->GetNumberOfConstituents());
      
      if(jet1->GetTagStatus()<1 || !jet1->GetTaggedJet())
	continue;

      fh2PtJet1VsLeadPtTagged[fCentBin]->Fill(ptJet1,jet1->MaxTrackPt());
      fh2PtVsMassJet1Tagged[fCentBin]->Fill(ptJet1,jet1->M());
      fpPtVsMassJet1Tagged[fCentBin]->Fill(ptJet1,jet1->M());
      fh2MassVsAreaJet1Tagged[fCentBin]->Fill(jet1->M(),jet1->Area());
      fh2MassVsNConstJet1Tagged[fCentBin]->Fill(jet1->M(),jet1->GetNumberOfConstituents());
      
      Double_t Et2 = jet1->M()*jet1->M() + jet1->Pt()*jet1->Pt();
      Double_t Et = 0.;    Double_t massOverEtR = 0.;
      if(Et2>0.) Et = TMath::Sqrt(Et2);
      if((Et*jetCont->GetJetRadius())>0.) 
	massOverEtR = jet1->M()/(Et*jetCont->GetJetRadius());
      fh2EtMassOverEtRSq[fCentBin]->Fill(Et,massOverEtR*massOverEtR);
    }
  }

  return kTRUE;
}

//________________________________________________________________________
Bool_t AliAnalysisTaskEmcalJetMass::RetrieveEventObjects() {
  //
  // retrieve event objects
  //

  if (!AliAnalysisTaskEmcalJet::RetrieveEventObjects())
    return kFALSE;

  return kTRUE;

}

//_______________________________________________________________________
void AliAnalysisTaskEmcalJetMass::Terminate(Option_t *) 
{
  // Called once at the end of the analysis.
}
Commit	Line	Data
8612dfc8	1	//
	2	// Jet mass analysis task.
	3	//
	4	// Author: M.Verweij
	5
	6	#include <TClonesArray.h>
	7	#include <TH1F.h>
	8	#include <TH2F.h>
	9	#include <TH3F.h>
	10	#include <THnSparse.h>
	11	#include <TList.h>
	12	#include <TLorentzVector.h>
	13	#include <TProfile.h>
	14	#include <TChain.h>
	15	#include <TSystem.h>
	16	#include <TFile.h>
	17	#include <TKey.h>
	18
	19	#include "AliVCluster.h"
	20	#include "AliVTrack.h"
	21	#include "AliEmcalJet.h"
	22	#include "AliRhoParameter.h"
	23	#include "AliLog.h"
	24	#include "AliEmcalParticle.h"
	25	#include "AliMCEvent.h"
	26	#include "AliGenPythiaEventHeader.h"
	27	#include "AliAODMCHeader.h"
	28	#include "AliMCEvent.h"
	29	#include "AliAnalysisManager.h"
	30	#include "AliJetContainer.h"
	31
	32	#include "AliAODEvent.h"
	33
	34	#include "AliAnalysisTaskEmcalJetMass.h"
	35
	36	ClassImp(AliAnalysisTaskEmcalJetMass)
	37
	38	//________________________________________________________________________
	39	AliAnalysisTaskEmcalJetMass::AliAnalysisTaskEmcalJetMass() :
	40	AliAnalysisTaskEmcalJet("AliAnalysisTaskEmcalJetMass", kTRUE),
	41	fContainerBase(0),
ba7663ad	42	fMinFractionShared(0),
8612dfc8	43	fh2PtJet1VsLeadPtAllSel(0),
	44	fh2PtJet1VsLeadPtTagged(0),
	45	fh2PtVsMassJet1All(0),
	46	fh2PtVsMassJet1Tagged(0),
	47	fpPtVsMassJet1All(0),
	48	fpPtVsMassJet1Tagged(0),
	49	fh2MassVsAreaJet1All(0),
02e33ee8	50	fh2MassVsAreaJet1Tagged(0),
ba7663ad	51	fh2MassVsNConstJet1All(0),
ba7663ad	52	fh2MassVsNConstJet1Tagged(0),
02e33ee8	53	fh2EtMassOverEtRSq(0)
8612dfc8	54	{
	55	// Default constructor.
	56
	57	fh2PtJet1VsLeadPtAllSel = new TH2F*[fNcentBins];
	58	fh2PtJet1VsLeadPtTagged = new TH2F*[fNcentBins];
	59	fh2PtVsMassJet1All = new TH2F*[fNcentBins];
	60	fh2PtVsMassJet1Tagged = new TH2F*[fNcentBins];
	61	fpPtVsMassJet1All = new TProfile*[fNcentBins];
	62	fpPtVsMassJet1Tagged = new TProfile*[fNcentBins];
	63	fh2MassVsAreaJet1All = new TH2F*[fNcentBins];
	64	fh2MassVsAreaJet1Tagged = new TH2F*[fNcentBins];
ba7663ad	65	fh2MassVsNConstJet1All = new TH2F*[fNcentBins];
ba7663ad	66	fh2MassVsNConstJet1Tagged = new TH2F*[fNcentBins];
02e33ee8	67	fh2EtMassOverEtRSq = new TH2F*[fNcentBins];
8612dfc8	68
	69	for (Int_t i = 0; i < fNcentBins; i++) {
	70	fh2PtJet1VsLeadPtAllSel[i] = 0;
	71	fh2PtJet1VsLeadPtTagged[i] = 0;
	72	fh2PtVsMassJet1All[i] = 0;
	73	fh2PtVsMassJet1Tagged[i] = 0;
	74	fpPtVsMassJet1All[i] = 0;
	75	fpPtVsMassJet1Tagged[i] = 0;
	76	fh2MassVsAreaJet1All[i] = 0;
	77	fh2MassVsAreaJet1Tagged[i] = 0;
ba7663ad	78	fh2MassVsNConstJet1All[i] = 0;
ba7663ad	79	fh2MassVsNConstJet1Tagged[i] = 0;
02e33ee8	80	fh2EtMassOverEtRSq[i] = 0;
8612dfc8	81	}
	82
	83	SetMakeGeneralHistograms(kTRUE);
	84
	85	}
	86
	87	//________________________________________________________________________
	88	AliAnalysisTaskEmcalJetMass::AliAnalysisTaskEmcalJetMass(const char *name) :
	89	AliAnalysisTaskEmcalJet(name, kTRUE),
	90	fContainerBase(0),
ba7663ad	91	fMinFractionShared(0),
8612dfc8	92	fh2PtJet1VsLeadPtAllSel(0),
	93	fh2PtJet1VsLeadPtTagged(0),
	94	fh2PtVsMassJet1All(0),
	95	fh2PtVsMassJet1Tagged(0),
	96	fpPtVsMassJet1All(0),
	97	fpPtVsMassJet1Tagged(0),
	98	fh2MassVsAreaJet1All(0),
02e33ee8	99	fh2MassVsAreaJet1Tagged(0),
ba7663ad	100	fh2MassVsNConstJet1All(0),
ba7663ad	101	fh2MassVsNConstJet1Tagged(0),
02e33ee8	102	fh2EtMassOverEtRSq(0)
8612dfc8	103	{
	104	// Standard constructor.
	105
	106	fh2PtJet1VsLeadPtAllSel = new TH2F*[fNcentBins];
	107	fh2PtJet1VsLeadPtTagged = new TH2F*[fNcentBins];
	108	fh2PtVsMassJet1All = new TH2F*[fNcentBins];
	109	fh2PtVsMassJet1Tagged = new TH2F*[fNcentBins];
	110	fpPtVsMassJet1All = new TProfile*[fNcentBins];
	111	fpPtVsMassJet1Tagged = new TProfile*[fNcentBins];
	112	fh2MassVsAreaJet1All = new TH2F*[fNcentBins];
	113	fh2MassVsAreaJet1Tagged = new TH2F*[fNcentBins];
ba7663ad	114	fh2MassVsNConstJet1All = new TH2F*[fNcentBins];
ba7663ad	115	fh2MassVsNConstJet1Tagged = new TH2F*[fNcentBins];
02e33ee8	116	fh2EtMassOverEtRSq = new TH2F*[fNcentBins];
8612dfc8	117
	118	for (Int_t i = 0; i < fNcentBins; i++) {
	119	fh2PtJet1VsLeadPtAllSel[i] = 0;
	120	fh2PtJet1VsLeadPtTagged[i] = 0;
	121	fh2PtVsMassJet1All[i] = 0;
	122	fh2PtVsMassJet1Tagged[i] = 0;
	123	fpPtVsMassJet1All[i] = 0;
	124	fpPtVsMassJet1Tagged[i] = 0;
	125	fh2MassVsAreaJet1All[i] = 0;
	126	fh2MassVsAreaJet1Tagged[i] = 0;
ba7663ad	127	fh2MassVsNConstJet1All[i] = 0;
ba7663ad	128	fh2MassVsNConstJet1Tagged[i] = 0;
02e33ee8	129	fh2EtMassOverEtRSq[i] = 0;
8612dfc8	130	}
	131
	132	SetMakeGeneralHistograms(kTRUE);
	133	}
	134
	135	//________________________________________________________________________
	136	AliAnalysisTaskEmcalJetMass::~AliAnalysisTaskEmcalJetMass()
	137	{
	138	// Destructor.
	139	}
	140
	141	//________________________________________________________________________
	142	void AliAnalysisTaskEmcalJetMass::UserCreateOutputObjects()
	143	{
	144	// Create user output.
	145
	146	AliAnalysisTaskEmcalJet::UserCreateOutputObjects();
	147
	148	Bool_t oldStatus = TH1::AddDirectoryStatus();
	149	TH1::AddDirectory(kFALSE);
	150
	151	const Int_t nBinsPt = 250;
	152	const Double_t minPt = -50.;
	153	const Double_t maxPt = 200.;
	154
	155	const Int_t nBinsArea = 100;
	156	const Double_t minArea = 0.;
	157	const Double_t maxArea = 1.;
	158
ba7663ad	159	const Int_t nBinsNConst = 100;
	160	const Double_t minNConst = 0.;
	161	const Double_t maxNConst = 500.;
	162
8612dfc8	163	TString histName = "";
	164	TString histTitle = "";
	165	for (Int_t i = 0; i < fNcentBins; i++) {
	166	histName = TString::Format("fh2PtJet1VsLeadPtAllSel_%d",i);
	167	histTitle = TString::Format("%s;#it{p}_{T,jet1};#it{p}_{T,lead trk}",histName.Data());
	168	fh2PtJet1VsLeadPtAllSel[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,20,0.,20.);
	169	fOutput->Add(fh2PtJet1VsLeadPtAllSel[i]);
	170
	171	histName = TString::Format("fh2PtJet1VsLeadPtTagged_%d",i);
	172	histTitle = TString::Format("%s;#it{p}_{T,jet1};#it{p}_{T,lead trk}",histName.Data());
	173	fh2PtJet1VsLeadPtTagged[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,20,0.,20.);
	174	fOutput->Add(fh2PtJet1VsLeadPtTagged[i]);
	175
	176	histName = TString::Format("fh2PtVsMassJet1All_%d",i);
	177	histTitle = TString::Format("%s;#it{p}_{T,jet1};#it{M}_{jet1}",histName.Data());
	178	fh2PtVsMassJet1All[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,nBinsPt,minPt,maxPt);
	179	fOutput->Add(fh2PtVsMassJet1All[i]);
	180
	181	histName = TString::Format("fh2PtVsMassJet1Tagged_%d",i);
	182	histTitle = TString::Format("%s;#it{p}_{T,jet1};#it{M}_{jet1}",histName.Data());
	183	fh2PtVsMassJet1Tagged[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,nBinsPt,minPt,maxPt);
	184	fOutput->Add(fh2PtVsMassJet1Tagged[i]);
	185
	186	histName = TString::Format("fpPtVsMassJet1All_%d",i);
	187	histTitle = TString::Format("%s;#it{p}_{T,jet1};Avg #it{M}_{jet1}",histName.Data());
	188	fpPtVsMassJet1All[i] = new TProfile(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt);
	189	fOutput->Add(fpPtVsMassJet1All[i]);
	190
	191	histName = TString::Format("fpPtVsMassJet1Tagged_%d",i);
	192	histTitle = TString::Format("%s;#it{p}_{T,jet1};Avg #it{M}_{jet1}",histName.Data());
	193	fpPtVsMassJet1Tagged[i] = new TProfile(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt);
	194	fOutput->Add(fpPtVsMassJet1Tagged[i]);
	195
	196	histName = TString::Format("fh2MassVsAreaJet1All_%d",i);
	197	histTitle = TString::Format("%s;#it{M}_{jet1};#it{A}",histName.Data());
	198	fh2MassVsAreaJet1All[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,nBinsArea,minArea,maxArea);
	199	fOutput->Add(fh2MassVsAreaJet1All[i]);
	200
	201	histName = TString::Format("fh2MassVsAreaJet1Tagged_%d",i);
	202	histTitle = TString::Format("%s;#it{M}_{jet1};#it{A}",histName.Data());
	203	fh2MassVsAreaJet1Tagged[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,nBinsArea,minArea,maxArea);
	204	fOutput->Add(fh2MassVsAreaJet1Tagged[i]);
	205
ba7663ad	206	histName = TString::Format("fh2MassVsNConstJet1All_%d",i);
	207	histTitle = TString::Format("%s;#it{M}_{jet1};#it{N}_{constituents}",histName.Data());
	208	fh2MassVsNConstJet1All[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,nBinsNConst,minNConst,maxNConst);
	209	fOutput->Add(fh2MassVsNConstJet1All[i]);
	210
	211	histName = TString::Format("fh2MassVsNConstJet1Tagged_%d",i);
	212	histTitle = TString::Format("%s;#it{M}_{jet1};#it{N}_{constituents}",histName.Data());
	213	fh2MassVsNConstJet1Tagged[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,nBinsNConst,minNConst,maxNConst);
	214	fOutput->Add(fh2MassVsNConstJet1Tagged[i]);
	215
02e33ee8	216	histName = TString::Format("fh2EtMassOverEtRSq_%d",i);
	217	histTitle = TString::Format("%s;#it{E}_{T};(#it{M}/(#it{E}_{T}#it{R}))^{2}",histName.Data());
	218	fh2EtMassOverEtRSq[i] = new TH2F(histName.Data(),histTitle.Data(),nBinsPt,minPt,maxPt,100,0.,1.);
	219	fOutput->Add(fh2EtMassOverEtRSq[i]);
8612dfc8	220	}
	221
	222	// =========== Switch on Sumw2 for all histos ===========
	223	for (Int_t i=0; i<fOutput->GetEntries(); ++i) {
	224	TH1 h1 = dynamic_cast<TH1>(fOutput->At(i));
	225	if (h1){
	226	h1->Sumw2();
	227	continue;
	228	}
	229	THnSparse hn = dynamic_cast<THnSparse>(fOutput->At(i));
	230	if(hn)hn->Sumw2();
	231	}
	232
	233	TH1::AddDirectory(oldStatus);
	234
	235	PostData(1, fOutput); // Post data for ALL output slots > 0 here.
	236	}
	237
	238	//________________________________________________________________________
	239	Bool_t AliAnalysisTaskEmcalJetMass::Run()
	240	{
	241	// Run analysis code here, if needed. It will be executed before FillHistograms().
	242
	243	return kTRUE;
	244	}
	245
	246	//________________________________________________________________________
	247	Bool_t AliAnalysisTaskEmcalJetMass::FillHistograms()
	248	{
	249	// Fill histograms.
	250
ba7663ad	251	AliEmcalJet* jet1 = NULL;
02e33ee8	252
	253	AliJetContainer *jetCont = GetJetContainer(fContainerBase);
	254	if(jetCont) {
	255	jetCont->ResetCurrentID();
	256	while((jet1 = jetCont->GetNextAcceptJet())) {
ba7663ad	257	Double_t fraction = jetCont->GetFractionSharedPt(jet1);
	258	if(fMinFractionShared>0. && fraction<fMinFractionShared) continue;
	259
02e33ee8	260	Double_t ptJet1 = jet1->Pt() - GetRhoVal(fContainerBase)*jet1->Area();//jetCont->GetJetPtCorr(jetCont->GetCurrentID());//
	261	fh2PtJet1VsLeadPtAllSel[fCentBin]->Fill(ptJet1,jet1->MaxTrackPt());
	262	fh2PtVsMassJet1All[fCentBin]->Fill(ptJet1,jet1->M());
	263	fpPtVsMassJet1All[fCentBin]->Fill(ptJet1,jet1->M());
	264	fh2MassVsAreaJet1All[fCentBin]->Fill(jet1->M(),jet1->Area());
ba7663ad	265	fh2MassVsNConstJet1All[fCentBin]->Fill(jet1->M(),jet1->GetNumberOfConstituents());
02e33ee8	266
ba7663ad	267	if(jet1->GetTagStatus()<1 \|\| !jet1->GetTaggedJet())
02e33ee8	268	continue;
ba7663ad	269
02e33ee8	270	fh2PtJet1VsLeadPtTagged[fCentBin]->Fill(ptJet1,jet1->MaxTrackPt());
	271	fh2PtVsMassJet1Tagged[fCentBin]->Fill(ptJet1,jet1->M());
	272	fpPtVsMassJet1Tagged[fCentBin]->Fill(ptJet1,jet1->M());
	273	fh2MassVsAreaJet1Tagged[fCentBin]->Fill(jet1->M(),jet1->Area());
ba7663ad	274	fh2MassVsNConstJet1Tagged[fCentBin]->Fill(jet1->M(),jet1->GetNumberOfConstituents());
02e33ee8	275
ef7f972e	276	Double_t Et2 = jet1->M()jet1->M() + jet1->Pt()jet1->Pt();
	277	Double_t Et = 0.; Double_t massOverEtR = 0.;
	278	if(Et2>0.) Et = TMath::Sqrt(Et2);
	279	if((Et*jetCont->GetJetRadius())>0.)
	280	massOverEtR = jet1->M()/(Et*jetCont->GetJetRadius());
	281	fh2EtMassOverEtRSq[fCentBin]->Fill(Et,massOverEtR*massOverEtR);
02e33ee8	282	}
8612dfc8	283	}
	284
	285	return kTRUE;
8612dfc8	286	}
8612dfc8	287
8612dfc8	288	//________________________________________________________________________
	289	Bool_t AliAnalysisTaskEmcalJetMass::RetrieveEventObjects() {
	290	//
	291	// retrieve event objects
	292	//
	293
	294	if (!AliAnalysisTaskEmcalJet::RetrieveEventObjects())
	295	return kFALSE;
	296
	297	return kTRUE;
	298
	299	}
	300
	301	//_______________________________________________________________________
	302	void AliAnalysisTaskEmcalJetMass::Terminate(Option_t *)
	303	{
	304	// Called once at the end of the analysis.
	305	}
	306