AliCorrectionManagerBase.h

Go to the documentation of this file.

// -*- mode: C++ -*-
#ifndef ALICORRECTIONMANAGERBASE_H
#define ALICORRECTIONMANAGERBASE_H
#include <TString.h>
#include <TNamed.h>
#include <TObjArray.h>
class AliOADBForward;
class TBrowser;
class TAxis;

class AliCorrectionManagerBase : public TObject
{
public:
  enum EConstants { 
    kIgnoreValue = 0,
    kIgnoreField = 999 // Must be synced with AliOADBForward::kInvalidField
  };
  enum EFields {
    kRun       = 0x01, 
    kSys       = 0x02, 
    kSNN       = 0x04, 
    kField     = 0x08, 
    kMC        = 0x10, 
    kSatellite = 0x20,
    kStandard  = kRun|kSys|kSNN|kField,
    kFull      = kStandard|kMC|kSatellite
  };
  virtual ~AliCorrectionManagerBase();
  virtual void SetPrefix(const TString& prefix);
  virtual void SetEnableFallBack(Bool_t use=true) { fFallBack = use; }

  virtual Bool_t Store(TObject*     o,
           ULong_t     runNo,
           UShort_t    sys, 
           UShort_t    sNN, 
           Short_t     field, 
           Bool_t      mc,
           Bool_t      sat, 
           const char* file,
           const char* meth="NEAR") const;
  virtual Bool_t Append(const TString& addition,
      const TString& destination="") const;
  virtual Bool_t CleanUp(const TString& destination,
       Bool_t         verb=false,
       Bool_t         all=false) const;
  /* @} */

  virtual void EnableCorrections(UInt_t what);
  virtual Bool_t CheckCorrections(UInt_t what, Bool_t verbose=true) const;
  Bool_t InitCorrections(ULong_t    run, 
       UShort_t   sys, 
       UShort_t   sNN, 
       Short_t    fld, 
       Bool_t     mc, 
       Bool_t     sat,
       Bool_t     force=false);
  /* @} */

  virtual const TAxis* GetVertexAxis() const { return 0; }
  virtual const TAxis* GetEtaAxis() const { return 0; }
  /* @} */

  virtual Bool_t IsInit() const { return fIsInit; }
  virtual void Print(Option_t* option="") const;
  virtual void Browse(TBrowser* b);
  virtual Bool_t IsFolder() const { return true; }
  virtual void SetDebug(Bool_t debug) { fDebug = debug; }
  /* @} */
  struct Correction : public TNamed
  {
    Correction();
    Correction(const TString& tableName, 
         const TString& fileName, 
         TClass*        cls,
         UShort_t       queryFields=kStandard,
         Bool_t         enabled=false);
    Correction(const Correction& o);
    Correction& operator=(const Correction& o);
    ~Correction() { delete fObject; }
    Bool_t ReadIt(AliOADBForward* db,
      ULong_t    run, 
      UShort_t   sys, 
      UShort_t   sNN, 
      Short_t    fld, 
      Bool_t     mc, 
      Bool_t     sat,
      Bool_t     vrb=false,
      Bool_t     fbk=false);
    Bool_t StoreIt(AliOADBForward* db,
       TObject*    o, 
       ULong_t    run, 
       UShort_t   sys, 
       UShort_t   sNN, 
       Short_t    fld, 
       Bool_t     mc, 
       Bool_t     sat,
       const char* file=0,
       const char* meth="NEAR") const;
    void Enable(Bool_t enabled=true) { fEnabled = enabled; }
    TObject* Get();
    const TObject* Get() const;
    void SetFile(const TString& fileName) { fTitle = fileName; }
    const TClass* TheClass() const;
    void Print(Option_t* option="") const;
    void Browse(TBrowser* b);
    Bool_t IsFolder() const { return true; }
    void MassageFields(ULong_t&  run,
           UShort_t& sys,
           UShort_t& sNN, 
           Short_t & fld, 
           Bool_t&   mc, 
           Bool_t&   sat) const;
    void CorrectFields(ULong_t&  run,
           UShort_t& sys,
           UShort_t& sNN, 
           Short_t & fld, 
           Bool_t&   mc, 
           Bool_t&   sat) const;
    Bool_t CleanIt(AliOADBForward* db, 
       const TString& dest, 
       Bool_t verb=false,
       Bool_t all=false) const;
    Bool_t OpenIt(AliOADBForward* db, 
      Bool_t vrb=false, 
      Bool_t fallback=false) const;

    mutable TClass*  fCls;      
    const TString fClientCls; // Class name
    UShort_t fQueryFields;    // Enabled query fields
    Bool_t   fEnabled;   // Whether we're in use 
    TString  fLastEntry; // Text representation of last entry
    TObject* fObject;    // The data 
    ClassDef(Correction,1) // Correction meta object
  };
  const char* GetObjectName(Int_t what) const;

protected:
  AliCorrectionManagerBase();
  AliCorrectionManagerBase(Bool_t notUsed);
  AliCorrectionManagerBase(const AliCorrectionManagerBase& o);
  AliCorrectionManagerBase& operator=(const AliCorrectionManagerBase& o);
  void RegisterCorrection(Int_t id, Correction* corr);
  void RegisterCorrection(Int_t id, 
        const TString& tableName, 
        const TString& fileName, 
        TClass*        cls,
        UShort_t       fields=kStandard,
        Bool_t         enabled=false);  
  void EnableCorrection(Int_t id, Bool_t enable=true);
  Correction* GetCorrection(Int_t id);
  const Correction* GetCorrection(Int_t id) const;
  void SetCorrectionFile(Int_t id, const TString& fileName) const;
  Int_t GetId(const TString& what) const;
  Int_t GetId(const TObject* obj) const;
  TObject* Get(Int_t id);
  const TObject* Get(Int_t id) const;
  Bool_t CheckConditions(ULong_t    run, 
       UShort_t   sys, 
       UShort_t   sNN, 
       Short_t    fld, 
       Bool_t     mc, 
       Bool_t     sat);
  Bool_t ReadCorrections(ULong_t    run, 
       UShort_t   sys, 
       UShort_t   sNN, 
       Short_t    fld, 
       Bool_t     mc, 
       Bool_t     sat);
  Bool_t ReadCorrection(Int_t      id,
      ULong_t    run, 
      UShort_t   sys, 
      UShort_t   sNN, 
      Short_t    fld, 
      Bool_t     mc, 
      Bool_t     sat);
  void SetCorrectionFile(Int_t i, const TString& file);
  
  TObjArray       fCorrections; // List of corrections
  Bool_t          fIsInit;      // Whether we're intialized or not 
  ULong_t         fRun;         // Cached run number
  UShort_t        fSys;         // Cached system (1:pp 2:PbPb 3:pPb)
  UShort_t        fSNN;         // Cached center of mass energy [GeV]
  Short_t         fField;       // Cached L3 magnetic field [kG]
  Bool_t          fMC;          // Cached Simulation flag
  Bool_t          fSatellite;   // Cached satellite interaction flat
  AliOADBForward* fDB;          
  Bool_t          fDebug;       // If true, do verbose queries 
  Bool_t          fFallBack;    // If true, enable fall-back queries 
  ClassDef(AliCorrectionManagerBase,2);
};

#endif