#ifndef __SEAM__CARVING_HPP__
#define __SEAM__CARVING_HPP__

#include <opencv2/core/core.hpp>
#define DEBUG 0

class SeamCarving {
  public:
    void showImage();
    const cv::Mat& getFinalImage();
    virtual void computeNewFinalImage(int pos);
    void setBlockUpdate(bool bUpdate);
    bool getBlockUpdateStatus();
    virtual void showSeamsImg();

  protected:
    SeamCarving(const cv::Mat &img, int seams, bool grow);
    void init();
    virtual cv::Mat drawSeam(const cv::Mat &frame, const std::vector<int> &seam) = 0;
    cv::Mat image;
    cv::Mat finalImage;
    int seams;
    bool grow;
    int sliderMax;
    int sliderPos;
    std::vector<std::vector<int>> vecSeams;

  private:
    cv::Mat GetEnergyImg(const cv::Mat &img);
    cv::Mat computeGradientMagnitude(const cv::Mat &frame);
    float intensity(float currIndex, int start, int end);
    cv::Mat computePathIntensityMat(const cv::Mat &rawEnergyMap);
    std::vector<int> getLeastImportantPath(const cv::Mat &importanceMap);
    cv::Mat removeLeastImportantPath(const cv::Mat &original, const std::vector<int> &seam);
    void removePixel(const cv::Mat &original, cv::Mat &outputMap, int row, int minCol);
    cv::Mat addLeastImportantPath(const cv::Mat &original, const std::vector<int> &seam);
    void addPixel(const cv::Mat &original, cv::Mat &outputMat, int row, int minCol);
    bool blockUpdate = false;

};

class SeamCarvingHorizontal : public SeamCarving
{
  public:
    SeamCarvingHorizontal(char* fileName, int seams=100, bool grow=false);
  protected:
    virtual cv::Mat drawSeam(const cv::Mat &frame, const std::vector<int> &seam) override;
};

class SeamCarvingVertical : public SeamCarving {
  public:
    SeamCarvingVertical(char* fileName, int seams=100, bool grow=false);
    virtual void computeNewFinalImage(int pos) override;
#if DEBUG
    virtual void showSeamsImg() override;
#endif
  protected:
    virtual cv::Mat drawSeam(const cv::Mat &frame, const std::vector<int> &seam) override;
};

#endif // __SEAM__CARVING_HPP__