src/seam_main.cc

   1 #include <iostream>
   2 #include <stack>
   3 #include <vector>
   4
   5 #include <boost/lexical_cast.hpp>
   6 #include <boost/foreach.hpp>
   7
   8 #include <mln/io/ppm/save.hh>
   9
  10 #include "all.hh"
  11
  12 using namespace mln;
  13 using namespace mln::value;
  14
  15 /// \file Find a vertical seam in an image.
  16
  17 int main (int argc, char** argv)
  18 {
  19   if (argc != 5)
  20   {
  21     std::cerr << "usage: " << *argv << " N in.ppm carved.ppm seams.ppm\n"
  22       "where N is an integer > 0: the number of seams to carve.\n";
  23     return 1;
  24   }
  25
  26   unsigned nseam = 1;
  27
  28   try {
  29     nseam = boost::lexical_cast<unsigned> (argv[1]);
  30   }
  31   catch (const boost::bad_lexical_cast&)
  32   {
  33     std::cerr << "invalid 1st argument" << std::endl;
  34     return 2;
  35   }
  36
  37   std::cerr << "initializing ... ";
  38   image2d<rgb8> img = load (argv[2]);
  39   image2d<float> nrg = get_nrg (img);
  40   std::stack<std::vector<unsigned> > seams;
  41
  42   for (unsigned i = 0; i < nseam; ++i)
  43   {
  44     std::cerr << (i + 1) << "..";
  45     seams.push (find_vertical_seam (nrg));
  46     img = carve_vertical_seam (img, seams.top ());
  47     nrg = carve_vertical_seam (nrg, seams.top ());
  48   }
  49   std::cerr << '.';
  50   io::ppm::save (img, argv[3]);
  51   for (unsigned i = 0; i < nseam; ++i)
  52   {
  53     img = uncarve_vertical_seam (img, seams.top ());
  54     seams.pop ();
  55   }
  56   std::cerr << '.';
  57   io::ppm::save (img, argv[4]);
  58   std::cerr << std::endl;
  59 }