wechatJump.cpp

#include "stdafx.h"
#include <iostream>
#include <vector>
#include <opencv2\core.hpp>
#include <opencv2\highgui.hpp>
#include <opencv2\imgproc.hpp>
#include <opencv2\features2d.hpp>
using namespace std;
using namespace cv;
int main()
{
	int img_cout = 1;//图片编号
for (size_t times = 0; times < 1000; times++)
{
	system("adb shell screencap -p /sdcard/jump.png && adb pull /sdcard/jump.png");//获取截图
	Mat imJumpRaw = imread("jump.png");//读取截图
	//裁剪顶部，大约400px
	Mat imJump(imJumpRaw.rows - 400, imJumpRaw.cols, imJumpRaw.type());
	for (size_t i = 0; i < imJump.rows; i++)
	{
		for (size_t j = 0; j < imJump.cols; j++)
		{
			imJump.at<Vec3b>(i, j) = imJumpRaw.at<Vec3b>(i+400, j);
		}
	}
	Mat imPlayer = imread("player.jpg");//读取棋子
	
	//模板匹配识别棋子位置
	Point playerLocation(0, 0);//旗子底部中心位置
	Mat res(imJump.rows-imPlayer.rows+1,imJump.cols-imPlayer.cols+1,CV_32FC1);
	matchTemplate(imJump, imPlayer, res, TM_CCOEFF_NORMED);
	double minv = 0, maxv = 0; Point minp, maxp;
	minMaxLoc(res, &minv, &maxv, &minp, &maxp);
	cout <<minv<<"  "<<maxv<< maxp <<minp<< endl;
	if (maxv > 0.5) { //中心改正
		playerLocation.x = maxp.x + 39;
		playerLocation.y = maxp.y + 189;
	}
	//高斯滤波，边缘提取
	Mat imTempCannied = imJump.clone();
	Canny(imJump, imTempCannied, 30, 15);
	//提取格子的最高处顶点
	Point2i girdTop(1000000,1000000);
	for (size_t i = 0; i < imTempCannied.rows; i++)
	{
		for (size_t j = 0; j < imTempCannied.cols; j++)
		{
			if (imTempCannied.at<uchar>(i, j) == 255)
			{
				if (i < girdTop.y) girdTop = Point2i(j, i);
			}
		}
	}
	//填充区域
	Rect boxRect(Point2i(0,0),Point2i(0,0));//boxRect即为格子面
	floodFill(imJump, Point2i(girdTop.x,girdTop.y+20), Scalar(255, 255, 255),&boxRect);
	//计算距离时间
	Point2i nextLocation(boxRect.x + boxRect.width / 2, boxRect.y + boxRect.height / 2);
	double dis = sqrt((playerLocation.x - nextLocation.x)*(playerLocation.x - nextLocation.x) + (playerLocation.y - nextLocation.y)*(playerLocation.y - nextLocation.y));
	double t = dis*1.395;
	string strCommand = "adb shell input swipe 500 1600 500 1602 "+to_string(int(t));
	cout << strCommand << endl;
	system(strCommand.c_str());

	//绘制定位点图
	line(imJump, playerLocation, nextLocation, Scalar(0, 0, 255), 2);
	line(imJump, Point2i(playerLocation.x, 0), Point2i(playerLocation.x, imJump.rows), Scalar(0, 255, 0));
	line(imJump, Point2i(0, playerLocation.y), Point2i(imJump.cols, playerLocation.y), Scalar(0, 255, 0));
	line(imJump, Point2i(nextLocation.x, 0), Point2i(nextLocation.x, imJump.rows), Scalar(0, 255, 0));
	line(imJump, Point2i(0, nextLocation.y), Point2i(imJump.cols, nextLocation.y), Scalar(0, 255, 0));
	circle(imJump, playerLocation, 20, Scalar(255, 0, 0));
	circle(imJump, nextLocation, 20, Scalar(255, 0, 0));
	imwrite("img/" + to_string(img_cout)+"_"+to_string(rand())+ ".png", imJump);
	img_cout++;
	system("sleep 1");
}
	system("pause");
    return 0;
}