Added functionality to dump files in respective bibNumber folder

bibnumber/Debug/subdir.mk

@@ -11,7 +11,8 @@ CPP_SRCS += \
 ../pipeline.cpp \
 ../textdetection.cpp \
 ../textrecognition.cpp \
-../train.cpp 
+../train.cpp \
+../dumpimages.cpp 
 
 OBJS += \
 ./batch.o \
@@ -21,7 +22,8 @@ OBJS += \
 ./pipeline.o \
 ./textdetection.o \
 ./textrecognition.o \
-./train.o 
+./train.o \
+./dumpimages.o 
 
 CPP_DEPS += \
 ./batch.d \
@@ -31,7 +33,8 @@ CPP_DEPS += \
 ./pipeline.d \
 ./textdetection.d \
 ./textrecognition.d \
-./train.d 
+./train.d \
+./dumpimages.d 
 
 
 # Each subdirectory must supply rules for building sources it contributes

bibnumber/batch.cpp

@@ -5,6 +5,7 @@
 #include <vector>
 #include <string>
 #include <boost/algorithm/string/predicate.hpp>
+#include <boost/lexical_cast.hpp>
 #include <boost/bimap.hpp>
 #include <boost/bimap/set_of.hpp>
 #include <boost/bimap/multiset_of.hpp>
@@ -16,6 +17,7 @@
 #include "batch.h"
 #include "pipeline.h"
 #include "log.h"
+#include "dumpimages.h"
 
 namespace bimaps = boost::bimaps;
 namespace fs = boost::filesystem;
@@ -243,10 +245,13 @@ int process(std::string inputName, std::string svmModel) {
 				bimaps::multiset_of<int> > imgTagBimap;
 
 		imgTagBimap tags;
+        bool file_dump_flag;
 
 		/* find images in directory */
 		img_paths = getImageFiles(inputName);
 
+        dumpimages::dumpFiles dumpObj;
+
 		/* process images */
 		for (int i = 0, j=img_paths.size(); i<j ; i++) {
 			std::vector<int> bibNumbers;
@@ -257,7 +262,18 @@ int process(std::string inputName, std::string svmModel) {
 			for (unsigned int k = 0; k < bibNumbers.size(); k++) {
 				tags.insert(
 						imgTagBimap::value_type(img_paths[i].string(), bibNumbers[k]));
+                        file_dump_flag = dumpObj.dumpImages(boost::lexical_cast<std::string>(bibNumbers[k]), img_paths[i]);
 			}
+
+            if(bibNumbers.size() == 0)
+            {
+                file_dump_flag = dumpObj.dumpImages(dumpObj.NOT_FOUND, img_paths[i]);
+            }
+
+            if(file_dump_flag)
+            {
+                std::cout << "Success in copying file." << std::endl;
+            }
 		}
 
 		/* save results to .csv file */

bibnumber/dumpimages.cpp

@@ -0,0 +1,34 @@
+
+#include "dumpimages.h"
+namespace fs = boost::filesystem;
+
+namespace dumpimages {
+
+    bool dumpFiles::dumpImages(std::string bibNumber, fs::path imageName)
+    {
+        fs::path pathname(imageName);
+        fs::path dirname = pathname.parent_path();
+        fs::path file(imageName);
+        fs::path full_path = dirname / file;
+        std::string bibPath = dirname.string() + "/"+ bibNumber;
+        if(!fs::exists(bibPath))
+        {
+            if(boost::filesystem::create_directory(bibPath))
+            {
+                std::cout << "Folder successfully created: " + bibNumber << std::endl;
+            }
+        }
+
+
+        fs::path bibPathDir(bibPath);
+        try {
+            fs::copy_file(pathname, bibPathDir / pathname.filename());
+        }
+        catch(fs::filesystem_error const & e)
+        {
+            std:: cerr << e.what() << '\n';
+            return false;
+        }
+        return true;
+    }
+} /* namespace dumpimages*/

bibnumber/dumpimages.h

@@ -0,0 +1,20 @@
+
+#ifndef BIBNUMBER_DUMPIMAGES_H
+#define BIBNUMBER_DUMPIMAGES_H
+#include <iostream>
+#include <string>
+#include <boost/filesystem.hpp>
+
+namespace dumpimages
+{
+    class dumpFiles {
+    public:
+        bool dumpImages(std::string bibNumber, boost::filesystem::path imageName);
+        std::string NOT_FOUND;
+
+        dumpFiles(){
+            NOT_FOUND = "NOT_FOUND";
+        }
+    };
+}
+#endif //BIBNUMBER_DUMPIMAGES_H