#include "AliITSLoader.h"
#include <AliRunLoader.h>

#include <TTree.h>
#include <TFile.h>

const TString AliITSLoader::fgkDefaultRawClustersContainerName = "TreeC";
const TString AliITSLoader::fgkDefaultBackTracksContainerName = "TreeB";
const TString AliITSLoader::fgkDefaultVerticesContainerName = "Vertex";
const TString AliITSLoader::fgkDefaultV0ContainerName = "V0";
const TString AliITSLoader::fgkDefaultCascadeContainerName = "Cascade";

ClassImp(AliITSLoader)

/*****************************************************************************/ 
AliITSLoader::AliITSLoader(const Char_t *name,const Char_t *topfoldername):
 AliLoader(name,topfoldername),
 fRawClustersDataLoader(fDetectorName + ".RawCl.root",fgkDefaultRawClustersContainerName,"Raw Clusters"),
 fBackTracksDataLoader(fDetectorName + ".BackTracks.root",fgkDefaultBackTracksContainerName,"Back Propagated Tracks"),
 fVertexDataLoader(fDetectorName + ".Vertex.root",fgkDefaultVerticesContainerName,"Primary Vertices","O"),
 fV0DataLoader(fDetectorName + ".V0s.root",fgkDefaultV0ContainerName,"V0 Vertices"),
  fCascadeDataLoader(fDetectorName + ".Cascades.root",fgkDefaultCascadeContainerName,"Cascades")
{
//ctor   
   fDataLoaders->Add(&fRawClustersDataLoader);
   fRawClustersDataLoader.SetEventFolder(fEventFolder);
   fRawClustersDataLoader.SetFolder(GetDetectorDataFolder());

   fDataLoaders->Add(&fBackTracksDataLoader);
   fBackTracksDataLoader.SetEventFolder(fEventFolder);
   fBackTracksDataLoader.SetFolder(GetDetectorDataFolder());

   fDataLoaders->Add(&fVertexDataLoader);
   fVertexDataLoader.SetEventFolder(fEventFolder);
   fVertexDataLoader.SetFolder(GetDetectorDataFolder());
   
   fDataLoaders->Add(&fV0DataLoader);
   fV0DataLoader.SetEventFolder(fEventFolder);
   fV0DataLoader.SetFolder(GetDetectorDataFolder());
   
   fDataLoaders->Add(&fCascadeDataLoader);
   fCascadeDataLoader.SetEventFolder(fEventFolder);
   fCascadeDataLoader.SetFolder(GetDetectorDataFolder());
   
}
/*****************************************************************************/ 

AliITSLoader::AliITSLoader(const Char_t *name,TFolder *topfolder):
 AliLoader(name,topfolder),
 fRawClustersDataLoader(fDetectorName + ".RawCl.root",fgkDefaultRawClustersContainerName,"Raw Clusters"),
 fBackTracksDataLoader(fDetectorName + ".BackTracks.root",fgkDefaultBackTracksContainerName,"Back Propagated Tracks"),
 fVertexDataLoader(fDetectorName + ".Vertex.root",fgkDefaultVerticesContainerName,"Primary Vertices","O"),
 fV0DataLoader(fDetectorName + ".V0.root",fgkDefaultV0ContainerName,"V0 Vertices"),
  fCascadeDataLoader(fDetectorName + ".Cascade.root",fgkDefaultCascadeContainerName,"Cascades")
{
//ctor   
   fDataLoaders->Add(&fRawClustersDataLoader);
   fRawClustersDataLoader.SetEventFolder(fEventFolder);
   fRawClustersDataLoader.SetFolder(GetDetectorDataFolder());

   fDataLoaders->Add(&fBackTracksDataLoader);
   fBackTracksDataLoader.SetEventFolder(fEventFolder);
   fBackTracksDataLoader.SetFolder(GetDetectorDataFolder());

   fDataLoaders->Add(&fVertexDataLoader);
   fVertexDataLoader.SetEventFolder(fEventFolder);
   fVertexDataLoader.SetFolder(GetDetectorDataFolder());

   fDataLoaders->Add(&fV0DataLoader);
   fV0DataLoader.SetEventFolder(fEventFolder);
   fV0DataLoader.SetFolder(GetDetectorDataFolder());
   
   fDataLoaders->Add(&fCascadeDataLoader);
   fCascadeDataLoader.SetEventFolder(fEventFolder);
   fCascadeDataLoader.SetFolder(GetDetectorDataFolder());
   
}
/*****************************************************************************/ 
AliITSLoader::~AliITSLoader()
{
 //destructor
  UnloadRawClusters();
  fDataLoaders->Remove(&fRawClustersDataLoader);

  UnloadBackTracks();
  fDataLoaders->Remove(&fBackTracksDataLoader);

  UnloadVertices();
  fDataLoaders->Remove(&fVertexDataLoader);

  UnloadV0s();
  fDataLoaders->Remove(&fV0DataLoader);

  UnloadCascades();
  fDataLoaders->Remove(&fCascadeDataLoader);

}

void AliITSLoader::MakeTree(Option_t *opt){
  // invokes AliLoader::MakeTree + specific ITS tree(s)
  // Valid options: H,S,D,R,T and C (C=raw clusters)
  AliLoader::MakeTree(opt);
  const char *oC = strstr(opt,"C");
  if (oC) MakeRawClustersContainer();

  const char *oB = strstr(opt,"B");
  if (oB) MakeBackTracksContainer();
  
  const char *oV0 = strstr(opt,"V0");
  if (oV0) MakeV0Container();
  
  const char *oX = strstr(opt,"X");
  if (oX) MakeCascadeContainer();
  
}