OneMaxEvalOp.cpp

#include <ecf/ECF.h>
#include "OneMaxEvalOp.h"
#include<fstream>
#include<string>
 
 
void OneMaxEvalOp::registerParameters(StateP state)
{
    state->getRegistry()->registerEntry("infile", (voidP) (new std::string), ECF::STRING);
}
 
 
bool OneMaxEvalOp::initialize(StateP state)
{
    s = state;
    height = width = 0;
 
    if(!state->getRegistry()->isModified("infile")) {
        state->getLogger()->log(1, "no input file defined (parameter 'infile'");
        return false;
    }
 
    voidP sptr = state->getRegistry()->getEntry("infile"); // get parameter value
    std::string filePath = *((std::string*) sptr.get()); // convert from voidP to user defined type
 
    ifstream inFile(filePath);
    std::string str;
 
    uint genLength = 0, row = 0;
 
    image.clear();
    while(getline(inFile, str)) {
        width = str.length();
        image.push_back(std::vector<int>(width));
        genLength += width;
        for(uint col = 0; col < width; col++)
            image[row][col] = (str[col] != ' ') ? 1 : 0;
        row++;
    }
 
    // set dimension for BitString genotype
    state->getRegistry()->modifyEntry("BitString.size", (voidP) new uint(genLength));
 
    // reinitialize population with updated size
    state->getPopulation()->initialize(state);
 
    bestVal = 0;
    height = row;
 
    return true;
}
 
// evaluate() receives a smart pointer to the individual to evaluate
FitnessP OneMaxEvalOp::evaluate(IndividualP individual)
{
    // evaluation creates a new fitness object using a smart pointer
    // in our case, we try to maximize the number of ones, so we use FitnessMax fitness (for maximization problems)
    FitnessP fitness (new FitnessMax);
 
    // Each individual is a vector of genotypes (defined in the configuration file).
    // We'll use BitString, and put it as the first and only genotype
    BitString::BitString* bitstr = (BitString::BitString*) individual->getGenotype().get();
    //BitStringP bitstr = boost::static_pointer_cast<BitString::BitString> (individual->getGenotype(0)); // don't need zero for the first one
        
    // count the ones; where are they?
    // BitString genotype contains a std::vector of bool's named 'bits'
    uint hits = 0;
    int row, col;
    for(uint i = 0; i<bitstr->bits.size(); i++){
            row = i / width;
            col = i % width;
            if(bitstr->bits[i] == true && image[row][col] > 0)
                hits++ ;
            else if(bitstr->bits[i] == false && image[row][col] == 0)
                hits++;
    }
    fitness->setValue(hits);
 
    if(hits > bestVal) {
        system("cls");
        char pix = 219;
        bestVal = hits;
        cout << endl;
        for(uint i = 0; i < height; i++) {
            for(uint j = 0; j < width; j++) {
                if(bitstr->bits[i*width + j] == true)
                    cout << pix;
                else
                    cout << " ";
            }
            cout << endl;
        }
        cout << endl;
    }
 
    if(bestVal == bitstr->bits.size()) {
        //s->setTerminateCond();
        for(uint i = 0; i < height; i++) 
            for(uint j = 0; j < width; j++) 
                image[i][j] = !image[i][j];     
        bestVal = 0;
    }
 
    // return the smart pointer to new fitness
    return fitness;
}