1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
16 // HFE correction framework container
17 // Contains many single containers
18 // Extra fuctionality like appending added
20 #ifndef ALIHFECONTAINER_H
21 #define ALIHFECONTAINER_H
27 #ifndef ROOT_TObjArray
28 #include <TObjArray.h>
38 typedef class THnSparseT<TArrayF> THnSparseF;
45 class AliHFEcontainer : public TNamed{
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);
54 virtual Long64_t Merge(TCollection *coll);
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, const Double_t * const content, Double_t weight = 1.) const;
62 void FillCFContainerStepname(const Char_t *name, const Char_t *step, const Double_t *const content, Double_t weight = 1.) const;
63 AliCFContainer *MakeMergedCFContainer(const Char_t *name, const Char_t *title, const Char_t *contnames) const;
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 void MakeUserDefinedBinning(UInt_t var, UInt_t nBins, const Double_t *binning);
75 void Sumw2(const char *contname) const;
77 virtual void Print(const Option_t * opt = NULL) const;
79 struct AliHFEvarInfo : public TObject{
81 AliHFEvarInfo(const Char_t *name);
82 AliHFEvarInfo(const AliHFEvarInfo &ref);
83 AliHFEvarInfo &operator=(const AliHFEvarInfo &ref);
86 UInt_t GetNumberOfBins() const { return fBinning->GetSize() ? fBinning->GetSize() - 1 : 0; };
87 const Double_t *GetBinning() const { return fBinning->GetArray(); };
88 TString *GetVarName() const { return fVarName; };
90 void SetVarName(const Char_t *name);
91 void SetBinning(UInt_t nBins, const Double_t *binning);
93 TString *fVarName; // Variable Name
94 TArrayD *fBinning; // Binning
96 ClassDef(AliHFEcontainer::AliHFEvarInfo, 1) // Variable information
100 THashList *fContainers; // TObjArray for Containers
101 THashList *fCorrelationMatrices; // Container for Correlation Matrices
102 TObjArray *fVariables; // Variable Information
103 UInt_t fNVars; // Number of Variables
104 Int_t fNEvents; // Number of Events
106 ClassDef(AliHFEcontainer, 1) // HFE Efficiency Container
109 //__________________________________________________________________
110 void AliHFEcontainer::SetBinning(UInt_t var, UInt_t nBins, Double_t *content){
112 // Set Binning for a given variable
114 if(var >= fNVars) return;
115 AliHFEvarInfo *inf = dynamic_cast<AliHFEvarInfo *>(fVariables->UncheckedAt(var));
117 inf->SetBinning(nBins, content);