Add 'nthreads' parameter to ctfindvar (#2713)

Josh Cardenzana · Jürgen Knödlseder
1 parent 24e385a1
Showing 3 changed files with 11 additions and 0 deletions
doc/source/users/reference_manual/ctfindvar.rst
src/ctfindvar/ctfindvar.cpp
src/ctfindvar/ctfindvar.par
@@ -108,6 +108,9 @@ Spatial binning parameters
 Standard parameters
 -------------------
  
+``(nthreads = 0) [integer]``
+    Number of parallel processes (0=use all available CPUs).
+
 ``(chatter = 2) [integer]``
     Verbosity of the executable:
      ``chatter = 0``: no information will be logged
@@ -610,6 +610,13 @@ void ctfindvar::free_members(void)
  ***************************************************************************/
 void ctfindvar::get_parameters(void)
 {
+    #ifdef _OPENMP
+    int nthreads = (*this)["nthreads"].integer();
+    if (nthreads > 0) {
+        omp_set_num_threads(nthreads);
+    }
+    #endif
+
     // Load the observations
     setup_observations(m_obs, false, true, false);
  
@@ -50,6 +50,7 @@ ysrc,     r, a, 22.51,-90,90, &quot;Second coordinate of the source of interest (DEC 
 #
 # Standard parameters
 #====================
+nthreads, i, h, 0,0,1000, "Number of parallel processes (0=use all available CPUs)"
 chatter,  i, h, 2,0,4, "Chattiness of output"
 clobber,  b, h, yes,,, "Overwrite existing output files with new output files?"
 debug,    b, h, no,,, "Activate debugging mode?"