PamVMC_update/include/PamVMCDetGeom.h

#ifndef PAMVMC_DET_GEOM_H
#define PAMVMC_DET_GEOM_H
#include <iostream>

#include <TMap.h>
#include <TObject.h>
#include <TString.h>
#include <TObjString.h>
#include <TList.h>

#include <Riostream.h>
#include <TVirtualMC.h>
#include <TGeoManager.h>
#include <TGeoMatrix.h>

#include "PamVMCMat.h"

struct pMotherProp : public TObject{

  TGeoVolume *fmother;
  Int_t fcopyNo;
  TGeoMatrix *fmothermatrix;
  

  pMotherProp():fmother(0),fcopyNo(-1),fmothermatrix(0){};
  pMotherProp(TGeoVolume *mother, Int_t copyNo, TGeoMatrix *mothermatrix):
    fmother(mother),fcopyNo(copyNo),fmothermatrix(mothermatrix){};

};


class PamVMCDetGeom  : public TObject {

 protected:

  enum kind {MAT,MED,VOL};
  TString fdname;
  TMap fdrot; //Map of geometry transformations
  TObjArray fmotherprop; // Vector of mother volume to be passed as node of other mother volumes
  TMap fmat; //Map of Material
  TMap fmix; // Mixtures
  TMap fmed; // Mediums
  TMap fvol;  // Volumes
  TGeoVolume *fmother; // Mother volume of the detector in case of any

 public:


   PamVMCDetGeom(const char *dname="NoDetGeom" ): 
     fdname(dname), fmother(0)
  {
    
  };

  
  virtual ~PamVMCDetGeom() { 
    fdrot.DeleteAll();
    fmotherprop.Clear(); 
    fmat.DeleteAll();
    fmix.DeleteAll();
    fmed.DeleteAll();
  }

 
  TGeoRotation* GetRot(const char* rot){
    return (TGeoRotation*)fdrot(rot);
  }

  Int_t GetMedID(const char *name){ return GetMed(name)->GetId(); }

  TGeoMaterial* GetMat(const char *name){ return (TGeoMaterial*)CheckDef(name,MAT); }
  
  TGeoMedium* GetMed(const char *name){ return (TGeoMedium*)CheckDef(name,MED); }
  
  TGeoVolume* GetVol(const char *name){ return (TGeoVolume*)CheckDef(name,VOL); }

  TObject* CheckDef(const char *name, kind k){
    switch(k){
    case MAT:
      return gGeoManager->GetMaterial(name);
    case MED:
      return gGeoManager->GetMedium(name);
    case VOL:
      return gGeoManager->GetVolume(name);
    default:
      return 0;
    }
  }


  void AddMotherProp(pMotherProp *mp) {SetMotherProp(mp);}

  void AddMotherProp(TObjArray *mp) {fmotherprop.AddAll(mp); }

  void SetMotherProp(pMotherProp *mp) {
    SetMotherProp(mp->fmother,mp->fcopyNo,mp->fmothermatrix);
  }

  void SetMotherProp(TGeoVolume *mv, Int_t cn, TGeoMatrix* mx){  

    fmotherprop.Add(new pMotherProp(mv,cn,mx) );
    }    

  void SetMotherVol(TGeoVolume *mv) { fmother=mv; }

  TObjArray* GetMotherProp(){ return &fmotherprop;}

  void RegisterNodes() { RegisterNodes(&fmotherprop); }

  void RegisterNodes( TObjArray * to ) {
    // Some exception or error printout 
    // should be added to signal that fmother is empty 
    if(fmother) {
      TIterator *p= (TIterator *)(*to).MakeIterator(); 
      pMotherProp *k; 
      while( (k=(pMotherProp *) p->Next())) {
        fmother->AddNode((*k).fmother,(*k).fcopyNo,(*k).fmothermatrix);
      }
      delete p;
    } 
  }

  ClassDef(PamVMCDetGeom,1)
};


