AsciiHeaderIO.hpp

Go to the documentation of this file.

// Copyright (c) 2013 The ComPWA Team.
// This file is part of the ComPWA framework, check
// https://github.com/ComPWA/ComPWA/license.txt for details.

#ifndef COMPWA_DATA_ASCIIHEADER_HPP_
#define COMPWA_DATA_ASCIIHEADER_HPP_

#include <string>
#include <vector>

namespace ComPWA {
namespace Data {
namespace Ascii {

class AsciiHeader {
public:
  AsciiHeader(std::vector<int> PIDs = {}, std::string Unit = "GeV",
              bool EnergyFirst = false)
      : PIDs(PIDs), Unit(Unit), EnergyFirst(EnergyFirst) {}

  static std::string extractHeaderContent(std::istream &InputStream);

  void importYAML(const std::string &HeaderContent);
  void importYAML(std::istream &InputStream) {
    importYAML(extractHeaderContent(InputStream));
  }

  void dumpToYAML(std::ostream &os) const;

  const std::vector<int> getFinalStatePIDs() const { return PIDs; }
  bool isEnergyFirst() const { return EnergyFirst; }
  const std::string &getUnit() const { return Unit; }

private:
  std::vector<int> PIDs;
  std::string Unit;
  bool EnergyFirst;
};

} // namespace Ascii
} // namespace Data
} // namespace ComPWA

#endif