Add method to apply offline zero suppression to SDD raw data

[u/mrichter/AliRoot.git] / ITS / AliITSOnlineSDD.cxx

diff --git a/ITS/AliITSOnlineSDD.cxx b/ITS/AliITSOnlineSDD.cxx
index a2075551ba1e9f6f493496a851b995a5d610f891..38a3b0d8c24dcea1edef4bf3e7915469de2b0d3b 100644 (file)
--- a/ITS/AliITSOnlineSDD.cxx
+++ b/ITS/AliITSOnlineSDD.cxx
@@ -21,6 +21,7 @@
 ///////////////////////////////////////////////////////////////////
 
 #include "AliITSOnlineSDD.h"
+#include "TH2F.h"
 
 ClassImp(AliITSOnlineSDD)
 //______________________________________________________________________
@@ -40,3 +41,52 @@ ClassImp(AliITSOnlineSDD)
   SetFirstGoodTB();
   SetLastGoodTB();
 }
+//______________________________________________________________________
+TH2F* AliITSOnlineSDD::ApplyZeroSuppression(TH2F* hRaw, Float_t basl, Int_t tL, Int_t tH){
+  // apply zero suppression 
+
+  Int_t nx=hRaw->GetNbinsX();
+  Float_t xmin=hRaw->GetXaxis()->GetXmin();
+  Float_t xmax=hRaw->GetXaxis()->GetXmax();
+  Int_t ny=hRaw->GetNbinsY();
+  Float_t ymin=hRaw->GetYaxis()->GetXmin();
+  Float_t ymax=hRaw->GetYaxis()->GetXmax();
+  TH2F* hZeroSupp=new TH2F(Form("%s_ZeroSupp",hRaw->GetName()),"",nx,xmin,xmax,ny,ymin,ymax);
+  for(Int_t ix=1; ix<=nx;ix++){
+    for(Int_t iy=1; iy<=ny;iy++){
+      //                     N
+      // Get "quintuple":   WCE
+      //                     S
+      Float_t cC=hRaw->GetBinContent(ix,iy)-basl;
+      Int_t nLow=0, nHigh=0;      
+      if(cC<tL){
+       hZeroSupp->SetBinContent(ix,iy,0.);
+       continue;
+      }
+      nLow++; // cC is greater than tL
+      if(cC>tH) nHigh++;
+      Float_t wW=0.;
+      if(iy>0) wW=hRaw->GetBinContent(ix,iy-1)-basl;
+      if(wW>tL) nLow++;
+      if(wW>tH) nHigh++;
+      Float_t eE=0.;
+      if(iy<ny) eE=hRaw->GetBinContent(ix,iy+1)-basl;
+      if(eE>tL) nLow++;
+      if(eE>tH) nHigh++;
+      Float_t nN=0.;
+      if(ix<nx) nN=hRaw->GetBinContent(ix+1,iy)-basl;
+      if(nN>tL) nLow++;
+      if(nN>tH) nHigh++;
+      Float_t sS=0.;
+      if(ix>0) sS=hRaw->GetBinContent(ix-1,iy)-basl;
+      if(sS>tL) nLow++;
+      if(sS>tH) nHigh++;
+      if(nLow>=2 && nHigh>=1){
+       hZeroSupp->SetBinContent(ix,iy,cC);
+      }else{
+       hZeroSupp->SetBinContent(ix,iy,0.);
+      }
+    }
+  }
+  return hZeroSupp;
+}