#ifndef RECORDER_H
#define RECORDER_H

#include <atomic>
#include <string>
#include <vector>

#include <gst/pbutils/pbutils.h>
#include <gst/app/gstappsrc.h>

class FrameBuffer;

/**
 * @brief The Recorder class defines the base class for all recorders
 * used to save images or videos from a frame buffer.
 *
 * The Mixer class calls addFrame() at each newly rendered frame for all of its recorder.
 */
class Recorder
{
public:
    Recorder();
    virtual ~Recorder() {}

    virtual void addFrame(FrameBuffer *frame_buffer, float dt) = 0;
    virtual void stop() { }
    virtual std::string info() { return ""; }

    inline bool finished() const { return finished_; }

protected:
    std::atomic<bool> finished_;
};

class PNGRecorder : public Recorder
{
    std::string     filename_;

public:

    PNGRecorder();
    void addFrame(FrameBuffer *frame_buffer, float);

};


class VideoRecorder : public Recorder
{
    std::string  filename_;

    // Frame buffer information
    FrameBuffer  *frame_buffer_;
    uint width_;
    uint height_;
    uint buf_size_;

    // operation
    std::atomic<bool> recording_;
    std::atomic<bool> accept_buffer_;

    // gstreamer pipeline
    GstElement   *pipeline_;
    GstAppSrc    *src_;
    GstBaseSink  *sink_;
    GstClockTime time_;
    GstClockTime timestamp_;
    GstClockTime frame_duration_;

    static void callback_need_data (GstAppSrc *src, guint length, gpointer user_data);
    static void callback_enough_data (GstAppSrc *src, gpointer user_data);

public:

    static const char* profile_name[4];
    static const std::vector<std::string> profile_description;

    VideoRecorder();
    ~VideoRecorder();

    void addFrame(FrameBuffer *frame_buffer, float dt);
    void stop() override;
    std::string info() override;

};


#endif // RECORDER_H