MUON/AliMUONPreClusterFinderV3.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 #include "AliMUONPreClusterFinderV3.h"
  19
  20 #include "AliLog.h"
  21 #include "AliMUONCluster.h"
  22 #include "AliMpVSegmentation.h"
  23 #include "TObjArray.h"
  24 #include "AliMpArea.h"
  25 #include "TVector2.h"
  26 #include "AliMUONPad.h"
  27 #include "AliMUONVDigit.h"
  28 #include "AliMUONVDigitStore.h"
  29 #include <Riostream.h>
  30 //#include "AliCodeTimer.h"
  31
  32 //-----------------------------------------------------------------------------
  33 /// \class AliMUONPreClusterFinderV3
  34 ///
  35 /// Implementation of AliMUONVClusterFinder
  36 ///
  37 /// This version uses a 2 steps approach :
  38 ///
  39 /// we first clusterize each cathode independently to form proto-preclusters,
  40 /// and then we try to "merge" proto-preclusters from the two cathodes
  41 /// when thoses proto-preclusters overlap, thus ending up with preclusters
  42 /// spanning the two cathodes.
  43 ///
  44 /// This implementation, on the contrary to PreClusterFinder or PreClusterFinderV2
  45 /// should not depend on the order of the input digits.
  46 ///
  47 /// \author Laurent Aphecetche
  48 //-----------------------------------------------------------------------------
  49
  50 using std::endl;
  51 using std::cout;
  52 ClassImp(AliMUONPreClusterFinderV3)
  53
  54 //_____________________________________________________________________________
  55 AliMUONPreClusterFinderV3::AliMUONPreClusterFinderV3()
  56 : AliMUONVClusterFinder(),
  57   fClusters(new TClonesArray("AliMUONCluster",10)),
  58   fkSegmentations(0x0),
  59   fPads(0x0),
  60   fDetElemId(0),
  61   fIterator(0x0)
  62 {
  63     /// ctor
  64   AliInfo("");
  65   for ( Int_t i = 0; i < 2; ++i )
  66   {
  67     fPreClusters[i] = new TClonesArray("AliMUONCluster",10);
  68   }
  69 }
  70
  71 //_____________________________________________________________________________
  72 AliMUONPreClusterFinderV3::~AliMUONPreClusterFinderV3()
  73 {
  74   /// dtor
  75   delete fClusters;
  76   for ( Int_t i = 0; i < 2; ++i )
  77   {
  78     delete fPreClusters[i];
  79   }
  80 }
  81
  82 //_____________________________________________________________________________
  83 Bool_t
  84 AliMUONPreClusterFinderV3::UsePad(const AliMUONPad& pad)
  85 {
  86   /// Add a pad to the list of pads to be considered
  87   if ( pad.DetElemId() != fDetElemId )
  88   {
  89     AliError(Form("Cannot add pad from DE %d to this cluster finder which is "
  90                   "currently dealing with DE %d",pad.DetElemId(),fDetElemId));
  91     return kFALSE;
  92   }
  93
  94   AliMUONPad* p = new AliMUONPad(pad);
  95   p->SetClusterId(-1);
  96   fPads[pad.Cathode()]->Add(p);
  97   return kTRUE;
  98 }
  99
 100 //_____________________________________________________________________________
 101 Bool_t
 102 AliMUONPreClusterFinderV3::Prepare(Int_t detElemId,
 103                                    TObjArray* pads[2],
 104                                    const AliMpArea& area,
 105                                    const AliMpVSegmentation* seg[2])
 106 {
 107   /// Prepare for clustering, by giving access to segmentations and digit lists
 108
 109   if ( area.IsValid() )
 110   {
 111     AliError("Handling of area not yet implemented for this class. Please check.");
 112   }
 113
 114   fkSegmentations = seg;
 115   fPads = pads;
 116
 117   fClusters->Clear("C");
 118   for ( Int_t i = 0; i < 2; ++i )
 119   {
 120     fPreClusters[i]->Clear("C");
 121   }
 122
 123   fDetElemId = detElemId;
 124
 125   if ( fPads[0]->GetLast() < 0 && fPads[1]->GetLast() < 0 )
 126   {
 127     // no pad at all, nothing to do...
 128     return kFALSE;
 129   }
 130
 131   MakeCathodePreClusters(0);
 132   MakeCathodePreClusters(1);
 133   MakeClusters();
 134
 135   delete fIterator;
 136   fIterator = fClusters->MakeIterator();
 137
 138   return kTRUE;
 139 }
 140
 141 //_____________________________________________________________________________
 142 void
 143 AliMUONPreClusterFinderV3::DumpPreClusters() const
 144 {
 145   /// Dump preclusters
 146   AliMUONCluster *c;
 147   TIter next0(fPreClusters[0]);
 148   TIter next1(fPreClusters[1]);
 149   cout << "Cath0" << endl;
 150   while ( ( c = static_cast<AliMUONCluster*>(next0())) )
 151   {
 152     cout << c->AsString().Data() << endl;
 153   }
 154   cout << "Cath1" << endl;
 155   while ( ( c = static_cast<AliMUONCluster*>(next1())) )
 156   {
 157     cout << c->AsString().Data() << endl;
 158   }
 159 }
 160
 161 //_____________________________________________________________________________
 162 void
 163 AliMUONPreClusterFinderV3::AddPreCluster(AliMUONCluster& cluster, AliMUONCluster* preCluster)
 164 {
 165   /// Add a pad to a cluster
 166
 167   AliMUONCluster a(*preCluster);
 168
 169   Int_t cathode = preCluster->Cathode();
 170   if ( cathode < 0 ) {
 171     AliError(Form("Cathod undefined: %d",cathode));
 172     AliFatal("");
 173     return;
 174   }
 175
 176   if ( cathode <=1 && !fPreClusters[cathode]->Remove(preCluster) )
 177   {
 178     AliError(Form("Could not remove %s from preclusters[%d]",
 179                   preCluster->AsString().Data(),cathode));
 180     StdoutToAliDebug(1,DumpPreClusters());
 181     AliFatal("");
 182     return;
 183   }
 184
 185   cluster.AddCluster(a);
 186
 187   // loop on the *other* cathode
 188   TIter next(fPreClusters[1-cathode]);
 189   AliMUONCluster* testCluster;
 190
 191   while ( ( testCluster = static_cast<AliMUONCluster*>(next())))
 192   {
 193     if ( AliMUONCluster::AreOverlapping(a,*testCluster) )
 194     {
 195       AddPreCluster(cluster,testCluster);
 196     }
 197   }
 198 }
 199
 200
 201 //_____________________________________________________________________________
 202 void
 203 AliMUONPreClusterFinderV3::AddPad(AliMUONCluster& cluster, AliMUONPad* pad)
 204 {
 205   /// Add a pad to a cluster
 206   AliMUONPad* addedPad = cluster.AddPad(*pad);
 207
 208   Int_t cathode = pad->Cathode();
 209   TObjArray& padArray = *fPads[cathode];
 210   delete padArray.Remove(pad);
 211
 212   TIter next(&padArray);
 213   AliMUONPad* testPad;
 214
 215   while ( ( testPad = static_cast<AliMUONPad*>(next())))
 216   {
 217     if ( AliMUONPad::AreNeighbours(*testPad,*addedPad) )
 218     {
 219       AddPad(cluster,testPad);
 220     }
 221   }
 222 }
 223
 224 //_____________________________________________________________________________
 225 AliMUONCluster*
 226 AliMUONPreClusterFinderV3::NextCluster()
 227 {
 228   /// Returns the next cluster
 229
 230   return static_cast<AliMUONCluster*>(fIterator->Next());
 231 }
 232
 233 //_____________________________________________________________________________
 234 void
 235 AliMUONPreClusterFinderV3::MakeClusters()
 236 {
 237   /// Associate (proto)preclusters to form (pre)clusters
 238
 239 //  AliCodeTimerAuto("",0)
 240
 241   for ( Int_t cathode = 0; cathode < 2; ++cathode )
 242   {
 243     TClonesArray& preclusters = *(fPreClusters[cathode]);
 244
 245     TIter next(&preclusters);
 246     AliMUONCluster* preCluster(0x0);
 247
 248     while ( ( preCluster = static_cast<AliMUONCluster*>(next()) ) )
 249     {
 250       Int_t id(fClusters->GetLast()+1);
 251       AliMUONCluster* cluster = new((*fClusters)[id]) AliMUONCluster;
 252       cluster->SetUniqueID(id);
 253       AddPreCluster(*cluster,preCluster);
 254     }
 255   }
 256 }
 257
 258 //_____________________________________________________________________________
 259 void
 260 AliMUONPreClusterFinderV3::MakeCathodePreClusters(Int_t cathode)
 261 {
 262   /// Build (proto)preclusters from digits on a given cathode
 263
 264 //  AliCodeTimerAuto(Form("Cathode %d",cathode),0)
 265
 266   while ( fPads[cathode]->GetLast() > 0  )
 267   {
 268     TIter next(fPads[cathode]);
 269     AliMUONPad* pad = static_cast<AliMUONPad*>(next());
 270
 271     if (!pad) AliFatal("");
 272
 273     Int_t id = fPreClusters[cathode]->GetLast()+1;
 274     AliMUONCluster* cluster = new ((*fPreClusters[cathode])[id]) AliMUONCluster;
 275     cluster->SetUniqueID(id);
 276
 277     // Builds (recursively) a cluster on first cathode only
 278     AddPad(*cluster,pad);
 279
 280     if ( cluster->Multiplicity() <= 1 )
 281     {
 282       if ( cluster->Multiplicity() == 0 )
 283       {
 284         // no pad is suspicious
 285         AliWarning("Got an empty cluster...");
 286       }
 287       // else only 1 pad (not suspicious, but kind of useless, probably noise)
 288       // so we remove it from our list
 289       fPreClusters[cathode]->Remove(cluster);
 290       // then proceed further
 291     }
 292   }
 293
 294 }