PWG3/hfe/AliHFEcontainer.h

   1 /**************************************************************************
   2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3 *                                                                        *
   4 * Author: The ALICE Off-line Project.                                    *
   5 * Contributors are mentioned in the code where appropriate.              *
   6 *                                                                        *
   7 * Permission to use, copy, modify and distribute this software and its   *
   8 * documentation strictly for non-commercial purposes is hereby granted   *
   9 * without fee, provided that the above copyright notice appears in all   *
  10 * copies and that both the copyright notice and this permission notice   *
  11 * appear in the supporting documentation. The authors make no claims     *
  12 * about the suitability of this software for any purpose. It is          *
  13 * provided "as is" without express or implied warranty.                  *
  14 **************************************************************************/
  15 //
  16 // HFE correction framework container
  17 // Contains many single containers
  18 // Extra fuctionality like appending added
  19 //
  20 #ifndef ALIHFECONTAINER_H
  21 #define ALIHFECONTAINER_H
  22
  23 #ifndef ROOT_TNamed
  24 #include <TNamed.h>
  25 #endif
  26
  27 #ifndef ROOT_TObjArray
  28 #include <TObjArray.h>
  29 #endif
  30
  31 #ifndef ROOT_TArrayD
  32 #include <TArrayD.h>
  33 #endif
  34
  35 class TArrayF;
  36 template <class X>
  37 class THnSparseT;
  38 typedef class THnSparseT<TArrayF> THnSparseF;
  39 class TCollection;
  40 class TList;
  41 class THashList;
  42 class TString;
  43 class AliCFContainer;
  44
  45 class AliHFEcontainer : public TNamed{
  46   public:
  47     AliHFEcontainer();
  48     AliHFEcontainer(const Char_t *name);
  49     AliHFEcontainer(const Char_t *name, UInt_t nVar);
  50     AliHFEcontainer(const AliHFEcontainer &ref);
  51     AliHFEcontainer& operator=(const AliHFEcontainer &ref);
  52     ~AliHFEcontainer();
  53
  54     virtual Long64_t Merge(TCollection *coll);
  55
  56     void CreateContainer(const Char_t *name, const Char_t *title, UInt_t nStep);
  57     void CreateCorrelationMatrix(const Char_t *name, const Char_t *title);
  58     AliCFContainer *GetCFContainer(const Char_t *name) const;
  59     THnSparseF *GetCorrelationMatrix(const Char_t *name) const;
  60     THashList *GetListOfCorrelationMatrices() const { return fCorrelationMatrices; }
  61     void FillCFContainer(const Char_t *name, UInt_t step, Double_t *content, Double_t weight = 1.) const;
  62     void FillCFContainerStepname(const Char_t *name, const Char_t *step, Double_t *content, Double_t weight = 1.) const;
  63     AliCFContainer *MakeMergedCFContainer(const Char_t *name, const Char_t *title, const Char_t *contnames) const;
  64
  65     Int_t GetNumberOfCFContainers() const;
  66     Int_t GetNumberOfEvents() const { return fNEvents; };
  67     void NewEvent() { fNEvents++; };
  68     void SetNumberOfVariables(UInt_t nVar);
  69     inline void SetBinning(UInt_t var, UInt_t nBins, Double_t *content);
  70     void SetVariableName(UInt_t var, const Char_t *varname);
  71     void SetStepTitle(const Char_t *contname, const Char_t *steptitle, UInt_t step);
  72     void MakeLinearBinning(UInt_t var, UInt_t nBins, Double_t begin, Double_t end);
  73     void MakeLogarithmicBinning(UInt_t var, UInt_t nBins, Double_t begin, Double_t end);
  74
  75     virtual void Print(const Option_t * opt = NULL) const;
  76
  77     struct AliHFEvarInfo : public TObject{
  78         AliHFEvarInfo();
  79         AliHFEvarInfo(const Char_t *name);
  80         AliHFEvarInfo(const AliHFEvarInfo &ref);
  81         AliHFEvarInfo &operator=(const AliHFEvarInfo &ref);
  82         ~AliHFEvarInfo();
  83
  84         UInt_t GetNumberOfBins() const { return fBinning->GetSize() ? fBinning->GetSize() - 1 : 0; };
  85         Double_t *GetBinning() const { return fBinning->GetArray(); };
  86         TString *GetVarName() const { return fVarName; };
  87
  88         void SetVarName(const Char_t *name);
  89         void SetBinning(UInt_t nBins, Double_t *binning);
  90       private:
  91         TString *fVarName;  // Variable Name
  92         TArrayD *fBinning;  // Binning
  93
  94         ClassDef(AliHFEcontainer::AliHFEvarInfo, 1)  // Variable information
  95     };
  96
  97   private:
  98     THashList *fContainers;     // TObjArray for Containers
  99     THashList *fCorrelationMatrices; // Container for Correlation Matrices
 100     TObjArray *fVariables;      // Variable Information
 101     UInt_t fNVars;              // Number of Variables
 102     Int_t fNEvents;             // Number of Events
 103
 104     ClassDef(AliHFEcontainer, 1)  // HFE Efficiency Container
 105 };
 106
 107 //__________________________________________________________________
 108 void AliHFEcontainer::SetBinning(UInt_t var, UInt_t nBins, Double_t *content){
 109   //
 110   // Set Binning for a given variable
 111   //
 112   if(var >= fNVars) return;
 113   AliHFEvarInfo *inf = dynamic_cast<AliHFEvarInfo *>(fVariables->UncheckedAt(var));
 114   if(!inf) return;
 115   inf->SetBinning(nBins, content);
 116 }
 117 #endif