--- /dev/null
+
+// Author: Artur Szostak
+// email: artur@alice.phy.uct.ac.za
+
+#include <TError.h>
+
+#include "AliRunLoader.h"
+#include "AliLoader.h"
+
+#include "AliMUONDataInterface.h"
+#include "AliMUONData.h"
+
+
+ClassImp(AliMUONDataInterface)
+
+
+AliMUONDataInterface::AliMUONDataInterface()
+ : TObject(), fData(NULL, "MUON", "MUON")
+{
+// Set all internal pointers to NULL and indices to -1.
+
+ Reset();
+};
+
+
+AliMUONDataInterface::~AliMUONDataInterface()
+{
+// Delete the runloader when done.
+// If the runloader is not to be deleted then call Reset just before
+// the destructor is called.
+
+ if (fRunloader != NULL)
+ delete fRunloader;
+};
+
+
+void AliMUONDataInterface::Reset()
+{
+// Sets all internal pointers to NULL and indices to -1.
+// Note: No resources are released!
+// Specificaly AliRunLoader is not deleted.
+
+ fRunloader = NULL;
+ fMuonloader = NULL;
+ fEventnumber = -1;
+ fTrack = -1;
+ fSCathode = -1;
+ fCathode = -1;
+ fHitAddressSet = kFALSE;
+ fSDigitAddressSet = kFALSE;
+ fDigitAddressSet = kFALSE;
+ fClusterAddressSet = kFALSE;
+ fTriggerAddressSet = kFALSE;
+};
+
+
+Bool_t AliMUONDataInterface::LoadLoaders(TString filename, TString foldername)
+{
+// Load the run and muon loaders from the specified file and folder.
+// kTRUE is returned on success and kFALSE on failure.
+
+ fRunloader = AliRunLoader::Open(filename, foldername, "READ");
+ if (fRunloader == NULL)
+ {
+ Error("LoadLoaders", "Could not find or load the run loader for the file: %s and folder: %s",
+ (const char*)filename, (const char*)foldername);
+ return kFALSE;
+ };
+ fMuonloader = fRunloader->GetLoader("MUONLoader");
+ if (fMuonloader == NULL)
+ {
+ Error("LoadLoaders", "Could not find the MUON loader in file: %s and folder: %s",
+ (const char*)filename, (const char*)foldername);
+ fRunloader = NULL;
+ return kFALSE;
+ };
+
+ // Need to connect the muon loader to the AliMUONData object,
+ // else class to fData will return NULL.
+ fData.SetLoader(fMuonloader);
+
+ fFilename = filename;
+ fFoldername = foldername;
+ fEventnumber = -1; // Reset the event number to force the event to be loaded.
+ return kTRUE;
+};
+
+
+Bool_t AliMUONDataInterface::FetchLoaders(TString filename, TString foldername)
+{
+// Fetch the run loader and muon loader objects from memory if they already exist,
+// or from memory if they do not.
+// If the currently loaded run loader (if any) is not refering to the file and folder
+// we are interested in then it is deleted and reopened with the required file and
+// folder names.
+
+ if (fRunloader == NULL)
+ {
+ fRunloader = AliRunLoader::GetRunLoader();
+ if (fRunloader == NULL)
+ return LoadLoaders(filename, foldername);
+
+ // Fetch the current file and folder names.
+ fFilename = fRunloader->GetFileName();
+ fFoldername = fRunloader->GetEventFolder()->GetName();
+ };
+
+ // If filename or foldername are not the same as the ones currently selected then
+ // reopen the file.
+ if ( filename.CompareTo(fFilename) != 0 || foldername.CompareTo(fFoldername) != 0 )
+ {
+ delete fRunloader;
+ return LoadLoaders(filename, foldername);
+ };
+ return kTRUE;
+};
+
+
+Bool_t AliMUONDataInterface::FetchEvent(Int_t event)
+{
+// Fetch the specified event from the runloader and reset all the track, cathode
+// and address flags to force them to be reloaded.
+
+ if (fEventnumber < 0)
+ {
+ fEventnumber = fRunloader->GetEventNumber();
+ fTrack = -1;
+ fSCathode = -1;
+ fCathode = -1;
+ fHitAddressSet = kFALSE;
+ fSDigitAddressSet = kFALSE;
+ fDigitAddressSet = kFALSE;
+ fClusterAddressSet = kFALSE;
+ fTriggerAddressSet = kFALSE;
+ };
+ if ( event != fEventnumber )
+ {
+ if ( fRunloader->GetEvent(event) < 0 ) return kFALSE;
+ fEventnumber = event;
+ fTrack = -1;
+ fSCathode = -1;
+ fCathode = -1;
+ fHitAddressSet = kFALSE;
+ fSDigitAddressSet = kFALSE;
+ fDigitAddressSet = kFALSE;
+ fClusterAddressSet = kFALSE;
+ fTriggerAddressSet = kFALSE;
+ };
+ return kTRUE;
+};
+
+
+Bool_t AliMUONDataInterface::FetchTreeK()
+{
+// Fetch the Kine tree from the current run loader.
+
+ if (fRunloader->TreeK() == NULL)
+ {
+ fRunloader->LoadKinematics("READ");
+ if (fRunloader->TreeK() == NULL)
+ {
+ Error("FetchTreeK", "Could not load TreeK.");
+ return kFALSE;
+ };
+ };
+ return kTRUE;
+};
+
+
+Bool_t AliMUONDataInterface::FetchTreeH()
+{
+// Fetch the Hits tree from the current muon loader.
+// Set all the required addresses etc...
+
+ if (fMuonloader->TreeH() == NULL)
+ {
+ fMuonloader->LoadHits("READ");
+ if (fMuonloader->TreeH() == NULL)
+ {
+ Error("FetchTreeH", "Could not load TreeH.");
+ return kFALSE;
+ };
+ fData.SetTreeAddress("H");
+ fHitAddressSet = kTRUE;
+ }
+ else if ( ! fHitAddressSet )
+ {
+ fData.SetTreeAddress("H");
+ fHitAddressSet = kTRUE;
+ };
+ return kTRUE;
+};
+
+
+Bool_t AliMUONDataInterface::FetchTreeS()
+{
+// Fetch the S-Digits tree from the current muon loader.
+// Set all the required addresses etc...
+
+ if (fMuonloader->TreeS() == NULL)
+ {
+ fMuonloader->LoadSDigits("READ");
+ if (fMuonloader->TreeS() == NULL)
+ {
+ Error("FetchTreeS", "Could not load TreeS.");
+ return kFALSE;
+ };
+ fData.SetTreeAddress("S");
+ fSDigitAddressSet = kTRUE;
+ }
+ else if ( ! fSDigitAddressSet )
+ {
+ fData.SetTreeAddress("S");
+ fSDigitAddressSet = kTRUE;
+ };
+ return kTRUE;
+};
+
+
+Bool_t AliMUONDataInterface::FetchTreeD()
+{
+// Fetch the digits tree from the current muon loader.
+// Set all the required addresses etc...
+
+ if (fMuonloader->TreeD() == NULL)
+ {
+ fMuonloader->LoadDigits("READ");
+ if (fMuonloader->TreeD() == NULL)
+ {
+ Error("FetchTreeD", "Could not load TreeD.");
+ return kFALSE;
+ };
+ fData.SetTreeAddress("D");
+ fDigitAddressSet = kTRUE;
+ }
+ else if ( ! fDigitAddressSet )
+ {
+ fData.SetTreeAddress("D");
+ fDigitAddressSet = kTRUE;
+ };
+ return kTRUE;
+};
+
+
+Bool_t AliMUONDataInterface::FetchTreeR()
+{
+// Fetch the reconstructed objects tree from the current muon loader.
+// Nore: The addresses must still be set.
+
+ if (fMuonloader->TreeR() == NULL)
+ {
+ fMuonloader->LoadRecPoints("READ");
+ if (fMuonloader->TreeR() == NULL)
+ {
+ Error("FetchTreeR", "Could not load TreeR.");
+ return kFALSE;
+ };
+
+ // Need to reset these flags so that the cluster and trigger address
+ // gets reset after this method.
+ fClusterAddressSet = kFALSE;
+ fTriggerAddressSet = kFALSE;
+ };
+ return kTRUE;
+};
+
+
+Int_t AliMUONDataInterface::NumberOfEvents(TString filename, TString foldername)
+{
+// Returns the number of events in the specified file/folder, and -1 on error.
+
+ if ( ! FetchLoaders(filename, foldername) ) return -1;
+ return fRunloader->GetNumberOfEvents();
+};
+
+
+Int_t AliMUONDataInterface::NumberOfParticles(TString filename, TString foldername, Int_t event)
+{
+// Returns the number of events in the specified file/folder, and -1 on error.
+
+ if ( ! FetchLoaders(filename, foldername) ) return -1;
+ if ( ! FetchEvent(event) ) return -1;
+ if ( ! FetchTreeK() ) return -1;
+ return (Int_t) fRunloader->TreeK()->GetEntriesFast();
+};
+
+
+TParticle* AliMUONDataInterface::Particle(
+ TString filename, TString foldername, Int_t event, Int_t particle
+ )
+{
+// Returns the specified particle in the given file, folder and event.
+// NULL is returned on error.
+
+ if ( ! FetchLoaders(filename, foldername) ) return NULL;
+ if ( ! FetchEvent(event) ) return NULL;
+ if ( ! FetchTreeK() ) return NULL;
+
+ TTree* treeK = fRunloader->TreeK();
+ TParticle* p = NULL;
+ treeK->GetBranch("Particles")->SetAddress(&p);
+ treeK->GetEvent(particle);
+ return p;
+};
+
+
+Int_t AliMUONDataInterface::NumberOfTracks(TString filename, TString foldername, Int_t event)
+{
+// Returns the number of tracks in the specified file/folder and event.
+// -1 is returned on error.
+
+ if ( ! FetchLoaders(filename, foldername) ) return -1;
+ if ( ! FetchEvent(event) ) return -1;
+ if ( ! FetchTreeH() ) return -1;
+ return fData.GetNtracks();
+};
+
+
+Int_t AliMUONDataInterface::NumberOfHits(
+ TString filename, TString foldername, Int_t event, Int_t track
+ )
+{
+// Returns the number of hits in the specified file/folder, event and track.
+// -1 is returned on error.
+
+ if ( ! FetchLoaders(filename, foldername) ) return -1;
+ if ( ! FetchEvent(event) ) return -1;
+ if ( ! FetchTreeH() ) return -1;
+
+ if (fTrack < 0 || fTrack != track)
+ {
+ fData.ResetHits();
+ fData.GetTrack(track);
+ fTrack = track;
+ };
+ return fData.Hits()->GetEntriesFast();
+};
+
+
+AliMUONHit* AliMUONDataInterface::Hit(
+ TString filename, TString foldername, Int_t event,
+ Int_t track, Int_t hit
+ )
+{
+// Returns the specified hit in the given file, folder, event and track.
+// NULL is returned on error.
+
+ if ( ! FetchLoaders(filename, foldername) ) return NULL;
+ if ( ! FetchEvent(event) ) return NULL;
+ if ( ! FetchTreeH() ) return NULL;
+
+ if (fTrack < 0 || fTrack != track)
+ {
+ fData.ResetHits();
+ fData.GetTrack(track);
+ fTrack = track;
+ };
+ return static_cast<AliMUONHit*>( fData.Hits()->At(hit) );
+};
+
+
+Int_t AliMUONDataInterface::NumberOfSDigits(
+ TString filename, TString foldername, Int_t event,
+ Int_t chamber, Int_t cathode
+ )
+{
+// Returns the number of s-digits in the given file, folder, event,
+// chamber and cathode. -1 is returned on error.
+
+ Assert( 0 <= chamber && chamber <= 13 );
+ Assert( 0 <= cathode && cathode <= 1 );
+
+ if ( ! FetchLoaders(filename, foldername) ) return -1;
+ if ( ! FetchEvent(event) ) return -1;
+ if ( ! FetchTreeS() ) return -1;
+
+ if ( fSCathode != cathode )
+ {
+ fData.ResetSDigits();
+ fData.GetCathodeS(cathode);
+ fSCathode = cathode;
+ };
+ return fData.SDigits(chamber)->GetEntriesFast();
+};
+
+
+AliMUONDigit* AliMUONDataInterface::SDigit(
+ TString filename, TString foldername, Int_t event,
+ Int_t chamber, Int_t cathode, Int_t sdigit
+ )
+{
+// Returns the specified s-digit in the given file, folder, event,
+// chamber and cathode. NULL is returned on error.
+
+ Assert( 0 <= chamber && chamber <= 13 );
+ Assert( 0 <= cathode && cathode <= 1 );
+
+ if ( ! FetchLoaders(filename, foldername) ) return NULL;
+ if ( ! FetchEvent(event) ) return NULL;
+ if ( ! FetchTreeS() ) return NULL;
+
+ if ( fSCathode != cathode )
+ {
+ fData.ResetSDigits();
+ fData.GetCathodeS(cathode);
+ fSCathode = cathode;
+ };
+ return static_cast<AliMUONDigit*>( fData.SDigits(chamber)->At(sdigit) );
+};
+
+
+Int_t AliMUONDataInterface::NumberOfDigits(
+ TString filename, TString foldername, Int_t event,
+ Int_t chamber, Int_t cathode
+ )
+{
+// Returns the number of digits in the given file, folder, event,
+// chamber and cathode. -1 is returned on error.
+ Assert( 0 <= chamber && chamber <= 13 );
+ Assert( 0 <= cathode && cathode <= 1 );
+
+ if ( ! FetchLoaders(filename, foldername) ) return -1;
+ if ( ! FetchEvent(event) ) return -1;
+ if ( ! FetchTreeD() ) return -1;
+
+ if ( fCathode != cathode )
+ {
+ fData.ResetDigits();
+ fData.GetCathode(cathode);
+ fCathode = cathode;
+ };
+ return fData.Digits(chamber)->GetEntriesFast();
+};
+
+
+AliMUONDigit* AliMUONDataInterface::Digit(
+ TString filename, TString foldername, Int_t event,
+ Int_t chamber, Int_t cathode, Int_t digit
+ )
+{
+// Returns the specified digit in the given file, folder, event,
+// chamber and cathode. NULL is returned on error.
+
+ Assert( 0 <= chamber && chamber <= 13 );
+ Assert( 0 <= cathode && cathode <= 1 );
+
+ if ( ! FetchLoaders(filename, foldername) ) return NULL;
+ if ( ! FetchEvent(event) ) return NULL;
+ if ( ! FetchTreeD() ) return NULL;
+
+ if ( fCathode != cathode )
+ {
+ fData.ResetDigits();
+ fData.GetCathode(cathode);
+ fCathode = cathode;
+ };
+ return static_cast<AliMUONDigit*>( fData.Digits(chamber)->At(digit) );
+};
+
+
+Int_t AliMUONDataInterface::NumberOfRawClusters(
+ TString filename, TString foldername, Int_t event, Int_t chamber
+ )
+{
+// Returns the number of raw clusters in the specified file, folder, event and chamber.
+// -1 is returned or error.
+
+ Assert( 0 <= chamber && chamber <= 13 );
+ if ( ! FetchLoaders(filename, foldername) ) return -1;
+ if ( ! FetchEvent(event) ) return -1;
+ if ( ! FetchTreeR() ) return -1;
+ if ( ! fClusterAddressSet )
+ {
+ // If the raw cluster address in TreeR is not set yet then set it now.
+ fData.SetTreeAddress("RC");
+ fData.ResetRawClusters();
+ fData.GetRawClusters();
+ fClusterAddressSet = kTRUE;
+ };
+ return fData.RawClusters(chamber)->GetEntriesFast();
+};
+
+
+AliMUONRawCluster* AliMUONDataInterface::RawCluster(
+ TString filename, TString foldername, Int_t event,
+ Int_t chamber, Int_t cluster
+ )
+{
+// Fetch the specified raw cluster from the given file, folder, event and chamber number.
+// NULL is returned on error.
+
+ Assert( 0 <= chamber && chamber <= 13 );
+ if ( ! FetchLoaders(filename, foldername) ) return NULL;
+ if ( ! FetchEvent(event) ) return NULL;
+ if ( ! FetchTreeR() ) return NULL;
+ if ( ! fClusterAddressSet )
+ {
+ // If the raw cluster address in TreeR is not set yet then set it now.
+ fData.SetTreeAddress("RC");
+ fData.ResetRawClusters();
+ fData.GetRawClusters();
+ fClusterAddressSet = kTRUE;
+ };
+ return static_cast<AliMUONRawCluster*>( fData.RawClusters(chamber)->At(cluster) );
+};
+
+
+Int_t AliMUONDataInterface::NumberOfLocalTriggers(TString filename, TString foldername, Int_t event)
+{
+// Return the number of local trigger objects in the specified file, folder and
+// event number. -1 is returned on error.
+
+ if ( ! FetchLoaders(filename, foldername) ) return -1;
+ if ( ! FetchEvent(event) ) return -1;
+ if ( ! FetchTreeR() ) return -1;
+ if ( ! fTriggerAddressSet )
+ {
+ // If the local trigger address in TreeR is not set yet then set it now.
+ fData.SetTreeAddress("GLT");
+ fData.ResetTrigger();
+ fData.GetTrigger();
+ fTriggerAddressSet = kTRUE;
+ };
+ return fData.LocalTrigger()->GetEntriesFast();
+};
+
+
+AliMUONLocalTrigger* AliMUONDataInterface::LocalTrigger(
+ TString filename, TString foldername, Int_t event, Int_t trigger
+ )
+{
+// Fetch the specified local trigger object from the given file, folder and event number.
+// NULL is returned on error.
+
+ if ( ! FetchLoaders(filename, foldername) ) return NULL;
+ if ( ! FetchEvent(event) ) return NULL;
+ if ( ! FetchTreeR() ) return NULL;
+ if ( ! fTriggerAddressSet )
+ {
+ // If the local trigger address in TreeR is not set yet then set it now.
+ fData.SetTreeAddress("GLT");
+ fData.ResetTrigger();
+ fData.GetTrigger();
+ fTriggerAddressSet = kTRUE;
+ };
+ return static_cast<AliMUONLocalTrigger*>( fData.LocalTrigger()->At(trigger) );
+};
+
+
+Bool_t AliMUONDataInterface::SetFile(TString filename, TString foldername)
+{
+// Set the current file and folder from which to fetch data.
+// kTRUE is returned if the run and muon loaders were found, else kFALSE.
+
+ return FetchLoaders(filename, foldername);
+};
+
+
+Bool_t AliMUONDataInterface::GetEvent(Int_t event)
+{
+// Select the current event from which to fetch data.
+// kTRUE is returned if the event was found, else kFALSE is returned.
+
+ return FetchEvent(event);
+};
+
+
+Int_t AliMUONDataInterface::NumberOfEvents()
+{
+// Get the number of events in the currently selected file.
+// -1 is returned on error.
+
+ if (fRunloader == NULL)
+ {
+ Error("NumberOfEvents", "File not set.");
+ return -1;
+ };
+ return fRunloader->GetNumberOfEvents();
+};
+
+
+Int_t AliMUONDataInterface::NumberOfParticles()
+{
+// Get the number of particles in the current event.
+// -1 is returned on error.
+
+ if (fRunloader == NULL)
+ {
+ Error("NumberOfParticles", "File not set.");
+ return -1;
+ };
+ if ( ! FetchTreeK() ) return -1;
+ return (Int_t) fRunloader->TreeK()->GetEntriesFast();
+};
+
+
+TParticle* AliMUONDataInterface::Particle(Int_t particle)
+{
+// Fetch the specified particle from the current event.
+// NULL is returned on error.
+
+ if (fRunloader == NULL)
+ {
+ Error("Particle", "File not set.");
+ return NULL;
+ };
+ if (fEventnumber < 0)
+ {
+ Error("Particle", "Event not chosen.");
+ return NULL;
+ };
+ if ( ! FetchTreeK() ) return NULL;
+ TTree* treeK = fRunloader->TreeK();
+ TParticle* p = NULL;
+ treeK->GetBranch("Particles")->SetAddress(&p);
+ treeK->GetEvent(particle);
+ return p;
+};
+
+
+Int_t AliMUONDataInterface::NumberOfTracks()
+{
+// Get the number of tracks in the current event.
+// -1 is returned on error.
+
+ if (fRunloader == NULL)
+ {
+ Error("NumberOfTracks", "File not set.");
+ return -1;
+ };
+ if (fEventnumber < 0)
+ {
+ Error("NumberOfTracks", "Event not chosen.");
+ return -1;
+ };
+ if ( ! FetchTreeH() ) return -1;
+ return fData.GetNtracks();
+};
+
+
+Int_t AliMUONDataInterface::NumberOfHits(Int_t track)
+{
+// Get the number of hits for the given track in the current event.
+// -1 is returned on error.
+
+ if (fRunloader == NULL)
+ {
+ Error("NumberOfHits", "File not set.");
+ return -1;
+ };
+ if (fEventnumber < 0)
+ {
+ Error("NumberOfHits", "Event not chosen.");
+ return -1;
+ };
+ if ( ! FetchTreeH() ) return -1;
+ if (fTrack < 0 || fTrack != track)
+ {
+ fData.ResetHits();
+ fData.GetTrack(track);
+ fTrack = track;
+ };
+ return fData.Hits()->GetEntriesFast();
+};
+
+
+AliMUONHit* AliMUONDataInterface::Hit(Int_t track, Int_t hit)
+{
+// Fetch the specified hit from the current event.
+// NULL is returned on error.
+
+ if (fRunloader == NULL)
+ {
+ Error("Hit", "File not set.");
+ return NULL;
+ };
+ if (fEventnumber < 0)
+ {
+ Error("Hit", "Event not chosen.");
+ return NULL;
+ };
+ if ( ! FetchTreeH() ) return NULL;
+ if (fTrack < 0 || fTrack != track)
+ {
+ fData.ResetHits();
+ fData.GetTrack(track);
+ fTrack = track;
+ };
+ return static_cast<AliMUONHit*>( fData.Hits()->At(hit) );
+};
+
+
+Int_t AliMUONDataInterface::NumberOfSDigits(Int_t chamber, Int_t cathode)
+{
+// Get the number of s-digits on the chamber, cathode in the current event.
+// -1 is returned on error.
+
+ Assert( 0 <= chamber && chamber <= 13 );
+ Assert( 0 <= cathode && cathode <= 1 );
+
+ if (fRunloader == NULL)
+ {
+ Error("NumberOfSDigits", "File not set.");
+ return -1;
+ };
+ if (fEventnumber < 0)
+ {
+ Error("NumberOfSDigits", "Event not chosen.");
+ return -1;
+ };
+
+ if ( ! FetchTreeS() ) return -1;
+ if ( fSCathode != cathode )
+ {
+ fData.ResetSDigits();
+ fData.GetCathodeS(cathode);
+ fSCathode = cathode;
+ };
+ return fData.SDigits(chamber)->GetEntriesFast();
+};
+
+
+AliMUONDigit* AliMUONDataInterface::SDigit(Int_t chamber, Int_t cathode, Int_t sdigit)
+{
+// Fetch the specified s-digits on the chamber, cathode from the current event.
+// NULL is returned on error.
+
+ Assert( 0 <= chamber && chamber <= 13 );
+ Assert( 0 <= cathode && cathode <= 1 );
+
+ if (fRunloader == NULL)
+ {
+ Error("SDigit", "File not set.");
+ return NULL;
+ };
+ if (fEventnumber < 0)
+ {
+ Error("SDigit", "Event not chosen.");
+ return NULL;
+ };
+
+ if ( ! FetchTreeS() ) return NULL;
+ if ( fSCathode != cathode )
+ {
+ fData.ResetSDigits();
+ fData.GetCathodeS(cathode);
+ fSCathode = cathode;
+ };
+ return static_cast<AliMUONDigit*>( fData.SDigits(chamber)->At(sdigit) );
+};
+
+
+Int_t AliMUONDataInterface::NumberOfDigits(Int_t chamber, Int_t cathode)
+{
+// Get the number of digits on the chamber, cathode in the current event.
+// -1 is returned on error.
+
+ Assert( 0 <= chamber && chamber <= 13 );
+ Assert( 0 <= cathode && cathode <= 1 );
+
+ if (fRunloader == NULL)
+ {
+ Error("NumberOfDigits", "File not set.");
+ return -1;
+ };
+ if (fEventnumber < 0)
+ {
+ Error("NumberOfDigits", "Event not chosen.");
+ return -1;
+ };
+
+ if ( ! FetchTreeD() ) return -1;
+ if ( fCathode != cathode )
+ {
+ fData.ResetDigits();
+ fData.GetCathode(cathode);
+ fCathode = cathode;
+ };
+ return fData.Digits(chamber)->GetEntriesFast();
+};
+
+
+AliMUONDigit* AliMUONDataInterface::Digit(Int_t chamber, Int_t cathode, Int_t digit)
+{
+// Fetch the specified digits on the chamber, cathode from the current event.
+// NULL is returned on error.
+
+ Assert( 0 <= chamber && chamber <= 13 );
+ Assert( 0 <= cathode && cathode <= 1 );
+
+ if (fRunloader == NULL)
+ {
+ Error("Digit", "File not set.");
+ return NULL;
+ };
+ if (fEventnumber < 0)
+ {
+ Error("Digit", "Event not chosen.");
+ return NULL;
+ };
+
+ if ( ! FetchTreeD() ) return NULL;
+ if ( fCathode != cathode )
+ {
+ fData.ResetDigits();
+ fData.GetCathode(cathode);
+ fCathode = cathode;
+ };
+ return static_cast<AliMUONDigit*>( fData.Digits(chamber)->At(digit) );
+};
+
+
+Int_t AliMUONDataInterface::NumberOfRawClusters(Int_t chamber)
+{
+// Get the number of raw clusters on the given chamber in the current event.
+// -1 is returned on error.
+
+ Assert( 0 <= chamber && chamber <= 13 );
+
+ if (fRunloader == NULL)
+ {
+ Error("NumberOfRawClusters", "File not set.");
+ return -1;
+ };
+ if (fEventnumber < 0)
+ {
+ Error("NumberOfRawClusters", "Event not chosen.");
+ return -1;
+ };
+
+ if ( ! FetchTreeR() ) return -1;
+ if ( ! fClusterAddressSet )
+ {
+ fData.SetTreeAddress("RC");
+ fData.ResetRawClusters();
+ fData.GetRawClusters();
+ fClusterAddressSet = kTRUE;
+ };
+ return fData.RawClusters(chamber)->GetEntriesFast();
+};
+
+
+AliMUONRawCluster* AliMUONDataInterface::RawCluster(Int_t chamber, Int_t cluster)
+{
+// Fetch the specified raw cluster on the given chamber from the current event.
+// NULL is returned on error.
+
+ Assert( 0 <= chamber && chamber <= 13 );
+
+ if (fRunloader == NULL)
+ {
+ Error("RawCluster", "File not set.");
+ return NULL;
+ };
+ if (fEventnumber < 0)
+ {
+ Error("RawCluster", "Event not chosen.");
+ return NULL;
+ };
+
+ if ( ! FetchTreeR() ) return NULL;
+ if ( ! fClusterAddressSet )
+ {
+ fData.SetTreeAddress("RC");
+ fData.ResetRawClusters();
+ fData.GetRawClusters();
+ fClusterAddressSet = kTRUE;
+ };
+ return static_cast<AliMUONRawCluster*>( fData.RawClusters(chamber)->At(cluster) );
+};
+
+
+Int_t AliMUONDataInterface::NumberOfLocalTriggers()
+{
+// Get the number of local trigger objects in the current event.
+// -1 is returned on error.
+
+ if (fRunloader == NULL)
+ {
+ Error("NumberOfLocalTriggers", "File not set.");
+ return -1;
+ };
+ if (fEventnumber < 0)
+ {
+ Error("NumberOfLocalTriggers", "Event not chosen.");
+ return -1;
+ };
+
+ if ( ! FetchTreeR() ) return -1;
+ if ( ! fTriggerAddressSet )
+ {
+ fData.SetTreeAddress("GLT");
+ fData.ResetTrigger();
+ fData.GetTrigger();
+ fTriggerAddressSet = kTRUE;
+ };
+ return fData.LocalTrigger()->GetEntriesFast();
+};
+
+
+AliMUONLocalTrigger* AliMUONDataInterface::LocalTrigger(Int_t trigger)
+{
+// Fetch the specified local trigger object from the current event.
+// NULL is returned on error.
+
+ if (fRunloader == NULL)
+ {
+ Error("LocalTrigger", "File not set.");
+ return NULL;
+ };
+ if (fEventnumber < 0)
+ {
+ Error("LocalTrigger", "Event not chosen.");
+ return NULL;
+ };
+
+ if ( ! FetchTreeR() ) return NULL;
+ if ( ! fTriggerAddressSet )
+ {
+ fData.SetTreeAddress("GLT");
+ fData.ResetTrigger();
+ fData.GetTrigger();
+ fTriggerAddressSet = kTRUE;
+ };
+ return static_cast<AliMUONLocalTrigger*>( fData.LocalTrigger()->At(trigger) );
+};
--- /dev/null
+#ifndef ALI_MUON_DATA_INTERFACE_CLASS_H
+#define ALI_MUON_DATA_INTERFACE_CLASS_H
+
+// Author: Artur Szostak
+// email: artur@alice.phy.uct.ac.za
+
+#include <TObject.h>
+#include <TString.h>
+#include <TParticle.h>
+
+#include "AliRunLoader.h"
+#include "AliLoader.h"
+#include "AliMUONData.h"
+
+#include "AliMUONHit.h"
+#include "AliMUONDigit.h"
+#include "AliMUONRawCluster.h"
+#include "AliMUONLocalTrigger.h"
+
+
+// An easy to use interface to the MUON module data stored in TreeK, TreeH, TreeS, TreeD and TreeR
+// One can fetch any of the data objects with all the calls to runloader, muon loader and AliMUONData
+// done behind the scenes and automatically.
+// This interface in not necessarily the fastest way to fetch the data but it is the easiest.
+// Note: If independant calls to the run loader, muon loader or AliMUONData objects are interspersed
+// with calls to the AliMUONDataInterface to fetch data, one might need to call the Reset method
+// between these method calls at some point to prevent AliMUONDataInterface from getting confused.
+// This is necessary since this object assumes the state of runloader, muon loader nor AliMUONData
+// has not changed between calls. If the state has changes then one must call Reset so that
+// AliMUONDataInterface refreshes what it knows about the state of the loader and AliMUONData objects.
+//
+class AliMUONDataInterface : public TObject
+{
+public:
+
+ AliMUONDataInterface();
+ ~AliMUONDataInterface();
+
+ // Sets all internal pointers to NULL without releasing the current runloader.
+ void Reset();
+
+ Int_t NumberOfEvents(TString filename, TString foldername);
+
+ Int_t NumberOfParticles(TString filename, TString foldername, Int_t event);
+ TParticle* Particle(TString filename, TString foldername, Int_t event, Int_t particle);
+
+ Int_t NumberOfTracks(TString filename, TString foldername, Int_t event);
+ Int_t NumberOfHits(TString filename, TString foldername, Int_t event, Int_t track);
+ AliMUONHit* Hit(TString filename, TString foldername, Int_t event, Int_t track, Int_t hit);
+
+ Int_t NumberOfSDigits(TString filename, TString foldername, Int_t event, Int_t chamber, Int_t cathode);
+ AliMUONDigit* SDigit(TString filename, TString foldername, Int_t event, Int_t chamber, Int_t cathode, Int_t sdigit);
+
+ Int_t NumberOfDigits(TString filename, TString foldername, Int_t event, Int_t chamber, Int_t cathode);
+ AliMUONDigit* Digit(TString filename, TString foldername, Int_t event, Int_t chamber, Int_t cathode, Int_t digit);
+
+ Int_t NumberOfRawClusters(TString filename, TString foldername, Int_t event, Int_t chamber);
+ AliMUONRawCluster* RawCluster(TString filename, TString foldername, Int_t event, Int_t chamber, Int_t cluster);
+
+ Int_t NumberOfLocalTriggers(TString filename, TString foldername, Int_t event);
+ AliMUONLocalTrigger* LocalTrigger(TString filename, TString foldername, Int_t event, Int_t trigger);
+
+ Bool_t SetFile(TString filename = "galice.root", TString foldername = "MUONFolder");
+ Bool_t GetEvent(Int_t event = 0);
+
+ Int_t NumberOfEvents();
+
+ Int_t NumberOfParticles();
+ TParticle* Particle(Int_t particle);
+
+ Int_t NumberOfTracks();
+ Int_t NumberOfHits(Int_t track);
+ AliMUONHit* Hit(Int_t track, Int_t hit);
+
+ Int_t NumberOfSDigits(Int_t chamber, Int_t cathode);
+ AliMUONDigit* SDigit(Int_t chamber, Int_t cathode, Int_t sdigit);
+
+ Int_t NumberOfDigits(Int_t chamber, Int_t cathode);
+ AliMUONDigit* Digit(Int_t chamber, Int_t cathode, Int_t digit);
+
+ Int_t NumberOfRawClusters(Int_t chamber);
+ AliMUONRawCluster* RawCluster(Int_t chamber, Int_t cluster);
+
+ Int_t NumberOfLocalTriggers();
+ AliMUONLocalTrigger* LocalTrigger(Int_t trigger);
+
+
+ // Returns the name of the currently selected file.
+ TString CurrentFile() { return fFilename; };
+
+ // Returns the name of the currently selected folder.
+ TString CurrentFolder() { return fFoldername; };
+
+ // Returns the number of the currently selected event.
+ Int_t CurrentEvent() { return fEventnumber; };
+
+ // Returns the currently selected track.
+ Int_t CurrentTrack() { return fTrack; };
+
+ // Returns the currently selected cathode in TreeS.
+ Int_t CurrentSCathode() { return fSCathode; };
+
+ // Returns the currently selected cathode in TreeD.
+ Int_t CurrentDCathode() { return fCathode; };
+
+private:
+
+ Bool_t LoadLoaders(TString filename, TString foldername);
+ Bool_t FetchLoaders(TString filename, TString foldername);
+ Bool_t FetchEvent(Int_t event);
+ Bool_t FetchTreeK();
+ Bool_t FetchTreeH();
+ Bool_t FetchTreeS();
+ Bool_t FetchTreeD();
+ Bool_t FetchTreeR();
+
+
+ Bool_t fHitAddressSet; //! Flag specifying if the TTree address for the hit tree was set.
+ Bool_t fSDigitAddressSet; //! Flag specifying if the TTree address for the s-digit tree was set.
+ Bool_t fDigitAddressSet; //! Flag specifying if the TTree address for the digit tree was set.
+ Bool_t fClusterAddressSet; //! Flag specifying if the TTree address for the cluster tree was set.
+ Bool_t fTriggerAddressSet; //! Flag specifying if the TTree address for the trigger tree was set.
+
+ AliRunLoader* fRunloader; //! Pointer to the runloader object used.
+ AliLoader* fMuonloader; //! Pointer to the muon loader object used.
+ AliMUONData fData; //! Pointer to the muon raw data interface.
+ TString fFilename; //! The file name from which we are fetching data.
+ TString fFoldername; //! The folder name from which we are fetching data.
+ Int_t fEventnumber; //! The currently selected event.
+ Int_t fTrack; //! The currently selected track.
+ Int_t fSCathode; //! The currently selected cathode in TreeS.
+ Int_t fCathode; //! The currently selected cathode in TreeD.
+
+ ClassDef(AliMUONDataInterface, 0) // A easy to use interface to data in the MUON module.
+};
+
+
+#endif // ALI_MUON_DATA_INTERFACE_CLASS_H
git://git.uio.no / u / mrichter / AliRoot.git / commitdiff