writer.cpp

Go to the documentation of this file.

/**
    @file
    @author Alexander Sherikov
 
    @copyright 2018-2020 Alexander Sherikov, Licensed under the Apache License, Version 2.0.
    (see @ref LICENSE or http://www.apache.org/licenses/LICENSE-2.0)
 
    @brief
*/
 
#include <boost/math/special_functions.hpp>
#include <ariles2/visitors/yaml_cpp.h>
#include <yaml-cpp/yaml.h>
 
 
namespace ariles2
{
    namespace ns_yaml_cpp
    {
        namespace impl
        {
            class Writer : public write::FileVisitorImplementation
            {
            public:
                using EmitterPtr = std::shared_ptr<YAML::Emitter>;
 
 
            public:
                /// instance of YAML emitter, is destroyed and reinitialized by flush()
                EmitterPtr emitter_;
 
                std::size_t map_depth_;
                bool skip_root_map_;
 
 
            public:
                void initEmitter()
                {
                    CPPUT_TRACE_FUNCTION;
                    emitter_ = std::make_shared<YAML::Emitter>();
                    emitter_->SetDoublePrecision(std::numeric_limits<double>::digits10);
                    if (output_stream_->tellp() != 0)
                    {
                        *emitter_ << YAML::Newline;
                    }
                    *emitter_ << YAML::BeginMap;
                    map_depth_ = 0;
                    skip_root_map_ = false;
                }
 
 
                void destroyEmitter()
                {
                    CPPUT_TRACE_FUNCTION;
                    *emitter_ << YAML::EndMap;
                    *output_stream_ << emitter_->c_str();
                    emitter_.reset();
                }
 
            public:
                template <class... t_Args>
                explicit Writer(t_Args &&...args) : FileVisitorImplementation(std::forward<t_Args>(args)...)
                {
                    initEmitter();
                }
 
 
                void flush()
                {
                    destroyEmitter();
                    *output_stream_ << "\n";
                    output_stream_->flush();
                    initEmitter();
                }
            };
        }  // namespace impl
    }      // namespace ns_yaml_cpp
}  // namespace ariles2
 
 
namespace ariles2
{
    namespace ns_yaml_cpp
    {
        Writer::Writer(const std::string &file_name)
        {
            makeImplPtr(file_name);
        }
 
 
        Writer::Writer(std::ostream &output_stream)
        {
            makeImplPtr(output_stream);
        }
 
 
 
        void Writer::startMap(const Parameters &, const std::size_t /*num_entries*/)
        {
            CPPUT_TRACE_FUNCTION;
            if (impl_->map_depth_ > 0 or not impl_->skip_root_map_)
            {
                *impl_->emitter_ << YAML::BeginMap;
            }
            ++impl_->map_depth_;
        }
 
        void Writer::startMapEntry(const std::string &map_name)
        {
            CPPUT_TRACE_FUNCTION;
            CPPUT_TRACE_VALUE(map_name);
            *impl_->emitter_ << YAML::Key << map_name;
            *impl_->emitter_ << YAML::Value;
        }
 
        void Writer::endMap()
        {
            CPPUT_TRACE_FUNCTION;
            CPPUT_ASSERT(impl_->map_depth_ > 0, "Internal logic error.");
            --impl_->map_depth_;
            if (impl_->map_depth_ > 0 or not impl_->skip_root_map_)
            {
                *impl_->emitter_ << YAML::EndMap;
            }
        }
 
 
        void Writer::flush()
        {
            CPPUT_TRACE_FUNCTION;
            impl_->flush();
        }
 
 
        void Writer::startArray(const std::size_t /*size*/, const bool compact)
        {
            CPPUT_TRACE_FUNCTION;
            if (compact)
            {
                *impl_->emitter_ << YAML::Flow;
            }
            *impl_->emitter_ << YAML::BeginSeq;
        }
 
 
        void Writer::endArray()
        {
            CPPUT_TRACE_FUNCTION;
            *impl_->emitter_ << YAML::EndSeq;
        }
 
 
        void Writer::startRoot(const std::string &name, const Parameters &)
        {
            CPPUT_TRACE_FUNCTION;
            CPPUT_TRACE_VALUE(name);
            if (name.empty())
            {
                impl_->skip_root_map_ = true;
            }
            else
            {
                startMapEntry(name);
            }
        }
 
        void Writer::endRoot(const std::string &name)
        {
            CPPUT_TRACE_FUNCTION;
            if (not name.empty())
            {
                endMapEntry();
            }
            impl_->skip_root_map_ = false;
        }
 
 
#define ARILES2_BASIC_TYPE(type)                                                                                       \
    void Writer::writeElement(const type &element, const Parameters &)                                                 \
    {                                                                                                                  \
        *impl_->emitter_ << element;                                                                                   \
    }
 
        CPPUT_MACRO_SUBSTITUTE(ARILES2_BASIC_INTEGER_TYPES_LIST)
 
#undef ARILES2_BASIC_TYPE
 
 
#define ARILES2_BASIC_TYPE(type)                                                                                       \
    void Writer::writeElement(const type &element, const Parameters &)                                                 \
    {                                                                                                                  \
        if (boost::math::isnan(element))                                                                               \
        {                                                                                                              \
            *impl_->emitter_ << ".nan";                                                                                \
        }                                                                                                              \
        else                                                                                                           \
        {                                                                                                              \
            if (boost::math::isinf(element))                                                                           \
            {                                                                                                          \
                if (element < 0.0)                                                                                     \
                {                                                                                                      \
                    *impl_->emitter_ << "-.inf";                                                                       \
                }                                                                                                      \
                else                                                                                                   \
                {                                                                                                      \
                    *impl_->emitter_ << ".inf";                                                                        \
                }                                                                                                      \
            }                                                                                                          \
            else                                                                                                       \
            {                                                                                                          \
                *impl_->emitter_ << static_cast<double>(element);                                                      \
            }                                                                                                          \
        }                                                                                                              \
    }
 
        CPPUT_MACRO_SUBSTITUTE(ARILES2_BASIC_REAL_TYPES_LIST)
 
#undef ARILES2_BASIC_TYPE
 
 
 
        void Writer::writeElement(const std::string &element, const Parameters &)
        {
            *impl_->emitter_ << element;
        }
 
        void Writer::writeElement(const bool &element, const Parameters &)
        {
            *impl_->emitter_ << element;
        }
    }  // namespace ns_yaml_cpp
}  // namespace ariles2