output.h

/*********************************************************************************************************************************************/
/*  DocWire SDK: Award-winning modern data processing in C++20. SourceForge Community Choice & Microsoft support. AI-driven processing.      */
/*  Supports nearly 100 data formats, including email boxes and OCR. Boost efficiency in text extraction, web data extraction, data mining,  */
/*  document analysis. Offline processing possible for security and confidentiality                                                          */
/*                                                                                                                                           */
/*  Copyright (c) SILVERCODERS Ltd, http://silvercoders.com                                                                                  */
/*  Project homepage: https://github.com/docwire/docwire                                                                                     */
/*                                                                                                                                           */
/*  SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-DocWire-Commercial                                                                  */
/*********************************************************************************************************************************************/
 
#ifndef DOCWIRE_OUTPUT_H
#define DOCWIRE_OUTPUT_H
 
#include "chain_element.h"
#include <concepts>
#include <ostream>
#include "parsing_chain.h"
#include <type_traits>
#include <variant>
#include <vector>
 
namespace docwire
{
 
template<class T>
concept OStreamDerived = std::derived_from<T, std::ostream>;
 
template<typename T>
concept ostream_derived_ref_qualified = OStreamDerived<std::remove_reference_t<T>>;
 
class DOCWIRE_CORE_EXPORT output_chain_element : public chain_element
{
public:
  output_chain_element(ref_or_owned<std::ostream> out_stream)
    : m_out_obj{out_stream}
  {}
 
  output_chain_element(ref_or_owned<std::vector<message_ptr>> out_vector)
    : m_out_obj{out_vector}
  {}
 
  bool is_leaf() const override
  {
    return true;
  }
 
  virtual continuation operator()(message_ptr msg, const message_callbacks& emit_message) override;
 
private:
  std::variant<ref_or_owned<std::ostream>, ref_or_owned<std::vector<message_ptr>>> m_out_obj;
};
 
inline parsing_chain operator|(ref_or_owned<chain_element> element, ref_or_owned<std::ostream> stream)
{
  return element | output_chain_element(stream);
}
 
inline parsing_chain& operator|=(parsing_chain& chain, ref_or_owned<std::ostream> stream)
{
  return chain |= output_chain_element(stream);
}
 
inline parsing_chain operator|(ref_or_owned<chain_element> element, ref_or_owned<std::vector<message_ptr>> vector)
{
  return element | output_chain_element(vector);
}
 
inline parsing_chain& operator|=(parsing_chain& chain, ref_or_owned<std::vector<message_ptr>> vector)
{
  return chain |= output_chain_element(vector);
}
 
} // namespace docwire
 
#endif //DOCWIRE_OUTPUT_H