samples/cpp/starter_imagelist.cpp

   1 /*
   2  * starter_imagelist.cpp
   3  *
   4  *  Created on: Nov 23, 2010
   5  *      Author: Ethan Rublee
   6  *
   7  * A starter sample for using opencv, load up an imagelist
   8  * that was generated with imagelist_creator.cpp
   9  * easy as CV_PI right?
  10  */
  11 #include "opencv2/imgcodecs.hpp"
  12 #include "opencv2/highgui/highgui.hpp"
  13 #include <iostream>
  14 #include <vector>
  15
  16 using namespace cv;
  17 using namespace std;
  18
  19 //hide the local functions in an unnamed namespace
  20 namespace
  21 {
  22 void help(char** av)
  23 {
  24   cout << "\nThis program gets you started being able to read images from a list in a file\n"
  25           "Usage:\n./" << av[0] << " image_list.yaml\n"
  26        << "\tThis is a starter sample, to get you up and going in a copy pasta fashion.\n"
  27        << "\tThe program reads in an list of images from a yaml or xml file and displays\n"
  28        << "one at a time\n"
  29        << "\tTry running imagelist_creator to generate a list of images.\n"
  30         "Using OpenCV version %s\n" << CV_VERSION << "\n" << endl;
  31 }
  32
  33 bool readStringList(const string& filename, vector<string>& l)
  34 {
  35   l.resize(0);
  36   FileStorage fs(filename, FileStorage::READ);
  37   if (!fs.isOpened())
  38     return false;
  39   FileNode n = fs.getFirstTopLevelNode();
  40   if (n.type() != FileNode::SEQ)
  41     return false;
  42   FileNodeIterator it = n.begin(), it_end = n.end();
  43   for (; it != it_end; ++it)
  44     l.push_back((string)*it);
  45   return true;
  46 }
  47
  48 int process(vector<string> images)
  49 {
  50     namedWindow("image", WINDOW_KEEPRATIO); //resizable window;
  51     for (size_t i = 0; i < images.size(); i++)
  52     {
  53         Mat image = imread(images[i], IMREAD_GRAYSCALE); // do grayscale processing?
  54         imshow("image",image);
  55         cout << "Press a key to see the next image in the list." << endl;
  56         waitKey(); // wait indefinitely for a key to be pressed
  57     }
  58     return 0;
  59 }
  60
  61 }
  62
  63 int main(int ac, char** av)
  64 {
  65
  66   if (ac != 2)
  67   {
  68     help(av);
  69     return 1;
  70   }
  71   std::string arg = av[1];
  72   vector<string> imagelist;
  73
  74   if (!readStringList(arg,imagelist))
  75   {
  76     cerr << "Failed to read image list\n" << endl;
  77     help(av);
  78     return 1;
  79   }
  80
  81   return process(imagelist);
  82 }