export.h

Go to the documentation of this file.

#pragma once
 
#include <boost/optional.hpp>
#include <iostream>
 
#include "storm/io/file.h"
#include "storm/utility/macros.h"
 
namespace storm {
namespace io {
 
template<typename DataType, typename Header1Type = DataType, typename Header2Type = DataType>
inline void exportDataToCSVFile(std::string filepath, std::vector<std::vector<DataType>> const& data,
                                boost::optional<std::vector<Header1Type>> const& header1 = boost::none,
                                boost::optional<std::vector<Header2Type>> const& header2 = boost::none) {
    std::ofstream filestream;
    storm::io::openFile(filepath, filestream);
 
    if (header1) {
        for (auto columnIt = header1->begin(); columnIt != header1->end(); ++columnIt) {
            if (columnIt != header1->begin()) {
                filestream << ",";
            }
            filestream << *columnIt;
        }
        filestream << '\n';
    }
 
    if (header2) {
        for (auto columnIt = header2->begin(); columnIt != header2->end(); ++columnIt) {
            if (columnIt != header2->begin()) {
                filestream << ",";
            }
            filestream << *columnIt;
        }
        filestream << '\n';
    }
 
    for (auto const& row : data) {
        for (auto columnIt = row.begin(); columnIt != row.end(); ++columnIt) {
            if (columnIt != row.begin()) {
                filestream << ",";
            }
            filestream << *columnIt;
        }
        filestream << '\n';
    }
    storm::io::closeFile(filestream);
}
 
template<typename Container>
inline void outputFixedWidth(std::ostream& stream, Container const& output, size_t maxWidth = 30) {
    size_t curLength = 0;
    for (auto s : output) {
        if (curLength > 0) {
            stream << ", ";
            curLength += 2;
        }
        stream << s;
        curLength += s.length();
        if (maxWidth > 0 && curLength >= maxWidth) {
            stream << '\n';
            curLength = 0;
        }
    }
}
}  // namespace io
}  // namespace storm