[u/mrichter/AliRoot.git] / PWG / EMCAL / AliEmcalSetupTask.cxx

// $Id$
//
// Task to setup emcal related global objects.
//
// Author: C.Loizides

#include "AliEmcalSetupTask.h"
#include <TClonesArray.h>
#include <TGeoGlobalMagField.h>
#include <TGeoManager.h>
#include <TRandom.h>
#include "AliAODEvent.h"
#include "AliAnalysisManager.h"
#include "AliCDBManager.h"
#include "AliEMCALGeometry.h"
#include "AliESDEvent.h"
#include "AliGRPManager.h"
#include "AliGeomManager.h"
#include "AliMagF.h"
#include "AliOADBContainer.h"
#include "AliTender.h"

ClassImp(AliEmcalSetupTask)

//________________________________________________________________________
AliEmcalSetupTask::AliEmcalSetupTask() : 
  AliAnalysisTaskSE(),
  fOcdbPath("uselocal"),
  fOadbPath("$ALICE_ROOT/OADB/EMCAL"),
  fGeoPath("$ALICE_ROOT/OADB/EMCAL"),
  fObjs("GRP ITS TPC TRD EMCAL"),
  fNoOCDB(kFALSE),
  fIsInit(kFALSE),
  fLocalOcdb(),
  fLocalOcdbStor()
{
  // Constructor.
}

//________________________________________________________________________
AliEmcalSetupTask::AliEmcalSetupTask(const char *name) : 
  AliAnalysisTaskSE(name),
  fOcdbPath("uselocal"),
  fOadbPath("$ALICE_ROOT/OADB/EMCAL"),
  fGeoPath("$ALICE_ROOT/OADB/EMCAL"),
  fObjs("GRP ITS TPC TRD EMCAL"),
  fNoOCDB(kFALSE),
  fIsInit(kFALSE),
  fLocalOcdb(),
  fLocalOcdbStor()
{
  // Constructor.
  fBranchNames = "ESD:AliESDHeader.,AliESDRun.";
}

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

//________________________________________________________________________
void AliEmcalSetupTask::ConnectInputData(Option_t *option)
{
  // Connect input data

  AliAnalysisTaskSE::ConnectInputData(option);

  if (fOcdbPath.Length()==0)
    return;

  AliCDBManager *man = AliCDBManager::Instance();
  if (man->IsDefaultStorageSet()) 
    return;

  if (fIsInit)
    return;

  AliAnalysisManager *am = AliAnalysisManager::GetAnalysisManager();
  if (!am)
    return;

  TObjArray *tasks = am->GetTasks();
  if (!tasks)
    return;

  AliTender *tender = 0;
  for (Int_t i=0; i<tasks->GetEntries(); ++i) {
    tender = dynamic_cast<AliTender*>(tasks->At(i));
    if (tender)
      break;
  }

  if (!tender)
    return;

  if (fOcdbPath != "uselocal") {
    tender->SetDefaultCDBStorage(fOcdbPath);
    return;
  }

  Int_t runno = AliAnalysisManager::GetAnalysisManager()->GetRunFromPath();
  if (runno<=0) {
    AliWarning(Form("Disabling tender, ignore possible message from tender below"));
    tender->SetDefaultCDBStorage("donotuse");
    return;
  }

  AliWarning(Form("Intercepting tender for run %d, ignore possible message from tender below", runno));
  Setup(runno);
  tender->SetDefaultCDBStorage(fLocalOcdbStor);
}

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

  if (fIsInit)
    return;

  AliAnalysisManager *am = AliAnalysisManager::GetAnalysisManager();
  if (!am) {
    AliError("Manager zero, returning");
    return;
  }

  am->LoadBranch("AliESDRun.");
  am->LoadBranch("AliESDHeader.");

  Int_t runno = InputEvent()->GetRunNumber();
  Setup(runno);
}

