io_util.h

Go to the documentation of this file.

#ifndef IO_UTIL_H
#define IO_UTIL_H

#include <vector>
#include <map> 
#include <string>
#include <boost/filesystem.hpp>
#include <opencv2/opencv.hpp>

#ifndef DIRECTORY_SEPARATOR
    #if defined(WIN32) || defined(_WIN32)
        #define DIRECTORY_SEPARATOR "\\"
    #else
        #define DIRECTORY_SEPARATOR "/"
    #endif
#endif

class IOUtil {
public:
    static int readMat(boost::filesystem::path file, cv::Mat &mat, std::string field = "mat");
    
    static int writeMat(boost::filesystem::path file, const cv::Mat& mat, 
            std::string field = "mat");
    
    template<typename T>
    static int writeArrayCSV(boost::filesystem::path file, T** array, int rows, int cols, 
            std::string separator = ",", int precision = 6);
    
    template<typename T>
    static int writeMatCSV(boost::filesystem::path file, const cv::Mat& mat, 
            std::string separator = ",", int precision = 6);
    
    static int readMatCSVInt(boost::filesystem::path file, cv::Mat &result);
    
    static int readMatCSVInt(boost::filesystem::path file, int rows, int cols, cv::Mat &result);
    
    static int readMatCSVFloat(boost::filesystem::path file, cv::Mat &result);
    
    static int readCSVHeaderString(boost::filesystem::path file, std::vector<std::string> &header);
    
    static int readCSVSummary(boost::filesystem::path file, std::vector<std::string>& row_headers, 
            std::vector<std::string>& col_headers, cv::Mat &data);
    
    static void listSubdirectories(boost::filesystem::path directory, 
            std::multimap<std::string, boost::filesystem::path> subdirectories);
    
    static void readDirectory(boost::filesystem::path directory, 
        std::vector<std::string> extensions, 
        std::multimap<std::string, boost::filesystem::path> &files,
        std::string prefix = "", std::string suffix = "",
        std::vector<std::string> exclude = std::vector<std::string>{});
        
    static void readDirectory(boost::filesystem::path directory,
        std::multimap<std::string, boost::filesystem::path> &files,
        std::string prefix = "", std::string suffix = "");
    
    static void getImageExtensions(std::vector<std::string> &extensions);
    
    static void getTxtExtensions(std::vector<std::string> &extensions);
    
    static void getCSVExtensions(std::vector<std::string> &extensions);
    
};

#endif  /* IO_UTIL_H */