Fixing minor bug recognizing diffractive events in simulation

[u/mrichter/AliRoot.git] / MUON / AliMUONPreClusterFinder.h

diff --git a/MUON/AliMUONPreClusterFinder.h b/MUON/AliMUONPreClusterFinder.h
index 6ff9f7f4914f656901a9c6e0ac73e2d55b2f78a4..9983f95f0e3b032177209a7d1a8e823834fc0b66 100644 (file)
--- a/MUON/AliMUONPreClusterFinder.h
+++ b/MUON/AliMUONPreClusterFinder.h
@@ -18,9 +18,13 @@
 #ifndef ALI_MP_AREA_H
 #  include "AliMpArea.h"
 #endif
+#ifndef ROOT_TClonesArray
+#  include <TClonesArray.h>
+#endif
 
 class TStopwatch;
 class AliMUONPad;
+class TObjArray;
 
 class AliMUONPreClusterFinder : public AliMUONVClusterFinder
 {
@@ -31,7 +35,7 @@ public:
   using AliMUONVClusterFinder::Prepare;
 
   virtual Bool_t Prepare(Int_t detElemId,
-                         TClonesArray* pads[2],
+                         TObjArray* pads[2],
                          const AliMpArea& area);
   
   virtual AliMUONCluster* NextCluster();
@@ -51,14 +55,17 @@ private:
   /// Whether we should stop working...
   virtual Bool_t ShouldAbort() const { return fShouldAbort; }
   
+  AliMUONCluster* NewCluster();
+  void RemoveCluster(AliMUONCluster* cluster);
+  
 private:
-  TClonesArray* fClusters; //!< the clusters we've found (owner)
-  TClonesArray** fPads; //!< the pads corresponding to the digits (not owner)
+  TClonesArray fClusters; //!< the clusters we've found (owner)
+  TObjArray** fPads; //!< the pads corresponding to the digits (not owner)
   Int_t fDetElemId; //!< which DE we're considering
   AliMpArea fArea; //!< area into which to consider pads to *start* a cluster
   Bool_t fShouldAbort; //!< to indicate clustering should stop right now
   
-  ClassDef(AliMUONPreClusterFinder,3) // A basic pre-cluster finder
+  ClassDef(AliMUONPreClusterFinder,4) // A basic pre-cluster finder
 };
 
 #endif