IO.h

Go to the documentation of this file.

#ifndef __D_CV_IO__
#define __D_CV_IO__

#include <iostream>
#include <vector>
#include <opencv/cv.h>
#include <opencv2/legacy/legacy.hpp>
#include <string>

namespace DUtilsCV
{

class IO
{
public:

  static void saveKeyPoints(const std::string &filename,
    const std::vector<cv::KeyPoint> &keys,
    const std::string &nodename = "keys");
  
  static void loadKeyPoints(const std::string &filename,
    std::vector<cv::KeyPoint> &keys,
    const std::string &nodename = "keys");
  
  static void print(const cv::Mat &m,
    const std::string &name = "", std::ostream &f = std::cout);
  
  template<class T>
  static void print(const cv::Mat &m, 
    const std::string &name = "", std::ostream &f = std::cout);
  
  template<class T>
  static void save(const std::string &filename, const T& c,
    const std::string &nodename = "data");

  template<class T>
  static void load(const std::string &filename, T& c,
    const std::string &nodename = "data");
  
  inline static void save(const std::string &filename, 
    const cv::FernClassifier &c,
    const std::string &nodename = "fern_classifier");

  inline static void load(const std::string &filename, 
    cv::FernClassifier &c,
    const std::string &nodename = "fern_classifier");
  
};

// ---------------------------------------------------------------------------

template<class T>
void IO::print(const cv::Mat &m, const std::string &name, std::ostream &f)
{
  if(!name.empty()) f << name << " = ";
  f << "[ ";
  for(int r = 0; r < m.rows; ++r)
  {
    for(int c = 0; c < m.cols; ++c)
    {
      f << m.at<T>(r, c) << " ";
    }
    f << std::endl;
  }
  f << "]" << std::endl;
}

// ---------------------------------------------------------------------------

template<class T>
void IO::save(const std::string &filename, const T& c,
  const std::string &nodename)
{
  cv::FileStorage fs(filename, cv::FileStorage::WRITE);
  c.write(fs, nodename);
}

// ---------------------------------------------------------------------------

template<class T>
void IO::load(const std::string &filename, T& c,
  const std::string &nodename)
{
  cv::FileStorage fs(filename, cv::FileStorage::READ);
  c.read(fs[nodename]);
}

// ---------------------------------------------------------------------------

inline void IO::save(const std::string &filename, 
    const cv::FernClassifier &c,
    const std::string &nodename)
{
  IO::save<cv::FernClassifier>(filename, c, nodename);
}

// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -

inline void IO::load(const std::string &filename, 
    cv::FernClassifier &c,
    const std::string &nodename)
{
  IO::load<cv::FernClassifier>(filename, c, nodename);
}

// ---------------------------------------------------------------------------

}

#endif