D:/Projekt/ECF_trunk/ECF/StatCalc.cpp

#include "ECF_base.h"
#include "ECF_macro.h"
#include <cmath>


StatCalc::StatCalc() 
{
    statNo = 0;
    statsFileName_ = "";
}


void StatCalc::registerParameters(StateP state)
{
    state->getRegistry()->registerEntry("stats.file", (voidP) (new std::string("")), ECF::STRING);
}


bool StatCalc::initialize(StateP state)
{
    state_ = state;
    average_.clear();
    stdDev_.clear();
    max_.clear();
    min_.clear();
    time_.clear();
    sampleSize_.clear();

    if(state->getRegistry()->isModified("stats.file")) {
        voidP sptr = state->getRegistry()->getEntry("stats.file");
        statsFileName_ = *((std::string*) sptr.get());

        statsFile_.open(statsFileName_.c_str());
        if(!statsFile_) {
            throw std::string("Error: can't open stats file (") + statsFileName_ + ")";
        }
    }
    return true;
}


double StatCalc::getFitnessMin(int generation)
{
    if(generation == -1)
        generation = statNo;
    return min_[generation];
}


double StatCalc::getFitnessMax(int generation)
{
    if(generation == -1)
        generation = statNo;
    return max_[generation];
}


bool StatCalc::operate(const std::vector<IndividualP>& pool)
{
    std::vector<double> fitnessTemp;

    sampleSize_.push_back((uint)pool.size());
    fitnessTemp.resize(pool.size());

    for(uint i = 0; i < pool.size(); i++) 
        fitnessTemp[i] = pool[i]->fitness->getValue();

    max_.push_back(fitnessTemp[0]);
    min_.push_back(fitnessTemp[0]);
    statNo = (uint) max_.size() - 1;
    double sum = 0;

    for(uint i = 0; i < fitnessTemp.size(); i++) {
        if(fitnessTemp[i] > max_[statNo]) {
            max_[statNo] = fitnessTemp[i];
        }
        if(fitnessTemp[i] < min_[statNo]) {
            min_[statNo] = fitnessTemp[i];
        }
        sum += fitnessTemp[i];
    }

    average_.push_back(sum / fitnessTemp.size());

    if(fitnessTemp.size() > 1) {
        double numerator = 0, denominator;
        for(uint i = 0; i < fitnessTemp.size(); i++) {
            numerator += ((fitnessTemp[i] - average_[statNo]) * (fitnessTemp[i] - average_[statNo]));
        }
        denominator = (fitnessTemp.size() - 1);
        stdDev_.push_back(sqrt(numerator / denominator));
    }
    else
        stdDev_.push_back(fitnessTemp[0]);

    time_.push_back(state_->getElapsedTime());

    return true;
}


bool StatCalc::update(std::vector<double> stats)
{
    double outMin = stats[0];
    double outMax = stats[1];
    double outAvg = stats[2];
    double outDev = stats[3];
    uint outSize = (uint) stats[4];

    if(outMin < min_.back())
        min_.back() = outMin;
    
    if(outMax > max_.back())
        max_.back() = outMax;

    uint mySize = sampleSize_.back();
    if(mySize > 1) {
        sampleSize_.back() = mySize + outSize;
        double myAvg = average_.back();
        double newAvg = (myAvg * mySize + outAvg * outSize) / sampleSize_.back();
        average_.back() = newAvg;

        uint totSize = sampleSize_.back();
        double mySum = myAvg * mySize;
        double outSum = outAvg * outSize;
        double totSum = mySum + outSum;
        double mySumSqr = pow(stdDev_.back(), 2) * mySize + mySum / mySize * mySum;
        double outSumSqr = outDev * outDev * outSize + outSum / outSize * outSum;
        double newDev = sqrt((mySumSqr + outSumSqr - (totSum / totSize * totSum)) / totSize);
        stdDev_.back() = newDev;
    }
    else {
        average_.back() = outAvg;
        stdDev_.back() = outDev;
        sampleSize_.back() = outSize;
    }

    return true;
}


std::vector<double> StatCalc::getStats(int gen)
{
    if(gen == -1)
        gen = statNo;

    std::vector<double> stats;
    stats.push_back(min_[gen]);
    stats.push_back(max_[gen]);
    stats.push_back(average_[gen]);
    stats.push_back(stdDev_[gen]);
    stats.push_back(sampleSize_[gen]);
    stats.push_back(time_[gen]);

    return stats;
}


void StatCalc::log(int generation)
{
    if(generation == -1)
        generation = statNo;

    ECF_LOG(state_, 3, "Stats: fitness\n\tmax: " + dbl2str(max_[generation]) + "\n\tmin: " +
        dbl2str(min_[generation]) + "\n\tavg: " + dbl2str(average_[generation]) + "\n\tstdev: " + 
        dbl2str(stdDev_[generation]));
}


// ispis tijeka min fitnesa - za potrebe mjerenja
void StatCalc::output(uint step)
{
    std::stringstream log;
    for(uint i = 0; i < min_.size(); i += step)
        log << i << "\t" << min_[i] << std::endl;
    ECF_LOG(state_, 2, log.str());
}