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 #include <mln/level/fill.hh>
  10
  11 #include "all.hh"
  12
  13 using namespace mln;
  14 using namespace mln::value;
  15
  16 static inline
  17 void
  18 add_bias (image2d<float> nrg, const char* path)
  19 {
  20   image2d<rgb8> tmp = load (path);
  21   assert (tmp.domain () == nrg.domain ());
  22   image2d<rgb8>::fwd_piter p (tmp.domain ());
  23
  24   for_all (p)
  25     nrg (p) += tmp (p).red () - tmp (p).green ();
  26 }
  27
  28 /// \file Find a vertical seam in an image.
  29
  30 int main (int argc, char** argv)
  31 {
  32   if (argc != 5 && argc != 6)
  33   {
  34     std::cerr << "usage: " << *argv
  35               << " N in.ppm carved.ppm seams.ppm [bias.ppm]\n"
  36       "where N is an integer > 0: the number of seams to carve.\n";
  37     return 1;
  38   }
  39
  40   unsigned nseam = 1;
  41
  42   try {
  43     nseam = boost::lexical_cast<unsigned> (argv[1]);
  44   }
  45   catch (const boost::bad_lexical_cast&)
  46   {
  47     std::cerr << "invalid 1st argument" << std::endl;
  48     return 2;
  49   }
  50
  51   std::cerr << "initializing ... ";
  52   image2d<rgb8> img = load (argv[2]);
  53   image2d<float> nrg = get_nrg (img);
  54   if (argc >= 6)
  55     add_bias (nrg, argv[5]);
  56   std::stack<std::vector<unsigned> > seams;
  57
  58   for (unsigned i = 0; i < nseam; ++i)
  59   {
  60     std::cerr << (i + 1) << "..";
  61     seams.push (find_vertical_seam (nrg));
  62     img = carve_vertical_seam (img, seams.top ());
  63     nrg = carve_vertical_seam (nrg, seams.top ());
  64   }
  65   std::cerr << '.';
  66   io::ppm::save (img, argv[3]);
  67   for (unsigned i = 0; i < nseam; ++i)
  68   {
  69     img = uncarve_vertical_seam (img, seams.top ());
  70     seams.pop ();
  71   }
  72   std::cerr << '.';
  73   io::ppm::save (img, argv[4]);
  74   std::cerr << std::endl;
  75 }