SVOutStream.h

Go to the documentation of this file.

// Copyright (c) 2002-present, OpenMS Inc. -- EKU Tuebingen, ETH Zurich, and FU Berlin
// SPDX-License-Identifier: BSD-3-Clause
//
// --------------------------------------------------------------------------
// $Maintainer: Hendrik Weisser $
// $Authors: Hendrik Weisser $
// --------------------------------------------------------------------------
 
#pragma once
 
#include <OpenMS/DATASTRUCTURES/StringUtils.h>
 
#include <ostream>
#include <fstream>      // std::ofstream
#include <sstream>
#include <boost/math/special_functions/fpclassify.hpp> // because isfinite not supported on Mac
 
namespace OpenMS
{
  enum Newline {nl};
 
  class OPENMS_DLLAPI SVOutStream :
    public std::ostream
  {
public:
 
    SVOutStream(const std::string& file_out,
                const std::string& sep = "\t",
                const std::string& replacement = "_",
                OpenMS::QuotingMethod quoting = OpenMS::QuotingMethod::DOUBLE);
 
    SVOutStream(std::ostream& out,
                const std::string& sep = "\t",
                const std::string& replacement = "_",
                OpenMS::QuotingMethod quoting = OpenMS::QuotingMethod::DOUBLE);
 
    ~SVOutStream() override;
 
    SVOutStream& operator<<(std::string str);    // use call-by-value here
 
 
 
 
    SVOutStream& operator<<(const char* c_str);
 
 
    SVOutStream& operator<<(const char c);
 
    SVOutStream& operator<<(std::ostream& (*fp)(std::ostream&));
 
    SVOutStream& operator<<(enum Newline);
 
    template<typename T>
    typename std::enable_if<std::is_arithmetic<typename std::remove_reference<T>::type>::value, SVOutStream&>::type operator<<(const T& value)
    {
      if (!newline_) static_cast<std::ostream&>(*this) << sep_;
      else newline_ = false;
      static_cast<std::ostream&>(*this) << StringUtils::toStr(value);
      return *this;
    };
 
    template<typename T>
    typename std::enable_if<!std::is_arithmetic<typename std::remove_reference<T>::type>::value, SVOutStream&>::type operator<<(const T& value)
    {
      if (!newline_)
        static_cast<std::ostream &>(*this) << sep_;
      else
        newline_ = false;
      static_cast<std::ostream &>(*this) << value;
      return *this;
    };
 
    SVOutStream& write(const std::string& str);   // write unmodified string
 
 
    bool modifyStrings(bool modify);
 
 
    template <typename NumericT>
    SVOutStream& writeValueOrNan(NumericT thing)
    {
      if ((boost::math::isfinite)(thing)) return operator<<(thing);
 
      bool old = modifyStrings(false);
      if ((boost::math::isnan)(thing)) 
      {
        operator<<(nan_);
      }
      else if (thing < 0) 
      {
        operator<<("-" + inf_);
      }
      else 
      {
        operator<<(inf_);
      }
      modifyStrings(old);
      return *this;
    }
 
protected:
    std::ofstream* ofs_;
 
    std::string sep_;
 
    std::string replacement_;
 
    std::string nan_;
 
    std::string inf_;
 
    OpenMS::QuotingMethod quoting_;
 
    bool modify_strings_;
 
    bool newline_;
 
    std::stringstream ss_;
  };
 
}