SDDSColumn.cpp

Go to the documentation of this file.

//
// Class SDDSColumn
//   This class writes column entries of SDDS files.
//
// Copyright (c) 2019, Christof Metzger-Kraus, Open Sourcerer
// All rights reserved
//
// This file is part of OPAL.
//
// OPAL is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// You should have received a copy of the GNU General Public License
// along with OPAL. If not, see <https://www.gnu.org/licenses/>.
//
#include "Structure/SDDSColumn.h"
#include "Utilities/OpalException.h"
 
#include <array>
#include <iomanip>
#include <ios>
 
SDDSColumn::SDDSColumn(
    const std::string& name, const std::string& type, const std::string& unit,
    const std::string& desc, std::ios_base::fmtflags flags, unsigned short prec)
    : name_m(name),
      description_m(std::make_tuple(type, unit, desc)),
      writeFlags_m(flags),
      writePrecision_m(prec),
      set_m(false) {
    constexpr std::array<std::ios_base::fmtflags, 3> numericalBase{
        std::ios_base::dec, std::ios_base::hex, std::ios_base::oct};
 
    constexpr std::array<std::ios_base::fmtflags, 2> floatFormat{
        std::ios_base::fixed, std::ios_base::scientific};
 
    constexpr std::array<std::ios_base::fmtflags, 3> adjustmentFlags{
        std::ios_base::internal, std::ios_base::left, std::ios_base::right};
 
    // This code ensures that for each group of flags only one flag is given
    for (std::ios_base::fmtflags flag : numericalBase) {
        if (writeFlags_m & flag) {
            writeFlags_m = (flag | (writeFlags_m & ~std::ios_base::basefield));
            break;
        }
    }
    for (std::ios_base::fmtflags flag : floatFormat) {
        if (writeFlags_m & flag) {
            writeFlags_m = (flag | (writeFlags_m & ~std::ios_base::floatfield));
            break;
        }
    }
    for (std::ios_base::fmtflags flag : adjustmentFlags) {
        if (writeFlags_m & flag) {
            writeFlags_m = (flag | (writeFlags_m & ~std::ios_base::adjustfield));
            break;
        }
    }
}
 
void SDDSColumn::writeHeader(
    std::ostream& os, unsigned int colNr, const std::string& indent) const {
    os << "&column\n"
       << indent << "name=" << name_m << ",\n"
       << indent << "type=" << std::get<0>(description_m) << ",\n";
 
    if (std::get<1>(description_m) != "")
        os << indent << "units=" << std::get<1>(description_m) << ",\n";
 
    os << indent << "description=\"" << colNr << " " << std::get<2>(description_m) << "\"\n"
       << "&end\n";
}
 
void SDDSColumn::writeValue(std::ostream& os) const {
    if (!set_m) {
        throw OpalException(
            "SDDSColumn::writeValue", "value for column '" + name_m + "' isn't set");
    }
 
    os.flags(writeFlags_m);
    os.precision(writePrecision_m);
 
    std::visit(
        [&os](auto&& arg) {
            using T = std::decay_t<decltype(arg)>;
            if constexpr (std::is_arithmetic_v<T>) {
                // float, double, int
                os << static_cast<double>(arg);
            } else {
                // string, char
                os << arg;
            }
            os << std::setw(10) << "\t";
        },
        value_m);
 
    set_m = false;
}
 
std::ostream& operator<<(std::ostream& os, const SDDSColumn& col) {
    col.writeValue(os);
 
    return os;
}