#include <iostream>
#include <opencv2/opencv.hpp>

using namespace cv;
using namespace std;

void show(Mat src, const char wd[], float i = 0.3) {

    int maxs = src.cols > src.rows ? src.cols : src.rows;
    float ra = 400 / (maxs + DBL_MIN);
    i = ra;
    //------展示图像操作------
    namedWindow(wd, WINDOW_FREERATIO);
    if (i > 100)
    {
        resizeWindow(wd, i, i);//窗口尺寸调节
    }
    else
    {
        resizeWindow(wd, int(i * src.cols), int(i * src.rows));//窗口尺寸调节
    }
    Mat src1 = src.clone();
    src1.convertTo(src1, 5);
    normalize(src1, src1, 0, 1, NORM_MINMAX);

    imshow(wd, src1);


    waitKey(0);
}

int applyMask(unsigned short* input, unsigned char* pmask, int Width, int Height, unsigned short* output)
{
    if (input == nullptr)
    {
        return 0;
    }
    if (pmask == nullptr)
    {
        return 0;
    }

    Mat src(Size(Width, Height), CV_16U, input);
    Mat mask(Size(Width, Height), CV_8U, pmask);
    Mat result;
    bitwise_and(src, src, result, mask);
  
    if (result.empty() || result.rows != Height || result.cols != Width )
    {
        return 0;
    }
    memcpy(output, result.data, Width * Height * sizeof(unsigned short));

    return 1;
}

int getMaskRect(unsigned char* pmask, int Width, int Height, int &x,int &y,int &RectWidth,int &RectHeight)
{
    if (pmask == nullptr)
    {
        return 0;
    }

    Mat mask(Size(Width, Height), CV_8U, pmask);

    if (!sum(mask)[0])
    {
        return 0;
    }

    cv::Mat nonZeroPoints;
    cv::findNonZero(mask, nonZeroPoints);
    cv::Rect rect = cv::boundingRect(nonZeroPoints);

    x = rect.x;
    y = rect.y;
    RectWidth = rect.width;
    RectHeight = rect.height;
    
    return 1;
}

int cropImg(unsigned short *input, unsigned short *output, int Width, int Height, int x, int y, int RectWidth, int RectHeight)
{
    if (input == nullptr)
    {
        return 0;
    }
    if (output == nullptr)
    {
        return 0;
    }

    Mat src(Size(Width, Height), CV_16U, input);
    memcpy(output, src(Rect(x, y, RectWidth, RectHeight)).clone().data, RectWidth * RectHeight * sizeof(unsigned short));

    return 1;
}


int main()
{
    Mat input = imread(R"(Image_FinalImage.tif)",-1);
    Mat mask = imread(R"(mask.tif)", -1);
   
    int x = 0;
    int y = 0;
    int width = 0;
    int height = 0;
    getMaskRect(mask.ptr<uchar>(), input.cols, input.rows, x, y, width, height);
    
    unsigned short* output = new unsigned short[width * height];
    cropImg(input.ptr<ushort>(), output,input.cols,input.rows,x,y,width,height);

    Mat result(Size(width, height), CV_16U, output);
    show(result, "result");

	return 0;
}