Gabriele Cosmo
131 lines
3.1 KiB
Plaintext
131 lines
3.1 KiB
Plaintext
// Copyright (C) 2010, Guy Barrand. All rights reserved.
|
|
// See the file tools.license for terms.
|
|
|
|
#ifndef tools_vec3d
|
|
#define tools_vec3d
|
|
|
|
#include "vec3"
|
|
#include "../S_STRING"
|
|
#include <cmath>
|
|
|
|
namespace tools {
|
|
|
|
class vec3d : public vec3<double> {
|
|
typedef vec3<double> parent;
|
|
public:
|
|
TOOLS_SCLASS(tools::vec3d) //for stype()
|
|
public:
|
|
vec3d():parent(){}
|
|
vec3d(const double a_vec[3]):parent(a_vec){}
|
|
vec3d(double a0,double a1,double a2):parent(a0,a1,a2){}
|
|
virtual ~vec3d() {}
|
|
public:
|
|
vec3d(const vec3d& a_from):parent(a_from){}
|
|
vec3d& operator=(const vec3d& a_from){
|
|
parent::operator=(a_from);
|
|
return *this;
|
|
}
|
|
|
|
vec3d(const parent& a_from):parent(a_from){}
|
|
|
|
public: //operators
|
|
vec3d operator*(double a_v) const {
|
|
return vec3d(m_data[0]*a_v,
|
|
m_data[1]*a_v,
|
|
m_data[2]*a_v);
|
|
}
|
|
vec3d operator+(const vec3d& a_v) const {
|
|
return vec3d(m_data[0]+a_v.m_data[0],
|
|
m_data[1]+a_v.m_data[1],
|
|
m_data[2]+a_v.m_data[2]);
|
|
}
|
|
vec3d operator-(const vec3d& a_v) const {
|
|
return vec3d(m_data[0]-a_v.m_data[0],
|
|
m_data[1]-a_v.m_data[1],
|
|
m_data[2]-a_v.m_data[2]);
|
|
}
|
|
vec3d& operator+=(const vec3d& a_v) {
|
|
m_data[0] += a_v.m_data[0];
|
|
m_data[1] += a_v.m_data[1];
|
|
m_data[2] += a_v.m_data[2];
|
|
return *this;
|
|
}
|
|
vec3d& operator-=(const vec3d& a_v) {
|
|
m_data[0] -= a_v.m_data[0];
|
|
m_data[1] -= a_v.m_data[1];
|
|
m_data[2] -= a_v.m_data[2];
|
|
return *this;
|
|
}
|
|
vec3d& operator*=(double a_v) {
|
|
m_data[0] *= a_v;
|
|
m_data[1] *= a_v;
|
|
m_data[2] *= a_v;
|
|
return *this;
|
|
}
|
|
vec3d operator-() const {
|
|
return vec3d(-m_data[0],-m_data[1],-m_data[2]);
|
|
}
|
|
public:
|
|
double length() const {return parent::length(::sqrt);}
|
|
double normalize() {return parent::normalize(::sqrt);}
|
|
public: //for g4dum/G4ThreeVector :
|
|
double mag2() const {return m_data[0]*m_data[0]+m_data[1]*m_data[1]+m_data[2]*m_data[2];}
|
|
void set(double a_0,double a_1,double a_2) {
|
|
m_data[0] = a_0;
|
|
m_data[1] = a_1;
|
|
m_data[2] = a_2;
|
|
}
|
|
void rotateUz(const vec3d& a_new_uz_vector) { //from CLHEP/ThreeVector.cc.
|
|
// a_new_uz_vector must be normalized !
|
|
double& dx = m_data[0];
|
|
double& dy = m_data[1];
|
|
double& dz = m_data[2];
|
|
double u1 = a_new_uz_vector.x();
|
|
double u2 = a_new_uz_vector.y();
|
|
double u3 = a_new_uz_vector.z();
|
|
double up = u1*u1 + u2*u2;
|
|
if (up>0) {
|
|
up = std::sqrt(up);
|
|
double px = dx, py = dy, pz = dz;
|
|
dx = (u1*u3*px - u2*py)/up + u1*pz;
|
|
dy = (u2*u3*px + u1*py)/up + u2*pz;
|
|
dz = -up*px + u3*pz;
|
|
} else if (u3 < 0.) { // phi=0 teta=pi
|
|
dx = -dx; dz = -dz;
|
|
} else {
|
|
}
|
|
//return *this;
|
|
}
|
|
|
|
|
|
private:static void check_instantiation() {vec3d v(0,0,0);v.set_value(1,1,1);}
|
|
};
|
|
|
|
inline vec3d operator*(double a_f,const vec3d& a_v) {
|
|
vec3d res(a_v);
|
|
res *= a_f;
|
|
return res;
|
|
}
|
|
|
|
}
|
|
|
|
|
|
#include <vector>
|
|
|
|
namespace tools {
|
|
|
|
#ifndef SWIG
|
|
//for sf, mf :
|
|
inline bool set_from_vec(vec3d& a_v,const std::vector<double>& a_sv) {
|
|
if(a_sv.size()!=3) return false;
|
|
a_v[0] = a_sv[0];
|
|
a_v[1] = a_sv[1];
|
|
a_v[2] = a_sv[2];
|
|
return true;
|
|
}
|
|
#endif
|
|
|
|
}
|
|
|
|
#endif
|