1 | #ifndef G4Lab_DigitsCollectionAccessor_h |
---|
2 | #define G4Lab_DigitsCollectionAccessor_h |
---|
3 | |
---|
4 | // Inheritance : |
---|
5 | #include <OnX/Inventor/InventorAccessor.h> |
---|
6 | |
---|
7 | #include <G4Transform3D.hh> |
---|
8 | |
---|
9 | class G4VDigiCollection; |
---|
10 | class G4VDigi; |
---|
11 | class G4LogicalVolume; |
---|
12 | class G4Colour; |
---|
13 | |
---|
14 | namespace Slash {namespace Core {class ISession;}} |
---|
15 | namespace Slash {namespace Core {class IWriter;}} |
---|
16 | |
---|
17 | #include <map> |
---|
18 | #include <G4AttDef.hh> |
---|
19 | #include <G4String.hh> |
---|
20 | |
---|
21 | namespace G4Lab { |
---|
22 | |
---|
23 | class DigitsCollectionBaseAccessor : public OnX::InventorAccessor { |
---|
24 | public: //Slash::Data::IAccessor |
---|
25 | virtual std::string name() const; |
---|
26 | virtual Slash::Core::IValue* findValue(Slash::Data::IAccessor::Data,const std::string&,void*); |
---|
27 | public: //Slash::Data::IVisualizer |
---|
28 | virtual void beginVisualize(); |
---|
29 | virtual void visualize(Slash::Data::IAccessor::Data,void*); |
---|
30 | public: |
---|
31 | DigitsCollectionBaseAccessor(Slash::Core::ISession&,const std::string& = ""); |
---|
32 | virtual ~DigitsCollectionBaseAccessor(); |
---|
33 | const std::string& DCName() const; |
---|
34 | protected: |
---|
35 | static G4VDigiCollection* getCollection(Slash::Core::IWriter&,const std::string&); |
---|
36 | static G4LogicalVolume* digiLogicalVolume(const G4VDigi&); |
---|
37 | static G4Transform3D* digiTransform3D(const G4VDigi&); |
---|
38 | static bool digiColor(const G4VDigi&,G4Colour&); |
---|
39 | void setProperties(G4VDigi&); |
---|
40 | protected: |
---|
41 | std::string fType; |
---|
42 | std::map<G4String,G4AttDef>* fAttDefs; |
---|
43 | std::string fDC; |
---|
44 | }; |
---|
45 | |
---|
46 | } |
---|
47 | |
---|
48 | #include <G4VDigi.hh> |
---|
49 | #include <G4TDigiCollection.hh> |
---|
50 | #include <Slash/Data/IIterator.h> |
---|
51 | |
---|
52 | namespace G4Lab { |
---|
53 | |
---|
54 | template <class T> |
---|
55 | class DigitsCollectionAccessor : public DigitsCollectionBaseAccessor { |
---|
56 | public: //Slash::Data::IAccessor |
---|
57 | virtual Slash::Data::IIterator* iterator() { |
---|
58 | const std::vector<std::string>& args = iteratorArguments(); |
---|
59 | if(args.size()) fDC = args[0]; |
---|
60 | clearProperties(); |
---|
61 | fAttDefs = 0; |
---|
62 | G4VDigiCollection* vdc = getCollection(printer(),fDC); |
---|
63 | if(!vdc) return 0; |
---|
64 | typedef G4TDigiCollection<T> Collection; |
---|
65 | Collection* dc = dynamic_cast<Collection*>(vdc); |
---|
66 | if(!dc) return 0; |
---|
67 | if(dc->entries()) { |
---|
68 | G4VDigi* digi = (*dc)[0]; |
---|
69 | if(digi) setProperties(*digi); |
---|
70 | } |
---|
71 | class Iterator : public virtual Slash::Data::IIterator { |
---|
72 | public: //Slash::Data::IIterator |
---|
73 | virtual Slash::Data::IAccessor::Data object() { |
---|
74 | if(fCurrent>=fVector.entries()) return 0; |
---|
75 | G4VDigi* digi = fVector[fCurrent]; |
---|
76 | return digi; |
---|
77 | } |
---|
78 | virtual void next() { fCurrent++;} |
---|
79 | virtual void* tag() { return 0;} |
---|
80 | public: |
---|
81 | Iterator(Collection& aVector):fVector(aVector),fCurrent(0) {} |
---|
82 | virtual ~Iterator() {} |
---|
83 | private: |
---|
84 | Collection& fVector; |
---|
85 | int fCurrent; |
---|
86 | }; |
---|
87 | return new Iterator(*dc); |
---|
88 | } |
---|
89 | public: |
---|
90 | DigitsCollectionAccessor(Slash::Core::ISession& aSession,const std::string& aName = "") |
---|
91 | :DigitsCollectionBaseAccessor(aSession,aName){} |
---|
92 | virtual ~DigitsCollectionAccessor(){} |
---|
93 | }; |
---|
94 | |
---|
95 | |
---|
96 | } |
---|
97 | |
---|
98 | #endif |
---|