Change Counter

#ifndef _TEST_APP
#define _TEST_APP
 
#define WHITE_THRESHOLD 244
#define AREA_THESHOLD 50
#define COLOR_THESHOLD 10
 
#include "ofMain.h"
 
class coin {
    public:
        void    load(string imgName);
 
        int     area;
        int     color;
};
 
class change {
    public:
        void    load(string imgName);
        void    seedFill(int x, int y, int tag);
        int     count(coin penny, coin nickel, coin dime, coin quarter);
 
        unsigned char* colorPixels;
        unsigned char* grayPixels;
        unsigned char* tagPixels;
 
        ofImage img;
        int     currTag;
};
 
class testApp : public ofBaseApp {
    public:
        void    setup();
 
        coin    penny;
        coin    nickel;
        coin    dime;
        coin    quarter;
 
        change  inMyPocket;
};
 
#endif

#include "testApp.h"
 
//--------------------------------------------------------------
void coin::load(string imgName) {
    // load the image
    ofImage img;
    img.loadImage(imgName);
    unsigned char* pixels = img.getPixels();
 
    // calculate the area
    area = 0;
    color = 0;
    int numPixels = img.width * img.height;
    for (int i=0; i < numPixels; i++) {
        int pixelAvg = pixels[3*i+2];
        if (pixelAvg < WHITE_THRESHOLD) {
            area++;
            color += pixelAvg;
        }
    }
    color /= area;
 
    printf("[area: %i  color: %i]\n", area, color);
}
 
//--------------------------------------------------------------
void change::load(string imgName) {
    // load the image
    img.loadImage(imgName);
    colorPixels = img.getPixels();
 
    // init the grayscale and the tags
    grayPixels = new unsigned char[img.width*img.height];
    tagPixels = new unsigned char[img.width*img.height];
    for (int i=0; i < img.width * img.height; i++) {
        int pixelAvg = (colorPixels[3*i] + colorPixels[3*i+1] + colorPixels[3*i+2]) / 3;
        grayPixels[i] = pixelAvg;            
        tagPixels[i] = 0;
    }
 
    // find the blobs
    printf("Coin Detection\n");
    currTag = 0;
    for (int i=1; i < img.width-1; i++) {
        for (int j=1; j < img.height-1; j++) {
            if (grayPixels[j*img.width + i] < WHITE_THRESHOLD && tagPixels[j*img.width + i] == 0) {
                currTag++;
                printf("> Found a coin with color %i at (%i, %i)!\n", grayPixels[j*img.width + i], i, j);
                seedFill(i, j, currTag);                
            }
        }
    }
}
 
//--------------------------------------------------------------
void change::seedFill(int x, int y, int tag) {
    if (grayPixels[y*img.width + x] < WHITE_THRESHOLD) {
        tagPixels[y*img.width + x] = tag;
 
        if (y > 0 && tagPixels[(y-1)*img.width + x] == 0)             // north
            seedFill(x, y-1, tag);  
        if (y < img.height-1 && tagPixels[(y+1)*img.width + x] == 0)  // south
            seedFill(x, y+1, tag);
        if (x < img.width-1 && tagPixels[y*img.width + (x+1)] == 0)   // east
            seedFill(x+1, y, tag);
        if (x > 0 && tagPixels[y*img.width + (x-1)] == 0)             // west 
            seedFill(x-1, y, tag);
    }
}
 
//--------------------------------------------------------------
int change::count(coin penny, coin nickel, coin dime, coin quarter) {
    // identify the coins
    int** coinAreas = new int*[currTag];
    for (int i=0; i < currTag; i++) {
        coinAreas[i] = new int[2];
        coinAreas[i][0] = 0;
        coinAreas[i][1] = 0;
    }
    for (int i=0; i < img.width * img.height; i++) {
        if (tagPixels[i] != 0) {
            coinAreas[tagPixels[i]-1][0]++;
            int pixelAvg = colorPixels[3*i+2];
            coinAreas[tagPixels[i]-1][1] += pixelAvg;
        }
    }    
    for (int i=0; i < currTag; i++) {
        coinAreas[i][1] /= coinAreas[i][0];
    }
 
    // count the value of all the change
    printf("Coin Identification\n");
    int value = 0;
    for (int i=0; i < currTag; i++) {
        if (ABS(coinAreas[i][0] - quarter.area) < AREA_THESHOLD) {
            printf("> Found a quarter");
            value += 25;
        }
        else if (ABS(coinAreas[i][0] - dime.area) < AREA_THESHOLD) {
            printf("> Found a dime");
            value += 10;
        }
        else if (ABS(coinAreas[i][1] - penny.color) < COLOR_THESHOLD) {
            printf("> Found a penny");
            value += 1;
        } 
        else {
            printf("> Found a nickel");
            value += 5;
        }
        printf(" [%i - %i]\n", coinAreas[i][0], coinAreas[i][1]);
    }
 
    printf("Found %i coins worth %i cents!\n", currTag, value); 
 
    return value;
}
 
//--------------------------------------------------------------
void testApp::setup() {
    printf("Coin Reference\n");
    printf("> Penny: ");
    penny.load("image1.jpg");
    printf("> Nickel: ");
    nickel.load("image2.jpg");
    printf("> Dime: ");
    dime.load("image3.jpg");
    printf("> Quarter: ");
    quarter.load("image4.jpg");
 
    printf("---------------------------\n");
    inMyPocket.load("image5.jpg");
    inMyPocket.count(penny, nickel, dime, quarter);
 
    printf("---------------------------\n");
    inMyPocket.load("image6.jpg");
    inMyPocket.count(penny, nickel, dime, quarter);
 
    printf("---------------------------\n");
    inMyPocket.load("image7.jpg");
    inMyPocket.count(penny, nickel, dime, quarter);
 
    printf("---------------------------\n");
    inMyPocket.load("image8.jpg");
    inMyPocket.count(penny, nickel, dime, quarter);
 
    std::exit(0);
}