Add check for hwp buffer size

hpc4cmb · May 6, 2020 · 43c2cf9 · 43c2cf9
1 parent 7c70ea8
commit 43c2cf9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/toast/_libtoast_tod_pointing.cpp b/src/toast/_libtoast_tod_pointing.cpp
@@ -41,6 +41,13 @@ void init_tod_pointing(py::module & m) {
                   auto hwpbuf = py::cast <py::buffer> (hwpang);
                   pybuffer_check_1D <double> (hwpbuf);
                   py::buffer_info info_hwpang = hwpbuf.request();
+                  if (info_hwpang.size != n) {
+                      auto log = toast::Logger::get();
+                      std::ostringstream o;
+                      o << "HWP buffer size is not consistent.";
+                      log.error(o.str().c_str());
+                      throw std::runtime_error(o.str().c_str());
+                  }
                   rawhwpang = reinterpret_cast <double *> (info_hwpang.ptr);
               }
               toast::pointing_matrix_healpix(hpix, nest, eps, cal, mode, n,

diff --git a/src/toast/todmap/pointing.py b/src/toast/todmap/pointing.py
@@ -164,7 +164,7 @@ def exec(self, data):
             try:
                 hwpang = tod.local_hwp_angle()
             except:
-                pass
+                hwpang = None
 
             # read the common flags and apply bitmask