//________________________________________________________________________
void AliEmcalSetupTask::Setup(Int_t runno) 
{
  // Setup everything

  // Setup AliEMCALGeometry corresponding to year
  TString geoname("EMCAL_COMPLETE12SMV1");
  Int_t year = 2013;
  if (runno>0 && runno<=139517) {
    year = 2010;
    geoname = "EMCAL_FIRSTYEARV1";
  } else if ((runno>139517) && (runno<=170593)) {
    year = 2011;
    geoname = "EMCAL_COMPLETEV1";
  } else if ((runno>170593) && (runno<=193766)) {
    year = 2012;
    geoname = "EMCAL_COMPLETE12SMV1";
  } else if ((runno>193766) && (runno <= 199161)) {
    geoname = "EMCAL_COMPLETE12SMV1";
    year = 2013;
  } else if (runno>199161) { //MV: is this the last run of run 1?
    geoname = "EMCAL_COMPLETE12SMV1_DCAL_8SM";
    year = 2015;
  }
  AliEMCALGeometry *geom = AliEMCALGeometry::GetInstance(geoname);
  if (!geom) {
    AliFatal(Form("Can not create geometry: %s",geoname.Data()));
    return;
  }

  if (runno<0)  // Run number 0 can occur for MC
    return;

  // Setup CDB manager
  AliCDBManager *man = 0;
  if (!fNoOCDB) {
    man = AliCDBManager::Instance();
    if (!man)
      AliFatal(Form("Did not get pointer to CDB manager"));

    if (man->IsDefaultStorageSet()) {
      AliInfo(Form("Default OCDB storage already set"));
    } else {
      if (fOcdbPath.Length()==0) {
	man = 0; // do not use OCDB
      } else if (fOcdbPath != "uselocal") {
	AliInfo(Form("Setting up OCDB to point to %s",fOcdbPath.Data()));
	man->SetDefaultStorage(fOcdbPath);
      } else { // use local copy of OCDB
	TString tmpdir=gSystem->WorkingDirectory();
	if (gSystem->AccessPathName(tmpdir))
	  tmpdir = "/tmp";
	tmpdir+="/";
	tmpdir+=gSystem->GetUid();
	tmpdir+="-";
	TDatime t;
	tmpdir+=t.Get();
	tmpdir+="-";
	Int_t counter = 0;
	fLocalOcdb = tmpdir;
	fLocalOcdb += Form("%d%d%d",gRandom->Integer(999999999),gRandom->Integer(999999999),gRandom->Integer(999999999));
	while (!gSystem->AccessPathName(fLocalOcdb)) {
	  fLocalOcdb = tmpdir;
	  fLocalOcdb += Form("%d%d%d",gRandom->Integer(999999999),gRandom->Integer(999999999),gRandom->Integer(999999999));
	  counter++;
	  if (counter>100) {
	    AliFatal(Form("Could not create local directory for OCDB at %s",tmpdir.Data()));
	  }
	}
	gSystem->MakeDirectory(fLocalOcdb);
	TString filename(Form("$ALICE_ROOT/PWG/EMCAL/data/%d.dat",year));
	TString cmd(Form("cd %s && tar -xf %s",fLocalOcdb.Data(),filename.Data()));
	Int_t ret = gSystem->Exec(cmd);
	if (ret==0) {
	  TString locdb("local://");
	  locdb+=fLocalOcdb;
	  locdb+="/";
	  locdb+=year;
	  AliInfo(Form("Setting up local OCDB at %s",locdb.Data()));
	  man->SetDefaultStorage(locdb);
	  fLocalOcdbStor = locdb;
	} else {
	  AliFatal(Form("Could not set up local OCDB at %s",fLocalOcdb.Data()));
	}
      }
    }
  }

  // Load geometry from OCDB 
  if (man) {
    if (man->GetRun()!=runno)
      man->SetRun(runno);
    AliInfo(Form("Loading grp data from OCDB for run %d", runno));
    AliGRPManager GRPManager;
    GRPManager.ReadGRPEntry();
    GRPManager.SetMagField();
    AliInfo(Form("Loading geometry from OCDB"));
    AliGeomManager::LoadGeometry();
    if (!fObjs.IsNull())
      AliGeomManager::ApplyAlignObjsFromCDB(fObjs);
  }

  // Load geometry from file (does not use misalignment of ITS/TPC!)
  TGeoManager *geo = AliGeomManager::GetGeometry();
  if (!geo && fGeoPath.Length()>0) {
    TString fname(gSystem->ExpandPathName(Form("%s/geometry_%d.root", fGeoPath.Data(), year)));
    if (gSystem->AccessPathName(fname)==0) {
      AliInfo(Form("Loading geometry from file %s (should be avoided!)", fname.Data()));
      AliGeomManager::LoadGeometry(fname);
      geo = AliGeomManager::GetGeometry();
    }
  }

  // Lock geometry
  if (geo) {
    AliInfo(Form("Locking geometry"));
    geo->LockGeometry();
  }

  // Construct field map
  if (!TGeoGlobalMagField::Instance()->GetField()) { 
    InputEvent()->InitMagneticField();
  }

  // Apply mis-alignment matrices from OADB
  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;
}

