ECCE @ EIC Software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
pyG4ThreeVector.cc
Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file pyG4ThreeVector.cc
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 // pyG4ThreeVector.cc
28 //
29 // 2005 Q
30 // ====================================================================
31 #include <boost/python.hpp>
32 #include "G4ThreeVector.hh"
33 #include "G4RotationMatrix.hh"
34 
35 using namespace boost::python;
36 using namespace CLHEP;
37 
38 typedef G4ThreeVector XXX; // ...
39 
40 // ====================================================================
41 // thin wrappers
42 // ====================================================================
43 namespace pyG4ThreeVector {
44 
45 G4double(XXX::*f1_theta)() const= &XXX::theta;
46 G4double(XXX::*f2_theta)(const XXX&) const = &XXX::theta;
47 
48 G4double(XXX::*f1_cosTheta)() const= &XXX::cosTheta;
49 G4double(XXX::*f2_cosTheta)(const XXX&) const = &XXX::cosTheta;
50 
51 G4double(XXX::*f1_cos2Theta)() const= &XXX::cos2Theta;
52 G4double(XXX::*f2_cos2Theta)(const XXX&) const = &XXX::cos2Theta;
53 
54 G4double(XXX::*f1_perp2)() const= &XXX::perp2;
55 G4double(XXX::*f2_perp2)(const XXX&) const = &XXX::perp2;
56 
57 G4double(XXX::*f1_perp)() const= &XXX::perp;
58 G4double(XXX::*f2_perp)(const XXX&) const = &XXX::perp;
59 
60 G4double(XXX::*f1_angle)() const= &XXX::angle;
61 G4double(XXX::*f2_angle)(const XXX&) const = &XXX::angle;
62 
63 G4double(XXX::*f1_eta)() const= &XXX::eta;
64 G4double(XXX::*f2_eta)(const XXX&) const = &XXX::eta;
65 
66 XXX(XXX::*f1_project)() const= &XXX::project;
67 XXX(XXX::*f2_project)(const XXX&) const = &XXX::project;
68 
69 XXX(XXX::*f1_perpPart)() const= &XXX::perpPart;
70 XXX(XXX::*f2_perpPart)(const XXX&) const = &XXX::perpPart;
71 
72 G4double(XXX::*f1_rapidity)() const= &XXX::rapidity;
73 G4double(XXX::*f2_rapidity)(const XXX&) const = &XXX::rapidity;
74 
75 G4double(XXX::*f1_polarAngle)(const XXX&) const= &XXX::polarAngle;
76 G4double(XXX::*f2_polarAngle)(const XXX&, const XXX&) const = &XXX::polarAngle;
77 
78 G4double(XXX::*f1_azimAngle)(const XXX&) const= &XXX::azimAngle;
79 G4double(XXX::*f2_azimAngle)(const XXX&, const XXX&) const = &XXX::azimAngle;
80 
81 XXX&(XXX::*f1_rotate)(G4double, const XXX&)= &XXX::rotate;
82 XXX&(XXX::*f2_rotate)(const XXX&, G4double)= &XXX::rotate;
83 XXX&(XXX::*f3_rotate)(const HepAxisAngle&)= &XXX::rotate;
84 XXX&(XXX::*f4_rotate)(const HepEulerAngles&)= &XXX::rotate;
85 XXX&(XXX::*f5_rotate)(G4double, G4double, G4double)= &XXX::rotate;
86 
87 BOOST_PYTHON_MEMBER_FUNCTION_OVERLOADS(f_isNear, isNear, 1, 2)
88 BOOST_PYTHON_MEMBER_FUNCTION_OVERLOADS(f_isParallel, isParallel, 1, 2)
89 BOOST_PYTHON_MEMBER_FUNCTION_OVERLOADS(f_isOrthogonal, isOrthogonal, 1, 2)
90 
91 }
92 
93 using namespace pyG4ThreeVector;
94 
95 // ====================================================================
96 // module definition
97 // ====================================================================
98 void export_G4ThreeVector()
99 {
100  class_<G4ThreeVector>("G4ThreeVector", "general 3-vector")
101  // constructors
102  .def(init<G4double>())
103  .def(init<G4double, G4double>())
104  .def(init<G4double, G4double, G4double>())
105  .def(init<const XXX&>())
106 
107  // property
108  .add_property("x", &XXX::x, &XXX::setX)
109  .add_property("y", &XXX::y, &XXX::setY)
110  .add_property("z", &XXX::z, &XXX::setZ)
111 
112  // methods
113  .def("set", &XXX::set)
114  .def("phi", &XXX::phi)
115  .def("mag", &XXX::mag)
116  .def("mag2", &XXX::mag2)
117  .def("setPhi", &XXX::setPhi)
118  .def("setTheta", &XXX::setTheta)
119  .def("setMag", &XXX::setMag)
120  .def("setPerp", &XXX::setPerp)
121  .def("setCylTheta", &XXX::setCylTheta)
122  .def("howNear", &XXX::howNear)
123  .def("deltaR", &XXX::deltaR)
124  .def("unit", &XXX::unit)
125  .def("orthogonal", &XXX::orthogonal)
126  .def("dot", &XXX::dot)
127  .def("cross", &XXX::cross)
128  .def("pseudoRapidity", &XXX::pseudoRapidity)
129  .def("setEta", &XXX::setEta)
130  .def("setCylEta",&XXX::setCylEta)
131  .def("setRThetaPhi", &XXX::setRThetaPhi)
132  .def("setREtaPhi", &XXX::setREtaPhi)
133  .def("setRhoPhiZ", &XXX::setRhoPhiZ)
134  .def("setRhoPhiEta", &XXX::setRhoPhiEta)
135  .def("getX", &XXX::getX)
136  .def("getY", &XXX::getY)
137  .def("getZ", &XXX::getZ)
138  .def("getR", &XXX::getR)
139  .def("getTheta", &XXX::getTheta)
140  .def("getPhi", &XXX::getPhi)
141  .def("r", &XXX::r)
142  .def("rho", &XXX::rho)
143  .def("getRho", &XXX::getRho)
144  .def("getEta", &XXX::getEta)
145  .def("setR", &XXX::setR)
146  .def("setRho", &XXX::setRho)
147  .def("compare", &XXX::compare)
148  .def("diff2", &XXX::diff2)
149  .def("setTolerance", &XXX::setTolerance)
150  .staticmethod("setTolerance")
151  .def("getTolerance", &XXX::getTolerance)
152  .staticmethod("getTolerance")
153  .def("isNear", &XXX::isNear, f_isNear())
154  .def("isParallel", &XXX::isParallel, f_isParallel())
155  .def("isOrthogonal", &XXX::isOrthogonal, f_isOrthogonal())
156  .def("howParallel", &XXX::howParallel)
157  .def("howOrthogonal", &XXX::howOrthogonal)
158  .def("beta", &XXX::beta)
159  .def("gamma", &XXX::gamma)
160  .def("deltaPhi", &XXX::deltaPhi)
161  .def("coLinearRapidity", &XXX::coLinearRapidity)
162  .def("theta", f1_theta)
163  .def("theta", f2_theta)
164  .def("cosTheta", f1_cosTheta)
165  .def("cosTheta", f2_cosTheta)
166  .def("cos2Theta", f1_cos2Theta)
167  .def("cos2Theta", f2_cos2Theta)
168  .def("perp2", f1_perp2)
169  .def("perp2", f2_perp2)
170  .def("angle", f1_angle)
171  .def("angle", f2_angle)
172  .def("eta", f1_eta)
173  .def("eta", f2_eta)
174  .def("project", f1_project)
175  .def("project", f2_project)
176  .def("perpPart", f1_perpPart)
177  .def("perpPart", f2_perpPart)
178  .def("rapidity", f1_rapidity)
179  .def("rapidity", f2_rapidity)
180  .def("polarAngle",f1_polarAngle)
181  .def("polarAngle",f2_polarAngle)
182  .def("azimAngle", f1_azimAngle)
183  .def("azimAngle", f2_azimAngle)
184  .def("rotateX", &XXX::rotateX,
185  return_value_policy<reference_existing_object>())
186  .def("rotateY", &XXX::rotateY,
187  return_value_policy<reference_existing_object>())
188  .def("rotateZ", &XXX::rotateZ,
189  return_value_policy<reference_existing_object>())
190  .def("rotateUz", &XXX::rotateUz,
191  return_value_policy<reference_existing_object>())
192  .def("transform",&XXX::transform,
193  return_value_policy<reference_existing_object>())
194  .def("rotate", f1_rotate,
195  return_value_policy<reference_existing_object>())
196  .def("rotate", f2_rotate,
197  return_value_policy<reference_existing_object>())
198  .def("rotate", f5_rotate,
199  return_value_policy<reference_existing_object>())
200 
201  // operators
202  .def(self_ns::str(self))
203  .def(self == self)
204  .def(self != self)
205  .def(self += self)
206  .def(self -= self)
207  .def(self - self)
208  .def(self + self)
209  .def(self * self)
210  .def(self * G4double())
211  .def(self / G4double())
212  .def(G4double() * self)
213  .def(self *= G4double())
214  .def(self /= G4double())
215  .def(self > self)
216  .def(self < self)
217  .def(self >= self)
218  .def(self <= self)
219  ;
220 }
221