| author | ecalot
<ecalot> 2005-07-03 19:12:35 UTC |
| committer | ecalot
<ecalot> 2005-07-03 19:12:35 UTC |
| parent | 8ed5d58b593cda9a4541a90539c3206255dca223 |
| poprecog/poprecog.c | +688 | -655 |
diff --git a/poprecog/poprecog.c b/poprecog/poprecog.c index f4e4709..0c7e11f 100644 --- a/poprecog/poprecog.c +++ b/poprecog/poprecog.c @@ -1,21 +1,21 @@ /* Poprecog - Prince of Persia image recognizer - Copyright (C) 2005 Princed Development Team + Copyright (C) 2005 Princed Development Team - This program is free software; you can redistribute it and/or modify - it under the terms of the GNU General Public License as published by - the Free Software Foundation; either version 2 of the License, or - (at your option) any later version. + This program is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 2 of the License, or + (at your option) any later version. - This program is distributed in the hope that it will be useful, - but WITHOUT ANY WARRANTY; without even the implied warranty of - MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - GNU General Public License for more details. + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. - You should have received a copy of the GNU General Public License - along with this program; if not, write to the Free Software - Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + You should have received a copy of the GNU General Public License + along with this program; if not, write to the Free Software + Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA - The authors of this program may be contacted at http://forum.princed.com.ar + The authors of this program may be contacted at http://forum.princed.com.ar */ /* @@ -27,12 +27,12 @@ poprecog.c: Prince of Persia Screenshots Recognizer */ /* - Changelog: - - 2005.06.15 - I've started coding this pretty tool ;) - - 2005.06.16 - Second day of coding (few hours)... - - 2005.06.17 - Third day of coding (few hours)... Some problems with POP1 compability. - - ... - - 2005.07.01 - :) + Changelog: + - 2005.06.15 - I've started coding this pretty tool ;) + - 2005.06.16 - Second day of coding (few hours)... + - 2005.06.17 - Third day of coding (few hours)... Some problems with POP1 compability. + - ... + - 2005.07.01 - :) */ /* Header files */ @@ -53,13 +53,13 @@ poprecog.c: Prince of Persia Screenshots Recognizer #include <sys/stat.h> #ifndef WIN32 - #define defmkdir(a) mkdir (a,(mode_t)0755) + #define defmkdir(a) mkdir (a,(mode_t)0755) #define osIndepGetCharacter() getchar() #define SEPS "\\" #define SEPC '\\' #else #include <direct.h> /* mkdir */ - #define defmkdir(a) mkdir (a) + #define defmkdir(a) mkdir (a) #define osIndepGetCharacter() getche() #define SEPS "/" #define SEPC '/' @@ -78,18 +78,18 @@ poprecog.c: Prince of Persia Screenshots Recognizer /* Types and global variables */ struct sScreenShotList { - char *fileName; + char *fileName; } screenShotList[MAX_SCREENSHOTS]; int screenShotsNumber; char screenShotsDir[100]; typedef struct sImage { - BITMAP *bitmap; - char *filePath; - char direction; - int pixelsNumber; - int dirID; + BITMAP *bitmap; + char *filePath; + char direction; + int pixelsNumber; + int dirID; } tImage; tImage image[MAX_IMAGES]; @@ -101,27 +101,27 @@ int DEBUGY = 0; struct sDirInfo { - char dirName[16]; - int recognizedNumber; - int optTwoDirections; - int optMaxRecognizedNumber; - int optMinImagePercent; - int optAllowTransparency; - /* int optVolatile; */ + char dirName[16]; + int recognizedNumber; + int optTwoDirections; + int optMaxRecognizedNumber; + int optMinImagePercent; + int optAllowTransparency; + /* int optVolatile; */ } dirInfo[MAX_DIRS]; int dirsNumber; typedef struct sRecognized { - int imageID; - int posX, posY; - int goodPixels; - int goodPixels2; - int layer; - int pixelsNumber; - int upperLayers; - int goodPixelsPercent; - int ownedPixels; /* for recognizeMap */ + int imageID; + int posX, posY; + int goodPixels; + int goodPixels2; + int layer; + int pixelsNumber; + int upperLayers; + int goodPixelsPercent; + int ownedPixels; /* for recognizeMap */ } tRecognized; tRecognized recognized[MAX_RECOGNIZED_ITEMS]; int recognizedNumber; @@ -131,8 +131,8 @@ int actualLayer; int recognizeMap[320][200]; /* here are stored information which 'recognize result' have this pixel */ struct sSharedPixels { - int recognizedID; - int number; + int recognizedID; + int number; } sharedPixels[MAX_RECOGNIZED_ITEMS]; FILE *outputFile, *outputSmallFile; @@ -145,699 +145,732 @@ int optMaxLayers; int match(char *pat, char *str) { - switch(*pat) - { - case '\0': return !*str; - case '*': return match(pat + 1, str) || *str && match(pat, str + 1); - case '?': return *str && match(pat + 1, str + 1); - default: return *pat == *str && match(pat + 1, str + 1); - } + switch(*pat) + { + case '\0': return !*str; + case '*': return match(pat + 1, str) || *str && match(pat, str + 1); + case '?': return *str && match(pat + 1, str + 1); + default: return *pat == *str && match(pat + 1, str + 1); + } } int pstrcmp(const void *p1, const void *p2) { - return strcmp(*(char * const *)p1, *(char * const *)p2); + return strcmp(*(char * const *)p1, *(char * const *)p2); } int cmptImage(void *a, void *b) { - register int pxa = ((tImage*)(a))->pixelsNumber; - register int pxb = ((tImage*)(b))->pixelsNumber; - if (pxa > pxb) - return -1; - if (pxa < pxb) - return 1; - return 0; + register int pxa = ((tImage*)(a))->pixelsNumber; + register int pxb = ((tImage*)(b))->pixelsNumber; + if (pxa > pxb) + return -1; + if (pxa < pxb) + return 1; + return 0; } inline int findImageOnScreenShotInPosition(int imageID, int posX, int posY) { - register int i, j, x, y; - register short transparentPixel = makecol16(0, 0, 0); - register short screenShotTransparentPixel = makecol16(255, 0, 255); - register int goodPixels; - BITMAP *bitmap = image[imageID].bitmap; - register int wPer2 = bitmap->w / 2; - register int hPer2 = bitmap->h / 2; - - sharedPixels[0].recognizedID = -1; - - goodPixels = 0; - for (x = 0; x < bitmap->w; x++) - for (y = 0; y < bitmap->h; y++) - { - if ((((short *)bitmap->line[y])[x] == transparentPixel) && (dirInfo[image[imageID].dirID].optAllowTransparency)) - continue; - if (((short *)bitmap->line[y])[x] == ((short *)screenShot->line[posY+y])[posX+x]) - { - if (recognizeMap[posX+x][posY+y] != -1) - { - for (i=0; sharedPixels[i].recognizedID != -1; i++) - { - if (sharedPixels[i].recognizedID == recognizeMap[posX+x][posY+y]) - { - if (sharedPixels[i].number >= 0) sharedPixels[i].number++; - break; - } - } - if (sharedPixels[i].recognizedID == -1) - { - sharedPixels[i].recognizedID = recognizeMap[posX+x][posY+y]; - sharedPixels[i].number = 1; - sharedPixels[i+1].recognizedID = -1; - } - } - else - { - goodPixels++; - } - continue; - } - if (((short *)transparentScreenShot->line[posY+y])[posX+x] == screenShotTransparentPixel) - { - if (recognizeMap[posX+x][posY+y] != -1) - { - for (i=0; sharedPixels[i].recognizedID != -1; i++) - { - if (sharedPixels[i].recognizedID == recognizeMap[posX+x][posY+y]) - { - if (sharedPixels[i].number >= 0) sharedPixels[i].number = -1; - break; - } - } - if (sharedPixels[i].recognizedID == -1) - { - sharedPixels[i].recognizedID = recognizeMap[posX+x][posY+y]; - sharedPixels[i].number = -1; - sharedPixels[i+1].recognizedID = -1; - } - } - continue; - } - return 0; - } - - for (i=0; sharedPixels[i].recognizedID != -1; i++) - { - if (sharedPixels[i].number > 0) goodPixels += sharedPixels[i].number; - } - - if (((goodPixels*100)/image[imageID].pixelsNumber) < dirInfo[image[imageID].dirID].optMinImagePercent) return 0; - - return goodPixels; + register int i, j, x, y; + register short transparentPixel = makecol16(0, 0, 0); + register short screenShotTransparentPixel = makecol16(255, 0, 255); + register int goodPixels; + BITMAP *bitmap = image[imageID].bitmap; + register int wPer2 = bitmap->w / 2; + register int hPer2 = bitmap->h / 2; + + sharedPixels[0].recognizedID = -1; + + goodPixels = 0; + for (x = 0; x < bitmap->w; x++) + for (y = 0; y < bitmap->h; y++) + { + if ((((short *)bitmap->line[y])[x] == transparentPixel) && (dirInfo[image[imageID].dirID].optAllowTransparency)) + continue; + if (((short *)bitmap->line[y])[x] == ((short *)screenShot->line[posY+y])[posX+x]) + { + if (recognizeMap[posX+x][posY+y] != -1) + { + for (i=0; sharedPixels[i].recognizedID != -1; i++) + { + if (sharedPixels[i].recognizedID == recognizeMap[posX+x][posY+y]) + { + if (sharedPixels[i].number >= 0) sharedPixels[i].number++; + break; + } + } + if (sharedPixels[i].recognizedID == -1) + { + sharedPixels[i].recognizedID = recognizeMap[posX+x][posY+y]; + sharedPixels[i].number = 1; + sharedPixels[i+1].recognizedID = -1; + } + } + else + { + goodPixels++; + } + continue; + } + if (((short *)transparentScreenShot->line[posY+y])[posX+x] == screenShotTransparentPixel) + { + if (recognizeMap[posX+x][posY+y] != -1) + { + for (i=0; sharedPixels[i].recognizedID != -1; i++) + { + if (sharedPixels[i].recognizedID == recognizeMap[posX+x][posY+y]) + { + if (sharedPixels[i].number >= 0) sharedPixels[i].number = -1; + break; + } + } + if (sharedPixels[i].recognizedID == -1) + { + sharedPixels[i].recognizedID = recognizeMap[posX+x][posY+y]; + sharedPixels[i].number = -1; + sharedPixels[i+1].recognizedID = -1; + } + } + continue; + } + return 0; + } + + for (i=0; sharedPixels[i].recognizedID != -1; i++) + { + if (sharedPixels[i].number > 0) goodPixels += sharedPixels[i].number; + } + + if (((goodPixels*100)/image[imageID].pixelsNumber) < dirInfo[image[imageID].dirID].optMinImagePercent) return 0; + + return goodPixels; } inline int putImageOnRecognizeMap(BITMAP *bitmap, int posX, int posY, int recognizedID) { - register int x, y, value; - register short transparentPixel = makecol16(0, 0, 0); - register short screenShotTransparentPixel = makecol16(255, 0, 255); - - for (x = 0; x < bitmap->w; x++) - for (y = 0; y < bitmap->h; y++) - { - value = recognizeMap[posX+x][posY+y]; - if (((((short *)bitmap->line[y])[x] != transparentPixel) || (!dirInfo[image[recognized[recognizedID].imageID].dirID].optAllowTransparency)) && - (((short *)bitmap->line[y])[x] == ((short *)screenShot->line[posY+y])[posX+x])) - { - if (value == -1) - { - recognizeMap[posX+x][posY+y] = recognizedID; - recognized[recognizedID].ownedPixels++; - } - else - if (recognized[value].goodPixels < recognized[recognizedID].goodPixels) - { - recognized[value].ownedPixels--; - recognizeMap[posX+x][posY+y] = recognizedID; - recognized[recognizedID].ownedPixels++; - } - } - } - return recognized[recognizedID].ownedPixels; + register int x, y, value; + register short transparentPixel = makecol16(0, 0, 0); + register short screenShotTransparentPixel = makecol16(255, 0, 255); + + for (x = 0; x < bitmap->w; x++) + for (y = 0; y < bitmap->h; y++) + { + value = recognizeMap[posX+x][posY+y]; + if (((((short *)bitmap->line[y])[x] != transparentPixel) || (!dirInfo[image[recognized[recognizedID].imageID].dirID].optAllowTransparency)) && + (((short *)bitmap->line[y])[x] == ((short *)screenShot->line[posY+y])[posX+x])) + { + if (value == -1) + { + recognizeMap[posX+x][posY+y] = recognizedID; + recognized[recognizedID].ownedPixels++; + } + else + if (recognized[value].goodPixels < recognized[recognizedID].goodPixels) + { + recognized[value].ownedPixels--; + recognizeMap[posX+x][posY+y] = recognizedID; + recognized[recognizedID].ownedPixels++; + } + } + } + return recognized[recognizedID].ownedPixels; } int cutImageFromTransparentScreenShot(int recognizedID, int posX, int posY) { - register int x, y, c; - register int transparentPixel = makecol16(0, 0, 0); - register int screenShotTransparentPixel = makecol16(255, 0, 255); - char buf[100]; - BITMAP *bitmap = image[recognized[recognizedID].imageID].bitmap; - - for (x = 0; x < bitmap->w; x++) - for (y = 0; y < bitmap->h; y++) - { - if (((((short *)bitmap->line[y])[x] != transparentPixel) || (!dirInfo[image[recognized[recognizedID].imageID].dirID].optAllowTransparency)) && - (((short *)bitmap->line[y])[x] == ((short *)screenShot->line[posY+y])[posX+x])) - ((short *)transparentScreenShot->line[posY+y])[posX+x] = screenShotTransparentPixel; - } + register int x, y, c; + register int transparentPixel = makecol16(0, 0, 0); + register int screenShotTransparentPixel = makecol16(255, 0, 255); + char buf[100]; + BITMAP *bitmap = image[recognized[recognizedID].imageID].bitmap; + + for (x = 0; x < bitmap->w; x++) + for (y = 0; y < bitmap->h; y++) + { + if (((((short *)bitmap->line[y])[x] != transparentPixel) || (!dirInfo[image[recognized[recognizedID].imageID].dirID].optAllowTransparency)) && + (((short *)bitmap->line[y])[x] == ((short *)screenShot->line[posY+y])[posX+x])) + ((short *)transparentScreenShot->line[posY+y])[posX+x] = screenShotTransparentPixel; + } } int cutImageFromScreenShot(int recognizedID, int posX, int posY) { - register int x, y, c; - register int transparentPixel = makecol16(0, 0, 0); - register int screenShotTransparentPixel = makecol16(255, 0, 255); - char buf[100]; - BITMAP *bitmap = image[recognized[recognizedID].imageID].bitmap; - - for (x = 0; x < bitmap->w; x++) - for (y = 0; y < bitmap->h; y++) - { - if (((((short *)bitmap->line[y])[x] != transparentPixel) || (!dirInfo[image[recognized[recognizedID].imageID].dirID].optAllowTransparency)) && - (((short *)bitmap->line[y])[x] == ((short *)screenShot->line[posY+y])[posX+x])) - ((short *)screenShot->line[posY+y])[posX+x] = screenShotTransparentPixel; - } + register int x, y, c; + register int transparentPixel = makecol16(0, 0, 0); + register int screenShotTransparentPixel = makecol16(255, 0, 255); + char buf[100]; + BITMAP *bitmap = image[recognized[recognizedID].imageID].bitmap; + + for (x = 0; x < bitmap->w; x++) + for (y = 0; y < bitmap->h; y++) + { + if (((((short *)bitmap->line[y])[x] != transparentPixel) || (!dirInfo[image[recognized[recognizedID].imageID].dirID].optAllowTransparency)) && + (((short *)bitmap->line[y])[x] == ((short *)screenShot->line[posY+y])[posX+x])) + ((short *)screenShot->line[posY+y])[posX+x] = screenShotTransparentPixel; + } } int cutImageFromDEBUGScreenShot(int recognizedID, int posX, int posY) { - register int x, y, c, i; - register int transparentPixel = makecol16(0, 0, 0); - register int randomColour = makecol16(128+rand()%128, 128+rand()%128, 128+rand()%128); - char buf[100]; - BITMAP *bitmap = image[recognized[recognizedID].imageID].bitmap; - - if (DEBUGY < 300) - { - textprintf(DEBUGScreenShot, font, 320+2, 30+DEBUGY, makecol(255, 255, 255), "%s", image[recognized[recognizedID].imageID].filePath); - textprintf(DEBUGScreenShot, font, 320+2+MIN(bitmap->w, 50)+1, 30+9+DEBUGY, makecol(255, 255, 255), "P(%d;%d) D%c L%d", recognized[recognizedID].posX, recognized[recognizedID].posY, image[recognized[recognizedID].imageID].direction, recognized[recognizedID].layer); - textprintf(DEBUGScreenShot, font, 320+2+MIN(bitmap->w, 50)+1, 30+18+DEBUGY, makecol(255, 255, 255), "D(%d;%d) Per:%d%%", bitmap->w, bitmap->h, (recognized[recognizedID].goodPixels*100)/recognized[recognizedID].pixelsNumber); - } - - for (x = 0; x < bitmap->w; x++) - for (y = 0; y < bitmap->h; y++) - { - if (((((short *)bitmap->line[y])[x] != transparentPixel) || (!dirInfo[image[recognized[recognizedID].imageID].dirID].optAllowTransparency)) && - (((short *)bitmap->line[y])[x] == ((short *)screenShot->line[posY+y])[posX+x])) - { - ((short *)DEBUGScreenShot->line[posY+y])[posX+x] = randomColour; - } - if ((((short *)bitmap->line[y])[x] != transparentPixel) && (DEBUGY < 300) && (x < 50)) - ((short *)DEBUGScreenShot->line[40+DEBUGY+y])[320+2+x] = randomColour; - } - DEBUGY += MAX(bitmap->h+20, 40); + register int x, y, c, i; + register int transparentPixel = makecol16(0, 0, 0); + register int randomColour = makecol16(128+rand()%128, 128+rand()%128, 128+rand()%128); + char buf[100]; + BITMAP *bitmap = image[recognized[recognizedID].imageID].bitmap; + + if (DEBUGY < 300) + { + textprintf(DEBUGScreenShot, font, 320+2, 30+DEBUGY, makecol(255, 255, 255), "%s", image[recognized[recognizedID].imageID].filePath); + textprintf(DEBUGScreenShot, font, 320+2+MIN(bitmap->w, 50)+1, 30+9+DEBUGY, makecol(255, 255, 255), "P(%d;%d) D%c L%d", recognized[recognizedID].posX, recognized[recognizedID].posY, image[recognized[recognizedID].imageID].direction, recognized[recognizedID].layer); + textprintf(DEBUGScreenShot, font, 320+2+MIN(bitmap->w, 50)+1, 30+18+DEBUGY, makecol(255, 255, 255), "D(%d;%d) Per:%d%%", bitmap->w, bitmap->h, (recognized[recognizedID].goodPixels*100)/recognized[recognizedID].pixelsNumber); + } + + for (x = 0; x < bitmap->w; x++) + for (y = 0; y < bitmap->h; y++) + { + if (((((short *)bitmap->line[y])[x] != transparentPixel) || (!dirInfo[image[recognized[recognizedID].imageID].dirID].optAllowTransparency)) && + (((short *)bitmap->line[y])[x] == ((short *)screenShot->line[posY+y])[posX+x])) + { + ((short *)DEBUGScreenShot->line[posY+y])[posX+x] = randomColour; + } + if ((((short *)bitmap->line[y])[x] != transparentPixel) && (DEBUGY < 300) && (x < 50)) + ((short *)DEBUGScreenShot->line[40+DEBUGY+y])[320+2+x] = randomColour; + } + DEBUGY += MAX(bitmap->h+20, 40); } int findImageOnScreenShot(int imageID) { - int numberOfRecognizedImages = 0; - register int x, y; - int i, alreadyExist; - int posX, posY; - int tmp; - short transparentPixel = makecol(0, 0, 0); - register short screenShotTransparentPixel = makecol(255, 0, 255); - BITMAP *bitmapToFind = image[imageID].bitmap; - int cp1x; int cp1y; short cp1c; - int cp2x; int cp2y; short cp2c; - int cp3x; int cp3y; short cp3c; - int cp4x; int cp4y; short cp4c; - int cp5x; int cp5y; short cp5c; - int newRecognized; - - do - { - cp1x = rand()%(bitmapToFind->w); - cp1y = rand()%(bitmapToFind->h); - cp1c = ((short *)bitmapToFind->line[cp1y])[cp1x]; - } while (cp1c == transparentPixel); - do - { - cp2x = rand()%(bitmapToFind->w); - cp2y = rand()%(bitmapToFind->h); - cp2c = ((short *)bitmapToFind->line[cp2y])[cp2x]; - } while (cp2c == transparentPixel); - do - { - cp3x = rand()%(bitmapToFind->w); - cp3y = rand()%(bitmapToFind->h); - cp3c = ((short *)bitmapToFind->line[cp3y])[cp3x]; - } while (cp3c == transparentPixel); - do - { - cp4x = rand()%(bitmapToFind->w); - cp4y = rand()%(bitmapToFind->h); - cp4c = ((short *)bitmapToFind->line[cp4y])[cp4x]; - } while (cp4c == transparentPixel); - do - { - cp5x = rand()%(bitmapToFind->w); - cp5y = rand()%(bitmapToFind->h); - cp5c = ((short *)bitmapToFind->line[cp5y])[cp5x]; - } while (cp5c == transparentPixel); + int numberOfRecognizedImages = 0; + register int x, y; + int i, alreadyExist; + int posX, posY; + int tmp; + short transparentPixel = makecol(0, 0, 0); + register short screenShotTransparentPixel = makecol(255, 0, 255); + BITMAP *bitmapToFind = image[imageID].bitmap; + int cp1x; int cp1y; short cp1c; + int cp2x; int cp2y; short cp2c; + int cp3x; int cp3y; short cp3c; + int cp4x; int cp4y; short cp4c; + int cp5x; int cp5y; short cp5c; + int newRecognized; + + do + { + cp1x = rand()%(bitmapToFind->w); + cp1y = rand()%(bitmapToFind->h); + cp1c = ((short *)bitmapToFind->line[cp1y])[cp1x]; + } while (cp1c == transparentPixel); + do + { + cp2x = rand()%(bitmapToFind->w); + cp2y = rand()%(bitmapToFind->h); + cp2c = ((short *)bitmapToFind->line[cp2y])[cp2x]; + } while (cp2c == transparentPixel); + do + { + cp3x = rand()%(bitmapToFind->w); + cp3y = rand()%(bitmapToFind->h); + cp3c = ((short *)bitmapToFind->line[cp3y])[cp3x]; + } while (cp3c == transparentPixel); + do + { + cp4x = rand()%(bitmapToFind->w); + cp4y = rand()%(bitmapToFind->h); + cp4c = ((short *)bitmapToFind->line[cp4y])[cp4x]; + } while (cp4c == transparentPixel); + do + { + cp5x = rand()%(bitmapToFind->w); + cp5y = rand()%(bitmapToFind->h); + cp5c = ((short *)bitmapToFind->line[cp5y])[cp5x]; + } while (cp5c == transparentPixel); #if 0 /* this code below will probably not be used, but maybe ... ;) */ - for (i = 0; i < NUMBER_OF_CONTROL_PIXELS; i++) - { - do - { - controlPixel[i].posX = rand()%(bitmapToFind->w); - controlPixel[i].posY = rand()%(bitmapToFind->h); - controlPixel[i].color = ((short *)bitmapToFind->line[controlPixel[i].posY])[controlPixel[i].posX]; - } - while (controlPixel[i].color == transparentPixel); - } + for (i = 0; i < NUMBER_OF_CONTROL_PIXELS; i++) + { + do + { + controlPixel[i].posX = rand()%(bitmapToFind->w); + controlPixel[i].posY = rand()%(bitmapToFind->h); + controlPixel[i].color = ((short *)bitmapToFind->line[controlPixel[i].posY])[controlPixel[i].posX]; + } + while (controlPixel[i].color == transparentPixel); + } #endif - for (x = 0; x < 321-(bitmapToFind->w); x++) - for (y = 0; y < 201-(bitmapToFind->h); y++) - { - if ((((short *)screenShot->line[y+cp1y])[x+cp1x] != cp1c) && - (((short *)transparentScreenShot->line[y+cp1y])[x+cp1x] != screenShotTransparentPixel)) - continue; - if ((((short *)screenShot->line[y+cp2y])[x+cp2x] != cp2c) && - (((short *)transparentScreenShot->line[y+cp2y])[x+cp2x] != screenShotTransparentPixel)) - continue; - if ((((short *)screenShot->line[y+cp3y])[x+cp3x] != cp3c) && - (((short *)transparentScreenShot->line[y+cp3y])[x+cp3x] != screenShotTransparentPixel)) - continue; - if ((((short *)screenShot->line[y+cp4y])[x+cp4x] != cp4c) && - (((short *)transparentScreenShot->line[y+cp4y])[x+cp4x] != screenShotTransparentPixel)) - continue; - if ((((short *)screenShot->line[y+cp5y])[x+cp5x] != cp5c) && - (((short *)transparentScreenShot->line[y+cp5y])[x+cp5x] != screenShotTransparentPixel)) - continue; + for (x = 0; x < 321-(bitmapToFind->w); x++) + for (y = 0; y < 201-(bitmapToFind->h); y++) + { + if ((((short *)screenShot->line[y+cp1y])[x+cp1x] != cp1c) && + (((short *)transparentScreenShot->line[y+cp1y])[x+cp1x] != screenShotTransparentPixel)) + continue; + if ((((short *)screenShot->line[y+cp2y])[x+cp2x] != cp2c) && + (((short *)transparentScreenShot->line[y+cp2y])[x+cp2x] != screenShotTransparentPixel)) + continue; + if ((((short *)screenShot->line[y+cp3y])[x+cp3x] != cp3c) && + (((short *)transparentScreenShot->line[y+cp3y])[x+cp3x] != screenShotTransparentPixel)) + continue; + if ((((short *)screenShot->line[y+cp4y])[x+cp4x] != cp4c) && + (((short *)transparentScreenShot->line[y+cp4y])[x+cp4x] != screenShotTransparentPixel)) + continue; + if ((((short *)screenShot->line[y+cp5y])[x+cp5x] != cp5c) && + (((short *)transparentScreenShot->line[y+cp5y])[x+cp5x] != screenShotTransparentPixel)) + continue; #if 0 - for (i = 0; i < NUMBER_OF_CONTROL_PIXELS; i++) - { - if ((((short *)actualScreenShot->line[y + controlPixel[i].posY])[x + controlPixel[i].posX] == controlPixel[i].color) || - (((short *)actualScreenShot->line[y + controlPixel[i].posY])[x + controlPixel[i].posX] == screenShotTransparentPixel)) - } + for (i = 0; i < NUMBER_OF_CONTROL_PIXELS; i++) + { + if ((((short *)actualScreenShot->line[y + controlPixel[i].posY])[x + controlPixel[i].posX] == controlPixel[i].color) || + (((short *)actualScreenShot->line[y + controlPixel[i].posY])[x + controlPixel[i].posX] == screenShotTransparentPixel)) + } #endif - posX = x; - posY = y; - - tmp = findImageOnScreenShotInPosition(imageID, posX, posY); - - if (tmp) - { - newRecognized = recognizedNumber; - alreadyExist = 0; - for (i = 0; i < recognizedNumber; i++) - { - if ((recognized[i].imageID == imageID) && - (recognized[i].posX == posX) && - (recognized[i].posY == posY)) - { - alreadyExist = 1; - newRecognized = i; - break; - } - } - recognized[newRecognized].imageID = imageID; - recognized[newRecognized].posX = posX; - recognized[newRecognized].posY = posY; - recognized[newRecognized].goodPixels = tmp; - recognized[newRecognized].goodPixels2 = tmp; - recognized[newRecognized].goodPixelsPercent = (recognized[newRecognized].goodPixels*65536)/image[recognized[newRecognized].imageID].pixelsNumber; - recognized[newRecognized].ownedPixels = 0; - recognized[newRecognized].pixelsNumber = image[imageID].pixelsNumber; - if (!alreadyExist) - { - recognized[newRecognized].layer = actualLayer; - recognizedNumber++; - dirInfo[image[imageID].dirID].recognizedNumber++; - } - if (dirInfo[image[imageID].dirID].recognizedNumber > dirInfo[image[imageID].dirID].optMaxRecognizedNumber) - return 0; - if (recognizedNumber > 99900) - { - printf("[ERROR] %s\n", "Too many recognized items!"); - while (1) {}; - } - } - } - return numberOfRecognizedImages; + posX = x; + posY = y; + + tmp = findImageOnScreenShotInPosition(imageID, posX, posY); + + if (tmp) + { + newRecognized = recognizedNumber; + alreadyExist = 0; + for (i = 0; i < recognizedNumber; i++) + { + if ((recognized[i].imageID == imageID) && + (recognized[i].posX == posX) && + (recognized[i].posY == posY)) + { + alreadyExist = 1; + newRecognized = i; + break; + } + } + recognized[newRecognized].imageID = imageID; + recognized[newRecognized].posX = posX; + recognized[newRecognized].posY = posY; + recognized[newRecognized].goodPixels = tmp; + recognized[newRecognized].goodPixels2 = tmp; + recognized[newRecognized].goodPixelsPercent = (recognized[newRecognized].goodPixels*65536)/image[recognized[newRecognized].imageID].pixelsNumber; + recognized[newRecognized].ownedPixels = 0; + recognized[newRecognized].pixelsNumber = image[imageID].pixelsNumber; + if (!alreadyExist) + { + recognized[newRecognized].layer = actualLayer; + recognizedNumber++; + dirInfo[image[imageID].dirID].recognizedNumber++; + } + if (dirInfo[image[imageID].dirID].recognizedNumber > dirInfo[image[imageID].dirID].optMaxRecognizedNumber) + return 0; + if (recognizedNumber > 99900) + { + printf("[ERROR] %s\n", "Too many recognized items!"); + while (1) {}; + } + } + } + return numberOfRecognizedImages; } void recognizeScreenShot(int screenShotID) { - char buf[100]; - int x, y; - int i, j, k, l, tmp; - short transparentPixel = makecol(0, 0, 0); - register short screenShotTransparentPixel = makecol(255, 0, 255); - int maxPixelsNumber, maxPixelsID, maxTotalPixelsNumber; - int recognizedNow, recognizedBefore; - int timeBefore, timeAfter; + char buf[100]; + int x, y; + int i, j, k, l, tmp; + short transparentPixel = makecol(0, 0, 0); + register short screenShotTransparentPixel = makecol(255, 0, 255); + int maxPixelsNumber, maxPixelsID, maxTotalPixelsNumber; + int recognizedNow, recognizedBefore; + int timeBefore, timeAfter; - timeBefore = time(0); - printf("\nRecognizing %s (shot %d of %d)\n", screenShotList[screenShotID].fileName, screenShotID+1, screenShotsNumber); - fprintf(outputFile, "\nRecognizing %s (shot %d of %d)\n", screenShotList[screenShotID].fileName, screenShotID+1, screenShotsNumber); - fprintf(outputSmallFile, "Recognizing %s\n", screenShotList[screenShotID].fileName); - sprintf(buf, "%s" SEPS "%s", screenShotsDir, screenShotList[screenShotID].fileName); - screenShot = load_bmp(buf, 0); - transparentScreenShot = load_bmp(buf, 0); - DEBUGScreenShot2 = load_bmp(buf, 0); - DEBUGScreenShot = create_bitmap(500, 400); - clear_to_color(DEBUGScreenShot, makecol(50, 50, 50)); - blit(screenShot, DEBUGScreenShot, 0, 0, 0, 200, 320, 200); - line(DEBUGScreenShot, 320, 0, 320, 399, makecol(255, 255, 255)); - totalNumberOfRecognizedImages = 0; - recognizedNumber = 0; - actualLayer = 0; - DEBUGY = 0; - for (i = 0; i < 320; i++) - for (j = 0; j < 200; j++) - { - recognizeMap[i][j] = -1; - } - for (i = 0; i < dirsNumber; i++) - dirInfo[i].recognizedNumber = 0; - - recognizedNow = 0; - do - { - recognizedBefore = recognizedNow; - recognizedNow = 0; - if ((optMaxLayers != 0) && (actualLayer+1 > optMaxLayers)) break; - - recognizedBefore = recognizedNumber; - printf(" Checking layer nr. %d ... ", actualLayer); - - for (x = 0; x < imagesNumber; x++) - { - if (dirInfo[image[x].dirID].recognizedNumber < dirInfo[image[x].dirID].optMaxRecognizedNumber) - printf("\b\b\b\b% 3d%%", (x*100)/imagesNumber); - findImageOnScreenShot(x); - } - printf("\b\b\b\bDone\n"); - - for (i = 0; i < 320; i++) - for (j = 0; j < 200; j++) - { - recognizeMap[i][j] = -1; - } - - for (i = 0; i < recognizedNumber; i++) - { - cutImageFromTransparentScreenShot(i, recognized[i].posX, recognized[i].posY); - putImageOnRecognizeMap(image[recognized[i].imageID].bitmap, recognized[i].posX, recognized[i].posY, i); - } - - recognizedNow = recognizedNumber; - - actualLayer++; - } - while (recognizedNow - recognizedBefore != 0); - - for (i = 0; i < recognizedNumber; i++) - recognized[i].ownedPixels = 0; - - for (;;) - { - maxPixelsNumber = 0; - maxTotalPixelsNumber = 0; - maxPixelsID = -1; - for (i = 0; i < recognizedNumber; i++) - { - if (recognized[i].goodPixels > maxPixelsNumber) - { - maxTotalPixelsNumber = 0; - maxPixelsNumber = recognized[i].goodPixels; - maxPixelsID = i; - } - else - if ((recognized[i].goodPixels == maxPixelsNumber) && (recognized[i].pixelsNumber > maxTotalPixelsNumber)) - { - maxTotalPixelsNumber = recognized[i].pixelsNumber; - maxPixelsID = i; - } - } - if (maxPixelsNumber != 0) - { - tmp = findImageOnScreenShotInPosition(recognized[maxPixelsID].imageID, recognized[maxPixelsID].posX, recognized[maxPixelsID].posY); - if (recognized[maxPixelsID].goodPixels != tmp) - { - recognized[maxPixelsID].goodPixels = tmp; - continue; - } - totalNumberOfRecognizedImages++; - cutImageFromDEBUGScreenShot(maxPixelsID, recognized[maxPixelsID].posX, recognized[maxPixelsID].posY); - cutImageFromScreenShot(maxPixelsID, recognized[maxPixelsID].posX, recognized[maxPixelsID].posY); - putImageOnRecognizeMap(image[recognized[maxPixelsID].imageID].bitmap, recognized[maxPixelsID].posX, recognized[maxPixelsID].posY, maxPixelsID); - printf(" Found %s d%c l%d x%d y%d (OW:%d / GO:%d / TO:%d)\n", image[recognized[maxPixelsID].imageID].filePath, image[recognized[maxPixelsID].imageID].direction, recognized[maxPixelsID].layer, recognized[maxPixelsID].posX, recognized[maxPixelsID].posY, recognized[maxPixelsID].ownedPixels, recognized[maxPixelsID].goodPixels, image[recognized[maxPixelsID].imageID].pixelsNumber); - fprintf(outputFile, " Found %s d%c l%d x%d y%d (OW:%d / GO:%d / TO:%d)\n", image[recognized[maxPixelsID].imageID].filePath, image[recognized[maxPixelsID].imageID].direction, recognized[maxPixelsID].layer, recognized[maxPixelsID].posX, recognized[maxPixelsID].posY, recognized[maxPixelsID].ownedPixels, recognized[maxPixelsID].goodPixels, image[recognized[maxPixelsID].imageID].pixelsNumber); - fprintf(outputSmallFile, "Found %s d %c l %d x %d y %d w %d h %d fpx %d fpy %d\n", image[recognized[maxPixelsID].imageID].filePath, image[recognized[maxPixelsID].imageID].direction, recognized[maxPixelsID].layer, recognized[maxPixelsID].posX, recognized[maxPixelsID].posY, image[recognized[maxPixelsID].imageID].bitmap->w, image[recognized[maxPixelsID].imageID].bitmap->h, (image[recognized[maxPixelsID].imageID].direction == '-') ? (recognized[maxPixelsID].posX) : (recognized[maxPixelsID].posX+image[recognized[maxPixelsID].imageID].bitmap->w), recognized[maxPixelsID].posY+image[recognized[maxPixelsID].imageID].bitmap->h); - recognized[maxPixelsID].goodPixels = 0; - } - else - break; - } - - textprintf(DEBUGScreenShot, font, 320+2, 1, makecol(255, 255, 255), "%s (%d/%d)", screenShotList[screenShotID], screenShotID+1, screenShotsNumber); - textprintf(DEBUGScreenShot, font, 320+2, 10, makecol(255, 255, 255), "Recognized number: %d", totalNumberOfRecognizedImages); - sprintf(buf, "%s/trans_%s.bmp", optResultsDir, screenShotList[screenShotID]); - save_bitmap(buf, transparentScreenShot, 0); - sprintf(buf, "%s/rec_%s.bmp", optResultsDir, screenShotList[screenShotID]); - save_bitmap(buf, DEBUGScreenShot, 0); - timeAfter = time(0); - - printf(" %d recognized images in about %d seconds\n", totalNumberOfRecognizedImages, timeAfter - timeBefore); - fprintf(outputFile, " %d recognized images in about %d seconds\n", totalNumberOfRecognizedImages, timeAfter - timeBefore); - destroy_bitmap(screenShot); - destroy_bitmap(transparentScreenShot); - destroy_bitmap(DEBUGScreenShot); - destroy_bitmap(DEBUGScreenShot2); + timeBefore = time(0); + printf("\nRecognizing %s (shot %d of %d)\n", screenShotList[screenShotID].fileName, screenShotID+1, screenShotsNumber); + fprintf(outputFile, "\nRecognizing %s (shot %d of %d)\n", screenShotList[screenShotID].fileName, screenShotID+1, screenShotsNumber); + fprintf(outputSmallFile, "Recognizing %s\n", screenShotList[screenShotID].fileName); + sprintf(buf, "%s" SEPS "%s", screenShotsDir, screenShotList[screenShotID].fileName); + screenShot = load_bmp(buf, 0); + transparentScreenShot = load_bmp(buf, 0); + DEBUGScreenShot2 = load_bmp(buf, 0); + DEBUGScreenShot = create_bitmap(500, 400); + clear_to_color(DEBUGScreenShot, makecol(50, 50, 50)); + blit(screenShot, DEBUGScreenShot, 0, 0, 0, 200, 320, 200); + line(DEBUGScreenShot, 320, 0, 320, 399, makecol(255, 255, 255)); + totalNumberOfRecognizedImages = 0; + recognizedNumber = 0; + actualLayer = 0; + DEBUGY = 0; + for (i = 0; i < 320; i++) + for (j = 0; j < 200; j++) + { + recognizeMap[i][j] = -1; + } + for (i = 0; i < dirsNumber; i++) + dirInfo[i].recognizedNumber = 0; + + recognizedNow = 0; + do + { + recognizedBefore = recognizedNow; + recognizedNow = 0; + if ((optMaxLayers != 0) && (actualLayer+1 > optMaxLayers)) break; + + recognizedBefore = recognizedNumber; + printf(" Checking layer nr. %d ... ", actualLayer); + + for (x = 0; x < imagesNumber; x++) + { + if (dirInfo[image[x].dirID].recognizedNumber < dirInfo[image[x].dirID].optMaxRecognizedNumber) + printf("\b\b\b\b% 3d%%", (x*100)/imagesNumber); + findImageOnScreenShot(x); + } + printf("\b\b\b\bDone\n"); + + for (i = 0; i < 320; i++) + for (j = 0; j < 200; j++) + { + recognizeMap[i][j] = -1; + } + + for (i = 0; i < recognizedNumber; i++) + { + cutImageFromTransparentScreenShot(i, recognized[i].posX, recognized[i].posY); + putImageOnRecognizeMap(image[recognized[i].imageID].bitmap, recognized[i].posX, recognized[i].posY, i); + } + + recognizedNow = recognizedNumber; + + actualLayer++; + } + while (recognizedNow - recognizedBefore != 0); + + for (i = 0; i < recognizedNumber; i++) + recognized[i].ownedPixels = 0; + + for (;;) + { + maxPixelsNumber = 0; + maxTotalPixelsNumber = 0; + maxPixelsID = -1; + for (i = 0; i < recognizedNumber; i++) + { + if (recognized[i].goodPixels > maxPixelsNumber) + { + maxTotalPixelsNumber = 0; + maxPixelsNumber = recognized[i].goodPixels; + maxPixelsID = i; + } + else + if ((recognized[i].goodPixels == maxPixelsNumber) && (recognized[i].pixelsNumber > maxTotalPixelsNumber)) + { + maxTotalPixelsNumber = recognized[i].pixelsNumber; + maxPixelsID = i; + } + } + if (maxPixelsNumber != 0) + { + tmp = findImageOnScreenShotInPosition(recognized[maxPixelsID].imageID, recognized[maxPixelsID].posX, recognized[maxPixelsID].posY); + if (recognized[maxPixelsID].goodPixels != tmp) + { + recognized[maxPixelsID].goodPixels = tmp; + continue; + } + totalNumberOfRecognizedImages++; + cutImageFromDEBUGScreenShot(maxPixelsID, recognized[maxPixelsID].posX, recognized[maxPixelsID].posY); + cutImageFromScreenShot(maxPixelsID, recognized[maxPixelsID].posX, recognized[maxPixelsID].posY); + putImageOnRecognizeMap(image[recognized[maxPixelsID].imageID].bitmap, recognized[maxPixelsID].posX, recognized[maxPixelsID].posY, maxPixelsID); + + /* Print results */ + printf(" Found %s d%c l%d x%d y%d (OW:%d / GO:%d / TO:%d)\n", + image[recognized[maxPixelsID].imageID].filePath, + image[recognized[maxPixelsID].imageID].direction, + recognized[maxPixelsID].layer, + recognized[maxPixelsID].posX, + recognized[maxPixelsID].posY, + recognized[maxPixelsID].ownedPixels, + recognized[maxPixelsID].goodPixels, + image[recognized[maxPixelsID].imageID].pixelsNumber + ); + fprintf(outputFile, " Found %s d%c l%d x%d y%d (OW:%d / GO:%d / TO:%d)\n", + image[recognized[maxPixelsID].imageID].filePath, + image[recognized[maxPixelsID].imageID].direction, + recognized[maxPixelsID].layer, + recognized[maxPixelsID].posX, + recognized[maxPixelsID].posY, + recognized[maxPixelsID].ownedPixels, + recognized[maxPixelsID].goodPixels, + image[recognized[maxPixelsID].imageID].pixelsNumber + ); + fprintf(outputSmallFile, "Found %s d %c l %d x %d y %d w %d h %d fpx %d fpy %d\n", + image[recognized[maxPixelsID].imageID].filePath, + image[recognized[maxPixelsID].imageID].direction, + recognized[maxPixelsID].layer, + recognized[maxPixelsID].posX, + recognized[maxPixelsID].posY, + image[recognized[maxPixelsID].imageID].bitmap->w, + image[recognized[maxPixelsID].imageID].bitmap->h, + (image[recognized[maxPixelsID].imageID].direction=='-') ? + (recognized[maxPixelsID].posX) : + (recognized[maxPixelsID].posX+image[recognized[maxPixelsID].imageID].bitmap->w + ), + recognized[maxPixelsID].posY+image[recognized[maxPixelsID].imageID].bitmap->h + ); + recognized[maxPixelsID].goodPixels = 0; + } + else + break; + } + + textprintf(DEBUGScreenShot, font, 320+2, 1, makecol(255, 255, 255), "%s (%d/%d)", screenShotList[screenShotID], screenShotID+1, screenShotsNumber); + textprintf(DEBUGScreenShot, font, 320+2, 10, makecol(255, 255, 255), "Recognized number: %d", totalNumberOfRecognizedImages); + sprintf(buf, "%s/trans_%s.bmp", optResultsDir, screenShotList[screenShotID]); + save_bitmap(buf, transparentScreenShot, 0); + sprintf(buf, "%s/rec_%s.bmp", optResultsDir, screenShotList[screenShotID]); + save_bitmap(buf, DEBUGScreenShot, 0); + timeAfter = time(0); + + printf(" %d recognized images in about %d seconds\n", totalNumberOfRecognizedImages, timeAfter - timeBefore); + fprintf(outputFile, " %d recognized images in about %d seconds\n", totalNumberOfRecognizedImages, timeAfter - timeBefore); + destroy_bitmap(screenShot); + destroy_bitmap(transparentScreenShot); + destroy_bitmap(DEBUGScreenShot); + destroy_bitmap(DEBUGScreenShot2); } void sortListOfScreenShots() { - char *tmpScreenShotsList[MAX_SCREENSHOTS]; - int tmpScreenShotsNumber; - int i; - DIR *dir = opendir(screenShotsDir); - struct dirent *file; - - tmpScreenShotsNumber = 0; - while (file = readdir(dir)) - { - if (match("*.bmp", file->d_name) || match("*.BMP", file->d_name)) - { - tmpScreenShotsList[tmpScreenShotsNumber] = (char *) malloc(strlen(file->d_name)+1); - strcpy(tmpScreenShotsList[tmpScreenShotsNumber], file->d_name); - tmpScreenShotsNumber++; - } - } - closedir(dir); - - qsort(tmpScreenShotsList, tmpScreenShotsNumber, sizeof(char *), pstrcmp); - - for (i = 0; i < tmpScreenShotsNumber; i++) - { - screenShotList[i].fileName = tmpScreenShotsList[i]; - } - screenShotsNumber = tmpScreenShotsNumber; + char *tmpScreenShotsList[MAX_SCREENSHOTS]; + int tmpScreenShotsNumber; + int i; + DIR *dir = opendir(screenShotsDir); + struct dirent *file; + + tmpScreenShotsNumber = 0; + while (file = readdir(dir)) + { + if (match("*.bmp", file->d_name) || match("*.BMP", file->d_name)) + { + tmpScreenShotsList[tmpScreenShotsNumber] = (char *) malloc(strlen(file->d_name)+1); + strcpy(tmpScreenShotsList[tmpScreenShotsNumber], file->d_name); + tmpScreenShotsNumber++; + } + } + closedir(dir); + + qsort(tmpScreenShotsList, tmpScreenShotsNumber, sizeof(char *), pstrcmp); + + for (i = 0; i < tmpScreenShotsNumber; i++) + { + screenShotList[i].fileName = tmpScreenShotsList[i]; + } + screenShotsNumber = tmpScreenShotsNumber; } void freeListOfScreenShots() { - int i; - - for (i = 0; i < screenShotsNumber; i++) - { - free(screenShotList[i].fileName); - } - screenShotsNumber = 0; + int i; + + for (i = 0; i < screenShotsNumber; i++) + { + free(screenShotList[i].fileName); + } + screenShotsNumber = 0; } int countPixels(BITMAP *bitmap) { - int x, y, number; - int transparentPixel = makecol16(0, 0, 0); - - number = 0; - for (x = 0; x < bitmap->w; x++) - for (y = 0; y < bitmap->h; y++) - { - if (((short *)bitmap->line[y])[x] != transparentPixel) - number++; - } - return number; + int x, y, number; + int transparentPixel = makecol16(0, 0, 0); + + number = 0; + for (x = 0; x < bitmap->w; x++) + for (y = 0; y < bitmap->h; y++) + { + if (((short *)bitmap->line[y])[x] != transparentPixel) + number++; + } + return number; } void readDir(int dirID) { - char buf[100]; - int i; - DIR *dir = opendir(dirInfo[dirID].dirName); - struct dirent *file; - FILE *optFile; - - sprintf(buf, "%s" SEPS "bitmaps.cfg", dirInfo[dirID].dirName); - if (optFile = fopen(buf, "r")) - { - fscanf(optFile, "%d", &dirInfo[dirID].optTwoDirections); - fscanf(optFile, "%d", &dirInfo[dirID].optMaxRecognizedNumber); - fscanf(optFile, "%d", &dirInfo[dirID].optMinImagePercent); - fscanf(optFile, "%d", &dirInfo[dirID].optAllowTransparency); - fclose(optFile); - } - else - { - dirInfo[dirID].optTwoDirections = 1; - dirInfo[dirID].optMaxRecognizedNumber = 99999; - dirInfo[dirID].optMinImagePercent = 30; - dirInfo[dirID].optAllowTransparency = 1; - } - - while (file = readdir(dir)) - { - if (match("*.bmp", file->d_name) || match("*.BMP", file->d_name)) - { - strcpy(buf, dirInfo[dirID].dirName); - strcat(buf, SEPS); - strcat(buf, file->d_name); - image[imagesNumber].filePath = (char *) malloc(strlen(buf)+1); - strcpy(image[imagesNumber].filePath, buf); - image[imagesNumber].bitmap = load_bmp(image[imagesNumber].filePath, 0); - image[imagesNumber].direction = '-'; - image[imagesNumber].pixelsNumber = countPixels(image[imagesNumber].bitmap); - image[imagesNumber].dirID = dirID; - imagesNumber++; - - if (dirInfo[dirID].optTwoDirections) - { - image[imagesNumber].filePath = (char *) malloc(strlen(buf)+1); - strcpy(image[imagesNumber].filePath, buf); - image[imagesNumber].bitmap = create_bitmap(image[imagesNumber-1].bitmap->w, image[imagesNumber-1].bitmap->h); - draw_sprite_h_flip(image[imagesNumber].bitmap, image[imagesNumber-1].bitmap, 0, 0); - image[imagesNumber].direction = 'M'; - image[imagesNumber].pixelsNumber = image[imagesNumber-1].pixelsNumber; - image[imagesNumber].dirID = dirID; - imagesNumber++; - } - } - } - closedir(dir); + char buf[100]; + int i; + DIR *dir = opendir(dirInfo[dirID].dirName); + struct dirent *file; + FILE *optFile; + + sprintf(buf, "%s" SEPS "bitmaps.cfg", dirInfo[dirID].dirName); + if (optFile = fopen(buf, "r")) + { + fscanf(optFile, "%d", &dirInfo[dirID].optTwoDirections); + fscanf(optFile, "%d", &dirInfo[dirID].optMaxRecognizedNumber); + fscanf(optFile, "%d", &dirInfo[dirID].optMinImagePercent); + fscanf(optFile, "%d", &dirInfo[dirID].optAllowTransparency); + fclose(optFile); + } + else + { + dirInfo[dirID].optTwoDirections = 1; + dirInfo[dirID].optMaxRecognizedNumber = 99999; + dirInfo[dirID].optMinImagePercent = 30; + dirInfo[dirID].optAllowTransparency = 1; + } + + while (file = readdir(dir)) + { + if (match("*.bmp", file->d_name) || match("*.BMP", file->d_name)) + { + strcpy(buf, dirInfo[dirID].dirName); + strcat(buf, SEPS); + strcat(buf, file->d_name); + image[imagesNumber].filePath = (char *) malloc(strlen(buf)+1); + strcpy(image[imagesNumber].filePath, buf); + image[imagesNumber].bitmap = load_bmp(image[imagesNumber].filePath, 0); + image[imagesNumber].direction = '-'; + image[imagesNumber].pixelsNumber = countPixels(image[imagesNumber].bitmap); + image[imagesNumber].dirID = dirID; + imagesNumber++; + + if (dirInfo[dirID].optTwoDirections) + { + image[imagesNumber].filePath = (char *) malloc(strlen(buf)+1); + strcpy(image[imagesNumber].filePath, buf); + image[imagesNumber].bitmap = create_bitmap(image[imagesNumber-1].bitmap->w, image[imagesNumber-1].bitmap->h); + draw_sprite_h_flip(image[imagesNumber].bitmap, image[imagesNumber-1].bitmap, 0, 0); + image[imagesNumber].direction = 'M'; + image[imagesNumber].pixelsNumber = image[imagesNumber-1].pixelsNumber; + image[imagesNumber].dirID = dirID; + imagesNumber++; + } + } + } + closedir(dir); } void sortListOfImages() { - int i; + int i; - qsort(image, imagesNumber, sizeof(tImage), cmptImage); + qsort(image, imagesNumber, sizeof(tImage), cmptImage); } void freeListOfImages() { - int i; - - for (i = 0; i < imagesNumber; i++) - { - free(image[i].filePath); - destroy_bitmap(image[i].bitmap); - } - imagesNumber = 0; + int i; + + for (i = 0; i < imagesNumber; i++) + { + free(image[i].filePath); + destroy_bitmap(image[i].bitmap); + } + imagesNumber = 0; } void readParameters() { - int i; - - strcpy(output, POPRECOG_ABOUT); - printf(output); - - printf("Step 1. Type dir where screenshots are stored.\nThis should be 320x200 bmp files with 256 colour palette.\n"); - strcat(output, "Step 1. Type dir where screenshots are stored.\nThis should be 320x200 bmp files with 256 colour palette.\n"); - scanf("%s", screenShotsDir); - sprintf(output, "%s%s\n", output, screenShotsDir); - - printf("\nStep 2. Type dirs where are stored bitmaps to recognize.\nPlease type in this format: [dirname] [max images on screenshot][ENTER].\nWhen you'll finish type END[ENTER].\n"); - strcat(output, "\nStep 2. Type dirs where are stored bitmaps to recognize.\nPlease type in this format: [dirname] [max images on screenshot][ENTER].\nWhen you'll finish type END[ENTER].\n"); - for (;;) - { - scanf("%s", dirInfo[dirsNumber].dirName); - sprintf(output, "%s%s\n", output, dirInfo[dirsNumber].dirName); - if (strcmp(dirInfo[dirsNumber].dirName, "END")) - { - dirsNumber++; - } - else - break; - } - - printf("\nStep 3. Type number of maximum layers\nHINT:\nIf you'll type 0, poprecog will automatically detect the number of layers,\nbut it is not recommended for tiles, walls, cinematic etc.\n"); - strcat(output, "\nStep 3. Type number of maximum layers\nHINT:\nIf you'll type 0, poprecog will automatically detect the number of layers,\nbut it is not recommended for tiles, walls, cinematic etc.\n"); - scanf("%d", &optMaxLayers); - sprintf(output, "%s%d\n", output, optMaxLayers); - - printf("\nStep 4. Type output dir, in which you'll see recognized things\n"); - strcat(output, "\nStep 4. Type output dir, in which you'll see recognized things\n"); - scanf("%s", optResultsDir); - sprintf(output, "%s%s\n", output, optResultsDir); - - printf("\n"); - strcat(output, "\n"); + int i; + + strcpy(output, POPRECOG_ABOUT); + printf(output); + + printf("Step 1. Type dir where screenshots are stored.\nThis should be 320x200 bmp files with 256 colour palette.\n"); + strcat(output, "Step 1. Type dir where screenshots are stored.\nThis should be 320x200 bmp files with 256 colour palette.\n"); + scanf("%s", screenShotsDir); + sprintf(output, "%s%s\n", output, screenShotsDir); + + printf("\nStep 2. Type dirs where are stored bitmaps to recognize.\nPlease type in this format: [dirname] [max images on screenshot][ENTER].\nWhen you'll finish type END[ENTER].\n"); + strcat(output, "\nStep 2. Type dirs where are stored bitmaps to recognize.\nPlease type in this format: [dirname] [max images on screenshot][ENTER].\nWhen you'll finish type END[ENTER].\n"); + for (;;) + { + scanf("%s", dirInfo[dirsNumber].dirName); + sprintf(output, "%s%s\n", output, dirInfo[dirsNumber].dirName); + if (strcmp(dirInfo[dirsNumber].dirName, "END")) + { + dirsNumber++; + } + else + break; + } + + printf("\nStep 3. Type number of maximum layers\nHINT:\nIf you'll type 0, poprecog will automatically detect the number of layers,\nbut it is not recommended for tiles, walls, cinematic etc.\n"); + strcat(output, "\nStep 3. Type number of maximum layers\nHINT:\nIf you'll type 0, poprecog will automatically detect the number of layers,\nbut it is not recommended for tiles, walls, cinematic etc.\n"); + scanf("%d", &optMaxLayers); + sprintf(output, "%s%d\n", output, optMaxLayers); + + printf("\nStep 4. Type output dir, in which you'll see recognized things\n"); + strcat(output, "\nStep 4. Type output dir, in which you'll see recognized things\n"); + scanf("%s", optResultsDir); + sprintf(output, "%s%s\n", output, optResultsDir); + + printf("\n"); + strcat(output, "\n"); } int main(int argc, char *argv[]) { - int i; - long timeBefore, timeAfter; - char buf[100]; - - srand((unsigned int)time((time_t *)NULL)); - - allegro_init(); - set_color_depth(16); - set_color_conversion(COLORCONV_TOTAL); - - readParameters(); - - defmkdir(optResultsDir); - - sprintf(buf, "%s/%s", optResultsDir, "all_results.txt"); - outputFile = fopen(buf, "a"); + int i; + long timeBefore, timeAfter; + char buf[100]; + + srand((unsigned int)time((time_t *)NULL)); + + allegro_init(); + set_color_depth(16); + set_color_conversion(COLORCONV_TOTAL); + + readParameters(); + + defmkdir(optResultsDir); + + sprintf(buf, "%s/%s", optResultsDir, "all_results.txt"); + outputFile = fopen(buf, "a"); if (!outputFile) - { - printf("Cannot open %s for append\n",buf); - return -1; + { + printf("Cannot open %s for append\n",buf); + return -1; } - sprintf(buf, "%s/%s", optResultsDir, "small_results.txt"); - outputSmallFile = fopen(buf, "a"); + sprintf(buf, "%s/%s", optResultsDir, "small_results.txt"); + outputSmallFile = fopen(buf, "a"); if (!outputFile) - { - printf("Cannot open %s for append\n",buf); - return -1; + { + printf("Cannot open %s for append\n",buf); + return -1; } - fprintf(outputFile, output); - - for (i = 0; i < dirsNumber; i++) - { - printf("Loading bitmaps from dir %s\n", dirInfo[i].dirName); - fprintf(outputFile, "Loading bitmaps from dir %s\n", dirInfo[i].dirName); - readDir(i); - } - sortListOfImages(); - sortListOfScreenShots(); - - timeBefore = time(0); - - for (i = 0; i < screenShotsNumber; i++) - recognizeScreenShot(i); - - timeAfter = time(0); - printf("\nRecognizing %d frames last about %d seconds\n", screenShotsNumber, timeAfter - timeBefore); - fprintf(outputFile, "\nRecognizing %d frames last about %d seconds\n", screenShotsNumber, timeAfter - timeBefore); - - printf("Freeing memory\n"); - fprintf(outputFile, "Freeing memory\n"); - freeListOfScreenShots(); - freeListOfImages(); - - fclose(outputFile); - fclose(outputSmallFile); - - printf("Type sth and press ENTER key...\n"); - scanf("\n"); - - return 0; + fprintf(outputFile, output); + + for (i = 0; i < dirsNumber; i++) + { + printf("Loading bitmaps from dir %s\n", dirInfo[i].dirName); + fprintf(outputFile, "Loading bitmaps from dir %s\n", dirInfo[i].dirName); + readDir(i); + } + sortListOfImages(); + sortListOfScreenShots(); + + timeBefore = time(0); + + for (i = 0; i < screenShotsNumber; i++) + recognizeScreenShot(i); + + timeAfter = time(0); + printf("\nRecognizing %d frames last about %d seconds\n", screenShotsNumber, timeAfter - timeBefore); + fprintf(outputFile, "\nRecognizing %d frames last about %d seconds\n", screenShotsNumber, timeAfter - timeBefore); + + printf("Freeing memory\n"); + fprintf(outputFile, "Freeing memory\n"); + freeListOfScreenShots(); + freeListOfImages(); + + fclose(outputFile); + fclose(outputSmallFile); + + printf("Type sth and press ENTER key...\n"); + scanf("\n"); + + return 0; } END_OF_MAIN();