cpp_watermark.cpp

/*
*filename: cpp_watermark.cpp
**
***	Author: Mario Leonardo Salinas
***	Student Code: 502795
***	Master Degree curriculum: Data and Knowledge - Science and Technologies
**		
*/

/*
*	C++ implementation of a parallel watermark application.
*	std::thread and user-defined blocking queues have been used.
*	The blocking queue implementation can be found in lib/util.cpp.
*	Lines 28-35 contain some global variables defined in pipeline.cpp.
**/
#include "./lib/pipeline.cpp"

int main(int argc, char * argv[]){
    if(argc < 6) {
        std::cout << "Usage is: " << argv[0] << " source_folder result_folder watermark_file par_deg avg" <<
        std::endl << "*************************************************" << 
        std::endl << "*** Images and watermark must have SAME SIZE. ***" << 
        std::endl << "*************************************************" << std::endl;
        return(0);
    }

    
    path_in = std::string(argv[1]);
    path_out = std::string(argv[2]);
    watermark = cimg_library::CImg<int>(argv[3]);
    int par_deg = std::stol(argv[4]);
    aVg = to_bool(argv[5]);

	stage2task_queue = std::vector<queue<std::pair<std::string, cimg_library::CImg<int> > > >(par_deg);
	stage3task_queue = std::vector<queue<std::pair<std::string, cimg_library::CImg<int> > > >(par_deg);

  	std::vector<std::thread> tids;
  	std::cerr << "cpp_watermark\n";
	auto start   = std::chrono::high_resolution_clock::now();

	for(int i=0; i<par_deg; i++) {
		tids.push_back(std::thread(stage1, i));
		tids.push_back(std::thread(stage2, i));
		tids.push_back(std::thread(stage3, i));
	}
	tids.push_back(std::thread(Source, par_deg));

	for(std::thread& t: tids)
		t.join();

    auto elapsed = std::chrono::high_resolution_clock::now() - start;
	auto msec    = std::chrono::duration_cast<std::chrono::milliseconds>(elapsed).count();

	std::cout << "Time: " << msec << 
                " nImgs: " << nImgs << " par_deg: " <<
                par_deg << " source: " << path_in << "\n";
	return 0;
}