edoFileSnapshot.cpp

/*
The Evolving Distribution Objects framework (EDO) is a template-based,
ANSI-C++ evolutionary computation library which helps you to write your
own estimation of distribution algorithms.

This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.

This library 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
Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public
License along with this library; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA

Copyright (c) Marc Schoenauer, Maarten Keijzer and GeNeura Team, 2001
Copyright (C) 2010 Thales group
*/
/*
Authors:
    todos@geneura.ugr.es
    Marc Schoenauer <Marc.Schoenauer@polytechnique.fr>
    Martin Keijzer <mkeijzer@dhi.dk>
    Johann Dréo <johann.dreo@thalesgroup.com>
    Caner Candan <caner.candan@thalesgroup.com>
*/

#include <cstdlib>

#include <iostream>
#include <fstream>
#include <stdexcept>

#include <utils/edoFileSnapshot.h>
#include <utils/compatibility.h>
#include <utils/eoParam.h>

edoFileSnapshot::edoFileSnapshot(std::string dirname,
                               unsigned int frequency /*= 1*/,
                               std::string filename /*= "gen"*/,
                               std::string delim /*= " "*/,
                               unsigned int counter /*= 0*/,
                               bool rmFiles /*= true*/,
                               bool saveFilenames /*= true*/)
    : _dirname(dirname), _frequency(frequency),
      _filename(filename), _delim(delim),
      _counter(counter), _saveFilenames(saveFilenames),
      _descOfFiles( NULL ), _boolChanged(true)
{
    std::string s = "test -d " + _dirname;

    int res = system(s.c_str());

    // test for (unlikely) errors

    if ( (res == -1) || (res == 127) )
        {
            throw std::runtime_error("Problem executing test of dir in eoFileSnapshot");
        }

    // now make sure there is a dir without any genXXX file in it
    if (res)                    // no dir present
        {
            s = std::string("mkdir ") + _dirname;
        }
    else if (!res && rmFiles)
        {
            s = std::string("/bin/rm -f ") + _dirname+ "/" + _filename + "*";
        }
    else
        {
            s = " ";
        }

    int dummy;
    dummy = system(s.c_str());
    // all done

    _descOfFiles = new std::ofstream( std::string(dirname + "/list_of_files.txt").c_str() );

}

edoFileSnapshot::~edoFileSnapshot()
{
    delete _descOfFiles;
}

void edoFileSnapshot::setCurrentFileName()
{
    std::ostringstream oscount;
    oscount << _counter;
    _currentFileName = _dirname + "/" + _filename + oscount.str();
}

eoMonitor& edoFileSnapshot::operator()(void)
{
    if (_counter % _frequency)
        {
            _boolChanged = false;  // subclass with gnuplot will do nothing
            _counter++;
            return (*this);
        }
    _counter++;
    _boolChanged = true;
    setCurrentFileName();

    std::ofstream os(_currentFileName.c_str());

    if (!os)
        {
            std::string str = "edoFileSnapshot: Could not open " + _currentFileName;
            throw std::runtime_error(str);
        }

    if ( _saveFilenames )
        {
            *_descOfFiles << _currentFileName.c_str() << std::endl;
        }

    return operator()(os);
}

eoMonitor& edoFileSnapshot::operator()(std::ostream& os)
{
    iterator it = vec.begin();

    os << (*it)->getValue();

    for ( ++it; it != vec.end(); ++it )
        {
            os << _delim.c_str() << (*it)->getValue();
        }

    os << '\n';

    return *this;
}