TSPlug
/
TSFindPicture.h

////////////////////////////////////////////////////////////////////////////////////

#if !defined(MYFINDPICTURECLASS_H)
#define MYFINDPICTURECLASS_H

#include <windows.h>
#include "d3d9.h"
#include <GdiPlus.h>
#include <GdiplusBitmap.h>
#include <vector>

class MyFindPictureClass
{
public:
	MyFindPictureClass(void);
	~MyFindPictureClass(void);

	enum RUNTYPE{TurnUP=0,TurnDOWN,TurnLEFT,TurnRIGHT,TurnERRORR};
	//LPDIRECT3DDEVICE9 m_pDxdevice;
	//HWND              m_Wnd;
	bool isprocessColor;//说明是找色或者取色
	int loadWidth;
	int loadHeight;
	HWND m_parenthwnd;
	HBITMAP hDXBitmap;
	HDC hDXMemDC;
	HWND m_hwnd;
	DWORD m_displayMode;//图色模式
	bool IsLastFindStr;//是否最后一次查找

	//COLORREF pWndBmpBuffer[2000][2000];
	COLORREF** pWndBmpBuffer;
	//DWORD ColorDataInfo[2000][2000];
	BYTE * pBuffer;
	//COLORREF** pColorBuffer;
	int wndWidth;
	int wndHeight;
	int nbuffersize;
	PVOID pbuffer;

	bool SaveGDIBitmap(HWND hWnd,RECT rect,wchar_t *savepath);

	bool SaveBitmapToFile(HBITMAP hBitmap, LPCWSTR lpFileName);

	bool CopyScreenToBitmap(LPRECT lpRect,wchar_t *savepath);

	DWORD GetBkMouseShape(HCURSOR bkmouseshape=NULL);//默认bkmouseshape为空时为前台模式, bkmouseshape有值的时候是后台模式

	bool loadBitmap(wchar_t* path);
	//////////////////找图   支持的是24位位图/////////////////////
	////  图片格式错误或者没找到图片，函数返回-1
	////  参数             说明
	////  hwnd           窗口的句柄,为NULL表示前台查找,不为NULL表示后台查找
	////  left           左上角x
	////  top            左上角y
	////  right          右下角x
	////  bottom         右下角y
	////  path           图片的路径,支持多张图片,使用"|"隔开
	////  colorOffset    色偏 例如"102030"
	////  simi           相似度
	////  dir            检索方式
	////  xpos           返回的图片的左上角x坐标
	////  ypos           返回的图片的左上角y坐标
	int findPicture(PVOID pbuffer,HWND hwnd,int left, int top, int right, int bottom, wchar_t* path, wchar_t *color, double simi, int dir, long &x,long &y,DWORD (*ColorDataInfo)[2000],wchar_t *retstring=NULL,int type=0);

	/////// 颜色内存处理。。 提高速度 //////////////////////
	bool processCreenWindow(int color,int left,int top,int right,int bottom,double simi,int dir,int &x,int &y,wchar_t* colorStr=NULL,bool isMulitColor=false);

	/////// 后台颜色内存处理。。 提高速度 //////////////////////
	int processColor(PVOID pbuffer,HWND hwnd,int &color,int left,int top,int right,int buttom,double simi,int dir,long &x,long &y,DWORD (*ColorDataInfo)[2000],wchar_t *colorstr=NULL,wchar_t *retstring=NULL,wchar_t *MultiColor=NULL);

	//将BMP图转JPG图
	bool   BMP2JPG(wchar_t*   strBMPFile,wchar_t*   strJPGFile);
	//将BMP图转PNG图
	bool   BMP2PNG(wchar_t*   strBMPFile,wchar_t*   strPNGFile);

	//////////////////////////////////////////////////////////////////////////////////////找字算法
	int   strColor;//字的颜色
	int   strColors[MAX_PATH];//字的颜色
	int   strColoroff[MAX_PATH];//字的色偏
	int ncount;//记录位图加载的点阵个数
	int addrxy[MAX_PATH*10][4];
	int dictindex;//记录当前字库的数量
	int usingindex;	//当前使用字库的下标
	std::vector<POINT> pWndBmpStr[2048];
	short **pWndBmpStrxy;//存储当前图片的所有坐标对于的整形值
	wchar_t SetPath[MAX_PATH];
	//wchar_t SetPicPwdString[MAX_PATH];
	//COLORREF **pLoadstrBuffer;
	//COLORREF pLoadstrBuffer[11][MAX_PATH-10];

