From 647a1a7340cf68bec05578cadadc3a22b554926f Mon Sep 17 00:00:00 2001
From: davidmcdonagh <davidnmcdonagh@gmail.com>
Date: Thu, 21 Mar 2024 16:16:21 +0000
Subject: [PATCH] Add missing export for laue ray predictor. Add missing
 include. Fix typo.

---
 .../spot_prediction/boost_python/ray_predictor.cc          | 7 +++++++
 .../spot_prediction/boost_python/reflection_predictor.cc   | 3 ++-
 .../algorithms/spot_prediction/reflection_predictor.h      | 1 +
 3 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/dials/algorithms/spot_prediction/boost_python/ray_predictor.cc b/src/dials/algorithms/spot_prediction/boost_python/ray_predictor.cc
index fd3494cc2a..0a224c652a 100644
--- a/src/dials/algorithms/spot_prediction/boost_python/ray_predictor.cc
+++ b/src/dials/algorithms/spot_prediction/boost_python/ray_predictor.cc
@@ -59,4 +59,11 @@ namespace dials { namespace algorithms { namespace boost_python {
       .def("__call__", &call_with_miller_index_array);
   }
 
+  void export_laue_ray_predictor() {
+    class_<LaueRayPredictor>("LaueRayPredictor", no_init)
+      .def(init<vec3<double>, mat3<double>, mat3<double> >(
+        (arg("unit_s0"), arg("fixed_rotation"), arg("setting_rotation"))))
+      .def("__call__", &LaueRayPredictor::operator(), (arg("miller_index"), arg("UB")));
+  }
+
 }}}  // namespace dials::algorithms::boost_python
diff --git a/src/dials/algorithms/spot_prediction/boost_python/reflection_predictor.cc b/src/dials/algorithms/spot_prediction/boost_python/reflection_predictor.cc
index d5699bb3db..9285a95040 100644
--- a/src/dials/algorithms/spot_prediction/boost_python/reflection_predictor.cc
+++ b/src/dials/algorithms/spot_prediction/boost_python/reflection_predictor.cc
@@ -173,8 +173,9 @@ namespace dials { namespace algorithms { namespace boost_python {
       const af::const_ref<std::size_t>&) = &Predictor::operator();
 
     class_<Predictor>("LaueReflectionPredictor", no_init)
-      .def(init<const vec3<double>&,
+      .def(init<const PolychromaticBeam&,
                 const Detector&,
+                const Goniometer&,
                 mat3<double>,
                 const cctbx::uctbx::unit_cell&,
                 const cctbx::sgtbx::space_group_type&,
diff --git a/src/dials/algorithms/spot_prediction/reflection_predictor.h b/src/dials/algorithms/spot_prediction/reflection_predictor.h
index d9748d9401..f06412bf58 100644
--- a/src/dials/algorithms/spot_prediction/reflection_predictor.h
+++ b/src/dials/algorithms/spot_prediction/reflection_predictor.h
@@ -28,6 +28,7 @@
 #include <dials/algorithms/spot_prediction/scan_varying_ray_predictor.h>
 #include <dials/algorithms/spot_prediction/stills_ray_predictor.h>
 #include <dials/algorithms/spot_prediction/ray_intersection.h>
+#include <cctbx/miller/index_generator.h>
 
 namespace dials { namespace algorithms {