include_starpu/skepu/src/data_collector.h

Go to the documentation of this file.

#ifndef DATA_COLLECTOR_H
#define DATA_COLLECTOR_H

#include <string>
#include <set>
#include <utility>
#include <fstream>
#include <iomanip>

#include <iostream>

namespace skepu
{

enum DataExportFormat
{
    GNUPLOT,
    PREDICTION_FILE
};

template <typename Tx, typename Ty>
class DataCollector2D
{

public:
    DataCollector2D(const std::string& _dataSetName, const std::string& _axisNameX, const std::string& _axisNameY);

    void addData(Tx x, Ty y);
    void clear();
    void writeDataToFile(const std::string& filename = "", DataExportFormat format = GNUPLOT);

private:
    std::string dataSetName;
    std::pair<std::string, std::string> axisNames;
    std::multiset< std::pair<Tx, Ty> > dataSet;

    void writeGnuPlotFile(const std::string& filename);

};

template <typename Tx, typename Ty>
DataCollector2D<Tx, Ty>::DataCollector2D(const std::string& _dataSetName, const std::string& _axisNameX, const std::string& _axisNameY)
{
    dataSetName = _dataSetName;
    axisNames.first = _axisNameX;
    axisNames.second = _axisNameY;
}

template <typename Tx, typename Ty>
void DataCollector2D<Tx, Ty>::addData(Tx x, Ty y)
{
    dataSet.insert(std::make_pair(x, y));
}

template <typename Tx, typename Ty>
void DataCollector2D<Tx, Ty>::clear()
{
    dataSet.clear();
}

template <typename Tx, typename Ty>
void DataCollector2D<Tx, Ty>::writeDataToFile(const std::string& filename, DataExportFormat format)
{
    std::string _filename;

    if(filename.empty())
    {
        _filename = dataSetName + ".dat";
    }
    else
    {
        _filename = filename;
    }

    //Call right exporter
    if(format == GNUPLOT)
    {
        writeGnuPlotFile(_filename);
    }
}

template <typename Tx, typename Ty>
void DataCollector2D<Tx, Ty>::writeGnuPlotFile(const std::string& filename)
{
    std::ofstream file(filename.c_str());
    int tabLength = axisNames.first.length()+20;
    file<<std::left;
    file<<std::fixed <<std::setprecision(5);
    if(file.is_open())
    {
        //First add name and axis names as comments
        file<<"# " <<dataSetName <<"\n";
        file<<"# " <<std::setw(tabLength) <<axisNames.first <<std::setw(tabLength) <<axisNames.second <<"\n";

        //Add data in two columns
        for(typename std::multiset< std::pair<Tx, Ty> >::iterator it = dataSet.begin(); it != dataSet.end(); ++it)
        {
            file<<"  " <<std::setw(tabLength) <<it->first <<std::setw(tabLength) <<it->second <<"\n";
        }

        file.close();
    }
}

}

#endif