producer.hpp

Go to the documentation of this file.

#ifndef OPENPOSE_PRODUCER_PRODUCER_HPP
#define OPENPOSE_PRODUCER_PRODUCER_HPP
 
#include <openpose/3d/cameraParameterReader.hpp>
#include <openpose/core/common.hpp>
#include <openpose/producer/enumClasses.hpp>
 
namespace op
{
    class OP_API Producer
    {
    public:
        explicit Producer(const ProducerType type, const std::string& cameraParameterPath, const bool undistortImage,
                          const int mNumberViews);
 
        virtual ~Producer();
 
        Matrix getFrame();
 
        std::vector<Matrix> getFrames();
 
        virtual std::vector<Matrix> getCameraMatrices();
 
        virtual std::vector<Matrix> getCameraExtrinsics();
 
        virtual std::vector<Matrix> getCameraIntrinsics();
 
        virtual std::string getNextFrameName() = 0;
 
        void setProducerFpsMode(const ProducerFpsMode fpsMode);
 
        inline ProducerType getType()
        {
            return mType;
        }
 
        virtual bool isOpened() const = 0;
 
        virtual void release() = 0;
 
        virtual double get(const int capProperty) = 0;
 
        virtual void set(const int capProperty, const double value) = 0;
 
        double get(const ProducerProperty property);
 
        void set(const ProducerProperty property, const double value);
 
    protected:
        void checkFrameIntegrity(Matrix& frame);
 
        void ifEndedResetOrRelease();
 
        void keepDesiredFrameRate();
 
        virtual Matrix getRawFrame() = 0;
 
        virtual std::vector<Matrix> getRawFrames() = 0;
 
    private:
        const ProducerType mType;
        ProducerFpsMode mProducerFpsMode;
        std::array<double, (int)ProducerProperty::Size> mProperties;
        unsigned int mNumberEmptyFrames;
        // For ProducerFpsMode::OriginalFps
        bool mTrackingFps;
        unsigned long long mFirstFrameTrackingFps;
        unsigned long long mNumberFramesTrackingFps;
        unsigned int mNumberSetPositionTrackingFps;
        std::chrono::high_resolution_clock::time_point mClockTrackingFps;
        // Camera parameters
        CameraParameterReader mCameraParameterReader;
 
        DELETE_COPY(Producer);
    };
 
    OP_API std::shared_ptr<Producer> createProducer(
        const ProducerType producerType = ProducerType::None, const std::string& producerString = "",
        const Point<int>& cameraResolution = Point<int>{-1,-1},
        const std::string& cameraParameterPath = "models/cameraParameters/", const bool undistortImage = true,
        const int numberViews = -1);
}
 
#endif // OPENPOSE_PRODUCER_PRODUCER_HPP