Set default PF pattern when not set properly in OCDB

[u/mrichter/AliRoot.git] / EMCAL / AliEMCALSurvey.h

diff --git a/EMCAL/AliEMCALSurvey.h b/EMCAL/AliEMCALSurvey.h
index 9356a3140fe40ef67a88ed020676ec491b953660..41619d9c15d74390d2a58146f3b4a12f5699572a 100644 (file)
--- a/EMCAL/AliEMCALSurvey.h
+++ b/EMCAL/AliEMCALSurvey.h
@@ -22,8 +22,13 @@ class AliEMCALGeometry;
 
 class AliEMCALSurvey : public TObject {
 public:
+
+  enum SurveyDataType_t { kSurvey = 0, //use real survey parameters
+                         kDummy = 1 //use dummy values for testing
+  };
+
   AliEMCALSurvey();
-  AliEMCALSurvey(const TString &txtFileName);
+  AliEMCALSurvey(const TString &txtFileName, const SurveyDataType_t dataType=kSurvey);
 
   virtual ~AliEMCALSurvey();
 
@@ -32,6 +37,9 @@ public:
   //Create AliAlignObjParams with null shifts and rotations.
   void CreateNullObjects(TClonesArray &alObj, const AliEMCALGeometry *geom)const;
 
+  void  SetDataType(const SurveyDataType_t dataType) { fDataType = dataType; }
+  Int_t GetDataType() const { return (Int_t)fDataType; }
+
 protected:
 
   struct AliEMCALSuperModuleDelta {
@@ -46,7 +54,9 @@ protected:
   Int_t          fNSuperModule; // Number of supermodules.
   AliEMCALSuperModuleDelta *fSuperModuleData; // Supermodule transformation data
 
-  void InitSuperModuleData(const Double_t *xReal, const Double_t *yReal, const Double_t *zReal);
+  void InitSuperModuleData(const Double_t *xReal, const Double_t *yReal, const Double_t *zReal, 
+                          const Double_t *psiReal, const Double_t *thetaReal, const Double_t *phiReal);
+  void InitSuperModuleData(const TObjArray* surveypoints);
 
 private:
   //Calculate shifts and rotations for supermodule.
@@ -55,7 +65,9 @@ private:
   AliEMCALSurvey(const AliEMCALSurvey &);
   AliEMCALSurvey &operator = (const AliEMCALSurvey &);
 
-  ClassDef(AliEMCALSurvey, 1) //Survey data reader
+  Int_t  fDataType; //! which date type (survey or dummy) to use
+
+  ClassDef(AliEMCALSurvey, 2) //Survey data reader
 };
 
 #endif