repos_assist/GameAssist/IMGProc/ImageLoc.h

#pragma once
#ifndef __IMAGELOC_H_
#define __IMAGELOC_H_
/*
常见的图像算法，例如图像查找，颜色序列匹配（多点着色）
由于ocr与图像查找类似，故也在类ImageLoc中实现
*/
#include <vector>
#include "optype.h"
#include <string>
//#include "Dict.h"
#include "color.h"
#include "ThreadPool.h"
#include "Image.hpp"

inline int HEX2INT(wchar_t c) {
	if (L'0' <= c && c <= L'9')
		return c - L'0';
	if (L'A' <= c && c <= L'Z')
		return c - L'A' + 10;
	if (L'a' <= c && c <= L'z')
		return c - L'a' + 10;
	return 0;
}


#define SET_BIT(x, idx) (x |= 1u << (idx))

#define GET_BIT(x, idx) ((x >> (idx)) & 1u)

using img_names = std::vector<TString>;
//检查是否为透明图
int check_transparent(Image* img);
//获取匹配点
void get_match_points(const Image& img, vector<int>& points);
//generate next index for kmp
void gen_next(const Image& img, vector<int>& next);
//sum of all pixels
int inline sum(uchar* begin, uchar* end) {
	int s = 0;
	while (begin != end)
		s += *begin++;
	return s;
}

void extractConnectivity(const ImageBin& src, int threshold, std::vector<ImageBin>& out);


struct gray_diff_t {
	gray_diff_t(color_df_t const& cd) :gray(cd.color.toGray()), diff(cd.df.toGray()) {

	}
	unsigned char gray;
	unsigned char diff;
};
/*
此类用于实现一些图像功能，如图像定位，简单ocr等
*/
class ImageBase
{
public:

	const static int _max_return_obj_ct = 1800;

	using vcolor_diff_t = vector<color_df_t>;//rgb color-diff
	using vgray_diff_t = vector<gray_diff_t>;//gray

	ImageBase();

	~ImageBase();

	//brief:输入图像，建立图形矩阵,在图像操作前调用
	//image_data:	4子节对齐的像素指针
	//widht:		图像宽度
	//hegith:		h
	//x1,y1,x2,y2 拷贝区域
	//type:			输入类型,type=0表示正常输入，为-1时表示倒置输入
	//long input_image(byte* psrc, int cols, int height,long x1,long y1,long x2,long y2, int type = 0);

	void set_offset(int x1, int y1);

	long is_valid(long x, long y)
	{
		return x >= 0 && y >= 0 && x < _src.width&& y < _src.height;
	}

	long GetPixel(long x, long y, color_t& cr);

	long CmpColor(color_t color, std::vector<color_df_t>& colors, double sim);

	long FindColor(std::vector<color_df_t>& colors, int dir, long& x, long& y);

	long FindColorEx(std::vector<color_df_t>& colors, TString& retstr);

	long FindMultiColor(std::vector<color_df_t>& first_color, std::vector<pt_cr_df_t>& offset_color, double sim, long dir, long& x, long& y);

	long FindMultiColorEx(std::vector<color_df_t>& first_color, std::vector<pt_cr_df_t>& offset_color, double sim, long dir, TString& retstr);

	long FindPic(std::vector<Image*>& pics, color_t dfcolor, double sim, long& x, long& y);

	long FindPicTh(std::vector<Image*>& pics, color_t dfcolor, double sim, long& x, long& y);

	long FindPicEx(std::vector<Image*>& pics, color_t dfcolor, double sim, vpoint_desc_t& vpd);

	long FindPicExTh(std::vector<Image*>& pics, color_t dfcolor, double sim, vpoint_desc_t& vpd);

	long FindColorBlock(double sim, long count, long height, long width, long& x, long& y);

	long FindColorBlockEx(double sim, long count, long height, long width, TString& retstr);

	//描述：查找目标图像中的直线
	//输入：精度
	//输出：outStr:直线描述[法线角度，直线到原点的距离];ret:该直线上的点的数量
	long FindLine(double sim, TString& outStr);
private:
	//rgb像素匹配
	template<bool nodfcolor>
	long simple_match(long x, long y, Image* timg, color_t dfcolor, int tnrom, double sim);
	//透明图匹配
	template<bool nodfcolor>
	long trans_match(long x, long y, Image* timg, color_t dfcolor, vector<uint>points, int max_error);
	//灰度匹配
	long real_match(long x, long y, ImageBin* timg, int tnorm, double sim);
	//记录和
	void record_sum(const ImageBin& gray);
	//[x1,x2),[y1,y2)
	int region_sum(int x1, int y1, int x2, int y2);

	int get_bk_color(inputbin bin);

	//垂直方向投影,投到x轴
	void binshadowx(const rect_t& rc, std::vector<rect_t>& out_put);
	//水平方向投影，投到(y)轴
	void binshadowy(const rect_t& rc, std::vector<rect_t>& out_put);
public:
	/*
	if(abs(cr-src)<=df) pixel=1;
	else pixel=0;
	*/
	void bgr2binary(vector<color_df_t>& colors);
	//二值化 auto
	void bgr2binarybk(const vector<color_df_t>& bk_colors);
	//图像裁剪
	void bin_image_cut(int min_word_h, const rect_t& inrc, rect_t& outrc);
	void get_rois(int min_word_h, std::vector<rect_t>& vroi);

public:
	Image _src;
	ImageBin _gray;
	ImageBin _record;
	ImageBin _binary;
	Image _sum;
private:
	//起始点
	int _x1, _y1;
	//偏移
	int _dx, _dy;
	ThreadPool m_threadPool;
};

#endif