Code:
#include <windows.h>
#include <iostream>
#define Winver 0x500
#include <tlhelp32.h>
#include <stdio.h>
#include <winable.h>
#include <fstream>
#include <istream>
#include <string>
#include <time.h>
#include <sys/timeb.h>
#include <vector>
#include <algorithm>
#include <sstream>
#include <iterator>
using namespace std;
int ScreenWidth; //the width of the screen
int ScreenHeight; //the height of the screen
RECT ava_rect;
HWND ava_wind;
ofstream pixel_file; //to-be-written to file that will hold the pixel colors
ifstream pixel_check; //to-be-read from file that holds the pixel colors
bool operator == (vector<unsigned int> const& v1, vector<unsigned int> const& v2){ //for comparing 2 vectors
return (v1[0] == v2[0]) && (v1[1] == v2[1]) && (v1[2] == v2[2]);
}
int getMilliCount(){ //timing function, used to test speed
timeb tb;
ftime(&tb);
int nCount = tb.millitm + (tb.time & 0xfffff) * 1000;
return nCount;
}
int getMilliSpan(int nTimeStart){ //timing function, used to test speed
int nSpan = getMilliCount() - nTimeStart;
if(nSpan < 0)
nSpan += 0x100000 * 1000;
return nSpan;
}
void mouseMove(float x_coord, float y_coord) //moves the mouse x and y amounts
{
//x_coord *= (65335/ScreenWidth);
//y_coord *= (65335/ScreenWidth);
//cout << x_coord << endl;
//cout << y_coord << endl;
INPUT mouse;
memset(&mouse, 0, sizeof(INPUT));
mouse.type = INPUT_MOUSE;
// flag for the mouse hook to tell that it's a synthetic event.
mouse.mi.dwExtraInfo = 0x200;
//POINT p;
//p.x = x_coord;
//p.y = y_coord;
//ClientToScreen(ava_wind, &p);
mouse.mi.dx = x_coord;
mouse.mi.dy = y_coord;
mouse.mi.dwFlags = mouse.mi.dwFlags | MOUSEEVENTF_MOVE;
SendInput(1, &mouse, sizeof(mouse));
}
int curr_color[2]; //the current pixel color
unsigned int pix_data[1000][2]; //reference radius for specific size
unsigned int found_pix[100][2]; //to avoid repeat colors
int found_pix_counter = 0;
int pix_pos_piece[1]; //a pixel x, y coordinate
int pix_pos_counter = 0; //the counter for all of the positions
void get_bitmap_ava(HWND ava_hwnd, int size_x, int size_y, bool get_pixel_color){ //grabs the bitma
int pix_pos_full[size_x * size_y][1]; //use max possible positions just in case
if(get_pixel_color){ //if the user pressed enter to grab pixels
pixel_file.open("pixels.dat"); //open the write-to file
}
int pos_x, pos_y; //the starting x and y position
pos_x = size_x / 2; //crosshairs are always center screen
pos_y = size_y / 2;
//POINT p; //used for special testing
//GetCursorPos(&p);
//pos_x = p.x;
//pos_y = p.y;
int radius = 10; //the amount of pixels to grab if saving pixel colors
//int start = getMilliCount(); //track screenread speed in milliseconds
HDC ava_dc = GetDC(ava_hwnd); //get the device context of the window
HDC ava_dc_cap = CreateCompatibleDC(ava_dc); //create a compatible dc out of that
HBITMAP hCaptureBitmap = CreateCompatibleBitmap(ava_dc, size_x, size_y); //make a bitmap
SelectObject(ava_dc_cap, hCaptureBitmap); //select it in memory
BitBlt(ava_dc_cap, 0, 0, size_x, size_y, ava_dc, 0, 0, SRCCOPY); //and bitblt it
//getting the size of the picture
BITMAP bm;
GetObject(hCaptureBitmap, sizeof(bm), &bm);
int width(bm.bmWidth), height(bm.bmHeight);
//creating a bitmapheader for getting the dibits
BITMAPINFOHEADER bminfoheader;
::ZeroMemory(&bminfoheader, sizeof(BITMAPINFOHEADER));
bminfoheader.biSize = sizeof(BITMAPINFOHEADER);
bminfoheader.biWidth = width;
bminfoheader.biHeight = -height;
bminfoheader.biPlanes = 1;
bminfoheader.biBitCount = 32;
bminfoheader.biCompression = BI_RGB;
bminfoheader.biSizeImage = width * 4 * height;
bminfoheader.biClrUsed = 0;
bminfoheader.biClrImportant = 0;
//ALL THAT IS COMMENTED OUT BELOW IS TO SAVE THE PICTURE TO A FILE
/*DWORD dwBmpSize = ((size_x * bminfoheader.biBitCount + 31) / 32) * 4 * size_y;
HANDLE hDIB = GlobalAlloc(GHND,dwBmpSize);
char *lpbitmap = (char *)GlobalLock(hDIB);
// Gets the "bits" from the bitmap and copies them into a buffer
// which is pointed to by lpbitmap.
GetDIBits(CreateCompatibleDC(0), hCaptureBitmap, 0,
(UINT)bm.bmHeight,
lpbitmap,
(BITMAPINFO *)&bminfoheader, DIB_RGB_COLORS);
// A file isz3 created, this is where we will save the screen capture.
HANDLE hFile = CreateFile("captureqwsx.bmp",
GENERIC_WRITE,
0,
NULL,
CREATE_ALWAYS,
FILE_ATTRIBUTE_NORMAL, NULL);
// Add the size of the headers to the size of the bitmap to get the total file size
DWORD dwSizeofDIB = dwBmpSize + sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER);
BITMAPFILEHEADER bmfh;
//Offset to where the actual bitmap bits start.
bmfh.bfOffBits = (DWORD)sizeof(BITMAPFILEHEADER) + (DWORD)sizeof(BITMAPINFOHEADER);
//Size of the file
bmfh.bfSize = dwSizeofDIB;
//bfType must always be BM for Bitmaps
bmfh.bfType = 0x4D42; //BM
DWORD dwBytesWritten = 0;
WriteFile(hFile, (LPSTR)&bmfh, sizeof(BITMAPFILEHEADER), &dwBytesWritten, NULL);
WriteFile(hFile, (LPSTR)&bminfoheader, sizeof(BITMAPINFOHEADER), &dwBytesWritten, NULL);
WriteFile(hFile, (LPSTR)lpbitmap, dwBmpSize, &dwBytesWritten, NULL);
//Unlock and Free the DIB from the heap
GlobalUnlock(hDIB);
GlobalFree(hDIB);
//Close the handle for the file that was created
CloseHandle(hFile);*/
//FILE SAVING PROCESS ENDS HERE
//create a buffer and let the GetDIBits fill in the buffer
unsigned char* pPixels = new unsigned char[(width * 4 * height)];
if( !GetDIBits(CreateCompatibleDC(0), hCaptureBitmap, 0, height, pPixels, (BITMAPINFO*) &bminfoheader, DIB_RGB_COLORS)) // load pixel info
{
//return if fails but first delete the resources
DeleteObject(hCaptureBitmap);
delete [] pPixels; // delete the array of objects
cout << "fail" << endl;
return;
}
int x, y; // fill the x and y coordinate for grabbing pixels
x = 0;
y = 0;
int total_found = 0; //the total amount of pixels matching pixels being searched for
int previous_x = -1; //set the initial prev_x and y to -1 to avoid issues
int previous_y = -1;
for(int iter = 0; iter < (size_x * size_y); iter++){ //loop through all of the pixels in the screen
if(x < size_x){ //if the pixels have not reached the right of the screen
x++; //keep going
}
else{
x = 0; //if they do reach the right, reset to the left
y++; //and go down one row
}
unsigned int r = pPixels[(width*y+x) * 4 + 2]; //the pixel color in r g b color format
unsigned int g = pPixels[(width*y+x) * 4 + 1];
unsigned int b = pPixels[(width*y+x) * 4 + 0];
curr_color[0] = r; //set the current color array values
curr_color[1] = g; //old values wrote over with each new pixel
curr_color[2] = b;
pix_pos_piece[0] = x; //set the current position
pix_pos_piece[1] = y; //wrote over with each new pixel
if(!get_pixel_color){ //if we are looking for pixels (not saving them)
bool done = false; //a bool to check if we have gone through the entire set of pixels to look for
for(unsigned int iter2 = 0; iter2 < sizeof(pix_data)/sizeof(pix_data[0]); iter2++){
if(abs(int(curr_color[0] - pix_data[iter2][0])) < 1 && abs(int(curr_color[1] - pix_data[iter2][1])) < 1 && abs(int(curr_color[2] - pix_data[iter2][2])) < 1){
found_pix[found_pix_counter][0] = curr_color[0];
found_pix[found_pix_counter][1] = curr_color[1];
found_pix[found_pix_counter][2] = curr_color[2];
found_pix_counter++;
//if all of the current pixel colors are within 1 color difference of the pixels being searched for
break; //exit the loop
}
if(sizeof(pix_data)/sizeof(unsigned int) - 1 == iter2){ //if it reaches the end of the loop
done = true;
}
}
if((abs(int(previous_x - x)) < 10 || abs(int(previous_y - y)) < 10) && (!done)){
//cout << "incr" << endl;
//cout << x << endl;
//cout << y << endl;
//cout << endl;
//if the found pixel was within 10 pixels of the last found one
total_found++; //increment the total found
pix_pos_full[pix_pos_counter][0] = pix_pos_piece[0]; //fill in the pix_pos array
pix_pos_full[pix_pos_counter][1] = pix_pos_piece[1]; //with the x, y values of the current pixel
}
else{ //if it was not found within 10 pixels or not the right color
//cout << "Reset" << endl;
total_found = 0; //reset the total found (needs to be a grouping to match a pattern)
pix_pos_counter = 0; //reset the pix_pos_counter
}
if(total_found > 6){
break;
}
}
if(get_pixel_color){ //if the user wants to save pixels
if((abs(int(100-r)) > 10) && (abs(int(100-g)) > 10) && (abs(int(100-b)) > 10)){
if(abs(x - pos_x) < radius && abs(y - pos_y) < radius){ //if the x and y positions of the pixels are within radius amount of the center of the screen
pixel_file << r << " " << g << " " << b << "\n"; //write the r g b to the file
}
}
}
}
//cout << total_found << endl; //for testing purposes, to see how many were found. Currently, either 0 or all of them are "found"
if(!get_pixel_color){ //make sure the user did not want to save pixels
if(total_found > 6){ //if more than 6 were found (arbitrary number, will be worked into percentage later)
int other_x_pos = pix_pos_full[(sizeof(pix_pos_full)/sizeof(int))/2][0]; //grab the center position
int other_y_pos = pix_pos_full[(sizeof(pix_pos_full)/sizeof(int))/2][1]; //of the pixel positions
//I am having trouble determining how much to move
//the mouse moves by x amount (the distance between pos_x and other_pos_x)
//but I can't determine it correctly
cout << other_x_pos - pos_x << endl;
cout << other_y_pos - pos_y << endl;
mouseMove(other_x_pos - pos_x, other_y_pos - pos_y);
}
}
total_found = 0; //reset the variables
pix_pos_counter = 0; //for the next screen grab
found_pix_counter = 0;
if(get_pixel_color){ //if the save-to file was opened
pixel_file.close(); //close it
}
//clean up the bitmap and buffer unless you still need it
DeleteObject(hCaptureBitmap);
delete [] pPixels; // delete the array of objects
ReleaseDC(ava_hwnd, ava_dc);
DeleteDC(ava_dc_cap);
DeleteObject(hCaptureBitmap);
//int end = getMilliSpan(start);
//cout << end << endl;
}
int main()
{
//use for testing
/*RECT desktop_rect;
HWND desktop_hwnd;
desktop_hwnd = GetDesktopWindow();
GetWindowRect(desktop_hwnd, &desktop_rect);
while(true){
if((GetKeyState(VK_RETURN) & 0x80) != 0){
get_bitmap_ava(desktop_hwnd, desktop_rect.right, desktop_rect.bottom, true);
Sleep(500);
}
if((GetKeyState(VK_LBUTTON) & 0x80) != 0){
get_bitmap_ava(desktop_hwnd, desktop_rect.right, desktop_rect.bottom);
}
if((GetKeyState(VK_END) & 0x80) != 0){
pixel_file.close();
break;
}
}
return 0;*/
pixel_check.open("pixels.dat"); //the file that contains pixels to look for
string line; //a line to hold each set up numbers
if(pixel_check.is_open()){ //if the file open ok
vector<string> tokens; //to hold the 3 numbers as strings
unsigned int tokints[2]; //to hold the 3 numbers as ints
int counter = 0; //indexing for the pix_data array
while(getline(pixel_check,line)){ //get each line in the file
istringstream iss(line); //create a stringstream
copy(istream_iterator<string>(iss), istream_iterator<string>(), back_inserter<vector<string> >(tokens)); //copy the sstream elements into tokens
for(unsigned int it = 0; it < sizeof(tokints)/sizeof(tokints[0]); it++){
tokints[it] = atoi(tokens.at(it).c_str()); //and convert/ copy tokens into tokints as ints
}
pix_data[counter][0] = tokints[0]; //fill pix_data
pix_data[counter][1] = tokints[1];
pix_data[counter][2] = tokints[2];
counter++;
}
}
pixel_check.close(); //close the file
ifstream ava_settings("C:/AeriaGames/AVA/avaGame/Config/AVAOptionSettings.ini");
size_t found;
int file_counter = 0;
if(ava_settings.is_open()){
while(getline(ava_settings,line) && file_counter < 3){
found = line.find("=");
if(found != string::npos){
line.replace(0, found+1,"");
if(file_counter == 1){
ScreenWidth = int(atof(line.c_str()));
}
else if(file_counter == 2){
ScreenHeight = int(atof(line.c_str()));
}
}
file_counter++;
}
ava_settings.close();
}
TCHAR title[500];
while(true){
if(GetWindowText(GetForegroundWindow(), title, 500) == 24){
ava_wind = GetActiveWindow();
break;
}
Sleep(100);
}
while(true){
if((GetKeyState(VK_RETURN) & 0x80) != 0){ //if the user presses enter
get_bitmap_ava(ava_wind, ScreenWidth, ScreenHeight, true); //screen read and save the pixels the center of the screen is on
Sleep(500);
}
if((GetKeyState(VK_LBUTTON) & 0x80) != 0){ //if the user left-clicks
get_bitmap_ava(ava_wind, ScreenWidth, ScreenHeight, false); //screen read and search for pixels
Sleep(100);
}
if((GetKeyState(VK_END) & 0x80) != 0){ //killswitch
break;
}
}
return 0;
}