simplified efficiency projection

[u/mrichter/AliRoot.git] / CORRFW / AliCFManager.h

diff --git a/CORRFW/AliCFManager.h b/CORRFW/AliCFManager.h
index 7d76ab11b391202826a25c14895e808a2ffc492a..7e5a8411adf4b8828e97ec19fd1a1f8c2eefc5f5 100644 (file)
--- 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