Eigen transforms (#744)

* Pull in Zack's implementation of inverse distortion

* sqrt fix for mac install

* Use math defines

* replace constexpr with const to fix stuff

* Including cmath so use_math_defines does something

* Move up the math defines because eigen probably include cmath already.

* Starting to port over eigen transforms

* setup.py

* Fixed the function I broke

* Update gvec_to_xy

* Update import

* Update tests/transforms/test_gvec_to_xy_from_file.py

Co-authored-by: Kevin Welsh <kevinwelsh132@gmail.com>

* Update hexrd/transforms/cpp_sublibrary/src/transforms.cpp

Co-authored-by: Kevin Welsh <kevinwelsh132@gmail.com>

* Apply suggestions from code review

Co-authored-by: Kevin Welsh <kevinwelsh132@gmail.com>

* Apply clang-format

* Change the name of the function at definition

* put str here instead of pathlib Path. I think there are some versions of setuptools or perhaps windows that doesn't accept Paths

---------

Co-authored-by: Kevin Lewis <kevindlewis23@gmail.com>
Co-authored-by: Kevin Welsh <kevinwelsh132@gmail.com>
Co-authored-by: Kevin Welsh <kevin.welsh@verdantevolution.com>

Author: 3 people

hexrd/transforms/cpp_sublibrary/src/inverse_distortion.cpp

@@ -1,31 +1,42 @@
-#define _USE_MATH_DEFINES 
-#include <pybind11/pybind11.h>
-#include <pybind11/eigen.h>
+#define _USE_MATH_DEFINES
 #include <Eigen/Core>
-#include <xsimd/xsimd.hpp>
 #include <cmath>
+#include <pybind11/eigen.h>
+#include <pybind11/pybind11.h>
+#include <xsimd/xsimd.hpp>
 
 namespace py = pybind11;
 const double FOUR_THIRDS_PI = M_PI * 4.0 / 3.0;
 const double N_THREE_HALVES_SQRT_3 = -3.0 / 2.0 * std::sqrt(3.0);
 const double TWO_OVER_SQRT_THREE = 2.0 / std::sqrt(3.0);
 
-Eigen::ArrayXXd ge_41rt_inverse_distortion(const Eigen::ArrayXXd& inputs, const double rhoMax, const Eigen::ArrayXd& params) {
+Eigen::ArrayXXd ge_41rt_inverse_distortion(const Eigen::ArrayXXd &inputs,
+                                           const double rhoMax,
+                                           const Eigen::ArrayXd &params) {
   Eigen::ArrayXd radii = inputs.matrix().rowwise().norm();
   Eigen::ArrayXd inverted_radii = radii.cwiseInverse();
   Eigen::ArrayXd cosines = inputs.col(0) * inverted_radii;
-  Eigen::ArrayXd cosine_double_angles = 2*cosines.square() - 1;
-  Eigen::ArrayXd cosine_quadruple_angles = 2*cosine_double_angles.square() - 1;
-  Eigen::ArrayXd sqrt_p_is = rhoMax / (-params[0]*cosine_double_angles - params[1]*cosine_quadruple_angles - params[2]).sqrt();
-  Eigen::ArrayXd solutions = TWO_OVER_SQRT_THREE*sqrt_p_is*(acos(N_THREE_HALVES_SQRT_3*radii/sqrt_p_is)/3 + FOUR_THIRDS_PI).cos();
-  Eigen::ArrayXXd results = solutions.rowwise().replicate(inputs.cols()).array() * inputs * inverted_radii.rowwise().replicate(inputs.cols()).array();
+  Eigen::ArrayXd cosine_double_angles = 2 * cosines.square() - 1;
+  Eigen::ArrayXd cosine_quadruple_angles =
+      2 * cosine_double_angles.square() - 1;
+  Eigen::ArrayXd sqrt_p_is =
+      rhoMax / (-params[0] * cosine_double_angles -
+                params[1] * cosine_quadruple_angles - params[2])
+                   .sqrt();
+  Eigen::ArrayXd solutions =
+      TWO_OVER_SQRT_THREE * sqrt_p_is *
+      (acos(N_THREE_HALVES_SQRT_3 * radii / sqrt_p_is) / 3 + FOUR_THIRDS_PI)
+          .cos();
+  Eigen::ArrayXXd results =
+      solutions.rowwise().replicate(inputs.cols()).array() * inputs *
+      inverted_radii.rowwise().replicate(inputs.cols()).array();
 
   return results;
 }
 
-PYBIND11_MODULE(inverse_distortion, m)
-{
-    m.doc() = "HEXRD inverse distribution plugin";
+PYBIND11_MODULE(inverse_distortion, m) {
+  m.doc() = "HEXRD inverse distribution plugin";
 
-    m.def("ge_41rt_inverse_distortion", &ge_41rt_inverse_distortion, "Inverse distortion for ge_41rt");
+  m.def("ge_41rt_inverse_distortion", &ge_41rt_inverse_distortion,
+        "Inverse distortion for ge_41rt");
 }