main.cpp

//	The heartbeat-seamcarving programme serves to segment lines of a given text block by means of the heartbeat-seamcarve algorithm as published by Mathias Seuret, Daniel Stökl Ben Ezra, Marcus Liwicki. Robust Heartbeat-based Line Segmentation Methods for Regular Texts and Paratextual Elements. HIP 2017 - Proceedings of the 4th International Workshop on Historical Document Imaging and Processing, Nov 2017, Kyoto, Japan. 〈hal-01677054〉. https://hal.archives-ouvertes.fr/hal-01677054

//    Copyright (C) 2017  Mathias Seuret, Daniel Stökl Ben Ezra, Marcus Liwicki. EPHE, PSL. U Fribourg.
//    Contact at: daniel.stoekl@ephe.psl.eu, mathias.seuret@unifr.ch, marcus.liwicki@unifr.ch.

//    This program is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License Version 3 as published by the Free Software Foundation.

//    This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more details.

//    You should have received a copy of the GNU Lesser General Public License along with this program.  If not, see <https://www.gnu.org/licenses/lgpl-3.0.en.html>.

#include "include/seam_carving.h"



using namespace cv;

using namespace std;



int main(int argc, char * argv [])

{

	//cv::namedWindow("for testing");

    if(argc <6) {

        std::cout << "Invalid number of arguments" << std::endl;

        std::cout << "Arguments are: " << std::endl;

        std::cout << "1- Entry Filename " << std::endl;

        std::cout << "2- Output Filename " << std::endl;

        std::cout << "3- Number of columns r (int)" << std::endl; 
		// division of the textblock into columns for the calculation of the separation lines.

        std::cout << "4- Cubic smoothing term b (float)" << std::endl;

        std::cout << "5- Gaussian sigma smoothing term (float) " << std::endl;

        std::cout << "6- Optional Line Completion (int 0 or 1) " << std::endl;
		// 1 with heartbeat, 0 without

        std::cout << "7- Optional SeamCarving Optimization size (int 1 or bigger) " << std::endl;
		// i.e. the number of pixels the seamcarve can climb or descend by 1 pixel width. 
        std::cout << "8- Optional csv output filename " << std::endl;

		//cv::waitKey(0);

        return 0;

    }

	std::cout << string(argv[1]) << std::endl;

    cv::Mat img = cv::imread(string(argv[1]), 1);

    if(img.rows<2) {

        std::cout << "Filename does not exist" << std::endl;

		//cv::waitKey(0);

        return 0;

    }

	//Seam Carving parameters



    std::string rs(argv[3]);

    std::string bs(argv[4]);

    std::string sigmas(argv[5]);

    std::string text_file="";

    std::string::size_type sz;

    bool complete_lines = false;

    int optim =1;

    if(argc>=7) {

        std::string comp(argv[6]);

        int compi = int(std::stof(comp,&sz));

        if(compi >=1) complete_lines = true;

    }

    if(argc>=8) {

        std::string optim_size(argv[7]);

        int optim_sizei = int(std::stof(optim_size,&sz));

        if(optim_sizei >=1) optim = optim_sizei;

    }

    if(argc>=9) {

        std::string file(argv[8]);

        text_file=file;

    }
    
    bool hb = true;
    if(argc>=10) {
        hb = argv[9][0] != '0';
    }

    int r = int(std::stof(rs,&sz));

    float b = std::stof(bs,&sz);

    float sigma = std::stof(sigmas,&sz);



    std::cout << "Computing seam carving with parameters: " << std::endl;

    std::cout << "r " << r << std::endl;

    std::cout << "b " << b << std::endl;

    std::cout << "sigma " << sigma << std::endl;
    
    std::cout << "heartbeat " << hb << std::endl;



    Carve::SeamCarving seam(r,b,sigma,img,complete_lines, optim,text_file, hb);
	seam.computeMedialSeam();
    seam.drawMedialSeam(img);
    seam.computeCarvedSeams();
    seam.drawCarvedSeam(img);
    seam.writeMatches();
    cv::imwrite(string(argv[2]), img);



    std::cout << "Seam Carving Done!!" << std::endl;

	//cv::waitKey(0);

	return 0;

}