X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=TRD%2FAliTRDCalibTask.h;h=e06e02e5554896a04adceb2add2bfe21cc782f29;hb=b14fe7f1773d1d7542bd13e937c0b25f07bfa1ee;hp=d2bf60430719d4e815a043ffd8e57a448c2131ea;hpb=cdf01180ec09afb07b5b92770433f329e1fba53f;p=u%2Fmrichter%2FAliRoot.git

diff --git a/TRD/AliTRDCalibTask.h b/TRD/AliTRDCalibTask.h
index d2bf6043071..e06e02e5554 100644
--- a/TRD/AliTRDCalibTask.h
+++ b/TRD/AliTRDCalibTask.h
@@ -28,17 +28,17 @@ class AliESDtrackCuts;
 class AliTRDCalDet;
 
 #include "TObjString.h"
-#include "AliAnalysisTask.h" 
+#include "AliAnalysisTaskSE.h" 
 #include "TMath.h"
 
-class AliTRDCalibTask : public AliAnalysisTask {
+class AliTRDCalibTask : public AliAnalysisTaskSE {
  public:
   AliTRDCalibTask(const char *name = "AliTRDCalibTask");
   virtual ~AliTRDCalibTask();
   
-  virtual void   ConnectInputData(Option_t *);
-  virtual void   CreateOutputObjects();
-  virtual void   Exec(Option_t *);
+  //  virtual void   ConnectInputData(Option_t *);
+  virtual void   UserCreateOutputObjects();
+  virtual void   UserExec(Option_t *);
   virtual void   Terminate(Option_t *);
   virtual Bool_t Load(const Char_t *filename);
   virtual Bool_t Load(TList *lister);
@@ -63,7 +63,16 @@ class AliTRDCalibTask : public AliAnalysisTask {
   void SetUseTPCVertex()                                               {fVtxTPC=kTRUE ; fVtxSPD=kFALSE;} 
   void SetUseSPDVertex()                                               {fVtxTPC=kFALSE; fVtxSPD=kTRUE ;} 
   void SetMinNbOfContributors(Int_t minNbOfContributors)               {fMinNbContributors = minNbOfContributors;};  
-  void SetRangePrimaryVertexZ(Double_t rangePrimaryVertexZ)            {fRangePrimaryVertexZ = TMath::Abs(rangePrimaryVertexZ);};  
+  void SetRangePrimaryVertexZ(Double_t rangePrimaryVertexZ)            {fRangePrimaryVertexZ = TMath::Abs(rangePrimaryVertexZ);}; 
+  void SetMinNbTracks(Int_t minNbTracks)                               {fMinNbTracks = minNbTracks;};
+  void SetMaxNbTracks(Int_t maxNbTracks)                               {fMaxNbTracks = maxNbTracks;};
+ 
+  void SetVersionGainUsed(Int_t versionGainUsed)                       { fVersionGainUsed = versionGainUsed;   }
+  void SetSubVersionGainUsed(Int_t subVersionGainUsed)                 { fSubVersionGainUsed = subVersionGainUsed;   }
+  void SetVersionGainLocalUsed(Int_t versionGainLocalUsed)             { fVersionGainLocalUsed = versionGainLocalUsed;   }
+  void SetSubVersionGainLocalUsed(Int_t subVersionGainLocalUsed)       { fSubVersionGainLocalUsed = subVersionGainLocalUsed;   }
+  void SetVersionVdriftUsed(Int_t versionVdriftUsed)                   { fVersionVdriftUsed = versionVdriftUsed;   }
+  void SetSubVersionVdriftUsed(Int_t subVersionVdriftUsed)             { fSubVersionVdriftUsed = subVersionVdriftUsed;   }
   
   void SetLow(Int_t low)                                               {fLow=low;};
   void SetHigh(Int_t high)                                             {fHigh=high;};
@@ -73,13 +82,16 @@ class AliTRDCalibTask : public AliAnalysisTask {
   void SetNbMaxCluster(Short_t nbMaxCluster)                           {fNbMaxCluster =  nbMaxCluster; }; 
   void SetOfflineTracks()                                              {fOfflineTracks=kTRUE; fStandaloneTracks=kFALSE; };
   void SetStandaloneTracks()                                           {fStandaloneTracks=kTRUE; fOfflineTracks=kFALSE; };
-
-  void SetCalDetGain(AliTRDCalDet * const calDetGain)                         {fCalDetGain = calDetGain;};  
+  
+  void SetCalDetGain(AliTRDCalDet * const calDetGain)                  {fCalDetGain = calDetGain;};  
 
   void SetMaxEvent(Int_t nbevents)                                     { fMaxEvent = nbevents; };
   void SetDebug(Int_t debug)                                           { fDebug = debug; };
 
  private:
+  Bool_t SetVersionSubversion();
+  Bool_t ParticleGood(int i) const;
+
   AliESDEvent  *fESD;                            //! ESD object
   AliESDfriend *fESDfriend;                      //! ESD friend
   const AliESDtrack *fkEsdTrack;                  //! ESD track
@@ -98,6 +110,7 @@ class AliTRDCalibTask : public AliAnalysisTask {
   TH1F        *fNbTRDTrackOffline;               //! nb ESD tracks with TRD clusters
   TH1F        *fNbTRDTrackStandalone;            //! nb ESD tracks with TRD clusters
   TH2F        *fNbTPCTRDtrack;                   //! nb TPC and TRD tracks when problems
+  TH2F        *fNbGoodTracks;                    //! nb of good tracks
    
   TH1F        *fNbTimeBin;                       //! nb Time Bin
   TH1F        *fNbTimeBinOffline;                //! nb Time Bin offline
@@ -133,6 +146,8 @@ class AliTRDCalibTask : public AliAnalysisTask {
   Bool_t      fVtxSPD;                           // Flag for use of SPD vertex
   Int_t       fMinNbContributors;                // Min number of contributors
   Double_t    fRangePrimaryVertexZ;              // Were the primary vertex is
+  Int_t       fMinNbTracks;                      // Min Nb Tracks
+  Int_t       fMaxNbTracks;                      // Max Nb Tracks
   
   Int_t       fLow;                              // lower limit of nb of TRD clusters per tracklet
   Int_t       fHigh;                             // higher limit of nb of TRD clusters per tracklet
@@ -144,6 +159,13 @@ class AliTRDCalibTask : public AliAnalysisTask {
   Bool_t      fOfflineTracks;                    // Only Offline refitted tracks
   Bool_t      fStandaloneTracks;                 // Take only standalone tracks
 
+  Int_t       fVersionGainUsed;                  // VersionGainUsed 
+  Int_t       fSubVersionGainUsed;               // SubVersionGainUsed
+  Int_t       fVersionGainLocalUsed;             // VersionGainLocalUsed 
+  Int_t       fSubVersionGainLocalUsed;          // SubVersionGainLocalUsed
+  Int_t       fVersionVdriftUsed;                // VersionVdriftUsed 
+  Int_t       fSubVersionVdriftUsed;             // SubVersionVdriftUsed
+
   AliTRDCalDet *fCalDetGain;                     // Calib object gain
 
   Int_t       fMaxEvent;                         // max events
@@ -157,3 +179,5 @@ class AliTRDCalibTask : public AliAnalysisTask {
 };
 
 #endif
+
+