New parametrization of the geometry and new geometry interface (Working week effort)

[u/mrichter/AliRoot.git] / MUON / AliMUONGeometryEnvelopeStore.h

diff --git a/MUON/AliMUONGeometryEnvelopeStore.h b/MUON/AliMUONGeometryEnvelopeStore.h
new file mode 100644 (file)
index 0000000..431ee53
--- /dev/null
+++ b/MUON/AliMUONGeometryEnvelopeStore.h
@@ -0,0 +1,147 @@
+// $Id$
+//
+// Class AliMUONGeometryEnvelopeStore
+// -----------------------------
+// Class for definititon of the temporary volume envelopes
+// used in geometry construction
+//
+// Author: Ivana Hrivnacova, IPN Orsay
+
+#ifndef ALI_MUON_GEOMETRY_ENVELOPE_STORE_H
+#define ALI_MUON_GEOMETRY_ENVELOPE_STORE_H
+
+#include <TObject.h>
+#include <TString.h>
+
+class TGeoTranslation;
+class TGeoRotation;
+class TGeoCombiTrans;
+class TObjArray;
+class TArrayI;
+
+class AliMUONChamber;
+class AliMUONGeometryEnvelope;
+class AliMUONGeometryTransformStore;
+
+class AliMUONGeometryEnvelopeStore : public TObject
+{
+  public:
+    AliMUONGeometryEnvelopeStore(AliMUONGeometryTransformStore* transforms);
+    AliMUONGeometryEnvelopeStore();
+    AliMUONGeometryEnvelopeStore(const AliMUONGeometryEnvelopeStore& rhs);
+    virtual ~AliMUONGeometryEnvelopeStore();
+
+    // operators  
+    AliMUONGeometryEnvelopeStore& operator = (const AliMUONGeometryEnvelopeStore& rhs);
+  
+    // methods
+        
+          // adding virtual envelopes  
+          // (not placed in MC geometry, only logical assembly of volumes,
+         //  cannot have more copies)  
+    void  AddEnvelope(const TString& name, Int_t id, 
+                      Bool_t isVirtual, const char* only="ONLY"); 
+    void  AddEnvelope(const TString& name, Int_t id,  
+                      Bool_t isVirtual,
+                      const TGeoTranslation& translation, 
+                     const char* only="ONLY"); 
+    void  AddEnvelope(const TString& name, Int_t id, 
+                      Bool_t isVirtual, 
+                      const TGeoTranslation& translation, 
+                     const TGeoRotation& rotation,
+                     const char* only="ONLY");
+    void  AddEnvelope(const TString& name, Int_t id,  
+                      Bool_t isVirtual,
+                      const TGeoCombiTrans& transform,
+                     const char* only="ONLY");
+                     
+          // adding non-virtual envelopes      
+          // (placed in MC geometry with transformation composed
+         //  of transformation of chamber and their transformation, 
+         //  can have more copies )    
+    void  AddEnvelope(const TString& name, Int_t id, 
+                      Int_t copyNo, const char* only="ONLY"); 
+    void  AddEnvelope(const TString& name, Int_t id, 
+                      Int_t copyNo, 
+                      const TGeoTranslation& translation,
+                     const char* only="ONLY"); 
+    void  AddEnvelope(const TString& name,  Int_t id, 
+                      Int_t copyNo,
+                      const TGeoTranslation& translation, 
+                     const TGeoRotation& rotation,
+                     const char* only="ONLY");
+    void  AddEnvelope(const TString& name,  Int_t id, 
+                      Int_t copyNo, 
+                      const TGeoCombiTrans& transform,
+                     const char* only="ONLY");
+
+          // adding constituents to virtual envelopes  
+          // (placed in MC geometry with transformation composed
+         //  of transformation of chamber, envelope and their own
+         //  transformation )  
+    void  AddEnvelopeConstituent(const TString& name, const TString& envName, 
+                      Int_t copyNo); 
+    void  AddEnvelopeConstituent(const TString& name, const TString& envName, 
+                      Int_t copyNo, const TGeoTranslation& translation); 
+    void  AddEnvelopeConstituent(const TString& name, const TString& envName, 
+                      Int_t copyNo, const TGeoTranslation& translation, 
+                     const TGeoRotation& rotation);
+    void  AddEnvelopeConstituent(const TString& name, const TString& envName, 
+                      Int_t copyNo, const TGeoCombiTrans& transform);
+                                     
+          // adding constituents to virtual envelopes with specified shape
+         // parameters
+          // (placed in MC geometry with transformation composed
+         //  of transformation of chamber, envelope and their own
+         //  transformation )  
+    void  AddEnvelopeConstituentParam(const TString& name, const TString& envName, 
+                      Int_t copyNo, Int_t npar, Double_t* param); 
+    void  AddEnvelopeConstituentParam(const TString& name, const TString& envName, 
+                      Int_t copyNo, const TGeoTranslation& translation,
+                     Int_t npar, Double_t* param); 
+    void  AddEnvelopeConstituentParam(const TString& name, const TString& envName, 
+                      Int_t copyNo, const TGeoTranslation& translation, 
+                     const TGeoRotation& rotation, Int_t npar, Double_t* param);
+    void  AddEnvelopeConstituentParam(const TString& name, const TString& envName, 
+                      Int_t copyNo, const TGeoCombiTrans& transform,
+                     Int_t npar, Double_t* param);
+                                     
+    void  SetDebug(Bool_t debug);
+
+    // Alignement
+    virtual Bool_t  GetAlign() const;
+    virtual void    SetAlign(Bool_t align);
+ 
+    // get methods
+    const TObjArray*  GetEnvelopes() const;
+
+  private:
+    // methods
+    AliMUONGeometryEnvelope* FindEnvelope(const TString& name) const;
+    Bool_t AlignEnvelope(AliMUONGeometryEnvelope* envelope) const;
+ 
+    // data members
+    AliMUONGeometryTransformStore* fDETransforms; // det elements transformations
+    TObjArray*  fEnvelopes; // the envelopes names and transformations
+                           // wrt to the chamber position in mother volume                                 
+    Bool_t      fDebug;     // Switch for debugging  
+    Bool_t      fAlign;     // option to read transformations from a file
+ 
+  ClassDef(AliMUONGeometryEnvelopeStore,1) // MUON envelope store
+};
+
+// inline functions
+
+inline void  AliMUONGeometryEnvelopeStore::SetDebug(Bool_t debug)
+{ fDebug = debug; }
+
+inline Bool_t  AliMUONGeometryEnvelopeStore::GetAlign() const
+{ return fAlign; }
+
+inline void AliMUONGeometryEnvelopeStore::SetAlign(Bool_t align)
+{ fAlign = align; }
+
+inline const TObjArray* AliMUONGeometryEnvelopeStore::GetEnvelopes() const
+{ return fEnvelopes; }
+
+#endif //ALI_MUON_CHAMBER_ENVELOPE_STORE_H