VideoPlayer.hpp

#pragma once

#include <functional>
#include <opencv2/highgui.hpp>
#include <wx/panel.h>
#include <wx/timer.h>
#include <wx/stattext.h>
#include <wx/button.h>

namespace xv{

    class ImagePanel;

class VideoPlayer : public wxPanel
{
    friend class Thread;
public:
    VideoPlayer(wxWindow * parent,
        wxWindowID id = wxID_ANY,
        const wxPoint &pos = wxDefaultPosition,
        const wxSize &size = wxDefaultSize,
        long style = wxTAB_TRAVERSAL,
        const wxString &name = wxPanelNameStr
        );

    VideoPlayer(cv::VideoCapture* videoCapture);

    virtual ~VideoPlayer();

    int getCurrentFrameIdx();

    ImagePanel& getImage();

#pragma region control

    virtual bool open(const cv::String &filename);

    virtual bool open(int device);

    virtual void play(); 
    
    virtual void play(
        std::function<void(cv::Mat &)> callback,
        std::function<void(cv::Mat &)> postProcessCallback = [](cv::Mat &){});

    virtual void pause();

    virtual void stop();

    virtual void seek(int);

#pragma endregion control

#pragma region event handling
    virtual void onSliderMove(wxScrollEvent&);

    virtual void onSliderClickDown(wxMouseEvent&);

    virtual void onSliderClickUp(wxMouseEvent&);

    virtual void onPlayClick(wxCommandEvent&);

    virtual void onMouseMove(wxCommandEvent& evt);
#pragma endregion event handling

private:
    template <typename _Tp>
    bool open_(const _Tp &deviceOrFilename);

    static const int LABEL_UPDATE_INTERVAL;

    static const char* LABEL_DEFAULT_TEXT;

    std::function<void(cv::Mat &)>
        m_preProcessCallback = [](cv::Mat &){},
        m_postProcessCallback = [](cv::Mat &){};

    int 
        m_fps = 0,
        m_frameCount = 0;

    wxTimer m_timer;

    void sizeEvent(wxSizeEvent&);
    void onTimer(wxTimerEvent&);

    void showTimeLabel(wxStaticText* label,int frame);

    std::function<void(cv::Mat &)> m_processCallback = [](cv::Mat &){};

    enum playbackState{
        idle,   //no video loaded
        playing,//ongoing playback
        paused, //video is paused
        seeking //looking for a frame
    } 
    m_state = paused,
    m_previousState = paused;

    void setState(playbackState);

    
    cv::VideoCapture* m_videoCapture = NULL;
    ImagePanel* m_image;
    wxButton* m_playButton;
    wxSlider* m_slider;
    wxStaticText
        *m_positionLabel,
        *m_durationLabel;
    wxSizer *m_sizer;

    wxMutex m_mutex;
    class Thread : public wxThread{
    public:
        Thread(VideoPlayer* const);
        virtual void *Entry();
    private:
        VideoPlayer *m_player;
    } *m_thread;

    void init();

    DECLARE_EVENT_TABLE()
};

}//namespace