174 lines
6.0 KiB
C++
174 lines
6.0 KiB
C++
#ifndef COORDS_H
|
|
#define COORDS_H
|
|
namespace AHFinderDirect
|
|
{
|
|
namespace local_coords
|
|
{
|
|
|
|
// compare if two angles are fuzzily equal mod 2*pi radians (360 degrees)
|
|
bool fuzzy_EQ_ang(fp ang1, fp ang2); // radians
|
|
bool fuzzy_EQ_dang(fp dang1, fp dang2); // degrees
|
|
|
|
// modulo-reduce {ang,dang} to be (fuzzily) within the range
|
|
// [min,max]_{ang,dang}, or error_exit() if no such value exists
|
|
fp modulo_reduce_ang(fp ang, fp min_ang, fp max_ang);
|
|
fp modulo_reduce_dang(fp dang, fp min_dang, fp max_dang);
|
|
|
|
} // close namespace local_coords::
|
|
|
|
namespace local_coords
|
|
{
|
|
// (r,(mu,nu,phi)) <--> (x,y,z)
|
|
void xyz_of_r_mu_nu(fp r, fp mu, fp nu, fp &x, fp &y, fp &z);
|
|
void xyz_of_r_mu_phi(fp r, fp mu, fp phi, fp &x, fp &y, fp &z);
|
|
void xyz_of_r_nu_phi(fp r, fp nu, fp phi, fp &x, fp &y, fp &z);
|
|
fp r_of_xyz(fp x, fp y, fp z);
|
|
fp mu_of_yz(fp y, fp z);
|
|
fp nu_of_xz(fp x, fp z);
|
|
fp phi_of_xy(fp x, fp y);
|
|
|
|
// ((mu,nu,phi)) --> the 3rd
|
|
fp phi_of_mu_nu(fp mu, fp nu);
|
|
fp nu_of_mu_phi(fp mu, fp phi);
|
|
fp mu_of_nu_phi(fp nu, fp phi);
|
|
|
|
// partial {x,y,z} / partial {mu,nu,phi}
|
|
void partial_xyz_wrt_r_mu_nu(fp r, fp mu, fp nu,
|
|
fp &partial_x_wrt_r, fp &partial_x_wrt_mu, fp &partial_x_wrt_nu,
|
|
fp &partial_y_wrt_r, fp &partial_y_wrt_mu, fp &partial_y_wrt_nu,
|
|
fp &partial_z_wrt_r, fp &partial_z_wrt_mu, fp &partial_z_wrt_nu);
|
|
void partial_xyz_wrt_r_mu_phi(fp r, fp mu, fp phi,
|
|
fp &partial_x_wrt_r, fp &partial_x_wrt_mu, fp &partial_x_wrt_phi,
|
|
fp &partial_y_wrt_r, fp &partial_y_wrt_mu, fp &partial_y_wrt_phi,
|
|
fp &partial_z_wrt_r, fp &partial_z_wrt_mu, fp &partial_z_wrt_phi);
|
|
void partial_xyz_wrt_r_nu_phi(fp r, fp nu, fp phi,
|
|
fp &partial_x_wrt_r, fp &partial_x_wrt_nu, fp &partial_x_wrt_phi,
|
|
fp &partial_y_wrt_r, fp &partial_y_wrt_nu, fp &partial_y_wrt_phi,
|
|
fp &partial_z_wrt_r, fp &partial_z_wrt_nu, fp &partial_z_wrt_phi);
|
|
|
|
// partial {mu,nu,phi} / partial {x,y,z}
|
|
fp partial_mu_wrt_y(fp y, fp z);
|
|
fp partial_mu_wrt_z(fp y, fp z);
|
|
fp partial_nu_wrt_x(fp x, fp z);
|
|
fp partial_nu_wrt_z(fp x, fp z);
|
|
fp partial_phi_wrt_x(fp x, fp y);
|
|
fp partial_phi_wrt_y(fp x, fp y);
|
|
|
|
// partial^2 {mu,nu,phi} / partial {x,y,z}{x,y,z}
|
|
fp partial2_mu_wrt_yy(fp y, fp z);
|
|
fp partial2_mu_wrt_yz(fp y, fp z);
|
|
fp partial2_mu_wrt_zz(fp y, fp z);
|
|
fp partial2_nu_wrt_xx(fp x, fp z);
|
|
fp partial2_nu_wrt_xz(fp x, fp z);
|
|
fp partial2_nu_wrt_zz(fp x, fp z);
|
|
fp partial2_phi_wrt_xx(fp x, fp y);
|
|
fp partial2_phi_wrt_xy(fp x, fp y);
|
|
fp partial2_phi_wrt_yy(fp x, fp y);
|
|
|
|
// usual polar spherical (r,theta,phi) <--> (x,y,z)
|
|
void xyz_of_r_theta_phi(fp r, fp theta, fp phi, fp &x, fp &y, fp &z);
|
|
void r_theta_phi_of_xyz(fp x, fp y, fp z, fp &r, fp &theta, fp &phi);
|
|
// ... already have r_of_xyz()
|
|
// ... already have phi_of_xy()
|
|
fp theta_of_xyz(fp x, fp y, fp z);
|
|
|
|
// ((mu,nu,phi)) <--> usual polar spherical (theta,phi)
|
|
// ... note phi is the same coordinate in both systems
|
|
void theta_phi_of_mu_nu(fp mu, fp nu, fp &ps_theta, fp &ps_phi);
|
|
void theta_phi_of_mu_phi(fp mu, fp phi, fp &ps_theta, fp &ps_phi);
|
|
void theta_phi_of_nu_phi(fp nu, fp phi, fp &ps_theta, fp &ps_phi);
|
|
void mu_nu_of_theta_phi(fp ps_theta, fp ps_phi, fp &mu, fp &nu);
|
|
void mu_phi_of_theta_phi(fp ps_theta, fp ps_phi, fp &mu, fp &phi);
|
|
void nu_phi_of_theta_phi(fp ps_theta, fp ps_phi, fp &nu, fp &phi);
|
|
|
|
// ((mu,nu,phi)) --> direction cosines (xcos,ycos,zcos)
|
|
void xyzcos_of_mu_nu(fp mu, fp nu, fp &xcos, fp &ycos, fp &zcos);
|
|
void xyzcos_of_mu_phi(fp mu, fp phi, fp &xcos, fp &ycos, fp &zcos);
|
|
void xyzcos_of_nu_phi(fp nu, fp phi, fp &xcos, fp &ycos, fp &zcos);
|
|
} // close namespace local_coords::
|
|
|
|
//*****************************************************************************
|
|
|
|
//
|
|
// ***** bit masks for coordinates ****
|
|
//
|
|
|
|
//
|
|
// We need to manipulate coordinates to do calculations like "which
|
|
// coordinate do these two patches have in common". We do these by
|
|
// Boolean operations on integers using the following bit masks:
|
|
//
|
|
|
|
namespace local_coords
|
|
{
|
|
|
|
typedef int coords_set;
|
|
|
|
enum
|
|
{
|
|
coords_set_mu = 0x1,
|
|
coords_set_nu = 0x2,
|
|
coords_set_phi = 0x4,
|
|
|
|
coords_set_empty = 0x0,
|
|
coords_set_all = coords_set_mu | coords_set_nu | coords_set_phi // no comma
|
|
};
|
|
|
|
// human-readable coordinate names for debugging etc
|
|
const char *name_of_coords_set(coords_set S);
|
|
|
|
// set complement of coordinates
|
|
inline coords_set coords_set_not(coords_set S)
|
|
{
|
|
return coords_set_all & ~S;
|
|
}
|
|
|
|
} // close namespace local_coords::
|
|
|
|
//******************************************************************************
|
|
|
|
//
|
|
// This class stores the origin point of our local coordinates, and
|
|
// provides conversions between local and global coordinates.
|
|
//
|
|
class global_coords
|
|
{
|
|
public:
|
|
// get global (x,y,z) coordinates of local origin point
|
|
fp origin_x() const { return origin_x_; }
|
|
fp origin_y() const { return origin_y_; }
|
|
fp origin_z() const { return origin_z_; }
|
|
|
|
// constructor: specify global (x,y,z) coordinates of local origin point
|
|
global_coords(fp origin_x_in, fp origin_y_in, fp origin_z_in)
|
|
: origin_x_(origin_x_in),
|
|
origin_y_(origin_y_in),
|
|
origin_z_(origin_z_in)
|
|
{
|
|
}
|
|
// destructor: compiler-generated no-op is ok
|
|
|
|
void recentering(fp x, fp y, fp z)
|
|
{
|
|
origin_x_ = x;
|
|
origin_y_ = y;
|
|
origin_z_ = z;
|
|
}
|
|
|
|
private:
|
|
// we forbid copying and passing by value
|
|
// by declaring the copy constructor and assignment operator
|
|
// private, but never defining them
|
|
global_coords(const global_coords &rhs);
|
|
global_coords &operator=(const global_coords &rhs);
|
|
|
|
private:
|
|
// global (x,y,z) coordinates of local origin point
|
|
fp origin_x_, origin_y_, origin_z_;
|
|
};
|
|
|
|
//******************************************************************************
|
|
|
|
} // namespace AHFinderDirect
|
|
#endif /* COORDS_H */
|