-
-// Author: Artur Szostak
-// email: artur@alice.phy.uct.ac.za
+/**************************************************************************
+ * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ * *
+ * Author: The ALICE Off-line Project. *
+ * Contributors are mentioned in the code where appropriate. *
+ * *
+ * Permission to use, copy, modify and distribute this software and its *
+ * documentation strictly for non-commercial purposes is hereby granted *
+ * without fee, provided that the above copyright notice appears in all *
+ * copies and that both the copyright notice and this permission notice *
+ * appear in the supporting documentation. The authors make no claims *
+ * about the suitability of this software for any purpose. It is *
+ * provided "as is" without express or implied warranty. *
+ **************************************************************************/
+
+/* $Id$ */
#include <TError.h>
#include <TParticle.h>
#include "AliMUONDataInterface.h"
#include "AliMUONLocalTrigger.h"
+#include "AliMUONGlobalTrigger.h"
#include "AliMUONHit.h"
#include "AliMUONDigit.h"
#include "AliMUONRawCluster.h"
+#include "AliMUONTrack.h"
#include "AliLog.h"
#include <iostream>
using std::endl;
using std::cout;
-
+///
+/// \class AliMUONDataInterface
+///
+/// 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.
+///
+/// \deprecated We have to revisit all this AliMUONData stuff anyway,
+/// and probably make a real AliMUONLoader instead...
+///
+/// \author Artur Szostak
+/// email: artur@alice.phy.uct.ac.za
+
+
+/// \cond CLASSIMP
ClassImp(AliMUONDataInterface)
-
+/// \endcond
AliMUONDataInterface::AliMUONDataInterface()
: TObject(), fData(NULL, "MUON", "MUON")
{
-// Set all internal pointers to NULL and indices to -1.
+/// Set all internal pointers to NULL and indices to -1.
Reset();
}
-AliMUONDataInterface::AliMUONDataInterface(const AliMUONDataInterface& rhs)
- : TObject(rhs)
-{
-// Protected copy constructor
-
- AliFatal("Not implemented.");
-}
-
AliMUONDataInterface::~AliMUONDataInterface()
{
-// Delete the runloader if we created it.
-// If the runloader is not to be deleted then call Reset just before
-// the destructor is called.
+/// Delete the runloader if we created it.
+/// If the runloader is not to be deleted then call Reset just before
+/// the destructor is called.
if (fRunloader != NULL && fCreatedRunLoader)
delete fRunloader;
}
-AliMUONDataInterface&
-AliMUONDataInterface::operator=(const AliMUONDataInterface& rhs)
-{
-// Protected assignement operator
-
- if (this == &rhs) return *this;
-
- AliFatal("Not implemented.");
-
- return *this;
-}
-
-
void AliMUONDataInterface::Reset()
{
-// Sets all internal pointers to NULL and indices to -1.
-// Note: No resources are released!
-// Specificaly AliRunLoader is not deleted.
+/// Sets all internal pointers to NULL and indices to -1.
+/// Note: No resources are released!
+/// Specificaly AliRunLoader is not deleted.
fCreatedRunLoader = kFALSE;
fRunloader = NULL;
fDigitAddressSet = kFALSE;
fClusterAddressSet = kFALSE;
fTriggerAddressSet = kFALSE;
+ fRecTracksAddressSet = kFALSE;
}
Bool_t AliMUONDataInterface::UseCurrentRunLoader()
{
-// Tries to fetch the current runloader with AliRunLoader::GetRunLoader. If nothing is
-// currently loaded then kFALSE is returned and AliMUONDataInterface is reset.
+/// Tries to fetch the current runloader with AliRunLoader::GetRunLoader. If nothing is
+/// currently loaded then kFALSE is returned and AliMUONDataInterface is reset.
Reset();
fRunloader = AliRunLoader::GetRunLoader();
Bool_t AliMUONDataInterface::FetchMuonLoader(TString filename, TString foldername)
{
+/// Fetches the muon loader for the given filename/foldername
+
fMuonloader = fRunloader->GetLoader("MUONLoader");
if (fMuonloader == NULL)
{
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.
+/// 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)
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.
+/// 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)
{
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 a negative event number is specified then the current runloader event
-// number is used.
+/// Fetch the specified event from the runloader and reset all the track, cathode
+/// and address flags to force them to be reloaded.
+/// If a negative event number is specified then the current runloader event
+/// number is used.
if (fEventnumber < 0)
{
fDigitAddressSet = kFALSE;
fClusterAddressSet = kFALSE;
fTriggerAddressSet = kFALSE;
+ fRecTracksAddressSet = kFALSE;
}
if ( event != fEventnumber )
{
fDigitAddressSet = kFALSE;
fClusterAddressSet = kFALSE;
fTriggerAddressSet = kFALSE;
+ fRecTracksAddressSet = kFALSE;
}
return kTRUE;
}
Bool_t AliMUONDataInterface::FetchTreeK()
{
-// Fetch the Kine tree from the current run loader.
+/// Fetch the Kine tree from the current run loader.
if (fRunloader->TreeK() == NULL)
{
Bool_t AliMUONDataInterface::FetchTreeH()
{
-// Fetch the Hits tree from the current muon loader.
-// Set all the required addresses etc...
+/// Fetch the Hits tree from the current muon loader.
+/// Set all the required addresses etc...
if (fMuonloader->TreeH() == NULL)
{
Bool_t AliMUONDataInterface::FetchTreeS()
{
-// Fetch the S-Digits tree from the current muon loader.
-// Set all the required addresses etc...
+/// Fetch the S-Digits tree from the current muon loader.
+/// Set all the required addresses etc...
if (fMuonloader->TreeS() == NULL)
{
Bool_t AliMUONDataInterface::FetchTreeD()
{
-// Fetch the digits tree from the current muon loader.
-// Set all the required addresses etc...
+/// Fetch the digits tree from the current muon loader.
+/// Set all the required addresses etc...
if (fMuonloader->TreeD() == NULL)
{
Bool_t AliMUONDataInterface::FetchTreeR()
{
-// Fetch the reconstructed objects tree from the current muon loader.
-// Note: The addresses must still be set.
-
- if (fMuonloader->TreeR() == NULL)
+/// Fetch the reconstructed objects tree from the current muon loader.
+/// Note: The addresses must still be set.
+
+ if (fMuonloader->TreeR() == NULL)
+ {
+ fMuonloader->LoadRecPoints("READ");
+ if (fMuonloader->TreeR() == NULL)
{
- fMuonloader->LoadRecPoints("READ");
- if (fMuonloader->TreeR() == NULL)
- {
- AliError("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;
+ AliError("Could not load TreeR.");
+ return kFALSE;
}
- return kTRUE;
+
+ // Need to reset these flags so that the cluster and trigger address
+ // gets reset after this method.
+ fClusterAddressSet = kFALSE;
+ fTriggerAddressSet = kFALSE;
+ }
+ return kTRUE;
}
-
+Bool_t AliMUONDataInterface::FetchTreeT()
+{
+/// fetch the reconstructed tracks tree from the current muon loader
+/// note : the addresses must still be set.
+ if (fMuonloader->TreeT() == NULL)
+ {
+ fMuonloader->LoadTracks("READ");
+ if (fMuonloader->TreeT() == NULL)
+ {
+ AliError("Could not load TreeT.");
+ return kFALSE;
+ }
+
+ // Need to reset these flags so that the rec tracks address
+ // gets reset after this method.
+ fRecTracksAddressSet = 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.
+/// 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.
+/// 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;
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.
+/// 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;
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.
+/// 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;
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.
+/// 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;
Int_t track, Int_t hit
)
{
-// Returns the specified hit in the given file, folder, event and track.
-// NULL is returned on error.
+/// 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;
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.
+/// 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 );
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.
+/// 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 );
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.
+/// 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 );
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.
+/// 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 );
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.
+/// 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;
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.
+/// 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;
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.
+/// 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;
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.
+/// 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;
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.
+/// 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.
+/// Select the current event from which to fetch data.
+/// kTRUE is returned if the event was found, else kFALSE is returned.
if (fRunloader == NULL)
{
Int_t AliMUONDataInterface::NumberOfEvents()
{
-// Get the number of events in the currently selected file.
-// -1 is returned on error.
+/// Get the number of events in the currently selected file.
+/// -1 is returned on error.
if (fRunloader == NULL)
{
Int_t AliMUONDataInterface::NumberOfParticles()
{
-// Get the number of particles in the current event.
-// -1 is returned on error.
+/// Get the number of particles in the current event.
+/// -1 is returned on error.
if (fRunloader == NULL)
{
TParticle* AliMUONDataInterface::Particle(Int_t particle)
{
-// Fetch the specified particle from the current event.
-// NULL is returned on error.
+/// Fetch the specified particle from the current event.
+/// NULL is returned on error.
if (fRunloader == NULL)
{
Int_t AliMUONDataInterface::NumberOfTracks()
{
-// Get the number of tracks in the current event.
-// -1 is returned on error.
+/// Get the number of tracks in the current event.
+/// -1 is returned on error.
if (fRunloader == NULL)
{
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.
+/// Get the number of hits for the given track in the current event.
+/// -1 is returned on error.
if (fRunloader == NULL)
{
AliMUONHit* AliMUONDataInterface::Hit(Int_t track, Int_t hit)
{
-// Fetch the specified hit from the current event.
-// NULL is returned on error.
+/// Fetch the specified hit from the current event.
+/// NULL is returned on error.
if (fRunloader == NULL)
{
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.
+/// 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 );
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.
+/// 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 );
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.
+/// 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 );
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.
+/// 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 );
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.
+/// Get the number of raw clusters on the given chamber in the current event.
+/// -1 is returned on error.
Assert( 0 <= chamber && chamber <= 13 );
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.
+/// Fetch the specified raw cluster on the given chamber from the current event.
+/// NULL is returned on error.
Assert( 0 <= chamber && chamber <= 13 );
Int_t AliMUONDataInterface::NumberOfLocalTriggers()
{
-// Get the number of local trigger objects in the current event.
-// -1 is returned on error.
+/// Get the number of local trigger objects in the current event.
+/// -1 is returned on error.
if (fRunloader == NULL)
{
AliMUONLocalTrigger* AliMUONDataInterface::LocalTrigger(Int_t trigger)
{
-// Fetch the specified local trigger object from the current event.
-// NULL is returned on error.
+/// Fetch the specified local trigger object from the current event.
+/// NULL is returned on error.
if (fRunloader == NULL)
{
}
return static_cast<AliMUONLocalTrigger*>( fData.LocalTrigger()->At(trigger) );
}
+
+Int_t AliMUONDataInterface::NumberOfGlobalTriggers()
+{
+/// Get the number of local trigger objects in the current event.
+/// -1 is returned on error.
+
+ if (fRunloader == NULL)
+ {
+ AliError("File not set.");
+ return -1;
+ }
+ if (fEventnumber < 0)
+ {
+ AliError("Event not chosen.");
+ return -1;
+ }
+
+ if ( ! FetchTreeD() ) return -1;
+ if ( ! fTriggerAddressSet )
+ {
+ fData.SetTreeAddress("GLT");
+ fData.ResetTrigger();
+ fData.GetTriggerD();
+ fTriggerAddressSet = kTRUE;
+ }
+ return fData.GlobalTrigger()->GetEntriesFast();
+}
+
+AliMUONGlobalTrigger* AliMUONDataInterface::GlobalTrigger(Int_t trigger)
+{
+/// Fetch the specified local trigger object from the current event.
+/// NULL is returned on error.
+
+ if (fRunloader == NULL)
+ {
+ AliError("File not set.");
+ return NULL;
+ }
+ if (fEventnumber < 0)
+ {
+ AliError( "Event not chosen.");
+ return NULL;
+ }
+
+ if ( ! FetchTreeD() ) return NULL;
+ if ( ! fTriggerAddressSet )
+ {
+ fData.SetTreeAddress("GLT");
+ fData.ResetTrigger();
+ fData.GetTriggerD();
+ fTriggerAddressSet = kTRUE;
+ }
+ return static_cast<AliMUONGlobalTrigger*>( fData.GlobalTrigger()->At(trigger) );
+}
+
+Int_t AliMUONDataInterface::NumberOfRecTracks()
+{
+/// Fetch the number of reconstructed tracks from the current event.
+/// NULL is returned on error.
+
+ if (fRunloader == NULL)
+ {
+ AliError("File not set.");
+ return -1;
+ }
+ if (fEventnumber < 0)
+ {
+ AliError( "Event not chosen.");
+ return -1;
+ }
+
+ if ( ! FetchTreeT() ) return -1;
+ if ( ! fRecTracksAddressSet )
+ {
+ fData.SetTreeAddress("RT");
+ fData.ResetRecTracks();
+ fData.GetRecTracks();
+ fRecTracksAddressSet = kTRUE;
+ }
+ return fData.RecTracks()->GetEntriesFast();
+}
+
+AliMUONTrack* AliMUONDataInterface::RecTrack(Int_t rectrack)
+{
+/// Fetch the specified reconstructed track object from the current event.
+/// NULL is returned on error.
+
+ if (fRunloader == NULL)
+ {
+ AliError("File not set.");
+ return NULL;
+ }
+ if (fEventnumber < 0)
+ {
+ AliError( "Event not chosen.");
+ return NULL;
+ }
+
+ if ( ! FetchTreeT() ) return NULL;
+ if ( ! fRecTracksAddressSet )
+ {
+ fData.SetTreeAddress("RT");
+ fData.ResetRecTracks();
+ fData.GetRecTracks();
+ fRecTracksAddressSet = kTRUE;
+ }
+ return static_cast<AliMUONTrack*>( fData.RecTracks()->At(rectrack) );
+ // return (AliMUONTrack*)(fData.RecTracks()->At(rectrack));
+}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff