mirror of
https://github.com/opencv/opencv.git
synced 2025-06-12 20:42:53 +08:00
5224d016e9
[GAPI] RMat integration into the framework * RMat integration * Added initialization of input mat in GArray initialization tests * Fixed klocwork warnings in RMat tests, changed argument order in EXPECT_EQ
275 lines
12 KiB
C++
275 lines
12 KiB
C++
#ifndef OPENCV_GAPI_COMMON_SERIALIZATION_HPP
|
|
#define OPENCV_GAPI_COMMON_SERIALIZATION_HPP
|
|
|
|
// This file is part of OpenCV project.
|
|
// It is subject to the license terms in the LICENSE file found in the top-level directory
|
|
// of this distribution and at http://opencv.org/license.html.
|
|
//
|
|
// Copyright (C) 2020 Intel Corporation
|
|
|
|
#include <iostream>
|
|
#include <fstream>
|
|
#include <string>
|
|
#include <map>
|
|
|
|
#include <ade/util/iota_range.hpp> // used in the vector<</>>
|
|
|
|
#include "compiler/gmodel.hpp"
|
|
#include "opencv2/gapi/render/render_types.hpp"
|
|
#include "opencv2/gapi/s11n.hpp" // basic interfaces
|
|
|
|
#if (defined _WIN32 || defined _WIN64) && defined _MSC_VER
|
|
#pragma warning(disable: 4702)
|
|
#endif
|
|
|
|
namespace cv {
|
|
namespace gapi {
|
|
namespace s11n {
|
|
|
|
struct GSerialized {
|
|
std::vector<cv::gimpl::Op> m_ops;
|
|
std::vector<cv::gimpl::Data> m_datas;
|
|
cv::gimpl::DataObjectCounter m_counter;
|
|
cv::gimpl::Protocol m_proto;
|
|
};
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// S11N operators
|
|
// Note: operators for basic types are defined in IIStream/IOStream
|
|
|
|
// G-API types /////////////////////////////////////////////////////////////////
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, cv::util::monostate );
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::util::monostate &);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, cv::GShape shape);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::GShape &shape);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, cv::detail::ArgKind k);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::detail::ArgKind &k);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, cv::detail::OpaqueKind k);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::detail::OpaqueKind &k);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, cv::gimpl::Data::Storage s);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::gimpl::Data::Storage &s);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::gimpl::DataObjectCounter &c);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::gimpl::DataObjectCounter &c);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::gimpl::Protocol &p);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::gimpl::Protocol &p);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::GArg &arg);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::GArg &arg);
|
|
|
|
//Forward declaration
|
|
//IOStream& operator<< (IOStream& os, const cv::GRunArg &arg);
|
|
//IIStream& operator>> (IIStream& is, cv::GRunArg &arg);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::GKernel &k);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::GKernel &k);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::GMatDesc &d);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::GMatDesc &d);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::GScalarDesc &);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::GScalarDesc &);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::GOpaqueDesc &);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::GOpaqueDesc &);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::GArrayDesc &);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::GArrayDesc &);
|
|
|
|
#if !defined(GAPI_STANDALONE)
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::UMat &);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::UMat &);
|
|
#endif // !defined(GAPI_STANDALONE)
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::RMat &r);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::RMat &r);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::gapi::wip::IStreamSource::Ptr &);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::gapi::wip::IStreamSource::Ptr &);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::detail::VectorRef &);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::detail::VectorRef &);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::detail::OpaqueRef &);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::detail::OpaqueRef &);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::gimpl::RcDesc &rc);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::gimpl::RcDesc &rc);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::gimpl::Op &op);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::gimpl::Op &op);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::gimpl::Data &op);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::gimpl::Data &op);
|
|
|
|
// Render types ////////////////////////////////////////////////////////////////
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::gapi::wip::draw::Text &t);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::gapi::wip::draw::Text &t);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream&, const cv::gapi::wip::draw::FText &);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream&, cv::gapi::wip::draw::FText &);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::gapi::wip::draw::Circle &c);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::gapi::wip::draw::Circle &c);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::gapi::wip::draw::Rect &r);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::gapi::wip::draw::Rect &r);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::gapi::wip::draw::Image &i);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::gapi::wip::draw::Image &i);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::gapi::wip::draw::Mosaic &m);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::gapi::wip::draw::Mosaic &m);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::gapi::wip::draw::Poly &p);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::gapi::wip::draw::Poly &p);
|
|
|
|
GAPI_EXPORTS IOStream& operator<< (IOStream& os, const cv::gapi::wip::draw::Line &l);
|
|
GAPI_EXPORTS IIStream& operator>> (IIStream& is, cv::gapi::wip::draw::Line &l);
|
|
|
|
// The top-level serialization routine.
|
|
// Note it is just a single function which takes a GModel and a list of nodes
|
|
// and writes the data to the stream (recursively)
|
|
GAPI_EXPORTS void serialize( IOStream& os
|
|
, const ade::Graph &g
|
|
, const std::vector<ade::NodeHandle> &nodes);
|
|
|
|
// The top-level serialization routine.
|
|
// Note it is just a single function which takes a GModel and a list of nodes
|
|
// and writes the data to the stream (recursively)
|
|
GAPI_EXPORTS void serialize( IOStream& os
|
|
, const ade::Graph &g
|
|
, const cv::gimpl::Protocol &p
|
|
, const std::vector<ade::NodeHandle> &nodes);
|
|
|
|
// The top-level deserialization routineS.
|
|
// Unfortunately the deserialization is a two-step process:
|
|
// 1. First we decode a stream into some intermediate representation
|
|
// (called "GSerialized");
|
|
// 2. Then we produce an ade::Graph from this intermediate representation.
|
|
//
|
|
// An ade::Graph can't be produced from the stream immediately
|
|
// since every GCompiled object has its own unique ade::Graph, so
|
|
// we can't do it once and for all since every compilation process
|
|
// is individual and _is_ altering the ade::Graph state (structure and metadata).
|
|
// At the same time, we can't hold the reference to "is" within the GComputation
|
|
// forever since this input stream may be associated with an external resource
|
|
// and have side effects.
|
|
//
|
|
// Summarizing, the `deserialize()` happens *once per GComputation* immediately
|
|
// during the cv::gapi::deserialize<GComputation>(), and `reconstruct()` happens
|
|
// on every compilation process issued for this GComputation.
|
|
GAPI_EXPORTS GSerialized deserialize(IIStream& is);
|
|
GAPI_EXPORTS void reconstruct(const GSerialized &s, ade::Graph &g);
|
|
|
|
// Generic: variant serialization //////////////////////////////////////////////
|
|
namespace detail { // FIXME: breaks old code
|
|
template<typename V>
|
|
IOStream& put_v(IOStream&, const V&, std::size_t) {
|
|
GAPI_Assert(false && "variant>>: requested index is invalid");
|
|
};
|
|
template<typename V, typename X, typename... Xs>
|
|
IOStream& put_v(IOStream& os, const V& v, std::size_t x) {
|
|
return (x == 0u)
|
|
? os << cv::util::get<X>(v)
|
|
: put_v<V, Xs...>(os, v, x-1);
|
|
}
|
|
template<typename V>
|
|
IIStream& get_v(IIStream&, V&, std::size_t, std::size_t) {
|
|
GAPI_Assert(false && "variant<<: requested index is invalid");
|
|
}
|
|
template<typename V, typename X, typename... Xs>
|
|
IIStream& get_v(IIStream& is, V& v, std::size_t i, std::size_t gi) {
|
|
if (i == gi) {
|
|
X x{};
|
|
is >> x;
|
|
v = std::move(x);
|
|
return is;
|
|
} else return get_v<V, Xs...>(is, v, i+1, gi);
|
|
}
|
|
} // namespace detail FIXME: breaks old code
|
|
|
|
template<typename... Ts>
|
|
IOStream& operator<< (IOStream& os, const cv::util::variant<Ts...> &v) {
|
|
os << (uint32_t)v.index();
|
|
return detail::put_v<cv::util::variant<Ts...>, Ts...>(os, v, v.index());
|
|
}
|
|
template<typename... Ts>
|
|
IIStream& operator>> (IIStream& is, cv::util::variant<Ts...> &v) {
|
|
int idx = -1;
|
|
is >> idx;
|
|
GAPI_Assert(idx >= 0 && idx < (int)sizeof...(Ts));
|
|
return detail::get_v<cv::util::variant<Ts...>, Ts...>(is, v, 0u, idx);
|
|
}
|
|
|
|
// FIXME: Basic Stream implementaions //////////////////////////////////////////
|
|
|
|
// Basic in-memory stream implementations.
|
|
class GAPI_EXPORTS ByteMemoryOutStream final: public IOStream {
|
|
std::vector<char> m_storage;
|
|
|
|
//virtual IOStream& operator << (uint32_t) override;
|
|
//virtual IOStream& operator<< (uint32_t) final;
|
|
public:
|
|
const std::vector<char>& data() const;
|
|
|
|
virtual IOStream& operator<< (bool) override;
|
|
virtual IOStream& operator<< (char) override;
|
|
virtual IOStream& operator<< (unsigned char) override;
|
|
virtual IOStream& operator<< (short) override;
|
|
virtual IOStream& operator<< (unsigned short) override;
|
|
virtual IOStream& operator<< (int) override;
|
|
//virtual IOStream& operator<< (std::size_t) override;
|
|
virtual IOStream& operator<< (float) override;
|
|
virtual IOStream& operator<< (double) override;
|
|
virtual IOStream& operator<< (const std::string&) override;
|
|
virtual IOStream& operator<< (uint32_t) override;
|
|
virtual IOStream& operator<< (uint64_t) override;
|
|
};
|
|
|
|
class GAPI_EXPORTS ByteMemoryInStream final: public IIStream {
|
|
const std::vector<char>& m_storage;
|
|
size_t m_idx = 0u;
|
|
|
|
void check(std::size_t n) { (void) n; GAPI_DbgAssert(m_idx+n-1 < m_storage.size()); }
|
|
uint32_t getU32() { uint32_t v{}; *this >> v; return v; };
|
|
|
|
//virtual IIStream& operator>> (uint32_t &) final;
|
|
|
|
public:
|
|
explicit ByteMemoryInStream(const std::vector<char> &data);
|
|
|
|
virtual IIStream& operator>> (bool &) override;
|
|
virtual IIStream& operator>> (std::vector<bool>::reference) override;
|
|
virtual IIStream& operator>> (char &) override;
|
|
virtual IIStream& operator>> (unsigned char &) override;
|
|
virtual IIStream& operator>> (short &) override;
|
|
virtual IIStream& operator>> (unsigned short &) override;
|
|
virtual IIStream& operator>> (int &) override;
|
|
virtual IIStream& operator>> (float &) override;
|
|
virtual IIStream& operator>> (double &) override;
|
|
//virtual IIStream& operator>> (std::size_t &) override;
|
|
virtual IIStream& operator >> (uint32_t &) override;
|
|
virtual IIStream& operator >> (uint64_t &) override;
|
|
virtual IIStream& operator>> (std::string &) override;
|
|
};
|
|
|
|
GAPI_EXPORTS void serialize(IOStream& os, const cv::GMetaArgs &ma);
|
|
GAPI_EXPORTS void serialize(IOStream& os, const cv::GRunArgs &ra);
|
|
GAPI_EXPORTS GMetaArgs meta_args_deserialize(IIStream& is);
|
|
GAPI_EXPORTS GRunArgs run_args_deserialize(IIStream& is);
|
|
|
|
} // namespace s11n
|
|
} // namespace gapi
|
|
} // namespace cv
|
|
|
|
#endif // OPENCV_GAPI_COMMON_SERIALIZATION_HPP
|