AliHadCorrTask* AddTaskHadCorr(
const char *nTracks = "EmcalTracks",
const char *nClusters = "EmcalClusters",
const char *outClusName = "CaloClustersCorr",
const Double_t hadcorr = 1,
const Double_t minPt = 0.15,
const Double_t phiMatch = 0.050,
const Double_t etaMatch = 0.025,
const Double_t Eexcl = 0,
const Bool_t trackClus = kTRUE,
const Bool_t histo = kFALSE,
const char *outputname = "AnalysisResults.root"
)
{
// Get the pointer to the existing analysis manager via the static access method.
//==============================================================================
AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
if (!mgr)
{
::Error("AddTaskHadCorr", "No analysis manager to connect to.");
return 0;
}
// Check the analysis type using the event handlers connected to the analysis manager.
//==============================================================================
AliVEventHandler* handler = mgr->GetInputEventHandler();
if (!handler)
{
::Error("AddTaskHadCorr", "This task requires an input event handler");
return 0;
}
enum EDataType_t {
kUnknown,
kESD,
kAOD
};
EDataType_t dataType = kUnknown;
if (handler->InheritsFrom("AliESDInputHandler")) {
dataType = kESD;
}
else if (handler->InheritsFrom("AliAODInputHandler")) {
dataType = kAOD;
}
//-------------------------------------------------------
// Init the task and do settings
//-------------------------------------------------------
TString trackName(nTracks);
TString clusName(nClusters);
if (trackName == "usedefault") {
if (dataType == kESD) {
trackName = "Tracks";
}
else if (dataType == kAOD) {
trackName = "tracks";
}
else {
trackName = "";
}
}
if (clusName == "usedefault") {
if (dataType == kESD) {
clusName = "CaloClusters";
}
else if (dataType == kAOD) {
clusName = "caloClusters";
}
else {
clusName = "";
}
}
TString name(Form("HadCorr_%s_%s", trackName.Data(), clusName.Data()));
if (strcmp(outClusName, "") != 0) name += Form("_%s", outClusName);
AliHadCorrTask *hcor = new AliHadCorrTask(name, histo);
hcor->SetOutClusName(outClusName);
hcor->SetPhiMatch(phiMatch);
hcor->SetEtaMatch(etaMatch);
hcor->SetHadCorr(hadcorr);
hcor->SetEexcl(Eexcl);
hcor->SetTrackClus(trackClus);
AliParticleContainer* trackCont = 0;
if (trackName == "Tracks" || trackName == "tracks") {
trackCont = hcor->AddTrackContainer(trackName);
}
else {
trackCont = hcor->AddParticleContainer(trackName);
}
trackCont->SetParticlePtCut(minPt);
AliClusterContainer *clusCont = hcor->AddClusterContainer(clusName);
if (clusCont) clusCont->SetClusNonLinCorrEnergyCut(minPt);
//-------------------------------------------------------
// Final settings, pass to manager and set the containers
//-------------------------------------------------------
mgr->AddTask(hcor);
// Create containers for input/output
mgr->ConnectInput (hcor, 0, mgr->GetCommonInputContainer() );
if (histo) {
AliAnalysisDataContainer *cohcor = mgr->CreateContainer(name,
TList::Class(),
AliAnalysisManager::kOutputContainer,
outputname);
mgr->ConnectOutput(hcor,1,cohcor);
}
return hcor;
}
AliAnalysisTaskJetExtractorHF* AddTaskJetExtractorHF(
const char *trackArray = "tracks",
const char *jetArray = "jets",
const char *rhoObject = "Rho",
Double_t jetRadius = 0.3,
Double_t minJetEta = 0.6,
Double_t minJetPt = 0.15,
Double_t minTrackPt = 0.15,
Double_t minJetAreaPerc = 0.557,
const char *suffix = ""
)
{
cout << " ############ MACRO EXECUTION STARTED: AddTaskJetExtractorHF.C ############\n";
//==============================================================================
// Prepare analysis manager, containers, etc.
AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
if (!mgr)
{
::Error("AddTaskJetExtractorHF", "No analysis manager to connect to.");
return NULL;
}
if (!mgr->GetInputEventHandler())
{
::Error("AddTaskJetExtractorHF", "This task requires an input event handler");
return NULL;
}
TString name("AliAnalysisTaskJetExtractorHF");
if (strcmp(jetArray,"")) {
name += "_";
name += jetArray;
}
if (strcmp(rhoObject,"")) {
name += "_";
name += rhoObject;
}
if (strcmp(suffix,"")) {
name += "_";
name += suffix;
}
AliAnalysisDataContainer* contHistos = mgr->CreateContainer(Form("%s_histos", name.Data()), TList::Class(), AliAnalysisManager::kOutputContainer, Form("%s:ChargedJetsHadronCF", AliAnalysisManager::GetCommonFileName()));
//==============================================================================
// Adding and configuring tasks
AliAnalysisTaskJetExtractorHF* jetTask = new AliAnalysisTaskJetExtractorHF(name);
jetTask->SetNeedEmcalGeom(kFALSE);
jetTask->SetVzRange(-10.,10.);
AliRDHFJetsCutsVertex* cuts = new AliRDHFJetsCutsVertex("jetCuts");
// vertexing
AliESDtrackCuts* esdTrackCuts = new AliESDtrackCuts("AliESDtrackCuts", "default");
esdTrackCuts->SetRequireSigmaToVertex(kFALSE);
esdTrackCuts->SetMinNClustersTPC(90);
esdTrackCuts->SetMaxChi2PerClusterTPC(4);
esdTrackCuts->SetRequireTPCRefit(kTRUE);
esdTrackCuts->SetRequireITSRefit(kTRUE);
esdTrackCuts->SetClusterRequirementITS(AliESDtrackCuts::kSPD, AliESDtrackCuts::kAny);
esdTrackCuts->SetMinDCAToVertexXY(0.);
esdTrackCuts->SetEtaRange(-0.8, 0.8);
esdTrackCuts->SetPtRange(1.0, 1.e10);
cuts->AddTrackCuts(esdTrackCuts);
cuts->SetNprongs(3);
cuts->SetMinPtHardestTrack(1.0);//default 0.3
cuts->SetSecVtxWithKF(kFALSE);//default with StrLinMinDist
cuts->SetImpParCut(0.);//default 0
cuts->SetDistPrimSec(0.);//default 0
cuts->SetCospCut(-1);//default -1
jetTask->SetVertexerCuts(cuts);
AliParticleContainer *trackCont = 0;
if(!strcmp(trackArray,"mctracks") || !strcmp(trackArray, "mcparticles"))
trackCont = jetTask->AddMCParticleContainer(trackArray);
else
trackCont = jetTask->AddTrackContainer(trackArray);
trackCont->SetParticlePtCut(minTrackPt);
AliJetContainer *jetCont = jetTask->AddJetContainer(jetArray,6,jetRadius);
if (jetCont) {
jetCont->SetRhoName(rhoObject);
jetCont->SetPercAreaCut(minJetAreaPerc);
jetCont->SetJetPtCut(minJetPt);
jetCont->SetLeadingHadronType(0);
jetCont->SetPtBiasJetTrack(minTrackPt);
jetCont->SetJetEtaLimits(-minJetEta, +minJetEta);
jetCont->ConnectParticleContainer(trackCont);
jetCont->SetMaxTrackPt(1000);
}
mgr->AddTask(jetTask);
//==============================================================================
// Finalization
mgr->ConnectInput (jetTask, 0, mgr->GetCommonInputContainer() );
mgr->ConnectOutput (jetTask, 1, contHistos );
cout << " ############ MACRO EXECUTION DONE: AddTaskJetExtractorHF.C ############\n";
return jetTask;
}
Bool_t AddTasksEmcalFjetSparseMaker()
{
AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
if (bIsInfoAOD) {
gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/train/AddAODHandler.C");
AliAODInputHandler *aodIH = AddAODHandler();
} else {
gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/train/AddESDHandler.C");
AliESDInputHandler *esdIH = AddESDHandler();
// esdIH->SetReadFriends(kFALSE);
}
if (bAnaInfoMC && (!bIsInfoAOD)) {
gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/train/AddMCHandler.C");
AliMCEventHandler *mctEH = AddMCHandler(kTRUE);
}
if (bOutputAOD) {
gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/train/AddAODOutputHandler.C");
AliAODHandler *aodH = AddAODOutputHandler();
aodH->SetOutputFileName("AliAOD.PicoJets.root");
aodH->SetFillAOD(kTRUE);
aodH->SetCreateNonStandardAOD();
mgr->SetOutputEventHandler(aodH);
}
//=============================================================================
if (bIsPhysSel) {
gROOT->LoadMacro("$ALICE_PHYSICS/PWG/EMCAL/macros/AddTaskEmcalPhysicsSelection.C");
AliEmcalPhysicsSelectionTask *taksPhysSel = AddTaskEmcalPhysicsSelection(kTRUE, kTRUE, wTriggerMask, 5., 5., 10., kTRUE, -1, -1, -1, -1);
if (bAnaInfoMC) {
AliEmcalPhysicsSelection *pPhysSel = static_cast<AliEmcalPhysicsSelection*>(taksPhysSel->GetPhysicsSelection());
if (!pPhysSel) return kTRUE; pPhysSel->SetAnalyzeMC();
}
}
if (bIsCentSel) {
gROOT->LoadMacro("$ALICE_PHYSICS/OADB/macros/AddTaskCentrality.C");
AliCentralitySelectionTask *taskCentSel = AddTaskCentrality(kTRUE, bIsAOD);
if (bAnaInfoMC) taskCentSel->SetMCInput();
}
if (bIsEvPnSel) {
gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskEventplane.C");
AliEPSelectionTask *taskEventPlane = AddTaskEventplane();
if (bAnaInfoMC) taskEventPlane->SetUseMCRP();
}
if (bIsRespPID) {
gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
AliAnalysisTaskSE *taskRespPID = AddTaskPIDResponse(bAnaInfoMC);
}
//=============================================================================
gROOT->LoadMacro("$ALICE_PHYSICS/PWG/EMCAL/macros/AddTaskEmcalSetup.C");
AliEmcalSetupTask *taskSetupEMCal = AddTaskEmcalSetup();
taskSetupEMCal->SetGeoPath("$ALICE_PHYSICS/OADB/EMCAL");
gROOT->LoadMacro("$ALICE_PHYSICS/PWG/EMCAL/macros/AddTaskEmcalPreparation.C");
AliAnalysisTaskSE *taskEmcMaker = AddTaskEmcalPreparation(sPeriodIn);
gROOT->LoadMacro("$ALICE_PHYSICS/PWGJE/EMCALJetTasks/macros/AddTaskJetPreparation.C");
AliAnalysisTaskSE *taskHadCorr = AddTaskJetPreparation(sPeriodIn.Data(),
sUsedTrksRD.Data(),
(bAnaInfoMC ? sUsedTrksMC.Data() : ""),
sInputClsRD.Data(),
sUsedClusRD.Data(),
2., 0., 0.03, 0.015, 0.15,
wTriggerMask,
kTRUE, // do trk clus
kTRUE, // histos
kTRUE, // make pico trk
kFALSE, // make trigger
kFALSE, // is EMCal train
1., // track eff
kTRUE, // do AOD track prop
kTRUE, // modify match
kFALSE, // use old bit
kFALSE, // trigger QA
4); // cent bin
//=============================================================================
gROOT->LoadMacro("$ALICE_PHYSICS/PWGJE/EMCALJetTasks/macros/AddTaskEmcalJet.C");
/*AliEmcalJetTask *taskAkt = AddTaskEmcalJet(sUsedTrksRD.Data(),
sUsedClusRD.Data(),
1, // anti-kT
dJetRadius,
nJetType,
dTrkPtCut,
dCluEnCut);*/
//=============================================================================
AliAnalysisTaskEmcalFjetSparseMaker *taskEmcalSM = new AliAnalysisTaskEmcalFjetSparseMaker("AliAnalysisTaskEmcalFjetSM");
//taskEmcalSM->SetForceBeamType(0);
//taskEmcalSM->SetIsPythia(kTRUE);
/*taskEmcalSM->SetCaloTriggerPatchInfoName("EmcalTriggers");
taskEmcalSM->SetTriggerTypeSel(AliAnalysisTaskEmcal::kJ2);
taskEmcalSM->SetMainPatchType(AliAnalysisTaskEmcal::kTriggerLevel1Jet);*/
AliParticleContainer *pContTrks = taskEmcalSM->AddParticleContainer(sUsedTrksRD.Data());
pContTrks->SetParticlePtCut(dTrkPtCut);
AliClusterContainer *pContClus = taskEmcalSM->AddClusterContainer(sUsedClusRD.Data());
pContClus->SetClusPtCut(dCluEnCut);
/*AliJetContainer *pContJets = taskEmcalSM->AddJetContainer(taskAkt->GetName(), sCutType.Data(), dJetRadius);
pContJets->SetPercAreaCut(dJetAreaCut);
pContJets->SetJetPtCut(dJetPtCut);
//pContJets->SetRhoName(sUsedRho.Data());
//pContJets->SetLocalRhoName();
pContJets->SetLeadingHadronType(nLeading);
pContJets->ConnectParticleContainer(taskEmcalSM->GetParticleContainer());
pContJets->ConnectClusterContainer(taskEmcalSM->GetClusterContainer());
pContJets->SetNameTitle(taskEmcalSM->GetNameJet().Data(),
taskEmcalSM->GetNameJet().Data());*/
mgr->AddTask(taskEmcalSM);
mgr->ConnectInput(taskEmcalSM, 0, mgr->GetCommonInputContainer());
mgr->ConnectOutput(taskEmcalSM, 1, mgr->CreateContainer("listGeneralH",
TList::Class(),
AliAnalysisManager::kOutputContainer,
AliAnalysisManager::GetCommonFileName()));
mgr->ConnectOutput(taskEmcalSM, 2, mgr->CreateContainer("listEmcalEvH",
TList::Class(),
AliAnalysisManager::kOutputContainer,
"AnalysisResults_EmcalEvH.root"));
mgr->ConnectOutput(taskEmcalSM, 3, mgr->CreateContainer("listEmcalTrk",
TList::Class(),
AliAnalysisManager::kOutputContainer,
"AnalysisResults_EmcalTrk.root"));
mgr->ConnectOutput(taskEmcalSM, 4, mgr->CreateContainer("listEmcalClu",
TList::Class(),
AliAnalysisManager::kOutputContainer,
"AnalysisResults_EmcalClu.root"));
/*mgr->ConnectOutput(taskEmcalSM, 5, mgr->CreateContainer("listEmcalJet",
TList::Class(),
AliAnalysisManager::kOutputContainer,
"AnalysisResults_EmcalJet.root"));*/
//=============================================================================
TObjArray *aTasks = mgr->GetTasks();
for (Int_t i=0; i<aTasks->GetEntries(); i++) {
AliAnalysisTaskEmcal *taskJE = dynamic_cast<AliAnalysisTaskEmcal*>(aTasks->At(i));
if (taskJE) if (taskJE->InheritsFrom("AliAnalysisTaskEmcal")) {
taskJE->SetVzRange(-10., 10.);
if (!sCentEsti.IsNull()) {
taskJE->SetCentralityEstimator(sCentEsti.Data());
taskJE->SetCentRange(dCentMin, dCentMax);
}
}
if (wTriggerMask) {
AliAnalysisTaskSE *taskSE = dynamic_cast<AliAnalysisTaskSE*>(aTasks->At(i));
if (taskSE) if (!(taskSE->InheritsFrom("AliPhysicsSelectionTask"))) {
taskSE->SelectCollisionCandidates(wTriggerMask);
}
}
}
//=============================================================================
return kFALSE;
}
EmcalHJetMassAnalysis::AliAnalysisTaskEmcalHJetMass* AddTaskEmcalHJetMass(const char * njetsBase,
const Double_t R,
const char * nrhoBase,
const char * ntracks,
const char * nclusters,
const char * type,
Double_t ptMinH = 5.,
Double_t maxDPhi = 0.6,
Int_t pSel,
const char * CentEst = "V0M",
TString trigClass = "",
TString kEmcalTriggers = "",
TString nJetsUnsub = "",
TString tag = "") {
AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
if (!mgr)
{
Error("AddTaskEmcalHJetMass","No analysis manager found.");
return 0;
}
Bool_t ismc=kFALSE;
ismc = (mgr->GetMCtruthEventHandler())?kTRUE:kFALSE;
// Check the analysis type using the event handlers connected to the analysis manager.
//==============================================================================
if (!mgr->GetInputEventHandler())
{
::Error("AddTaskEmcalHJetMass", "This task requires an input event handler");
return NULL;
}
TString wagonName = Form("HJetMass_%s_TC%s%s",njetsBase,trigClass.Data(),tag.Data());
//Configure jet tagger task
EmcalHJetMassAnalysis::AliAnalysisTaskEmcalHJetMass *task = new EmcalHJetMassAnalysis::AliAnalysisTaskEmcalHJetMass(wagonName.Data());
task->SetNCentBins(4);
task->SetMaxDeltaPhi(maxDPhi);
AliParticleContainer *trackCont = task->AddParticleContainer(ntracks);
trackCont->SetParticlePtCut(ptMinH);
AliClusterContainer *clusterCont = task->AddClusterContainer(nclusters);
task->SetJetContainerBase(0);
TString strType(type);
AliJetContainer *jetContBase = task->AddJetContainer(njetsBase,strType,R);
if(jetContBase) {
jetContBase->SetRhoName(nrhoBase);
jetContBase->ConnectParticleContainer(trackCont);
jetContBase->ConnectClusterContainer(clusterCont);
jetContBase->SetPercAreaCut(0.6);
}
if(!nJetsUnsub.IsNull()) {
AliJetContainer *jetContUS = task->AddJetContainer(nJetsUnsub.Data(),strType,R);
if(jetContUS) {
jetContUS->SetRhoName(nrhoBase);
jetContUS->ConnectParticleContainer(trackCont);
jetContUS->ConnectClusterContainer(clusterCont);
jetContUS->SetPercAreaCut(0.6);
}
}
task->SetCaloTriggerPatchInfoName(kEmcalTriggers.Data());
task->SetCentralityEstimator(CentEst);
task->SelectCollisionCandidates(pSel);
task->SetUseAliAnaUtils(kFALSE);
mgr->AddTask(task);
//Connnect input
mgr->ConnectInput (task, 0, mgr->GetCommonInputContainer() );
//Connect output
TString contName(wagonName);
TString outputfile = Form("%s",AliAnalysisManager::GetCommonFileName());
AliAnalysisDataContainer *coutput1 = mgr->CreateContainer(contName.Data(), TList::Class(),AliAnalysisManager::kOutputContainer,outputfile);
mgr->ConnectOutput(task,1,coutput1);
return task;
}
AliAnalysisTaskJetExtractor* AddTaskJetExtractor(
const char *trackArray = "tracks",
const char *jetArray = "jets",
const char *rhoObject = "Rho",
Double_t jetRadius = 0.3,
Double_t minJetEta = 0.6,
Double_t minJetPt = 0.15,
Double_t minTrackPt = 0.15,
Double_t minJetAreaPerc = 0.557,
const char *suffix = ""
)
{
cout << " ############ MACRO EXECUTION STARTED: AddTaskJetExtractor.C ############\n";
//==============================================================================
// Prepare analysis manager, containers, etc.
AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
if (!mgr)
{
::Error("AddTaskJetExtractor", "No analysis manager to connect to.");
return NULL;
}
if (!mgr->GetInputEventHandler())
{
::Error("AddTaskJetExtractor", "This task requires an input event handler");
return NULL;
}
TString name("AliAnalysisTaskJetExtractor");
if (strcmp(jetArray,"")) {
name += "_";
name += jetArray;
}
if (strcmp(rhoObject,"")) {
name += "_";
name += rhoObject;
}
if (strcmp(suffix,"")) {
name += "_";
name += suffix;
}
AliAnalysisDataContainer* contHistos = mgr->CreateContainer(Form("%s_histos", name.Data()), TList::Class(), AliAnalysisManager::kOutputContainer, Form("%s:ChargedJetsHadronCF", AliAnalysisManager::GetCommonFileName()));
//==============================================================================
// Adding and configuring tasks
AliAnalysisTaskJetExtractor* jetTask = new AliAnalysisTaskJetExtractor(name);
jetTask->SetNeedEmcalGeom(kFALSE);
jetTask->SetVzRange(-10.,10.);
AliParticleContainer *trackCont = 0;
if(!strcmp(trackArray,"mctracks") || !strcmp(trackArray, "mcparticles"))
trackCont = jetTask->AddMCParticleContainer(trackArray);
else
trackCont = jetTask->AddTrackContainer(trackArray);
trackCont->SetParticlePtCut(minTrackPt);
AliJetContainer *jetCont = jetTask->AddJetContainer(jetArray,6,jetRadius);
if (jetCont) {
jetCont->SetRhoName(rhoObject);
jetCont->SetPercAreaCut(minJetAreaPerc);
jetCont->SetJetPtCut(minJetPt);
jetCont->SetLeadingHadronType(0);
jetCont->SetPtBiasJetTrack(minTrackPt);
jetCont->SetJetEtaLimits(-minJetEta, +minJetEta);
jetCont->ConnectParticleContainer(trackCont);
jetCont->SetMaxTrackPt(1000);
}
mgr->AddTask(jetTask);
//==============================================================================
// Finalization
mgr->ConnectInput (jetTask, 0, mgr->GetCommonInputContainer() );
mgr->ConnectOutput (jetTask, 1, contHistos );
cout << " ############ MACRO EXECUTION DONE: AddTaskJetExtractor.C ############\n";
return jetTask;
}
AliAnalysisTaskDeltaPt* AddTaskDeltaPt(
const char *ntracks = "Tracks",
const char *nclusters = "CaloClusters",
const char *njets = "Jets",
const char *nembtracks = "TracksEmbedded",
const char *nembclusters = "CaloClustersEmbedded",
const char *nembjets = "EmbJets",
const char *nrandtracks = "TracksRandomized",
const char *nrandclusters = "CaloClustersRandomized",
const char *nrho = "Rho",
Double_t jetradius = 0.2,
Double_t jetareacut = 0.557,
Double_t trackptcut = 0.15,
Double_t clusptcut = 0.30,
const char *type = "TPC",
const char *taskname = "AliAnalysisTaskDeltaPt"
)
{
// Get the pointer to the existing analysis manager via the static access method.
//==============================================================================
AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
if (!mgr)
{
::Error("AddTaskDeltaPt", "No analysis manager to connect to.");
return NULL;
}
// Check the analysis type using the event handlers connected to the analysis manager.
//==============================================================================
if (!mgr->GetInputEventHandler())
{
::Error("AddTaskDeltaPt", "This task requires an input event handler");
return NULL;
}
//-------------------------------------------------------
// Init the task and do settings
//-------------------------------------------------------
TString name;
if (strcmp(ntracks, "") == 0 && strcmp(nclusters, "") == 0)
name = Form("%s_%s_R0%d_%s",taskname,nrho,(Int_t)floor(jetradius*100+0.5),type);
else if (strcmp(ntracks, "") == 0)
name = Form("%s_%s_%s_R0%d_%s",taskname,nclusters,nrho,(Int_t)floor(jetradius*100+0.5),type);
else if (strcmp(nclusters, "") == 0)
name = Form("%s_%s_%s_R0%d_%s",taskname,ntracks,nrho,(Int_t)floor(jetradius*100+0.5),type);
else
name = Form("%s_%s_%s_%s_R0%d_%s",taskname,ntracks,nclusters,nrho,(Int_t)floor(jetradius*100+0.5),type);
AliAnalysisTaskDeltaPt* jetTask = new AliAnalysisTaskDeltaPt(name);
jetTask->SetConeRadius(jetradius);
jetTask->SetRhoName(nrho,-1);
if (strcmp(type,"TPC")==0)
jetTask->SetConeEtaPhiTPC();
else if (strcmp(type,"EMCAL")==0)
jetTask->SetConeEtaPhiEMCAL();
AliParticleContainer *partCont = jetTask->AddParticleContainer(ntracks);
if (partCont) {
partCont->SetName("Tracks");
partCont->SetParticlePtCut(trackptcut);
}
AliClusterContainer *clusCont = jetTask->AddClusterContainer(nclusters);
if (clusCont) {
clusCont->SetName("CaloClusters");
clusCont->SetClusPtCut(clusptcut);
}
AliJetContainer *jetCont = jetTask->AddJetContainer(njets,type,jetradius);
if (jetCont) {
jetCont->SetName("Jets");
jetCont->SetPercAreaCut(jetareacut);
jetCont->SetRhoName(nrho);
jetCont->ConnectParticleContainer(partCont);
jetCont->ConnectClusterContainer(clusCont);
}
AliParticleContainer *embPartCont = jetTask->AddParticleContainer(nembtracks);
if (embPartCont) {
embPartCont->SetName("EmbTracks");
embPartCont->SetParticlePtCut(trackptcut);
}
AliClusterContainer *embClusCont = jetTask->AddClusterContainer(nembclusters);
if (embClusCont) {
embClusCont->SetName("EmbClusters");
embClusCont->SetClusPtCut(clusptcut);
}
AliJetContainer *embJetCont = jetTask->AddJetContainer(nembjets,type,jetradius);
if (embJetCont) {
embJetCont->SetName("EmbJets");
embJetCont->SetPercAreaCut(jetareacut);
embJetCont->SetRhoName(nrho);
embJetCont->ConnectParticleContainer(embPartCont);
embJetCont->ConnectClusterContainer(embClusCont);
}
AliParticleContainer *randPartCont = jetTask->AddParticleContainer(nrandtracks);
if (randPartCont) {
randPartCont->SetName("RandTracks");
randPartCont->SetParticlePtCut(trackptcut);
}
AliClusterContainer *randClusCont = jetTask->AddClusterContainer(nrandclusters);
if (randClusCont) {
randClusCont->SetName("RandClusters");
randClusCont->SetClusPtCut(clusptcut);
}
//-------------------------------------------------------
// Final settings, pass to manager and set the containers
//-------------------------------------------------------
mgr->AddTask(jetTask);
// Create containers for input/output
AliAnalysisDataContainer *cinput1 = mgr->GetCommonInputContainer();
TString contname(name);
contname += "_histos";
AliAnalysisDataContainer *coutput1 = mgr->CreateContainer(contname.Data(),
TList::Class(),AliAnalysisManager::kOutputContainer,
Form("%s", AliAnalysisManager::GetCommonFileName()));
mgr->ConnectInput(jetTask, 0, cinput1);
mgr->ConnectOutput(jetTask, 1, coutput1);
return jetTask;
}
AliAnalysisTaskPWGJEQA* AddTaskPWGJEQA(
const char* ntracks = "usedefault",
const char* nclusters = "usedefault",
const char* ncells = "usedefault",
const char *nGenLev = "mcparticles",
Bool_t doTrackQA = kTRUE,
Bool_t doCaloQA = kTRUE,
Bool_t doJetQA = kTRUE,
Bool_t doEventQA = kTRUE,
Double_t trackPtCut = 0.15,
Double_t clusECut = 0.30,
const char* suffix = ""
)
{
// Get the pointer to the existing analysis manager via the static access method
AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
if (!mgr) {
::Error("AddTaskPWGJEQA", "No analysis manager to connect to.");
return NULL;
}
// Check the analysis type using the event handlers connected to the analysis manager
AliVEventHandler* handler = mgr->GetInputEventHandler();
if (!handler) {
::Error("AddTaskPWGJEQA", "This task requires an input event handler");
return NULL;
}
enum EDataType_t {
kUnknown,
kESD,
kAOD
};
EDataType_t dataType = kUnknown;
if (handler->InheritsFrom("AliESDInputHandler")) {
dataType = kESD;
}
else if (handler->InheritsFrom("AliAODInputHandler")) {
dataType = kAOD;
}
// Init the task and do settings
TString trackName(ntracks);
TString clusName(nclusters);
TString cellName(ncells);
if (trackName == "usedefault") {
if (dataType == kESD) {
trackName = "Tracks";
}
else if (dataType == kAOD) {
trackName = "tracks";
}
else {
trackName = "";
}
}
if (clusName == "usedefault") {
if (dataType == kESD) {
clusName = "CaloClusters";
}
else if (dataType == kAOD) {
clusName = "caloClusters";
}
else {
clusName = "";
}
}
if (ncells == "usedefault") {
if (dataType == kESD) {
cellName = "EMCALCells";
}
else if (dataType == kAOD) {
cellName = "emcalCells";
}
else {
cellName = "";
}
}
TString name("AliAnalysisTaskPWGJEQA");
if (!trackName.IsNull()) {
name += "_";
name += trackName;
}
if (!clusName.IsNull()) {
name += "_";
name += clusName;
}
if (!cellName.IsNull()) {
name += "_";
name += cellName;
}
if (strcmp(suffix,"")) {
name += "_";
name += suffix;
}
AliAnalysisTaskPWGJEQA* qaTask = new AliAnalysisTaskPWGJEQA(name);
qaTask->SetVzRange(-10,10);
qaTask->SetNeedEmcalGeom(kFALSE);
qaTask->SetCaloCellsName(cellName);
qaTask->SetDetectorLevelName(trackName);
if (nGenLev && strcmp(nGenLev,"")!=0) qaTask->SetGeneratorLevelName(nGenLev);
qaTask->SetDoTrackQA(doTrackQA);
qaTask->SetDoCaloQA(doCaloQA);
qaTask->SetDoJetQA(doJetQA);
qaTask->SetDoEventQA(doEventQA);
// Add the detector-level track container
if (trackName == "tracks" || trackName == "Tracks") {
AliTrackContainer* trackCont = qaTask->AddTrackContainer(trackName);
trackCont->SetFilterHybridTracks(kTRUE);
}
else if (!trackName.IsNull()) {
qaTask->AddParticleContainer(trackName);
}
AliParticleContainer *partCont = qaTask->GetParticleContainer(trackName);
if (partCont) {
partCont->SetParticlePtCut(trackPtCut);
}
// Add the generator-level container, if specified
if (nGenLev && strcmp(nGenLev,"")!=0) {
AliMCParticleContainer* mcpartCont = qaTask->AddMCParticleContainer(nGenLev);
mcpartCont->SelectPhysicalPrimaries(kTRUE);
mcpartCont->SetParticlePtCut(0);
}
// Add the cluster container
AliClusterContainer *clusterCont = qaTask->AddClusterContainer(clusName);
if (clusterCont) {
clusterCont->SetClusECut(0.);
clusterCont->SetClusPtCut(0.);
clusterCont->SetClusHadCorrEnergyCut(clusECut);
clusterCont->SetDefaultClusterEnergy(AliVCluster::kHadCorr);
}
// Final settings, pass to manager and set the containers
mgr->AddTask(qaTask);
// Create containers for input/output
AliAnalysisDataContainer *cinput1 = mgr->GetCommonInputContainer() ;
TString contName = TString::Format("%s_histos", name.Data());
TString commonoutput;
commonoutput = mgr->GetCommonFileName();
AliAnalysisDataContainer *coutput1 = mgr->CreateContainer(contName.Data(),
TList::Class(),AliAnalysisManager::kOutputContainer,
commonoutput);
mgr->ConnectInput (qaTask, 0, cinput1 );
mgr->ConnectOutput (qaTask, 1, coutput1 );
return qaTask;
}
