Don't misalign geometry when creating snapshot.

[u/mrichter/AliRoot.git] / MUON / AliMUONRejectList.cxx

diff --git a/MUON/AliMUONRejectList.cxx b/MUON/AliMUONRejectList.cxx
index 29064319fb0dd8ede46b16b99fde809a0297713f..0d29e512b7a75b3eabf09e5b31beb88a216435ee 100644 (file)
--- a/MUON/AliMUONRejectList.cxx
+++ b/MUON/AliMUONRejectList.cxx
@@ -21,7 +21,9 @@
 /// Object to hold the probability to reject elements during reconstruction.
 ///
 /// Those elements are either channels, manus, 
-/// bus patches, detection elements, or all the four.
+/// bus patches, detection elements, pcbs (for slats) and HV channels,
+/// or all of them.
+///
 /// (we do not consider the next level, chamber, because if a full
 /// chamber is missing, we assume we'll remove that run from the
 /// list of usable runs anyway).
@@ -40,12 +42,25 @@
 
 #include "AliMUONRejectList.h"
 
+#include "AliLog.h"
+#include "AliMpArea.h"
 #include "AliMpConstants.h"
+#include "AliMpDCSNamer.h"
+#include "AliMpDDLStore.h"
+#include "AliMpDEStore.h"
+#include "AliMpDetElement.h"
+#include "AliMpMotifPosition.h"
+#include "AliMpPCB.h"
+#include "AliMpSegmentation.h"
+#include "AliMpSlat.h"
+#include "AliMpVSegmentation.h"
 #include "AliMUON2DMap.h"
 #include "AliMUONCalibParamNF.h"
 #include "Riostream.h"
 #include "TMath.h"
 
+using std::cout;
+using std::endl;
 /// \cond CLASSIMP
 ClassImp(AliMUONRejectList)
 /// \endcond
@@ -382,6 +397,60 @@ void AliMUONRejectList::SetChannelProbability(Int_t detElemId, Int_t manuId, Int
   ZeroOrOne(proba);
 }
 
+//_____________________________________________________________________________
+void AliMUONRejectList::SetPCBProbability(Int_t detElemId, Int_t pcbNumber, Float_t proba)
+{
+  /// Set the probability to reject all the manus of a given (slat) PCB
+  AliMpSegmentation* seg = AliMpSegmentation::Instance();
+  AliMp::CathodType ct[] = { AliMp::kCath0, AliMp::kCath1 };
+  
+  for ( Int_t i = 0; i < 2; ++i )
+  {
+    const AliMpVSegmentation* vseg = seg->GetMpSegmentation(detElemId,ct[i]);
+    if (!vseg)
+    {
+      AliError(Form("Could not get segmentation of DE %d",detElemId));
+      continue;
+    }
+    const AliMpSlat* slat = seg->GetSlat(vseg);
+    if (!slat)
+    {
+      AliError(Form("Could not get slat from DE %d",detElemId));
+      continue;      
+    }
+    AliMpPCB* pcb = slat->GetPCB(pcbNumber);
+    for ( Int_t j = 0; j < pcb->GetSize(); ++j )
+    {
+      AliMpMotifPosition* mp = pcb->GetMotifPosition(j);
+      SetManuProbability(detElemId,mp->GetID(),proba);
+    }
+  }
+}
+
+//_____________________________________________________________________________
+void AliMUONRejectList::SetHVProbability(const char* dcsName, Float_t proba)
+{
+  /// Set the probability to reject all the manus of a given HV part
+  /// Caution : the dcs string is a dcs NAME, _not_ an alias
+  
+  AliMpDCSNamer hv("TRACKER");
+
+  TString alias = hv.DCSAliasFromName(dcsName);
+
+  Int_t detElemId = hv.DetElemIdFromDCSAlias(alias.Data());
+  Int_t index = hv.DCSIndexFromDCSAlias(alias.Data());
+
+  AliMpDetElement* de = AliMpDDLStore::Instance()->GetDetElement(detElemId);
+
+  const AliMpArrayI* manus = de->ManusForHV(index);
+
+  for ( Int_t i = 0; i < manus->GetSize(); ++ i )
+  {
+    Int_t manuId = manus->GetValue(i);
+    SetManuProbability(detElemId,manuId,proba);
+  }
+}
+
 //_____________________________________________________________________________
 void 
 AliMUONRejectList::Print(Option_t* opt) const