imageDirectoryReader.hpp

Go to the documentation of this file.

#ifndef OPENPOSE_PRODUCER_IMAGE_DIRECTORY_READER_HPP
#define OPENPOSE_PRODUCER_IMAGE_DIRECTORY_READER_HPP
 
#include <openpose/core/common.hpp>
#include <openpose/producer/producer.hpp>
 
namespace op
{
    class OP_API ImageDirectoryReader : public Producer
    {
    public:
        explicit ImageDirectoryReader(
            const std::string& imageDirectoryPath, const std::string& cameraParameterPath = "",
            const bool undistortImage = false, const int numberViews = -1);
 
        virtual ~ImageDirectoryReader();
 
        std::string getNextFrameName();
 
        inline bool isOpened() const
        {
            return (mFrameNameCounter >= 0);
        }
 
        inline void release()
        {
            mFrameNameCounter = {-1ll};
        }
 
        double get(const int capProperty);
 
        void set(const int capProperty, const double value);
 
    private:
        const std::string mImageDirectoryPath;
        const std::vector<std::string> mFilePaths;
        Point<int> mResolution;
        long long mFrameNameCounter;
 
        Matrix getRawFrame();
 
        std::vector<Matrix> getRawFrames();
 
        DELETE_COPY(ImageDirectoryReader);
    };
}
 
#endif // OPENPOSE_PRODUCER_IMAGE_DIRECTORY_READER_HPP