	void hextobyte(char * hexStr,char * byteStr);//将十六进制字符串转换成二进制字符串
	//void byteTopic(short strvalue[MAX_PATH],int strwide,COLORREF pLoadstrBuffer[11][MAX_PATH-10]);//二进制数值还原成图片MAX_PATH
	void byteTopic(short strvalue[MAX_PATH],int strwide);//二进制数值还原成图片MAX_PATH
	//bool comparaStrTosim(COLORREF pLoadstrBuffer[11][MAX_PATH-10],int strwide,int strhight,int &RetX,int &RetY,int startX,int startY,float simi,int nstrcount);	//相似度匹配
	bool comparaStrTosim(int strwide,int strhight,int &RetX,int &RetY,int startX,int startY,float simi,int nstrcount);	//相似度匹配
	bool comparaStr(short pLoadstrBufferxy[MAX_PATH],int strwide,int strhight,int &RetX,int &RetY,int startX,int startY,float simi,int nstrcount);//匹配字符串
	void dicthextobyte(char * hexStr,char * byteStr,int &hexlen,short &nstrinfo);// 字库信息转二进制
	bool GetStringInfo();//获取位置点阵信息
	//获取位置点阵信息,和GetStringInfo区别是,多了一层X坐标的扫描,排除多余无效的点阵信息
	bool GetStringInfo1();
	bool MySetDict(int nindex,wchar_t *dictpath,pMyDictInfo *MyDictCount,int *nMyDictCountsize,int &NowUsingDictIndex);//设置字库 加载字库
	bool MyUseDict(int Useindex,pMyDictInfo *MyDictCount,int &NowUsingDictIndex);	   //使用字库
	bool MyDeleteDict(int deleteindex,pMyDictInfo *MyDictCount,int *nMyDictCountsize,int &NowUsingDictIndex);	//删除字库
	int ocr(PVOID pbuffer,HWND hwnd,int left, int top, int right, int bottom, wchar_t * color_format, double simi,wchar_t * retstr,pMyDictInfo *MyDictCount,int *nMyDictCountsize,int NowUsingDictIndex,DWORD (*ColorDataInfo)[2000],int type=0);
	int FindStr(PVOID pbuffer,HWND hwnd,int left, int top, int right, int bottom, wchar_t* strings, wchar_t* color, double simi, long &x,long &y,pMyDictInfo *MyDictCount,int *nMyDictCountsize,int NowUsingDictIndex,DWORD (*ColorDataInfo)[2000],wchar_t *retstring=NULL,int type=0);
	int FindStrFast(PVOID pbuffer,HWND hwnd,int left, int top, int right, int bottom, wchar_t* strings, wchar_t* color, double simi, long &x,long &y,pMyDictInfo *MyDictCount,int *nMyDictCountsize,int NowUsingDictIndex,DWORD (*ColorDataInfo)[2000],wchar_t *retstring=NULL,int type=0);
	//pMyDictInfo mydict;
	//////////////////////////////////////////////////////////////////////////////////////找字算法
private:

	bool IsFindPicNext;
	COLORREF** pLoadBmpBuffer;
	//COLORREF** pWndBmpBuffer;
	int		   ncolors;	     //色偏个数
	BYTE       m_colorOffsR; // R的偏移量
	BYTE       m_colorOffsG; // G的偏移量
	BYTE       m_colorOffsB; // B的偏移量
	BYTE       m_colorOffR[MAX_PATH]; // R的偏移量
	BYTE       m_colorOffG[MAX_PATH]; // G的偏移量
	BYTE       m_colorOffB[MAX_PATH]; // B的偏移量
	/*double     m_sim;        // 相似度*/  //未使用该变量
	int        m_Dir;        // 检索方式
	int		   m_simColor;	 //	相似度的颜色边界值

	//////////////////////////////存储传进来的四点的范围/////////////
	int        m_Left;       // 左上点
	int        m_Top ;       // 顶端点
	int        m_Right ;     // 右下点
	int        m_bottom ;    // 底端点

	int		xpos_offset;
	int		ypos_offset;

	///////////////////////////// 助记透明图的处理 /////////////
	int        m_helpXpoint;
	int        m_helpYpoint;
    //bool catchWindowBitmap();
 	//bool processBKWindows(HWND src_hWnd);

	//配合BMP2JPG使用
	int   GetEncoderClsid(const   WCHAR*   format,   CLSID*   pClsid);
	bool getGDIBitmap(HWND wnd,RECT rc,int typemode=0);//初始默认为找图	,typemode=0
	bool getDXBitmap(PVOID pbuffer,DWORD (*ColorDataInfo)[2000],int typemode=0);//初始默认为找图	,typemode=0
	bool findImage(wchar_t* path,long &x,long &y,int type=0);

	bool comparabmp(int startX,int startY,int endX,int endY);
	//////////////// 检索 方式 //////////////////
	//////////////////////////////////0:从左到右,从上倒下/////////////////////
	bool leftToRightFromTop(long &x,long &y);
	//////处理背景色为透明的图片////////////////
	bool comparaLeftToRightFromTop(int h,int w,int startX,int startY);
	//////处理背景色为不透明的图片////////////////
	bool comparaLeftToRightFromTopEx(int h,int w);

	////////////////////////////////////1:从左到右,从下倒上//////////////////
	bool leftToRightFromBottom(long &x,long &y);
	////////处理背景色为透明的图片////////////////
	bool comparaLeftToRightFromBottom(int h,int w,int startX,int startY);
	////////处理背景色为不透明的图片////////////////
	bool comparaLeftToRightFromBottomEx(int h,int w);

	////////////////////////////////////2:从右到左,从上倒下//////////////////
	bool rightToLeftFromTop(long &x,long &y);
	////////处理背景色为透明的图片////////////////
	bool comparaRightToLeftFromTop(int h,int w,int startX,int startY);
	////////处理背景色为不透明的图片////////////////
	bool comparaRightToLeftFromTopEx(int h,int w);

	//////////////////////////////////3:从右到左,从下倒上///////////////////
	bool rightToLeftFromBottom(long &x,long &y);
	////////处理背景色为透明的图片////////////////
	bool comparaRightToLeftFromBottom(int h,int w,int startX,int startY);
	////////处理背景色为不透明的图片////////////////
	bool comparaRightToLeftFromBottomEx(int h,int w);

	COLORREF RgbToBgb(__in COLORREF rgb);

	bool FindMultiColor(wchar_t *MultiColor,int i,int j,int Bottom,int right,int help_simi);

};
#endif