msg.h

#ifndef DBALLE_MSG_H
#define DBALLE_MSG_H
 
#include <dballe/core/defs.h>
#include <dballe/core/fwd.h>
#include <dballe/core/matcher.h>
#include <dballe/exporter.h>
#include <dballe/importer.h>
#include <dballe/message.h>
#include <dballe/msg/context.h>
#include <dballe/msg/fwd.h>
#include <dballe/var.h>
#include <iosfwd>
#include <memory>
#include <stdio.h>
#include <vector>
 
namespace dballe {
struct CSVReader;
struct CSVWriter;
 
namespace impl {

struct ImporterOptions : public dballe::ImporterOptions
{
    ImporterOptions() = default;
    ImporterOptions(const std::string& s) : dballe::ImporterOptions(s) {}
    ImporterOptions(const ImporterOptions&)            = default;
    ImporterOptions(ImporterOptions&&)                 = default;
    ImporterOptions& operator=(const ImporterOptions&) = default;
    ImporterOptions& operator=(ImporterOptions&&)      = default;
    using dballe::ImporterOptions::operator==;
    using dballe::ImporterOptions::operator!=;
};

struct ExporterOptions : public dballe::ExporterOptions
{
    ExporterOptions()                                  = default;
    ExporterOptions(const ExporterOptions&)            = default;
    ExporterOptions(ExporterOptions&&)                 = default;
    ExporterOptions& operator=(const ExporterOptions&) = default;
    ExporterOptions& operator=(ExporterOptions&&)      = default;
    using dballe::ExporterOptions::operator==;
    using dballe::ExporterOptions::operator!=;
};
 
// Compatibility/shortcut from old Messages implementation to new vector of
// shared_ptr
typedef std::vector<std::shared_ptr<dballe::Message>> Messages;
 
namespace msg {

Messages messages_from_csv(CSVReader& in);

void messages_to_csv(const Messages& msgs, CSVWriter& out);

unsigned messages_diff(const Messages& msgs1, const Messages& msgs2);

void messages_print(const Messages& msgs, FILE* out);
 
class Contexts
{
public:
    typedef std::vector<msg::Context>::const_iterator const_iterator;
    typedef std::vector<msg::Context>::iterator iterator;
    typedef std::vector<msg::Context>::const_reverse_iterator
        const_reverse_iterator;
    typedef std::vector<msg::Context>::reverse_iterator reverse_iterator;
 
protected:
    std::vector<msg::Context> m_contexts;
 
    iterator insert_new(const Level& level, const Trange& trange);
 
public:
    Contexts()                           = default;
    Contexts(const Contexts&)            = default;
    Contexts(Contexts&&)                 = default;
    Contexts& operator=(const Contexts&) = default;
    Contexts& operator=(Contexts&&)      = default;
 
    const_iterator begin() const { return m_contexts.begin(); }
    const_iterator end() const { return m_contexts.end(); }
    iterator begin() { return m_contexts.begin(); }
    iterator end() { return m_contexts.end(); }
    const_reverse_iterator rbegin() const { return m_contexts.rbegin(); }
    const_reverse_iterator rend() const { return m_contexts.rend(); }
    const_iterator cbegin() const { return m_contexts.cbegin(); }
    const_iterator cend() const { return m_contexts.cend(); }
 
    const_iterator find(const Level& level, const Trange& trange) const;
    iterator find(const Level& level, const Trange& trange);
 
    iterator obtain(const Level& level, const Trange& trange);
    bool drop(const Level& level, const Trange& trange);
 
    size_t size() const { return m_contexts.size(); }
    bool empty() const { return m_contexts.empty(); }
    void clear() { return m_contexts.clear(); }
    void reserve(typename std::vector<Value>::size_type size)
    {
        m_contexts.reserve(size);
    }
    iterator erase(iterator pos) { return m_contexts.erase(pos); }
    // iterator erase(const_iterator pos) { return m_contexts.erase(pos); }
};
 
} // namespace msg

class Message : public dballe::Message
{
protected:
    int find_index(const Level& lev, const Trange& tr) const;
 
    const wreport::Var* get_impl(const Level& lev, const Trange& tr,
                                 wreport::Varcode code) const override;
    void set_impl(const Level& lev, const Trange& tr,
                  std::unique_ptr<wreport::Var> var) override;
 
