New geometry v5

[u/mrichter/AliRoot.git] / TRD / AliTRDdigitizer.h

diff --git a/TRD/AliTRDdigitizer.h b/TRD/AliTRDdigitizer.h
index d6c589cf42d80f22a96a58f602996224a25dd010..bccbc3f2d3c5c3d002426e0c369f79f82158a70f 100644 (file)
--- a/TRD/AliTRDdigitizer.h
+++ b/TRD/AliTRDdigitizer.h
@@ -11,6 +11,7 @@ class TFile;
 class TF1;
 
 class AliRunDigitizer;
+class AliRunLoader;
 
 class AliTRD;
 class AliTRDdigitsManager;
@@ -37,7 +38,7 @@ class AliTRDdigitizer : public AliDigitizer {
   virtual Bool_t       InitDetector();
   virtual void         Exec(Option_t* option = 0);  
   virtual Bool_t       Open(const Char_t *file, Int_t nEvent = 0);
-  virtual Bool_t       MakeBranch(const Char_t *file = 0) const;
+  virtual Bool_t       MakeBranch(TTree* tree) const;
   virtual Bool_t       MakeDigits();
   virtual void         AddSDigitsManager(AliTRDdigitsManager *manager);
   virtual void         DeleteSDigitsManager();
@@ -46,6 +47,8 @@ class AliTRDdigitizer : public AliDigitizer {
   virtual Bool_t       SDigits2Digits();
   virtual Bool_t       WriteDigits() const;
 
+  void         InitOutput(Int_t iEvent);
+ 
   virtual void         SetCompress(Int_t c = 1)             { fCompress        = c;   };
   virtual void         SetDebug(Int_t v = 1)                { fDebug           = v;   };
   virtual void         SetSDigits(Int_t v = 1)              { fSDigits         = v;   };
@@ -55,6 +58,9 @@ class AliTRDdigitizer : public AliDigitizer {
   virtual void         SetGeometry(AliTRDgeometry *geo)     { fGeo             = geo; };
   virtual void         SetParameter(AliTRDparameter *par)   { fPar             = par; };
   virtual void         SetMergeSignalOnly(Bool_t m = kTRUE) { fMergeSignalOnly = m;   };
+  virtual void         SetSimple(Int_t v = 1)               { fSimpleSim       = v;
+                                                              fSimpleDet       = 12;
+                                                              fCompress        = kFALSE; };
 
   AliTRDdigitsManager *Digits()                       const { return fDigitsManager; };
 
@@ -62,10 +68,12 @@ class AliTRDdigitizer : public AliDigitizer {
           Bool_t       GetSDigits()                   const { return fSDigits;       };
           Float_t      GetSDigitsScale()              const { return fSDigitsScale;  };
   AliTRDparameter     *GetParameter()                 const { return fPar;           };
+          Bool_t       GetSimple()                    const { return fSimpleSim;     };
 
  protected:
 
-  TFile               *fInputFile;          //! ALIROOT-file
+  //TFile               *fInputFile;          //! ALIROOT-file
+  AliRunLoader        *fRunLoader;          //! Local pointer
   AliTRDdigitsManager *fDigitsManager;      //! Manager for the output digits
   AliTRDdigitsManager *fSDigitsManager;     //! Manager for the summed input s-digits
   TList               *fSDigitsManagerList; //! List of managers of input s-digits
@@ -79,7 +87,9 @@ class AliTRDdigitizer : public AliDigitizer {
   Bool_t               fSDigits;            //  Switch for the summable digits
   Float_t              fSDigitsScale;       //  Scale factor for the summable digits 
   Bool_t               fMergeSignalOnly;    //  Merge only detectors that contain a signal
-
+  Bool_t               fSimpleSim;          //  Switch for the simplified simulation
+  Int_t                fSimpleDet;          //  Detecttor number used in the simplified simulation
+ 
  private:
 
   virtual void         DeConvExp(Double_t *source, Double_t *target, Int_t n, Int_t nexp);