//________________________________________________________________________
void AliEmcalSetupTask::Terminate(Option_t *) 
{
  // Called at the end.

  if (fLocalOcdb.Length()>0) {
    TString cmd(Form("rm -rf %s", fLocalOcdb.Data()));
    gSystem->Exec(cmd);
  }
}
Commit	Line	Data
6ea93b32	1	// $Id$
6ea93b32	2	//
980821ba	3	// Task to setup emcal related global objects.
6ea93b32	4	//
cd231d42	5	// Author: C.Loizides
6ea93b32	6
b933ccf0	7	#include "AliEmcalSetupTask.h"
6ea93b32	8	#include <TClonesArray.h>
6ea93b32	9	#include <TGeoGlobalMagField.h>
dec5f256	10	#include <TGeoManager.h>
012dab22	11	#include <TRandom.h>
dec5f256	12	#include "AliAODEvent.h"
6ea93b32	13	#include "AliAnalysisManager.h"
	14	#include "AliCDBManager.h"
	15	#include "AliEMCALGeometry.h"
	16	#include "AliESDEvent.h"
eb0d359d	17	#include "AliGRPManager.h"
6ea93b32	18	#include "AliGeomManager.h"
dec5f256	19	#include "AliMagF.h"
6ea93b32	20	#include "AliOADBContainer.h"
810787c3	21	#include "AliTender.h"
6ea93b32	22
	23	ClassImp(AliEmcalSetupTask)
	24
	25	//________________________________________________________________________
	26	AliEmcalSetupTask::AliEmcalSetupTask() :
	27	AliAnalysisTaskSE(),
012dab22	28	fOcdbPath("uselocal"),
6ea93b32	29	fOadbPath("$ALICE_ROOT/OADB/EMCAL"),
e8481053 CL	30	fGeoPath("$ALICE_ROOT/OADB/EMCAL"),
e8481053 CL	31	fObjs("GRP ITS TPC TRD EMCAL"),
6b6a028a	32	fNoOCDB(kFALSE),
012dab22	33	fIsInit(kFALSE),
810787c3	34	fLocalOcdb(),
810787c3	35	fLocalOcdbStor()
6ea93b32	36	{
	37	// Constructor.
	38	}
	39
	40	//________________________________________________________________________
	41	AliEmcalSetupTask::AliEmcalSetupTask(const char *name) :
	42	AliAnalysisTaskSE(name),
012dab22	43	fOcdbPath("uselocal"),
6ea93b32	44	fOadbPath("$ALICE_ROOT/OADB/EMCAL"),
4b4fc849	45	fGeoPath("$ALICE_ROOT/OADB/EMCAL"),
e8481053	46	fObjs("GRP ITS TPC TRD EMCAL"),
6b6a028a	47	fNoOCDB(kFALSE),
012dab22	48	fIsInit(kFALSE),
810787c3	49	fLocalOcdb(),
810787c3	50	fLocalOcdbStor()
6ea93b32	51	{
	52	// Constructor.
	53	fBranchNames = "ESD:AliESDHeader.,AliESDRun.";
	54	}
	55
	56	//________________________________________________________________________
	57	AliEmcalSetupTask::~AliEmcalSetupTask()
	58	{
	59	// Destructor.
	60	}
	61
810787c3	62	//________________________________________________________________________
	63	void AliEmcalSetupTask::ConnectInputData(Option_t *option)
	64	{
	65	// Connect input data
	66
	67	AliAnalysisTaskSE::ConnectInputData(option);
	68
	69	if (fOcdbPath.Length()==0)
	70	return;
	71
3f0ec366	72	AliCDBManager *man = AliCDBManager::Instance();
	73	if (man->IsDefaultStorageSet())
	74	return;
	75
810787c3	76	if (fIsInit)
	77	return;
	78
	79	AliAnalysisManager *am = AliAnalysisManager::GetAnalysisManager();
	80	if (!am)
	81	return;
	82
	83	TObjArray *tasks = am->GetTasks();
	84	if (!tasks)
	85	return;
	86
	87	AliTender *tender = 0;
	88	for (Int_t i=0; i<tasks->GetEntries(); ++i) {
	89	tender = dynamic_cast<AliTender*>(tasks->At(i));
	90	if (tender)
	91	break;
	92	}
	93
	94	if (!tender)
	95	return;
	96
	97	if (fOcdbPath != "uselocal") {
	98	tender->SetDefaultCDBStorage(fOcdbPath);
	99	return;
	100	}
	101
	102	Int_t runno = AliAnalysisManager::GetAnalysisManager()->GetRunFromPath();
	103	if (runno<=0) {
	104	AliWarning(Form("Disabling tender, ignore possible message from tender below"));
	105	tender->SetDefaultCDBStorage("donotuse");
	106	return;
	107	}
	108
	109	AliWarning(Form("Intercepting tender for run %d, ignore possible message from tender below", runno));
	110	Setup(runno);
	111	tender->SetDefaultCDBStorage(fLocalOcdbStor);
	112	}
	113
