| Geant4 Cross Reference |
1 //
2 // ********************************************************************
3 // * License and Disclaimer *
4 // * *
5 // * The Geant4 software is copyright of the Copyright Holders of *
6 // * the Geant4 Collaboration. It is provided under the terms and *
7 // * conditions of the Geant4 Software License, included in the file *
8 // * LICENSE and available at http://cern.ch/geant4/license . These *
9 // * include a list of copyright holders. *
10 // * *
11 // * Neither the authors of this software system, nor their employing *
12 // * institutes,nor the agencies providing financial support for this *
13 // * work make any representation or warranty, express or implied, *
14 // * regarding this software system or assume any liability for its *
15 // * use. Please see the license in the file LICENSE and URL above *
16 // * for the full disclaimer and the limitation of liability. *
17 // * *
18 // * This code implementation is the result of the scientific and *
19 // * technical work of the GEANT4 collaboration. *
20 // * By using, copying, modifying or distributing the software (or *
21 // * any work based on the software) you agree to acknowledge its *
22 // * use in resulting scientific publications, and indicate your *
23 // * acceptance of all terms of the Geant4 Software license. *
24 // ********************************************************************
25 //
26 //
27 #ifndef MOLECULAR_OCTREE_NODE_HH
28 #define MOLECULAR_OCTREE_NODE_HH
29
30 #include "G4ThreeVector.hh"
31 #include "globals.hh"
32
33 #include <array>
34 #include <vector>
35
36 class G4VPhysicalVolume;
37
38 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
39
40 class OctreeNode
41 {
42 public:
43 // Uniform divisions constructor along each axis
44 OctreeNode(const G4ThreeVector&, const G4ThreeVector&, G4int, OctreeNode* parent = nullptr);
45
46 ~OctreeNode();
47
48 inline G4bool HasChildren() const { return (fChildren[0] != nullptr); }
49
50 inline OctreeNode* GetParent() const { return fParent; };
51
52 inline const auto& GetHalfLengths() const { return fHalfLengths; };
53
54 inline G4double GetHalfLengthsMag() const { return fHalfLengthsMag; };
55
56 inline const G4ThreeVector& GetPosition() const { return fPosition; };
57
58 inline const auto& GetChildren() const { return fChildren; };
59
60 const std::vector<G4VPhysicalVolume*> SearchOctree(const G4ThreeVector&,
61 G4double _rad = 0) const;
62
63 void SearchOctree(const G4ThreeVector& pos, std::vector<G4VPhysicalVolume*>& out,
64 G4double _rad = 0) const;
65
66 const std::vector<G4VPhysicalVolume*> SearchOctree(const G4ThreeVector&) const;
67
68 G4int GetNumberOfTerminalNodes();
69
70 void AddPhysicalVolume(G4VPhysicalVolume*);
71
72 std::vector<G4VPhysicalVolume*> GetContents() const;
73
74 inline G4int GetMaxContents() const { return fMaxContents; };
75
76 protected:
77 void Split();
78
79 const OctreeNode* GetChildFromPosition(G4ThreeVector const&) const;
80
81 OctreeNode* GetChildFromPosition(G4ThreeVector const& pos);
82
83 private:
84 G4ThreeVector fPosition, fHalfLengths;
85 G4int fMaxContents;
86
87 std::vector<G4VPhysicalVolume*> fContents;
88
89 OctreeNode* fParent;
90 // fChildren is arranged logically to save on queries
91 // The scheme is defined by quadrant as follows:
92 // X Y Z Index | X Y Z Index
93 // + + + 0 | - + + 4
94 // + + - 1 | - + - 5
95 // + - + 2 | - - + 6
96 // + - - 3 | - - - 7
97 std::array<OctreeNode*, 8> fChildren;
98 G4double fHalfLengthsMag;
99 };
100
101 //....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
102
103 #endif // MOLECULAR_OCTREE_NODE_HH
104