fix spline selection for MC for train tests - Extension

[u/mrichter/AliRoot.git] / EMCAL / AliEMCALClusterizerNxN.h

diff --git a/EMCAL/AliEMCALClusterizerNxN.h b/EMCAL/AliEMCALClusterizerNxN.h
index 007844d497a9a2f4e1bf3ba52894e90f02e61e37..a01ad867a0e07df8825ea2346b0f430798e5a8dc 100644 (file)
--- a/EMCAL/AliEMCALClusterizerNxN.h
+++ b/EMCAL/AliEMCALClusterizerNxN.h
@@ -37,6 +37,13 @@ public:
 
   virtual const char * Version() const { return "clu-NxN" ; }  
   
+  void SetNRowDiff(Int_t nd) { fNRowDiff = nd; }
+  void SetNColDiff(Int_t nd) { fNColDiff = nd; }
+  Int_t GetNRowDiff() const { return fNRowDiff; } 
+  Int_t GetNColDiff() const { return fNColDiff; } 
+  void SetEnergyGrad(Bool_t b) { fEnergyGrad= b; }
+  Bool_t GetEnergyGrad() const { return fEnergyGrad; }
+
 protected:
 
   virtual void   MakeClusters();            
@@ -45,8 +52,11 @@ private:
   AliEMCALClusterizerNxN(const AliEMCALClusterizerNxN &); //copy ctor
   AliEMCALClusterizerNxN & operator = (const AliEMCALClusterizerNxN &);
 
-   ClassDef(AliEMCALClusterizerNxN,2)   // Clusterizer implementation version 1
+  Int_t  fNRowDiff;  //how many neighbors to consider along row (phi)
+  Int_t  fNColDiff;  //how many neighbors to consider along col (eta)
+  Bool_t fEnergyGrad; //if true only cluster if neighboring cell has less energy
 
+  ClassDef(AliEMCALClusterizerNxN,4)   // Clusterizer implementation version 1
 };
 
 #endif // AliEMCALCLUSTERIZERNXN_H