changes in AddAnalysisTaskPIDFluctuation.C for only one output file in analysis train...

[u/mrichter/AliRoot.git] / PWGGA / EMCALTasks / AliEmcalSetupTask.cxx

// $Id$
//
// Task to setup emcal related global objects
//
//

#include <TClonesArray.h>
#include <TGeoManager.h>
#include <TGeoGlobalMagField.h>
#include "AliAnalysisManager.h"
#include "AliCDBManager.h"
#include "AliEMCALGeometry.h"
#include "AliESDEvent.h"
#include "AliEmcalSetupTask.h"
#include "AliGeomManager.h"
#include "AliOADBContainer.h"

ClassImp(AliEmcalSetupTask)

//________________________________________________________________________
AliEmcalSetupTask::AliEmcalSetupTask() : 
  AliAnalysisTaskSE(),
  fOcdbPath(),
  fOadbPath("$ALICE_ROOT/OADB/EMCAL"),
  fGeoPath("."),
  fEsdEv(0),
  fIsInit(kFALSE)
{
  // Constructor.
}

//________________________________________________________________________
AliEmcalSetupTask::AliEmcalSetupTask(const char *name) : 
  AliAnalysisTaskSE(name),
  fOcdbPath(),
  fOadbPath("$ALICE_ROOT/OADB/EMCAL"),
  fGeoPath("."),
  fEsdEv(0),
  fIsInit(kFALSE)
{
  // Constructor.
  fBranchNames = "ESD:AliESDHeader.,AliESDRun.";
}

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

//________________________________________________________________________
void AliEmcalSetupTask::UserExec(Option_t *) 
{
  // Main loop, called for each event.

  if (fIsInit)
    return;

  fEsdEv = dynamic_cast<AliESDEvent*>(InputEvent());
  if (!fEsdEv) {
    AliError("Task works only on ESD events, returning");
    return;
  }

  AliAnalysisManager *am = AliAnalysisManager::GetAnalysisManager();
  if (!am) {
    AliError("Manager zero, returning");
    return;
  }
  am->LoadBranch("AliESDRun.");
  am->LoadBranch("AliESDHeader.");

  Int_t runno = fEsdEv->GetRunNumber();
  TString geoname("EMCAL_FIRSTYEARV1");
  Int_t year = 2010;
  if (runno>139517) {
    year = 2011;
    geoname = "EMCAL_COMPLETEV1";
  } if (runno>170593) {
    year = 2012;
    geoname = "EMCAL_COMPLETE12SM";
  }

  AliEMCALGeometry *geom = AliEMCALGeometry::GetInstance(geoname);
  if (!geom) {
    AliFatal(Form("Can not create geometry: %s",geoname.Data()));
    return;
  }

  AliCDBManager *man = 0;
  if (fOcdbPath.Length()>0) {
    AliInfo(Form("Setting up OCDB"));
    man = AliCDBManager::Instance();
    man->SetDefaultStorage(fOcdbPath);
    man->SetRun(runno);
  }

  TGeoManager *geo = AliGeomManager::GetGeometry();
  if (!geo) {
    TString fname(gSystem->ExpandPathName(Form("%s/geometry_%d.root", fGeoPath.Data(), year)));
    if (gSystem->AccessPathName(fname)==0) {
      AliInfo(Form("Loading geometry from %s", fname.Data()));
      AliGeomManager::LoadGeometry(fname);
    } else if (man) {
      AliInfo(Form("Loading geometry from OCDB"));
      AliGeomManager::LoadGeometry();
    }
  }
  if (geo) {
    AliGeomManager::ApplyAlignObjsFromCDB("EMCAL");
    AliInfo(Form("Locking geometry"));
    geo->LockGeometry();
  }

  if (!TGeoGlobalMagField::Instance()->GetField()) { // construct field map
    AliInfo("Constructing field map from ESD run info");
    fEsdEv->InitMagneticField();
  }

  if (fOadbPath.Length()>0) {
    AliOADBContainer emcalgeoCont(Form("emcal"));
    emcalgeoCont.InitFromFile(Form("%s/EMCALlocal2master.root",fOadbPath.Data()),
                              Form("AliEMCALgeo"));
    TObjArray *mobj=dynamic_cast<TObjArray*>(emcalgeoCont.GetObject(runno,"EmcalMatrices"));
    if (mobj) {
      for(Int_t mod=0; mod < (geom->GetEMCGeometry())->GetNumberOfSuperModules(); mod++){
        //AliInfo(Form("Misalignment matrix %d", mod));
        geom->SetMisalMatrix((TGeoHMatrix*) mobj->At(mod),mod);
      } 
    }
  }

  fIsInit = kTRUE;
}