    void seti(const Level& lev, const Trange& tr, wreport::Varcode code,
              int val, int conf);
    void setd(const Level& lev, const Trange& tr, wreport::Varcode code,
              double val, int conf);
    void setc(const Level& lev, const Trange& tr, wreport::Varcode code,
              const char* val, int conf);
 
public:
    Message()                            = default;
    Message(const Message&)              = default;
    Message(Message&&)                   = default;
    Message& operator=(const Message& m) = default;
    Message& operator=(Message&& m)      = default;

    MessageType type = MessageType::GENERIC;
    Values station_data;
    msg::Contexts data;
 
    static std::shared_ptr<Message> create();

    static const Message& downcast(const dballe::Message& o);

    static Message& downcast(dballe::Message& o);

    static std::shared_ptr<Message>
    downcast(std::shared_ptr<dballe::Message> o);
 
    std::shared_ptr<dballe::Message> clone() const override;
    Datetime get_datetime() const override;
    Coords get_coords() const override;
    Ident get_ident() const override;
    std::string get_report() const override;
    MessageType get_type() const override { return type; }
    bool foreach_var(std::function<bool(const Level&, const Trange&,
                                        const wreport::Var&)>) const override;
    void print(FILE* out) const override;
    unsigned diff(const dballe::Message& msg) const override;

    void clear();
 
    using dballe::Message::get;
    using dballe::Message::set;

    const wreport::Var* get(const Shortcut& shortcut) const;

    void set(const Shortcut& shortcut, const wreport::Var& var);

    void set_datetime(const Datetime& dt);

    bool remove_context(const Level& lev, const Trange& tr);

    const msg::Context* find_context(const Level& lev, const Trange& tr) const;

    const Values& find_station_context() const;

    msg::Context* edit_context(const Level& lev, const Trange& tr);

    msg::Context& obtain_context(const Level& lev, const Trange& tr);

    wreport::Var* edit(wreport::Varcode code, const Level& lev,
                       const Trange& tr);
 
#if 0
    bool remove(wreport::Varcode code, const Level& lev, const Trange& tr);
#endif

    void sounding_pack_levels();

    bool from_csv(CSVReader& in);

    void to_csv(CSVWriter& out) const;
 
    std::shared_ptr<dballe::CursorStation>
    query_stations(const Query& query) const override;
    std::shared_ptr<dballe::CursorStationData>
    query_station_data(const Query& query) const override;
    std::shared_ptr<dballe::CursorData>
    query_data(const Query& query) const override;
    std::shared_ptr<dballe::CursorData>
    query_station_and_data(const Query& query) const;

    static void csv_header(CSVWriter& out);

    static MessageType type_from_repmemo(const char* repmemo);

    static const char* repmemo_from_type(MessageType type);
 
#include <dballe/msg/msg-extravars.h>
};

struct MatchedMsg : public Matched
{
    const impl::Message& m;
 
    MatchedMsg(const impl::Message& r);
    ~MatchedMsg();
 
    matcher::Result match_var_id(int val) const override;
    matcher::Result match_station_id(int val) const override;
    matcher::Result match_station_wmo(int block,
                                      int station = -1) const override;
    matcher::Result match_datetime(const DatetimeRange& range) const override;
    matcher::Result match_coords(const LatRange& latrange,
                                 const LonRange& lonrange) const override;
    matcher::Result match_rep_memo(const char* memo) const override;
};

struct MatchedMessages : public Matched
{
    const std::vector<std::shared_ptr<dballe::Message>>& m;
 
    MatchedMessages(const std::vector<std::shared_ptr<dballe::Message>>& m);
    ~MatchedMessages();
 
    matcher::Result match_var_id(int val) const override;
    matcher::Result match_station_id(int val) const override;
    matcher::Result match_station_wmo(int block,
                                      int station = -1) const override;
    matcher::Result match_datetime(const DatetimeRange& range) const override;
    matcher::Result match_coords(const LatRange& latrange,
                                 const LonRange& lonrange) const override;
    matcher::Result match_rep_memo(const char* memo) const override;
};
 
} // namespace impl
} // namespace dballe
#endif