MUON/AliMUONTrackStoreV1.cxx

   1 /**************************************************************************
   2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3 *                                                                        *
   4 * Author: The ALICE Off-line Project.                                    *
   5 * Contributors are mentioned in the code where appropriate.              *
   6 *                                                                        *
   7 * Permission to use, copy, modify and distribute this software and its   *
   8 * documentation strictly for non-commercial purposes is hereby granted   *
   9 * without fee, provided that the above copyright notice appears in all   *
  10 * copies and that both the copyright notice and this permission notice   *
  11 * appear in the supporting documentation. The authors make no claims     *
  12 * about the suitability of this software for any purpose. It is          *
  13 * provided "as is" without express or implied warranty.                  *
  14 **************************************************************************/
  15
  16 // $Id$
  17
  18 //-----------------------------------------------------------------------------
  19 /// \class AliMUONTrackStoreV1
  20 ///
  21 /// Implementation of AliMUONTrackStoreV1, which should be backward
  22 /// compatible, i.e. able to read old TreeT files
  23 ///
  24 /// \author Laurent Aphecetche, Subatech
  25 //-----------------------------------------------------------------------------
  26
  27 #include "AliMUONTrackStoreV1.h"
  28
  29 #include <TClonesArray.h>
  30 #include <TTree.h>
  31 #include "AliLog.h"
  32 #include "AliMUONTrack.h"
  33 #include "AliMUONTreeManager.h"
  34
  35 /// \cond CLASSIMP
  36 ClassImp(AliMUONTrackStoreV1)
  37 /// \endcond
  38
  39 //_____________________________________________________________________________
  40 AliMUONTrackStoreV1::AliMUONTrackStoreV1() : AliMUONVTrackStore(),
  41  fTracks(0x0)
  42 {
  43    /// Ctor
  44   CreateTracks();
  45 }
  46
  47 //_____________________________________________________________________________
  48 AliMUONTrackStoreV1::AliMUONTrackStoreV1(TRootIOCtor* /*dummy*/) : AliMUONVTrackStore(),
  49 fTracks(0x0)
  50 {
  51   /// Ctor
  52 }
  53
  54 //_____________________________________________________________________________
  55 AliMUONTrackStoreV1::~AliMUONTrackStoreV1()
  56 {
  57   /// dtor
  58   delete fTracks;
  59 }
  60
  61 //_____________________________________________________________________________
  62 AliMUONTrack*
  63 AliMUONTrackStoreV1::Add(const AliMUONTrack& track)
  64 {
  65   /// Add a track
  66
  67   if (!fTracks) CreateTracks();
  68
  69   return new((*fTracks)[fTracks->GetLast()+1]) AliMUONTrack(track);
  70 }
  71
  72 //_____________________________________________________________________________
  73 AliMUONTrack*
  74 AliMUONTrackStoreV1::Remove(AliMUONTrack& track)
  75 {
  76   /// Remove a track from the store
  77   AliMUONTrack* t = static_cast<AliMUONTrack*>(fTracks->Remove(&track));
  78   if (t) fTracks->Compress();
  79   return t;
  80 }
  81
  82 //_____________________________________________________________________________
  83 Bool_t
  84 AliMUONTrackStoreV1::Connect(TTree& tree, Bool_t alone) const
  85 {
  86   /// Connect this store to the tree
  87   AliMUONTreeManager tman;
  88
  89   Bool_t ok;
  90
  91   if ( tree.GetBranch("MUONTrack") )
  92   {
  93     if ( alone ) tman.UpdateBranchStatuses(tree,"MUONTrack");
  94     ok = tman.SetAddress(tree,"MUONTrack",TracksPtr());
  95   }
  96   else
  97   {
  98     ok = tman.MakeBranch(tree,ClassName(),"TClonesArray","MUONTrack",
  99                          TracksPtr());
 100   }
 101   return ok;
 102 }
 103
 104 //_____________________________________________________________________________
 105 TIterator*
 106 AliMUONTrackStoreV1::CreateIterator() const
 107 {
 108   /// Create an iterator to loop over tracks
 109   if ( fTracks ) return fTracks->MakeIterator();
 110   return 0x0;
 111 }
 112
 113 //_____________________________________________________________________________
 114 void
 115 AliMUONTrackStoreV1::Clear(Option_t*)
 116 {
 117   /// Reset
 118   if (fTracks) fTracks->Clear("C");
 119 }
 120
 121 //_____________________________________________________________________________
 122 void
 123 AliMUONTrackStoreV1::CreateTracks()
 124 {
 125   /// Allocate track container
 126   if (fTracks)
 127   {
 128     AliError("Cannot allocate again fTracks as it is there already !");
 129   }
 130   else
 131   {
 132     fTracks = new TClonesArray("AliMUONTrack",10);
 133   }
 134 }
 135
 136 //_____________________________________________________________________________
 137 Int_t
 138 AliMUONTrackStoreV1::GetSize() const
 139 {
 140   /// Return the number of tracks we hold
 141   if ( fTracks ) return fTracks->GetLast()+1;
 142   return 0;
 143 }