6ea93b32	114	//________________________________________________________________________
	115	void AliEmcalSetupTask::UserExec(Option_t *)
	116	{
	117	// Main loop, called for each event.
	118
	119	if (fIsInit)
	120	return;
	121
6ea93b32	122	AliAnalysisManager *am = AliAnalysisManager::GetAnalysisManager();
	123	if (!am) {
	124	AliError("Manager zero, returning");
	125	return;
	126	}
012dab22	127
6ea93b32	128	am->LoadBranch("AliESDRun.");
	129	am->LoadBranch("AliESDHeader.");
	130
dec5f256	131	Int_t runno = InputEvent()->GetRunNumber();
810787c3	132	Setup(runno);
	133	}
	134
	135	//________________________________________________________________________
	136	void AliEmcalSetupTask::Setup(Int_t runno)
	137	{
	138	// Setup everything
	139
810787c3	140	// Setup AliEMCALGeometry corresponding to year
b0aacf89	141	TString geoname("EMCAL_COMPLETE12SMV1");
b0aacf89	142	Int_t year = 2013;
c080ac89	143	if (runno>0 && runno<=139517) {
a1a56aea	144	year = 2010;
a1a56aea	145	geoname = "EMCAL_FIRSTYEARV1";
4b4fc849	146	} else if ((runno>139517) && (runno<=170593)) {
64a8dac7	147	year = 2011;
64a8dac7	148	geoname = "EMCAL_COMPLETEV1";
4b4fc849	149	} else if ((runno>170593) && (runno<=193766)) {
4b4fc849	150	year = 2012;
d9b3f983	151	geoname = "EMCAL_COMPLETE12SMV1";
de47dde0	152	} else if ((runno>193766) && (runno <= 199161)) {
	153	geoname = "EMCAL_COMPLETE12SMV1";
	154	year = 2013;
b0aacf89	155	} else if (runno>199161) { //MV: is this the last run of run 1?
	156	geoname = "EMCAL_COMPLETE12SMV1_DCAL_8SM";
	157	year = 2015;
6ea93b32	158	}
6ea93b32	159	AliEMCALGeometry *geom = AliEMCALGeometry::GetInstance(geoname);
	160	if (!geom) {
	161	AliFatal(Form("Can not create geometry: %s",geoname.Data()));
	162	return;
	163	}
	164
de47dde0	165	if (runno<0) // Run number 0 can occur for MC
c080ac89	166	return;
c080ac89	167
012dab22	168	// Setup CDB manager
6ea93b32	169	AliCDBManager *man = 0;
6b6a028a	170	if (!fNoOCDB) {
	171	man = AliCDBManager::Instance();
	172	if (!man)
	173	AliFatal(Form("Did not get pointer to CDB manager"));
	174
	175	if (man->IsDefaultStorageSet()) {
	176	AliInfo(Form("Default OCDB storage already set"));
	177	} else {
	178	if (fOcdbPath.Length()==0) {
	179	man = 0; // do not use OCDB
	180	} else if (fOcdbPath != "uselocal") {
	181	AliInfo(Form("Setting up OCDB to point to %s",fOcdbPath.Data()));
	182	man->SetDefaultStorage(fOcdbPath);
	183	} else { // use local copy of OCDB
	184	TString tmpdir=gSystem->WorkingDirectory();
	185	if (gSystem->AccessPathName(tmpdir))
	186	tmpdir = "/tmp";
	187	tmpdir+="/";
	188	tmpdir+=gSystem->GetUid();
	189	tmpdir+="-";
	190	TDatime t;
	191	tmpdir+=t.Get();
	192	tmpdir+="-";
	193	Int_t counter = 0;
012dab22	194	fLocalOcdb = tmpdir;
810787c3	195	fLocalOcdb += Form("%d%d%d",gRandom->Integer(999999999),gRandom->Integer(999999999),gRandom->Integer(999999999));
6b6a028a	196	while (!gSystem->AccessPathName(fLocalOcdb)) {
	197	fLocalOcdb = tmpdir;
	198	fLocalOcdb += Form("%d%d%d",gRandom->Integer(999999999),gRandom->Integer(999999999),gRandom->Integer(999999999));
	199	counter++;
	200	if (counter>100) {
	201	AliFatal(Form("Could not create local directory for OCDB at %s",tmpdir.Data()));
	202	}
	203	}
	204	gSystem->MakeDirectory(fLocalOcdb);
	205	TString filename(Form("$ALICE_ROOT/PWG/EMCAL/data/%d.dat",year));
	206	TString cmd(Form("cd %s && tar -xf %s",fLocalOcdb.Data(),filename.Data()));
	207	Int_t ret = gSystem->Exec(cmd);
	208	if (ret==0) {
	209	TString locdb("local://");
	210	locdb+=fLocalOcdb;
	211	locdb+="/";
	212	locdb+=year;
	213	AliInfo(Form("Setting up local OCDB at %s",locdb.Data()));
	214	man->SetDefaultStorage(locdb);
	215	fLocalOcdbStor = locdb;
	216	} else {
	217	AliFatal(Form("Could not set up local OCDB at %s",fLocalOcdb.Data()));
012dab22	218	}
012dab22	219	}
eb0d359d	220	}
6ea93b32	221	}
6b6a028a	222
012dab22	223	// Load geometry from OCDB
eb0d359d	224	if (man) {
012dab22 CL	225	if (man->GetRun()!=runno)
012dab22 CL	226	man->SetRun(runno);
af901268	227	AliInfo(Form("Loading grp data from OCDB for run %d", runno));
eb0d359d CL	228	AliGRPManager GRPManager;
	229	GRPManager.ReadGRPEntry();
	230	GRPManager.SetMagField();
	231	AliInfo(Form("Loading geometry from OCDB"));
	232	AliGeomManager::LoadGeometry();
e8481053 CL	233	if (!fObjs.IsNull())
e8481053 CL	234	AliGeomManager::ApplyAlignObjsFromCDB(fObjs);
eb0d359d	235	}
6ea93b32	236
012dab22	237	// Load geometry from file (does not use misalignment of ITS/TPC!)
6ea93b32	238	TGeoManager *geo = AliGeomManager::GetGeometry();
012dab22	239	if (!geo && fGeoPath.Length()>0) {
64a8dac7	240	TString fname(gSystem->ExpandPathName(Form("%s/geometry_%d.root", fGeoPath.Data(), year)));
6ea93b32	241	if (gSystem->AccessPathName(fname)==0) {
012dab22	242	AliInfo(Form("Loading geometry from file %s (should be avoided!)", fname.Data()));
6ea93b32	243	AliGeomManager::LoadGeometry(fname);
ba4cb34f	244	geo = AliGeomManager::GetGeometry();
6ea93b32	245	}
6ea93b32	246	}
012dab22 CL	247
012dab22 CL	248	// Lock geometry
eb0d359d CL	249	if (geo) {
	250	AliInfo(Form("Locking geometry"));
	251	geo->LockGeometry();
	252	}
6ea93b32	253
012dab22 CL	254	// Construct field map
012dab22 CL	255	if (!TGeoGlobalMagField::Instance()->GetField()) {
ddc5a41b	256	InputEvent()->InitMagneticField();
6ea93b32	257	}
6ea93b32	258
012dab22	259	// Apply mis-alignment matrices from OADB
6ea93b32	260	if (fOadbPath.Length()>0) {
	261	AliOADBContainer emcalgeoCont(Form("emcal"));
	262	emcalgeoCont.InitFromFile(Form("%s/EMCALlocal2master.root",fOadbPath.Data()),
	263	Form("AliEMCALgeo"));
	264	TObjArray mobj=dynamic_cast<TObjArray>(emcalgeoCont.GetObject(runno,"EmcalMatrices"));
	265	if (mobj) {
	266	for(Int_t mod=0; mod < (geom->GetEMCGeometry())->GetNumberOfSuperModules(); mod++){
	267	//AliInfo(Form("Misalignment matrix %d", mod));
	268	geom->SetMisalMatrix((TGeoHMatrix*) mobj->At(mod),mod);
	269	}
	270	}
	271	}
	272
	273	fIsInit = kTRUE;
	274	}
012dab22 CL	275
	276	//________________________________________________________________________
	277	void AliEmcalSetupTask::Terminate(Option_t *)
	278	{
	279	// Called at the end.
	280
	281	if (fLocalOcdb.Length()>0) {
	282	TString cmd(Form("rm -rf %s", fLocalOcdb.Data()));
	283	gSystem->Exec(cmd);
	284	}
	285	}