Forgotten commit

[u/mrichter/AliRoot.git] / TPC / AliTPCComposedCorrection.h

diff --git a/TPC/AliTPCComposedCorrection.h b/TPC/AliTPCComposedCorrection.h
index c4e6fe9eebb2dc484726f22484ae189bfd904b52..e5b7776e5fbe00387992b14b13af6b7da85c07b0 100644 (file)
--- a/TPC/AliTPCComposedCorrection.h
+++ b/TPC/AliTPCComposedCorrection.h
@@ -27,8 +27,10 @@
 ////////////////////////////////////////////////////////////////////////////////
 
 #include "AliTPCCorrection.h"
+#include "TVectorD.h"
 
 class TCollection;
+class TTimeStamp;
 
 class AliTPCComposedCorrection : public AliTPCCorrection {
 public:
@@ -41,7 +43,7 @@ public:
   void SetOmegaTauT1T2(Float_t omegaTau,Float_t t1,Float_t t2);
 
   TCollection* GetCorrections() const {return fCorrections;}
-  void SetCorrections(TCollection *corrections) {fCorrections=corrections;}
+  void SetCorrections(const TCollection *corrections) {fCorrections=(TCollection*)corrections;}
   CompositionType GetMode() const {return fMode;}
   void SetMode(CompositionType mode) {fMode=mode;}
 
@@ -53,16 +55,17 @@ public:
   // initialization and update functions
   virtual void Init();
   virtual void Update(const TTimeStamp &timeStamp);
-
+  void SetWeights(TVectorD * weights){fWeights= (TVectorD*) weights->Clone();}
+  const  TVectorD * GetWeights() const {return fWeights;}
 
 private:
   TCollection *fCorrections; // The corrections this one is composed of.
   CompositionType fMode;     // The way to apply the corrections (see general class documentation)
-
+  TVectorD        *fWeights;  // optional vector with weights - used for fit benchmarking
   AliTPCComposedCorrection & operator = (const AliTPCComposedCorrection);
   AliTPCComposedCorrection(const AliTPCComposedCorrection&); //dummy copy contructor
 
-  ClassDef(AliTPCComposedCorrection,1);
+  ClassDef(AliTPCComposedCorrection,2);
 };
 
 #endif