data_collector_3d.h

Go to the documentation of this file.

#ifndef DATA_COLLECTOR_H
#define DATA_COLLECTOR_H

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

#include <iostream>


enum DataExportFormat
{
    GNUPLOT,
    PREDICTION_FILE
};

template <typename Tx, typename Ty, typename Tz>
class DataCollector3D
{

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

    void addData(Tx x, Ty y, Tz z);
    void clear();
    void writeDataToFile(const std::string& filename = "", DataExportFormat format = GNUPLOT, int tabLength=-1);

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

    void writeGnuPlotFile(const std::string& filename, int tabLength=-1);

};

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

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

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

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

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

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

template <typename Tx, typename Ty, typename Tz>
void DataCollector3D<Tx, Ty, Tz>::writeGnuPlotFile(const std::string& filename, int tabLength)
{
    std::ofstream file(filename.c_str());
    
    if(tabLength<1)
        tabLength = axisNames.first.length()+10;
        
    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.first <<std::setw(tabLength) <<axisNames.second.second <<"\n";
        
        const Tx *pLastKey = NULL;

        //Add data in two columns
        for(typename std::multimap< Tx, std::pair<Ty, Tz> >::iterator it = dataSet.begin(); it != dataSet.end(); ++it)
        {
            if( (pLastKey!=NULL) && (*pLastKey==it->first) )
            {
                continue;
            }
            pLastKey = &(it->first);
            
            typename std::pair<typename std::multimap< Tx, std::pair<Ty, Tz> >::iterator,typename std::multimap< Tx, std::pair<Ty, Tz> >::iterator> range = dataSet.equal_range(*pLastKey);
            
            for(typename std::multimap< Tx, std::pair<Ty, Tz> >::iterator i = range.first; i != range.second; ++i)
            {
                file<<"  " <<std::setw(tabLength) <<(*pLastKey)<<std::setw(tabLength)<<i->second.first<<std::setw(tabLength)<<i->second.second<<"\n";
                
            }
            file<<"\n";    
        }

        file.close();
    }
}


#endif