#endif //PAMVMC_DET_GEOM_H
1	formato	1.1	#ifndef PAMVMC_DET_GEOM_H
2			#define PAMVMC_DET_GEOM_H
3			#include <iostream>
4
5			#include <TMap.h>
6			#include <TObject.h>
7			#include <TString.h>
8			#include <TObjString.h>
9			#include <TList.h>
10
11			#include <Riostream.h>
12			#include <TVirtualMC.h>
13			#include <TGeoManager.h>
14			#include <TGeoMatrix.h>
15
16			#include "PamVMCMat.h"
17
18			struct pMotherProp : public TObject{
19
20			TGeoVolume *fmother;
21			Int_t fcopyNo;
22			TGeoMatrix *fmothermatrix;
23
24
25			pMotherProp():fmother(0),fcopyNo(-1),fmothermatrix(0){};
26			pMotherProp(TGeoVolume mother, Int_t copyNo, TGeoMatrix mothermatrix):
27			fmother(mother),fcopyNo(copyNo),fmothermatrix(mothermatrix){};
28
29			};
30
31
32			class PamVMCDetGeom : public TObject {
33
34			protected:
35
36			enum kind {MAT,MED,VOL};
37			TString fdname;
38			TMap fdrot; //Map of geometry transformations
39			TObjArray fmotherprop; // Vector of mother volume to be passed as node of other mother volumes
40			TMap fmat; //Map of Material
41			TMap fmix; // Mixtures
42			TMap fmed; // Mediums
43			TMap fvol; // Volumes
44			TGeoVolume *fmother; // Mother volume of the detector in case of any
45
46			public:
47
48
49			PamVMCDetGeom(const char *dname="NoDetGeom" ):
50			fdname(dname), fmother(0)
51			{
52
53			};
54
55
56			virtual ~PamVMCDetGeom() {
57			fdrot.DeleteAll();
58			fmotherprop.Clear();
59			fmat.DeleteAll();
60			fmix.DeleteAll();
61			fmed.DeleteAll();
62			}
63
64
65
66
67			TGeoRotation* GetRot(const char* rot){
68			return (TGeoRotation*)fdrot(rot);
69			}
70
71			Int_t GetMedID(const char *name){ return GetMed(name)->GetId(); }
72
73			TGeoMaterial* GetMat(const char name){ return (TGeoMaterial)CheckDef(name,MAT); }
74
75			TGeoMedium* GetMed(const char name){ return (TGeoMedium)CheckDef(name,MED); }
76
77			TGeoVolume* GetVol(const char name){ return (TGeoVolume)CheckDef(name,VOL); }
78
79			TObject* CheckDef(const char *name, kind k){
80			switch(k){
81			case MAT:
82			return gGeoManager->GetMaterial(name);
83			case MED:
84			return gGeoManager->GetMedium(name);
85			case VOL:
86			return gGeoManager->GetVolume(name);
87			default:
88			return 0;
89			}
90			}
91
92
93			void AddMotherProp(pMotherProp *mp) {SetMotherProp(mp);}
94
95			void AddMotherProp(TObjArray *mp) {fmotherprop.AddAll(mp); }
96
97			void SetMotherProp(pMotherProp *mp) {
98			SetMotherProp(mp->fmother,mp->fcopyNo,mp->fmothermatrix);
99			}
100
101			void SetMotherProp(TGeoVolume mv, Int_t cn, TGeoMatrix mx){
102
103			fmotherprop.Add(new pMotherProp(mv,cn,mx) );
104			}
105
106			void SetMotherVol(TGeoVolume *mv) { fmother=mv; }
107
108			TObjArray* GetMotherProp(){ return &fmotherprop;}
109
110			void RegisterNodes() { RegisterNodes(&fmotherprop); }
111
112			void RegisterNodes( TObjArray * to ) {
113			// Some exception or error printout
114			// should be added to signal that fmother is empty
115			if(fmother) {
116			TIterator p= (TIterator )(*to).MakeIterator();
117			pMotherProp *k;
118			while( (k=(pMotherProp *) p->Next())) {
119			fmother->AddNode((k).fmother,(k).fcopyNo,(*k).fmothermatrix);
120			}
121			delete p;
122			}
123			}
124
125			ClassDef(PamVMCDetGeom,1)
126			};
127
128
129			#endif //PAMVMC_DET_GEOM_H