main - Copy.cpp

#include <ecf/ECF.h>
#include "SymbRegEvalOp.h"
 
#include <string>
uint showTree(Tree::Tree* tree, std::string& output, uint iNode = 0, uint prefix = 0)
{
    Tree::PrimitiveP prim = tree->at(iNode)->primitive_;
    int arity = prim->getNumberOfArguments();
 
    if(arity == 0) {
        cout << prim->getName();
    }
    else if(prim->getName() == "pow") {
        cout << "pow(";
            iNode++;
            iNode = showTree(tree, output, iNode, prefix);
        cout << ",";
            iNode++;
            iNode = showTree(tree, output, iNode, prefix);
        cout << ")";
    }
    else if(arity == 2) {
        cout << "(";
            iNode++;
            iNode = showTree(tree, output, iNode, prefix);
        cout << prim->getName();
            iNode++;
            iNode = showTree(tree, output, iNode, prefix);
        cout << ")";
    }
    else {
        cout << prim->getName();
        cout << "(";
        for(int child = 0; child < arity; child++) {
            iNode++;
            iNode = showTree(tree, output, iNode, prefix);
        }
        cout << ")";
    }
    return iNode;
}
 
 
int main(int argc, char **argv)
{
    StateP state (new State);
 
    // set the evaluation operator
    state->setEvalOp(new SymbRegEvalOp);
 
    state->initialize(argc, argv);
    state->run();
 
    // after the evolution: show best tree
    std::vector<IndividualP> hof = state->getHoF()->getBest();
    IndividualP ind = hof[0];
    std::cout << ind->toString();
    Tree::Tree* tree = (Tree::Tree*) ind->getGenotype().get();
    std::string out;
    showTree(tree, out);
    cout << out << endl;
 
    return 0;
}