File size: 1,938 Bytes
7fc5a59 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | #ifndef OPENPOSE_GUI_GUI_HPP
#define OPENPOSE_GUI_GUI_HPP
#include <atomic>
#include <openpose/core/common.hpp>
#include <openpose/core/renderer.hpp>
#include <openpose/gui/frameDisplayer.hpp>
#include <openpose/pose/poseExtractorNet.hpp>
#include <openpose/face/faceExtractorNet.hpp>
#include <openpose/hand/handExtractorNet.hpp>
namespace op
{
class OP_API Gui
{
public:
Gui(const Point<int>& outputSize, const bool fullScreen,
const std::shared_ptr<std::atomic<bool>>& isRunningSharedPtr,
const std::shared_ptr<std::pair<std::atomic<bool>, std::atomic<int>>>& videoSeekSharedPtr = nullptr,
const std::vector<std::shared_ptr<PoseExtractorNet>>& poseExtractorNets = {},
const std::vector<std::shared_ptr<FaceExtractorNet>>& faceExtractorNets = {},
const std::vector<std::shared_ptr<HandExtractorNet>>& handExtractorNets = {},
const std::vector<std::shared_ptr<Renderer>>& renderers = {},
const DisplayMode displayMode = DisplayMode::Display2D);
virtual ~Gui();
virtual void initializationOnThread();
void setImage(const Matrix& cvMatOutput);
void setImage(const std::vector<Matrix>& cvMatOutputs);
virtual void update();
protected:
std::shared_ptr<std::atomic<bool>> spIsRunning;
DisplayMode mDisplayMode;
DisplayMode mDisplayModeOriginal;
private:
// Frames display
FrameDisplayer mFrameDisplayer;
// Other variables
std::vector<std::shared_ptr<PoseExtractorNet>> mPoseExtractorNets;
std::vector<std::shared_ptr<FaceExtractorNet>> mFaceExtractorNets;
std::vector<std::shared_ptr<HandExtractorNet>> mHandExtractorNets;
std::vector<std::shared_ptr<Renderer>> mRenderers;
std::shared_ptr<std::pair<std::atomic<bool>, std::atomic<int>>> spVideoSeek;
};
}
#endif // OPENPOSE_GUI_GUI_HPP
|