Added render GCP

It draws the crops of the GCPs and tiles them together.

src/asp/ControlNetTK/Makefile.am

@@ -35,7 +35,10 @@ cnet_merge_LDADD     = $(CNETTK_LOCAL_LIBS)
 isis_position_extract_SOURCES = isis_position_extract.cc
 isis_position_extract_LDADD   = $(CNETTK_LOCAL_LIBS)
 
-bin_PROGRAMS = reduce_match cnet_build cnet_convert cnet_merge
+render_gcp_SOURCES = render_gcp.cc
+render_gcp_LDADD   = $(CNETTK_LOCAL_LIBS)
+
+bin_PROGRAMS = reduce_match cnet_build cnet_convert cnet_merge render_gcp
 libexec_PROGRAMS = isis_position_extract
 
 endif

src/asp/ControlNetTK/render_gcp.cc

@@ -0,0 +1,118 @@
+// Boost
+#include <boost/program_options.hpp>
+#include <boost/filesystem/operations.hpp>
+#include <boost/foreach.hpp>
+namespace po = boost::program_options;
+namespace fs = boost::filesystem;
+
+#include <vw/Core.h>
+#include <vw/Math/Vector.h>
+#include <vw/FileIO.h>
+#include <vw/Image/Algorithms.h>
+#include <asp/Core/Macros.h>
+#include <asp/Core/Common.h>
+using namespace vw;
+
+struct Options : public asp::BaseOptions {
+  std::vector<std::pair<std::string,Vector2> > input_files;
+  std::string gcp_file;
+  std::string output_file;
+  Vector2i grid_size;
+  size_t kernel_size;
+};
+
+void handle_arguments( int argc, char *argv[], Options& opt ) {
+  po::options_description general_options("");
+  general_options.add_options()
+    ("kernel-size",po::value(&opt.kernel_size)->default_value(25),
+     "Size of the individual crops thats make up the output image")
+    ("output,o",po::value(&opt.output_file),
+     "Output file for this process. If blank it will produce output with extension \"-render.tif\".");
+  general_options.add( asp::BaseOptionsDescription(opt) );
+
+  po::options_description positional("");
+  positional.add_options()
+    ("gcp-file", po::value(&opt.gcp_file));
+
+  po::positional_options_description positional_desc;
+  positional_desc.add("gcp-file", 1);
+
+  std::ostringstream usage;
+  usage << "Usage: " << argv[0] << " [options] <gcp file> \n";
+
+  po::variables_map vm =
+    asp::check_command_line( argc, argv, opt, general_options,
+                             positional, positional_desc, usage.str() );
+
+  if ( opt.gcp_file.empty() )
+    vw_throw( ArgumentErr() << "Missing input GCP file!\n"
+              << usage.str() << general_options );
+  if ( opt.output_file.empty() )
+    opt.output_file =
+      fs::path( opt.gcp_file ).stem()+"-render.tif";
+}
+
+int main( int argc, char *argv[] ) {
+
+  Options opt;
+  try {
+    handle_arguments( argc, argv, opt );
+
+    // Open GCP file
+    vw_out() << " -> Opening \"" << opt.gcp_file << "\".\n";
+    std::ifstream ifile( opt.gcp_file.c_str() );
+    if ( !ifile.is_open() )
+      vw_throw( ArgumentErr() << "Unable to open GCP file!\n" );
+    size_t count = 0;
+    while (!ifile.eof()) {
+      if ( count == 0 ) {
+        // Don't really care about this line
+        Vector3 eh, ei;
+        ifile >> eh[0] >> eh[1] >> eh[2] >> ei[0] >> ei[1] >> ei[2];
+      } else {
+        std::string file_name;
+        Vector2 location;
+        ifile >> file_name >> location[0] >> location[1];
+        opt.input_files.push_back( std::make_pair(file_name,location) );
+      }
+      count++;
+    }
+    ifile.close();
+
+    // Find and locate input images
+    vw_out() << " -> Checking input files.\n";
+    typedef std::pair<std::string,Vector2> ElementType;
+    BOOST_FOREACH( ElementType const& input, opt.input_files ) {
+      if ( !fs::exists(input.first) )
+        vw_throw( IOErr() << "Unable to open \"" << input.first << "\". Is it in the current working directory?" );
+    }
+    opt.grid_size[0] = ceil( sqrt( float(opt.input_files.size() ) ) );
+    opt.grid_size[1] = ceil( float(opt.input_files.size())/float(opt.grid_size[0] ) );
+
+    // Render image in memory
+    vw_out() << " -> Rasterizing \"" << opt.output_file << "\" in memory.\n";
+    ImageView<PixelGray<uint8> > output(opt.grid_size[0]*opt.kernel_size,
+                                        opt.grid_size[1]*opt.kernel_size);
+    count = 0;
+    BOOST_FOREACH( ElementType const& input, opt.input_files ) {
+      // Working out write location
+      Vector2i index;
+      index[0] = count - floor(float(count)/float(opt.grid_size[0]))*opt.grid_size[0];
+      index[1] = floor(float(count)/float(opt.grid_size[0]));
+
+      // Applying to image
+      DiskImageView<PixelGray<float> > image( input.first );
+      crop( output, index[0]*opt.kernel_size, index[1]*opt.kernel_size,
+            opt.kernel_size, opt.kernel_size ) =
+        channel_cast_rescale<uint8>(normalize(crop( image, input.second[0]-opt.kernel_size/2,
+                                                    input.second[1]-opt.kernel_size/2,
+                                                    opt.kernel_size, opt.kernel_size )));
+
+      count++;
+    }
+    write_image( opt.output_file, output );
+
+  } ASP_STANDARD_CATCHES;
+
+  return 0;
+}