command line option --ss_locs

simply reads the input files, generates the subset locations and exits
before the analysis is run

Author: dicengine

src/core/DICe_Main.cpp

@@ -170,6 +170,32 @@ int main(int argc, char *argv[]) {
     // let the schema know how many images there are in the sequence and the first frame id:
     schema->set_frame_range(first_frame_id,num_frames);
 
+    if(input_params->get<bool>(DICe::print_subset_locations_and_exit,false)){
+      // write out the subset locations for left camera and exit
+      Teuchos::RCP<MultiField> coords = schema->mesh()->get_field(DICe::mesh::field_enums::INITIAL_COORDINATES_FS);
+      const int_t ss_locs_size = proc_rank==0 ?
+          coords->get_map()->get_num_global_elements() : 0; // both x and y coords
+      Teuchos::Array<int_t> owned_ids(ss_locs_size);
+      for(int_t i=0;i<ss_locs_size;++i)
+          owned_ids[i] = i;
+      Teuchos::RCP<MultiField_Map> zero_map = Teuchos::rcp (new MultiField_Map(-1, owned_ids,0,*schema->mesh()->get_comm()));
+      Teuchos::RCP<MultiField> all_on_zero_coords = Teuchos::rcp( new MultiField(zero_map,1,true));
+      // all to zero gather of coordinates
+      MultiField_Exporter exporter(*zero_map,*coords->get_map());
+      // export the field to zero
+      all_on_zero_coords->do_import(coords,exporter);
+      //all_on_zero_coords->describe();
+      if(proc_rank==0){
+        std::FILE * ssFile = fopen("subset_locs.txt","w");
+        for(int_t i=0;i<ss_locs_size/2;++i){
+          fprintf(ssFile,"%i %i\n",(int_t)all_on_zero_coords->local_value(i*2+0),(int_t)all_on_zero_coords->local_value(i*2+1));
+        }
+        fclose(ssFile);
+      }
+      DICe::finalize();
+      return 0;
+    }
+
     bool has_motion_windows = false;
     if(schema->analysis_type()==LOCAL_DIC){
       has_motion_windows = schema->motion_window_params()->size()>0;
@@ -195,6 +221,9 @@ int main(int argc, char *argv[]) {
     std::string stereo_file_prefix = input_params->get<std::string>(DICe::output_prefix,"DICe_solution");
     stereo_file_prefix += "_stereo";
 
+
+
+
     // if the user selects predict_resolution_error option, an error analysis is performed and the actual analysis is skipped
     if(correlation_params->get<bool>(DICe::estimate_resolution_error,false)){
       file_prefix = "DICe_error_estimation_solution";

src/core/DICe_Parser.cpp

@@ -94,6 +94,7 @@ Teuchos::RCP<Teuchos::ParameterList> parse_command_line(int argc,
                        ("input,i",po::value<std::string>(),"XML input file name <filename>.xml")
                        ("generate,g",po::value<std::string>()->implicit_value("dice"),"Create XML input file templates")
                        ("stats,s","Print field statistics to screen")
+                       ("ss_locs","Print a file (ss_locs.txt) with the subset locations and exit before analysis")
                        ;
 
   // Parse the command line options
@@ -156,6 +157,11 @@ Teuchos::RCP<Teuchos::ParameterList> parse_command_line(int argc,
     inputParams->set(DICe::print_timing,true);
   }
 
+  // Print subset locations and exit?
+  if(vm.count("ss_locs")){
+    inputParams->set(DICe::print_subset_locations_and_exit,true);
+  }
+
   // Print timing statistics?
   if(vm.count("stats")){
     inputParams->set(DICe::print_stats,true);

src/core/DICe_Parser.h

@@ -81,6 +81,8 @@ const char* const image_edge_buffer_size = "image_edge_buffer_size";
 /// Input parameter
 const char* const print_timing = "print_timing";
 /// Input parameter
+const char* const print_subset_locations_and_exit = "print_subset_locations_and_exit";
+/// Input parameter
 const char* const print_stats = "print_stats";
 /// Input parameter
 const char* const output_stereo_files = "output_stereo_files";