155 lines
4.0 KiB
C++
155 lines
4.0 KiB
C++
// 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) 2018 Intel Corporation
|
|
|
|
|
|
#ifndef OPENCV_GAPI_FLUID_BUFFER_HPP
|
|
#define OPENCV_GAPI_FLUID_BUFFER_HPP
|
|
|
|
#include <list>
|
|
#include <numeric> // accumulate
|
|
#include <ostream> // ostream
|
|
#include <cstdint> // uint8_t
|
|
|
|
#include <opencv2/gapi/opencv_includes.hpp>
|
|
#include <opencv2/gapi/gmat.hpp>
|
|
|
|
#include <opencv2/gapi/util/optional.hpp>
|
|
|
|
namespace cv {
|
|
namespace gapi {
|
|
namespace fluid {
|
|
|
|
struct Border
|
|
{
|
|
// This constructor is required to support existing kernels which are part of G-API
|
|
Border(int _type, cv::Scalar _val) : type(_type), value(_val) {}
|
|
|
|
int type;
|
|
cv::Scalar value;
|
|
};
|
|
|
|
using BorderOpt = util::optional<Border>;
|
|
|
|
bool operator == (const Border& b1, const Border& b2);
|
|
|
|
class GAPI_EXPORTS Buffer;
|
|
|
|
class GAPI_EXPORTS View
|
|
{
|
|
public:
|
|
struct Cache
|
|
{
|
|
std::vector<const uint8_t*> m_linePtrs;
|
|
GMatDesc m_desc;
|
|
int m_border_size = 0;
|
|
|
|
inline const uint8_t* linePtr(int index) const
|
|
{
|
|
// "out_of_window" check:
|
|
// user must not request the lines which are outside of specified kernel window
|
|
GAPI_DbgAssert(index >= -m_border_size
|
|
&& index < -m_border_size + static_cast<int>(m_linePtrs.size()));
|
|
return m_linePtrs[index + m_border_size];
|
|
}
|
|
};
|
|
|
|
const inline uint8_t* InLineB(int index) const // -(w-1)/2...0...+(w-1)/2 for Filters
|
|
{
|
|
return m_cache->linePtr(index);
|
|
}
|
|
|
|
template<typename T> const inline T* InLine(int i) const
|
|
{
|
|
const uint8_t* ptr = this->InLineB(i);
|
|
return reinterpret_cast<const T*>(ptr);
|
|
}
|
|
|
|
inline operator bool() const { return m_priv != nullptr; }
|
|
bool ready() const;
|
|
inline int length() const { return m_cache->m_desc.size.width; }
|
|
int y() const;
|
|
|
|
inline const GMatDesc& meta() const { return m_cache->m_desc; }
|
|
|
|
class GAPI_EXPORTS Priv; // internal use only
|
|
Priv& priv(); // internal use only
|
|
const Priv& priv() const; // internal use only
|
|
|
|
View();
|
|
View(std::unique_ptr<Priv>&& p);
|
|
View(View&& v);
|
|
View& operator=(View&& v);
|
|
~View();
|
|
|
|
private:
|
|
std::unique_ptr<Priv> m_priv;
|
|
const Cache* m_cache = nullptr;
|
|
};
|
|
|
|
class GAPI_EXPORTS Buffer
|
|
{
|
|
public:
|
|
struct Cache
|
|
{
|
|
std::vector<uint8_t*> m_linePtrs;
|
|
GMatDesc m_desc;
|
|
};
|
|
|
|
// Default constructor (executable creation stage,
|
|
// all following initialization performed in Priv::init())
|
|
Buffer();
|
|
// Scratch constructor (user kernels)
|
|
Buffer(const cv::GMatDesc &desc);
|
|
|
|
// Constructor for intermediate buffers (for tests)
|
|
Buffer(const cv::GMatDesc &desc,
|
|
int max_line_consumption, int border_size,
|
|
int skew,
|
|
int wlpi,
|
|
BorderOpt border);
|
|
// Constructor for in/out buffers (for tests)
|
|
Buffer(const cv::Mat &data, bool is_input);
|
|
~Buffer();
|
|
Buffer& operator=(Buffer&&);
|
|
|
|
inline uint8_t* OutLineB(int index = 0)
|
|
{
|
|
return m_cache->m_linePtrs[index];
|
|
}
|
|
|
|
template<typename T> inline T* OutLine(int index = 0)
|
|
{
|
|
uint8_t* ptr = this->OutLineB(index);
|
|
return reinterpret_cast<T*>(ptr);
|
|
}
|
|
|
|
int y() const;
|
|
|
|
int linesReady() const;
|
|
void debug(std::ostream &os) const;
|
|
inline int length() const { return m_cache->m_desc.size.width; }
|
|
int lpi() const; // LPI for WRITER
|
|
|
|
inline const GMatDesc& meta() const { return m_cache->m_desc; }
|
|
|
|
View mkView(int borderSize, bool ownStorage);
|
|
void addView(const View* v);
|
|
|
|
class GAPI_EXPORTS Priv; // internal use only
|
|
Priv& priv(); // internal use only
|
|
const Priv& priv() const; // internal use only
|
|
|
|
private:
|
|
std::unique_ptr<Priv> m_priv;
|
|
const Cache* m_cache;
|
|
};
|
|
|
|
} // namespace cv::gapi::fluid
|
|
} // namespace cv::gapi
|
|
} // namespace cv
|
|
|
|
#endif // OPENCV_GAPI_FLUID_BUFFER_HPP
|