X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=CORRFW%2FAliCFManager.h;h=7e5a8411adf4b8828e97ec19fd1a1f8c2eefc5f5;hb=ac6b9aed8a1b7d3c1afc5cbdec9b5ea210158f22;hp=7d76ab11b391202826a25c14895e808a2ffc492a;hpb=00148604895ba731a3eaa2b5a36bb6ce35fa4b93;p=u%2Fmrichter%2FAliRoot.git

diff --git a/CORRFW/AliCFManager.h b/CORRFW/AliCFManager.h
index 7d76ab11b39..7e5a8411adf 100644
--- a/CORRFW/AliCFManager.h
+++ b/CORRFW/AliCFManager.h
@@ -66,17 +66,21 @@ class AliCFManager : public TNamed
   //pass the pointer to the correction container
   virtual void SetEventContainer(AliCFContainer* c) {
     fEvtContainer=c; 
-    fNStepEvt=c->GetNStep();
+    SetNStepEvent(c->GetNStep());
     AliWarning(Form("Please dont forget to set the cut list (event empty) for the %d event-selection step requested",fNStepEvt));
   }
   
   //pass the pointer to the correction container
   virtual void SetParticleContainer(AliCFContainer* c) {
     fPartContainer=c; 
-    fNStepPart=c->GetNStep();
+    SetNStepParticle(c->GetNStep());
     AliWarning(Form("Please dont forget to set the cut list (even empty) for the %d particle-selection step requested",fNStepPart));
   }
   
+  //Set the number of steps (already done if you have defined your containers)
+  virtual void SetNStepEvent   (Int_t nstep) {fNStepEvt  = nstep;}
+  virtual void SetNStepParticle(Int_t nstep) {fNStepPart = nstep;}
+
   //Setter for event-level selection cut list at selection step isel
   virtual void SetEventCutsList(Int_t isel, TObjArray* array) ;
   
@@ -113,8 +117,8 @@ class AliCFManager : public TNamed
  private:
   
   //number of steps
-  Int_t fNStepEvt; 
-  Int_t fNStepPart; 
+  Int_t fNStepEvt;  // number of steps in event selection
+  Int_t fNStepPart; // number of steps in particle selection
   //the correction grid
   AliCFContainer *fEvtContainer; //ptr to Evt-Level correction container
   //the correction grid