lisaanalysistools 1.0.7__cp312-cp312-macosx_10_9_x86_64.whl → 1.0.9__cp312-cp312-macosx_10_9_x86_64.whl

{lisaanalysistools-1.0.7.dist-info → lisaanalysistools-1.0.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lisaanalysistools
-Version: 1.0.7
+Version: 1.0.9
 Home-page: https://github.com/mikekatz04/LISAanalysistools
 Author: Michael Katz
 Author-email: mikekatz04@gmail.com
@@ -11,7 +11,7 @@ Classifier: Natural Language :: English
 Classifier: Programming Language :: C++
 Classifier: Programming Language :: Cython
 Classifier: Programming Language :: Python :: 3.12
-Requires-Python: >=3.12
+Requires-Python: >=3.6
 Description-Content-Type: text/markdown
 License-File: LICENSE
 
@@ -40,7 +40,7 @@ LISA Analysis Tools leverages conda environments to install and use necessary pa
 
 For an easy full install, follow these instructions.
 
-First, clone the repo and `cd` to the `LATW` directory.:
+First, clone the repo and `cd` to the `LISAanalysistools` directory.:
 ```
 git clone https://github.com/mikekatz04/LISAanalysistools.git
 cd LISAanalysistools/
@@ -56,7 +56,7 @@ If you want more flexibility, you can install each package given above separatel
 
 To install this software for use with NVIDIA GPUs (compute capability >5.0), you need the [CUDA toolkit](https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html) and [CuPy](https://cupy.chainer.org/). The CUDA toolkit must have cuda version >8.0. Be sure to properly install CuPy within the correct CUDA toolkit version. Make sure the nvcc binary is on `$PATH` or set it as the `CUDA_HOME` environment variable.
 
-We are currently working on building wheels and making the GPU version pip installable. For now, to work with GPUs, git clone the repository and install it from source. 
+We are currently working on building wheels and making the GPU version pip installable. For now, to work with GPUs, git clone the repository and install it from source. You must run `python scripts/prebuild.py` before running the install process.  
 
 ## Contributing
 
@@ -66,7 +66,7 @@ Please read [CONTRIBUTING.md](CONTRIBUTING.md) for details on our code of conduc
 
 We use [SemVer](http://semver.org/) for versioning. For the versions available, see the [tags on this repository](https://github.com/mikekatz04/LISAanalysistools/tags).
 
-Current Version: 1.0.7
+Current Version: 1.0.9
 
 ## Authors/Developers
 

lisaanalysistools-1.0.9.dist-info/RECORD

@@ -0,0 +1,43 @@
+lisatools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lisatools/_version.py,sha256=QbUYcRE1ARwVrFkV2kInId-i0dmrMze1qvmx5x5kC7A,123
+lisatools/analysiscontainer.py,sha256=kvTP0KSzfAlLBkeF3RNn0BcZs_K1Z0ntQA4xglfxqgI,15345
+lisatools/datacontainer.py,sha256=QVz0twD46Fl_J-ueGQUCXOJkkXUEum7yrwp9LaVeohU,9853
+lisatools/detector.py,sha256=M0bFi2sGd8RJHGlHblOfcWoJjE0HEPVPBB5SCaoq8kk,21035
+lisatools/diagnostic.py,sha256=o9vtfXbY3yMDk4cGNeOsTrlP7gYG_BRROw7gyuThDaM,34192
+lisatools/sensitivity.py,sha256=cUgYvfsfd-MrKCAUk2rwb-oQm6IfHI-IMndbRLOKqVs,27380
+lisatools/stochastic.py,sha256=CysAQ5BIVD349fLjAMpZAXaT0X0dFjrd4QFi0vfqops,9458
+lisatools/cutils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lisatools/cutils/detector_cpu.cpython-312-darwin.so,sha256=YllH2vtquGFNiYQghMveEtsH9HvvN_cejSNoCqvmKb4,154720
+lisatools/cutils/include/Detector.hpp,sha256=Ic37OgP-gvEg8qouhv9aFYf7vQP98tJ_i6PGu8RI1YQ,2050
+lisatools/cutils/include/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lisatools/cutils/include/global.hpp,sha256=3VPiqglTMRrIBXlEvDUJO2-CjKy_SLUXZt-9A1fH6lQ,572
+lisatools/cutils/src/Detector.cpp,sha256=JvAK8UCoYxgTACBhx3hi0EK9hT-HRYOxX2XozHuHI7E,7947
+lisatools/cutils/src/Detector.cu,sha256=JvAK8UCoYxgTACBhx3hi0EK9hT-HRYOxX2XozHuHI7E,7947
+lisatools/cutils/src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lisatools/cutils/src/pycppdetector.pyx,sha256=e0tz79VNS7Sxxwyx9RQwsFVdpuClBsUf7OArSkyRBa0,7613
+lisatools/sampling/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lisatools/sampling/likelihood.py,sha256=G2kAQ43qlhAzIFWvYsrSmHXd7WKJAzcCN2o07vRE8vc,29585
+lisatools/sampling/prior.py,sha256=1K1PMStpwO9WT0qG0aotKSyoNjuehXNbzTDtlk8Q15M,21407
+lisatools/sampling/stopping.py,sha256=Q8q7nM0wnJervhRduf2tBXZZHUVza5kJiAUAMUQXP5o,9682
+lisatools/sampling/utility.py,sha256=rOGotS0Aj8-DAWqsTVy2xWNsxsoz74BVrHEnG2mOkwU,14340
+lisatools/sampling/moves/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lisatools/sampling/moves/skymodehop.py,sha256=0nf72eFhFMGwi0dLJci6XZz-bIMGqco2B2_J72hQvf8,3348
+lisatools/sources/__init__.py,sha256=Fm085xHQ3VpRjqaSlws0bdVefFofAxNzVZyGQQYrQic,140
+lisatools/sources/defaultresponse.py,sha256=2SMbRf-UwgLNaA89tIANjw4BCKh1XEuRg5mgkvAg3-k,839
+lisatools/sources/utils.py,sha256=wuk2pHMTNN2r03K98uQCAYIh3eXZqk_SUcStyJZLJl8,14061
+lisatools/sources/waveformbase.py,sha256=JPLqLZd1e-6E3ySyXodO83nZGH8bVq2K_s8sF2Oy84w,845
+lisatools/sources/bbh/__init__.py,sha256=M3yP4eaScZZMhOucn88iiJ2WGL6zZTQI0xUEnmw4Nu8,37
+lisatools/sources/bbh/waveform.py,sha256=-xg2uYB8AWf233W34hxwLsnlSYPF9d7TqzNfXj95zaA,2599
+lisatools/sources/emri/__init__.py,sha256=3W0kL4bfTtZNm-wM5t6Y0TTjtZMh4fe_qDyZ6a5rLck,38
+lisatools/sources/emri/waveform.py,sha256=W7Dx_ibygF8zbz1neJzyddy7HGH0kDppDI3HKStdbqc,2422
+lisatools/sources/gb/__init__.py,sha256=GDlLi4M4QPoEdvScIyOyhTuqk0XvFM5ymhR1EE7FYXY,36
+lisatools/sources/gb/waveform.py,sha256=Plwym5CHCEZIDiQ4Fc7MrTh7T74hMNkqGqy2YDpd_OE,1777
+lisatools/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lisatools/utils/constants.py,sha256=r1kVwkpbZS13JTOxj2iRxT5sMgTYX30y-S0JdVmD5Oo,1354
+lisatools/utils/pointeradjust.py,sha256=2sT-7qeYWr1pd_sHk9leVHUTSJ7jJgYIRoWQOtYqguE,2995
+lisatools/utils/utility.py,sha256=MKqRsG8qRI1xCsj51mt6QRa80-deUUgedxibyMV3KD4,6776
+lisaanalysistools-1.0.9.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+lisaanalysistools-1.0.9.dist-info/METADATA,sha256=YnlE4nlJwhEHXdksYlclsuPIP1UPMRvfwhs6xFqTb2s,4200
+lisaanalysistools-1.0.9.dist-info/WHEEL,sha256=EahQufTUtBnyvrECh1gpiswxGRRuu1fCWmVm0K3Ige4,110
+lisaanalysistools-1.0.9.dist-info/top_level.txt,sha256=qXUn8Xi8yvK0vr3QH0vvT5cmoccjSj-UPeKGLAxdYGo,10
+lisaanalysistools-1.0.9.dist-info/RECORD,,

{lisaanalysistools-1.0.7.dist-info → lisaanalysistools-1.0.9.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (72.1.0)
+Generator: setuptools (72.2.0)
 Root-Is-Purelib: false
 Tag: cp312-cp312-macosx_10_9_x86_64
 

lisatools/_version.py

@@ -1,4 +1,4 @@
-__version__ = '1.0.7'
+__version__ = '1.0.9'
 __copyright__ = "Michael L. Katz 2024"
 __name__ = "lisaanalysistools"
 __author__ = "Michael L. Katz"

lisatools/analysiscontainer.py

@@ -112,17 +112,17 @@ class AnalysisContainer:
                     # char strain
                     ax[i * self.sens_mat.shape[1] + j].loglog(
                         self.data_res_arr.f_arr,
-                        self.data_res_arr.f_arr * self.data_res_arr[i],
+                        self.data_res_arr.f_arr * np.abs(self.data_res_arr[i]),
                     )
                     ax[i * self.sens_mat.shape[1] + j].loglog(
                         self.data_res_arr.f_arr,
-                        self.data_res_arr.f_arr * self.data_res_arr[j],
+                        self.data_res_arr.f_arr * np.abs(self.data_res_arr[j]),
                     )
         else:
             for i in range(self.sens_mat.shape[0]):
                 ax[i].loglog(
                     self.data_res_arr.f_arr,
-                    self.data_res_arr.f_arr * self.data_res_arr[i],
+                    self.data_res_arr.f_arr * np.abs(self.data_res_arr[i]),
                 )
         return (fig, ax)
 

lisatools/cutils/include/Detector.hpp

@@ -30,11 +30,12 @@ public:
     double *x_arr;
     int nlinks;
     int nspacecraft;
+    double armlength;
     int *links;
     int *sc_r;
     int *sc_e;
 
-    Orbits(double dt_, int N_, double *n_arr_, double *ltt_arr_, double *x_arr_, int *links_, int *sc_r_, int *sc_e_)
+    Orbits(double dt_, int N_, double *n_arr_, double *ltt_arr_, double *x_arr_, int *links_, int *sc_r_, int *sc_e_, double armlength_)
     {
         dt = dt_;
         N = N_;
@@ -47,15 +48,22 @@ public:
         sc_r = sc_r_;
         sc_e = sc_e_;
         links = links_;
+        armlength = armlength_;
+    };
 
-        // std::cout << " START " << std::endl;
-        // for (int i = 0; i < nlinks; i += 1)
-        // {
-        //     sc_r[i] = sc_r_[i];
-        //     sc_e[i] = sc_e_[i];
-        //     links[i] = links_[i];
-        //     // std::cout << i << " HAHAHAH " << sc_r_[i] << " " << sc_e_[i] << std::endl;
-        // }
+    int get_sc_r_from_arr(int i)
+    {
+        return sc_r[i];
+    };
+
+    int get_sc_e_from_arr(int i)
+    {
+        return sc_e[i];
+    };
+
+    int get_link_from_arr(int i)
+    {
+        return links[i];
     };
 
     CUDA_DEVICE int get_window(double t);
@@ -68,11 +76,9 @@ public:
     CUDA_DEVICE void get_normal_unit_vec_ptr(Vec *vec, double t, int link);
     CUDA_DEVICE void get_pos_ptr(Vec *vec, double t, int sc);
     void get_light_travel_time_arr(double *ltt, double *t, int *link, int num);
-    void dealloc(){
-        // delete[] links;
-        // delete[] sc_r;
-        // delete[] sc_e;
-    };
+    void get_pos_arr(double *pos_x, double *pos_y, double *pos_z, double *t, int *sc, int num);
+    void get_normal_unit_vec_arr(double *normal_unit_vec_x, double *normal_unit_vec_y, double *normal_unit_vec_z, double *t, int *link, int num);
+    void dealloc() {};
 };
 
 #endif // __DETECTOR_HPP__

lisatools/cutils/src/Detector.cpp

@@ -55,8 +55,7 @@ int Orbits::get_sc_ind(int sc)
         printf("BAD sc ind. Must be 1,2,3. %d\n", sc);
 #else
         std::ostringstream oss;
-        int xxxx = 1;
-        oss << "Bad sc ind. Must be 1,2,3. Input sc is " << sc << " " << xxxx;
+        oss << "Bad sc ind. Must be 1,2,3. Input sc is " << sc << " " << std::endl;
         std::string var = oss.str();
         throw std::invalid_argument(var);
 #endif // __CUDACC__
@@ -166,19 +165,20 @@ void Orbits::get_pos_ptr(Vec *vec, double t, int sc)
 
 #define NUM_THREADS 64
 
+
 CUDA_KERNEL
-void get_light_travel_time_kernel(double *ltt, double *t, int *link, int num, Orbits& orbits)
+void get_light_travel_time_kernel(double *ltt, double *t, int *link, int num, Orbits &orbits)
 {
     int start, end, increment;
-    #ifdef __CUDACC__
+#ifdef __CUDACC__
     start = blockIdx.x * blockDim.x + threadIdx.x;
     end = num;
     increment = gridDim.x * blockDim.x;
-    #else // __CUDACC__
+#else  // __CUDACC__
     start = 0;
     end = num;
     increment = 1;
-    #endif // __CUDACC__
+#endif // __CUDACC__
 
     for (int i = start; i < end; i += increment)
     {
@@ -188,7 +188,7 @@ void get_light_travel_time_kernel(double *ltt, double *t, int *link, int num, Or
 
 void Orbits::get_light_travel_time_arr(double *ltt, double *t, int *link, int num)
 {
-    #ifdef __CUDACC__
+#ifdef __CUDACC__
     int num_blocks = std::ceil((num + NUM_THREADS - 1) / NUM_THREADS);
 
     // copy self to GPU
@@ -199,12 +199,109 @@ void Orbits::get_light_travel_time_arr(double *ltt, double *t, int *link, int nu
     get_light_travel_time_kernel<<<num_blocks, NUM_THREADS>>>(ltt, t, link, num, *orbits_gpu);
     cudaDeviceSynchronize();
     gpuErrchk(cudaGetLastError());
-    
+
     gpuErrchk(cudaFree(orbits_gpu));
 
-    #else // __CUDACC__
+#else // __CUDACC__
 
     get_light_travel_time_kernel(ltt, t, link, num, *this);
 
-    #endif // __CUDACC__
-}
+#endif // __CUDACC__
+}
+
+
+CUDA_KERNEL
+void get_pos_kernel(double *pos_x, double *pos_y, double *pos_z, double *t, int *sc, int num, Orbits &orbits)
+{
+    int start, end, increment;
+#ifdef __CUDACC__
+    start = blockIdx.x * blockDim.x + threadIdx.x;
+    end = num;
+    increment = gridDim.x * blockDim.x;
+#else  // __CUDACC__
+    start = 0;
+    end = num;
+    increment = 1;
+#endif // __CUDACC__
+    Vec _tmp(0.0, 0.0, 0.0);
+
+    for (int i = start; i < end; i += increment)
+    {
+        _tmp = orbits.get_pos(t[i], sc[i]);
+        pos_x[i] = _tmp.x;
+        pos_y[i] = _tmp.y;
+        pos_z[i] = _tmp.z;
+    }
+}
+
+void Orbits::get_pos_arr(double *pos_x, double *pos_y, double *pos_z, double *t, int *sc, int num)
+{
+#ifdef __CUDACC__
+    int num_blocks = std::ceil((num + NUM_THREADS - 1) / NUM_THREADS);
+
+    // copy self to GPU
+    Orbits *orbits_gpu;
+    gpuErrchk(cudaMalloc(&orbits_gpu, sizeof(Orbits)));
+    gpuErrchk(cudaMemcpy(orbits_gpu, this, sizeof(Orbits), cudaMemcpyHostToDevice));
+
+    get_pos_kernel<<<num_blocks, NUM_THREADS>>>(pos_x, pos_y, pos_z, t, sc, num, *orbits_gpu);
+    cudaDeviceSynchronize();
+    gpuErrchk(cudaGetLastError());
+
+    gpuErrchk(cudaFree(orbits_gpu));
+
+#else // __CUDACC__
+
+    get_pos_kernel(pos_x, pos_y, pos_z, t, sc, num, *this);
+
+#endif // __CUDACC__
+}
+
+
+CUDA_KERNEL
+void get_normal_unit_vec_kernel(double *normal_unit_vec_x, double *normal_unit_vec_y, double *normal_unit_vec_z, double *t, int *link, int num, Orbits &orbits)
+{
+    int start, end, increment;
+#ifdef __CUDACC__
+    start = blockIdx.x * blockDim.x + threadIdx.x;
+    end = num;
+    increment = gridDim.x * blockDim.x;
+#else  // __CUDACC__
+    start = 0;
+    end = num;
+    increment = 1;
+#endif // __CUDACC__
+    Vec _tmp(0.0, 0.0, 0.0);
+
+    for (int i = start; i < end; i += increment)
+    {
+        _tmp = orbits.get_normal_unit_vec(t[i], link[i]);
+        normal_unit_vec_x[i] = _tmp.x;
+        normal_unit_vec_y[i] = _tmp.y;
+        normal_unit_vec_z[i] = _tmp.z;
+    }
+}
+
+void Orbits::get_normal_unit_vec_arr(double *normal_unit_vec_x, double *normal_unit_vec_y, double *normal_unit_vec_z, double *t, int *link, int num)
+{
+#ifdef __CUDACC__
+    int num_blocks = std::ceil((num + NUM_THREADS - 1) / NUM_THREADS);
+
+    // copy self to GPU
+    Orbits *orbits_gpu;
+    gpuErrchk(cudaMalloc(&orbits_gpu, sizeof(Orbits)));
+    gpuErrchk(cudaMemcpy(orbits_gpu, this, sizeof(Orbits), cudaMemcpyHostToDevice));
+
+    get_normal_unit_vec_kernel<<<num_blocks, NUM_THREADS>>>(normal_unit_vec_x, normal_unit_vec_y, normal_unit_vec_z, t, link, num, *orbits_gpu);
+    cudaDeviceSynchronize();
+    gpuErrchk(cudaGetLastError());
+
+    gpuErrchk(cudaFree(orbits_gpu));
+
+#else // __CUDACC__
+
+    get_normal_unit_vec_kernel(normal_unit_vec_x, normal_unit_vec_y, normal_unit_vec_z, t, link, num, *this);
+
+#endif // __CUDACC__
+}
+

lisatools/cutils/src/Detector.cu

@@ -55,8 +55,7 @@ int Orbits::get_sc_ind(int sc)
         printf("BAD sc ind. Must be 1,2,3. %d\n", sc);
 #else
         std::ostringstream oss;
-        int xxxx = 1;
-        oss << "Bad sc ind. Must be 1,2,3. Input sc is " << sc << " " << xxxx;
+        oss << "Bad sc ind. Must be 1,2,3. Input sc is " << sc << " " << std::endl;
         std::string var = oss.str();
         throw std::invalid_argument(var);
 #endif // __CUDACC__
@@ -166,19 +165,20 @@ void Orbits::get_pos_ptr(Vec *vec, double t, int sc)
 
 #define NUM_THREADS 64
 
+
 CUDA_KERNEL
-void get_light_travel_time_kernel(double *ltt, double *t, int *link, int num, Orbits& orbits)
+void get_light_travel_time_kernel(double *ltt, double *t, int *link, int num, Orbits &orbits)
 {
     int start, end, increment;
-    #ifdef __CUDACC__
+#ifdef __CUDACC__
     start = blockIdx.x * blockDim.x + threadIdx.x;
     end = num;
     increment = gridDim.x * blockDim.x;
-    #else // __CUDACC__
+#else  // __CUDACC__
     start = 0;
     end = num;
     increment = 1;
-    #endif // __CUDACC__
+#endif // __CUDACC__
 
     for (int i = start; i < end; i += increment)
     {
@@ -188,7 +188,7 @@ void get_light_travel_time_kernel(double *ltt, double *t, int *link, int num, Or
 
 void Orbits::get_light_travel_time_arr(double *ltt, double *t, int *link, int num)
 {
-    #ifdef __CUDACC__
+#ifdef __CUDACC__
     int num_blocks = std::ceil((num + NUM_THREADS - 1) / NUM_THREADS);
 
     // copy self to GPU
@@ -199,12 +199,109 @@ void Orbits::get_light_travel_time_arr(double *ltt, double *t, int *link, int nu
     get_light_travel_time_kernel<<<num_blocks, NUM_THREADS>>>(ltt, t, link, num, *orbits_gpu);
     cudaDeviceSynchronize();
     gpuErrchk(cudaGetLastError());
-    
+
     gpuErrchk(cudaFree(orbits_gpu));
 
-    #else // __CUDACC__
+#else // __CUDACC__
 
     get_light_travel_time_kernel(ltt, t, link, num, *this);
 
-    #endif // __CUDACC__
-}
+#endif // __CUDACC__
+}
+
+
+CUDA_KERNEL
+void get_pos_kernel(double *pos_x, double *pos_y, double *pos_z, double *t, int *sc, int num, Orbits &orbits)
+{
+    int start, end, increment;
+#ifdef __CUDACC__
+    start = blockIdx.x * blockDim.x + threadIdx.x;
+    end = num;
+    increment = gridDim.x * blockDim.x;
+#else  // __CUDACC__
+    start = 0;
+    end = num;
+    increment = 1;
+#endif // __CUDACC__
+    Vec _tmp(0.0, 0.0, 0.0);
+
+    for (int i = start; i < end; i += increment)
+    {
+        _tmp = orbits.get_pos(t[i], sc[i]);
+        pos_x[i] = _tmp.x;
+        pos_y[i] = _tmp.y;
+        pos_z[i] = _tmp.z;
+    }
+}
+
+void Orbits::get_pos_arr(double *pos_x, double *pos_y, double *pos_z, double *t, int *sc, int num)
+{
+#ifdef __CUDACC__
+    int num_blocks = std::ceil((num + NUM_THREADS - 1) / NUM_THREADS);
+
+    // copy self to GPU
+    Orbits *orbits_gpu;
+    gpuErrchk(cudaMalloc(&orbits_gpu, sizeof(Orbits)));
+    gpuErrchk(cudaMemcpy(orbits_gpu, this, sizeof(Orbits), cudaMemcpyHostToDevice));
+
+    get_pos_kernel<<<num_blocks, NUM_THREADS>>>(pos_x, pos_y, pos_z, t, sc, num, *orbits_gpu);
+    cudaDeviceSynchronize();
+    gpuErrchk(cudaGetLastError());
+
+    gpuErrchk(cudaFree(orbits_gpu));
+
+#else // __CUDACC__
+
+    get_pos_kernel(pos_x, pos_y, pos_z, t, sc, num, *this);
+
+#endif // __CUDACC__
+}
+
+
+CUDA_KERNEL
+void get_normal_unit_vec_kernel(double *normal_unit_vec_x, double *normal_unit_vec_y, double *normal_unit_vec_z, double *t, int *link, int num, Orbits &orbits)
+{
+    int start, end, increment;
+#ifdef __CUDACC__
+    start = blockIdx.x * blockDim.x + threadIdx.x;
+    end = num;
+    increment = gridDim.x * blockDim.x;
+#else  // __CUDACC__
+    start = 0;
+    end = num;
+    increment = 1;
+#endif // __CUDACC__
+    Vec _tmp(0.0, 0.0, 0.0);
+
+    for (int i = start; i < end; i += increment)
+    {
+        _tmp = orbits.get_normal_unit_vec(t[i], link[i]);
+        normal_unit_vec_x[i] = _tmp.x;
+        normal_unit_vec_y[i] = _tmp.y;
+        normal_unit_vec_z[i] = _tmp.z;
+    }
+}
+
+void Orbits::get_normal_unit_vec_arr(double *normal_unit_vec_x, double *normal_unit_vec_y, double *normal_unit_vec_z, double *t, int *link, int num)
+{
+#ifdef __CUDACC__
+    int num_blocks = std::ceil((num + NUM_THREADS - 1) / NUM_THREADS);
+
+    // copy self to GPU
+    Orbits *orbits_gpu;
+    gpuErrchk(cudaMalloc(&orbits_gpu, sizeof(Orbits)));
+    gpuErrchk(cudaMemcpy(orbits_gpu, this, sizeof(Orbits), cudaMemcpyHostToDevice));
+
+    get_normal_unit_vec_kernel<<<num_blocks, NUM_THREADS>>>(normal_unit_vec_x, normal_unit_vec_y, normal_unit_vec_z, t, link, num, *orbits_gpu);
+    cudaDeviceSynchronize();
+    gpuErrchk(cudaGetLastError());
+
+    gpuErrchk(cudaFree(orbits_gpu));
+
+#else // __CUDACC__
+
+    get_normal_unit_vec_kernel(normal_unit_vec_x, normal_unit_vec_y, normal_unit_vec_z, t, link, num, *this);
+
+#endif // __CUDACC__
+}
+

lisatools/cutils/src/pycppdetector.pyx

@@ -14,7 +14,7 @@ cdef extern from "../include/Detector.hpp":
         VecWrap(double x_, double y_, double z_) except+
 
     cdef cppclass OrbitsWrap "Orbits":
-        OrbitsWrap(double dt_, int N_, double *n_arr_, double *L_arr_, double *x_arr_, int *links_, int *sc_r_, int *sc_e_) except+
+        OrbitsWrap(double dt_, int N_, double *n_arr_, double *L_arr_, double *x_arr_, int *links_, int *sc_r_, int *sc_e_, double armlength_) except+
         int get_window(double t) except+
         void get_normal_unit_vec_ptr(VecWrap *vec, double t, int link)
         int get_link_ind(int link) except+
@@ -23,6 +23,11 @@ cdef extern from "../include/Detector.hpp":
         VecWrap get_pos_ptr(VecWrap* out, double t, int sc) except+
         void get_light_travel_time_arr(double *ltt, double *t, int *link, int num) except+
         void dealloc();
+        void get_pos_arr(double *pos_x, double *pos_y, double *pos_z, double *t, int *sc, int num) except+
+        void get_normal_unit_vec_arr(double *normal_unit_vec_x, double *normal_unit_vec_y, double *normal_unit_vec_z, double *t, int *link, int num) except+
+        int get_sc_r_from_arr(int i) except+
+        int get_sc_e_from_arr(int i) except+
+        int get_link_from_arr(int i) except+
         
 
 cdef class pycppDetector:
@@ -35,6 +40,7 @@ cdef class pycppDetector:
     cdef size_t links
     cdef size_t sc_r
     cdef size_t sc_e
+    cdef double armlength
 
     def __cinit__(self, 
         *args, 
@@ -48,7 +54,8 @@ cdef class pycppDetector:
             x_arr,
             links,
             sc_r, 
-            sc_e
+            sc_e,
+            armlength
         ), tkwargs = wrapper(*args, **kwargs)
 
         self.dt = dt
@@ -59,6 +66,7 @@ cdef class pycppDetector:
         self.links = links
         self.sc_r = sc_r
         self.sc_e = sc_e
+        self.armlength = armlength
 
         cdef size_t n_arr_in = n_arr
         cdef size_t L_arr_in = L_arr
@@ -75,7 +83,8 @@ cdef class pycppDetector:
             <double*> x_arr_in, 
             <int*> links_in, 
             <int*> sc_r_in, 
-            <int*> sc_e_in
+            <int*> sc_e_in,
+            armlength
         )
 
     def __dealloc__(self):
@@ -84,7 +93,7 @@ cdef class pycppDetector:
             del self.g
 
     def __reduce__(self):
-        return (rebuild, (self.dt, self.N, self.n_arr, self.L_arr, self.x_arr, self.links, self.sc_r, self.sc_e,))
+        return (rebuild, (self.dt, self.N, self.n_arr, self.L_arr, self.x_arr, self.links, self.sc_r, self.sc_e, self.armlength))
 
     def get_window(self, t: float) -> int:
         return self.g.get_window(t)
@@ -157,6 +166,42 @@ cdef class pycppDetector:
         elif isinstance(t, np.ndarray):
             return self.get_pos_arr(t, sc)
 
+    def get_pos_arr_wrap(self, *args, **kwargs):
+        (pos_x, pos_y, pos_z, t, sc, num), tkwargs = wrapper(*args, **kwargs)
+
+        cdef size_t pos_x_in = pos_x
+        cdef size_t pos_y_in = pos_y
+        cdef size_t pos_z_in = pos_z
+        cdef size_t t_in = t
+        cdef size_t sc_in = sc
+
+        self.g.get_pos_arr(
+            <double *>pos_x_in,
+            <double *>pos_y_in,
+            <double *>pos_z_in,
+            <double *>t_in,
+            <int *>sc_in,
+            num
+        )
+    
+    def get_normal_unit_vec_arr_wrap(self, *args, **kwargs):
+        (normal_unit_vec_x, normal_unit_vec_y, normal_unit_vec_z, t, sc, num), tkwargs = wrapper(*args, **kwargs)
+
+        cdef size_t normal_unit_vec_x_in = normal_unit_vec_x
+        cdef size_t normal_unit_vec_y_in = normal_unit_vec_y
+        cdef size_t normal_unit_vec_z_in = normal_unit_vec_z
+        cdef size_t t_in = t
+        cdef size_t sc_in = sc
+
+        self.g.get_normal_unit_vec_arr(
+            <double *>normal_unit_vec_x_in,
+            <double *>normal_unit_vec_y_in,
+            <double *>normal_unit_vec_z_in,
+            <double *>t_in,
+            <int *>sc_in,
+            num
+        )
+
     def get_light_travel_time_arr_wrap(self, *args, **kwargs):
         (ltt, t, link, num), tkwargs = wrapper(*args, **kwargs)
 
@@ -170,6 +215,15 @@ cdef class pycppDetector:
             <int *>link_in,
             num
         )
+
+    def get_sc_r_from_arr(self, i: int) -> int:
+        return self.g.get_sc_r_from_arr(i)
+
+    def get_sc_e_from_arr(self, i: int) -> int:
+        return self.g.get_sc_e_from_arr(i)
+
+    def get_link_from_arr(self, i: int) -> int:
+        return self.g.get_link_from_arr(i)
         
 
     @property
@@ -184,7 +238,8 @@ def rebuild(dt,
     x_arr,
     links,
     sc_r, 
-    sc_e
+    sc_e,
+    armlength
 ):
     c = pycppDetector(
         dt,
@@ -194,6 +249,7 @@ def rebuild(dt,
         x_arr,
         links,
         sc_r, 
-        sc_e
+        sc_e,
+        armlength
     )
     return c

lisatools/datacontainer.py

@@ -225,7 +225,7 @@ class DataResidualArray:
                 raise ValueError
 
         self.nchannels = len(new_out)
-        self._data_res_arr = np.asarray(list(new_out), dtype=complex)
+        self._data_res_arr = np.asarray(list(new_out), dtype=new_out[0].dtype)
 
     def __getitem__(self, index: tuple) -> np.ndarray:
         """Index this class directly in ``self.data_res_arr``."""