run_stats_table.h

Go to the documentation of this file.

#ifndef RUN_STATS_TABLE_H

#include"stats_recorder.h"
#include <fstream>

#include"arglib.h"

class RunStatsTable
{
public:
    
    RunStatsTable(const char* arg, const char* desc) : filename(arg,desc,""), stats_recorded(false)
    {}
    
    ~RunStatsTable()
    {
        if (filename.was_set()) {
            switch(flush_to_file())
            {
                case 1: cerr << "WARNING: number of cells in one or more rows does not "
                    << "match the number of column headers when reading the \""
                    << filename.get_value() << "\" run stats file." << endl; break;
                case -1: cerr << "WARNING: could not read the table headers when reading the \""
                    << filename.get_value() << "\" run stats file."; break;
                case -2: cerr << "WARNING: could not open the \"" << filename.get_value()
                    << "\" run stats file for read." << endl; break;
                case -3: cerr << "WARNING: could not open the \"" << filename.get_value()
                    << "\" run stats file for write." << endl; break;
                case -4: cerr << "WARNING: error when appending statistics to \"" << filename.get_value()
                    << "\" run stats file for write." << endl; break;
                default:
                    break;
            }
        }
    }
    
    void start() {if (filename.was_set()) { stats_recorded=true; stats.start(); }}
    void lap()   {if (filename.was_set()) { stats_recorded=true; stats.lap(); }}
    void stop()  {if (filename.was_set()) { stats_recorded=true; stats.stop(); }}
    
    int flush_to_file()
    {
        CSVStringTable table;
        int ret;
        
        if (!stats_recorded) return 2;
        
        /* Read the table. */
        if ((ret=read_from_file(table, filename.get_value()))) {
            if (ret != -2) return ret;
            // Continue if could not open for read (-2).
        }
        
        /* Update the table. */
        if (stats.append_to(table) < 0) {
            return -4;
        }
        
        stats.clear();
        stats_recorded = false;
        
        /* Write the table back. */
        return write_to_file(table, filename.get_value());
    }
    
    bool was_set() const {
        return filename.was_set();
    }
    
private:
    
    int read_from_file(CSVStringTable& table, string filename)
    {
        ifstream ifs(filename.c_str());
        
        if (ifs.is_open()) {
            int ret=table.read(ifs);
            ifs.close();
            return ret;
        }
        else
            return -2;
    }
    
    int write_to_file(const CSVStringTable& table, string filename)
    {
        ofstream ofs(filename.c_str());
        
        if (ofs.is_open()) {
            table.write(ofs);
            ofs.close();
            return 0;
        }
        else {
            return -3;
        }
    }
    
    RunStatsRecorder stats;
    clarg::argString filename;
    bool stats_recorded;
};


#endif // RUN_STATS_TABLE