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 "AliMUONTrack.h"
  32 #include "AliMUONTreeManager.h"
  33
  34 /// \cond CLASSIMP
  35 ClassImp(AliMUONTrackStoreV1)
  36 /// \endcond
  37
  38 //_____________________________________________________________________________
  39 AliMUONTrackStoreV1::AliMUONTrackStoreV1() : AliMUONVTrackStore(),
  40  fTracks(new TClonesArray("AliMUONTrack",10))
  41 {
  42    /// Ctor
  43 }
  44
  45 //_____________________________________________________________________________
  46 AliMUONTrackStoreV1::~AliMUONTrackStoreV1()
  47 {
  48   /// dtor
  49   delete fTracks;
  50 }
  51
  52 //_____________________________________________________________________________
  53 AliMUONTrack*
  54 AliMUONTrackStoreV1::Add(const AliMUONTrack& track)
  55 {
  56   /// Add a track
  57   return new((*fTracks)[fTracks->GetLast()+1]) AliMUONTrack(track);
  58 }
  59
  60 //_____________________________________________________________________________
  61 AliMUONTrack*
  62 AliMUONTrackStoreV1::Remove(AliMUONTrack& track)
  63 {
  64   /// Remove a track from the store
  65   AliMUONTrack* t = static_cast<AliMUONTrack*>(fTracks->Remove(&track));
  66   if (t) fTracks->Compress();
  67   return t;
  68 }
  69
  70 //_____________________________________________________________________________
  71 Bool_t
  72 AliMUONTrackStoreV1::Connect(TTree& tree, Bool_t alone) const
  73 {
  74   /// Connect this store to the tree
  75   AliMUONTreeManager tman;
  76   Bool_t ok;
  77
  78   if ( tree.GetBranch("MUONTrack") )
  79   {
  80     if ( alone ) tman.UpdateBranchStatuses(tree,"MUONTrack");
  81     ok = tman.SetAddress(tree,"MUONTrack",TracksPtr());
  82   }
  83   else
  84   {
  85     ok = tman.MakeBranch(tree,ClassName(),"TClonesArray","MUONTrack",
  86                          TracksPtr());
  87   }
  88   return ok;
  89 }
  90
  91 //_____________________________________________________________________________
  92 TIterator*
  93 AliMUONTrackStoreV1::CreateIterator() const
  94 {
  95   /// Create an iterator to loop over tracks
  96   return fTracks->MakeIterator();
  97 }
  98
  99 //_____________________________________________________________________________
 100 void
 101 AliMUONTrackStoreV1::Clear(Option_t*)
 102 {
 103   /// Reset
 104   fTracks->Clear("C");
 105 }
 106
 107 //_____________________________________________________________________________
 108 Int_t
 109 AliMUONTrackStoreV1::GetSize() const
 110 {
 111   /// Return the number of tracks we hold
 112   return fTracks->GetLast()+1;
 113 }