src/seam_main.cc

   1 #include <iostream>
   2 #include <vector>
   3
   4 #include <boost/foreach.hpp>
   5
   6 #include <mln/io/ppm/save.hh>
   7 #include <mln/literal/colors.hh>
   8
   9 #include "all.hh"
  10
  11 using namespace mln;
  12 using namespace mln::value;
  13
  14 std::vector<unsigned>
  15 carve_vertical_seam (const image2d<float> img);
  16
  17 int main (int argc, char** argv)
  18 {
  19   if (argc != 3)
  20   {
  21     std::cerr << "usage: " << *argv << " in.ppm out.ppm\n";
  22     return 1;
  23   }
  24
  25   image2d<rgb8> img = load (argv[1]);
  26   image2d<float> nrg = get_nrg (img);
  27   std::vector<unsigned> seam_path = carve_vertical_seam (nrg);
  28   unsigned row = 0;
  29   BOOST_FOREACH (unsigned col, seam_path)
  30   {
  31     img.at (row, col) = literal::red;
  32     ++row;
  33   }
  34   io::ppm::save (img, argv[2]);
  35 }