effect.h

/***

    Olive - Non-Linear Video Editor
    Copyright (C) 2019  Olive Team

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.

***/

#ifndef EFFECT_H
#define EFFECT_H

#include <QObject>
#include <QString>
#include <QVector>
#include <QColor>
#include <QOpenGLFunctions>
#include <QOpenGLShaderProgram>
#include <QOpenGLTexture>
#include <QMutex>
#include <QThread>
class QLabel;
class QWidget;
class CollapsibleWidget;
class QGridLayout;
class QPushButton;
class QMouseEvent;

class Clip;
class QXmlStreamReader;
class QXmlStreamWriter;
class Effect;
class EffectRow;
class CheckboxEx;

struct EffectMeta {
    QString name;
    QString category;
    QString filename;
    QString path;
    QString tooltip;
    int internal;
    int type;
    int subtype;
};
extern QVector<EffectMeta> effects;

double log_volume(double linear);
Effect* create_effect(Clip* c, const EffectMeta *em);
const EffectMeta* get_internal_meta(int internal_id, int type);

enum EffectType {
    EFFECT_TYPE_INVALID,
    EFFECT_TYPE_VIDEO,
    EFFECT_TYPE_AUDIO,
    EFFECT_TYPE_EFFECT,
    EFFECT_TYPE_TRANSITION
};

enum EffectKeyframeType {
    EFFECT_KEYFRAME_LINEAR,
    EFFECT_KEYFRAME_BEZIER,
    EFFECT_KEYFRAME_HOLD
};

enum EffectInternal {
    EFFECT_INTERNAL_TRANSFORM,
    EFFECT_INTERNAL_TEXT,
    EFFECT_INTERNAL_SOLID,
    EFFECT_INTERNAL_NOISE,
    EFFECT_INTERNAL_VOLUME,
    EFFECT_INTERNAL_PAN,
    EFFECT_INTERNAL_TONE,
    EFFECT_INTERNAL_SHAKE,
    EFFECT_INTERNAL_TIMECODE,
    EFFECT_INTERNAL_MASK,
    EFFECT_INTERNAL_FILLLEFTRIGHT,
    EFFECT_INTERNAL_VST,
    EFFECT_INTERNAL_CORNERPIN,
    EFFECT_INTERNAL_FREI0R,
    EFFECT_INTERNAL_COUNT
};

enum EffectBlendMode {
    BLEND_MODE_ADD,
    BLEND_MODE_AVERAGE,
    BLEND_MODE_COLORBURN,
    BLEND_MODE_COLORDODGE,
    BLEND_MODE_DARKEN,
    BLEND_MODE_DIFFERENCE,
    BLEND_MODE_EXCLUSION,
    BLEND_MODE_GLOW,
    BLEND_MODE_HARDLIGHT,
    BLEND_MODE_HARDMIX,
    BLEND_MODE_LIGHTEN,
    BLEND_MODE_LINEARBURN,
    BLEND_MODE_LINEARDODGE,
    BLEND_MODE_LINEARLIGHT,
    BLEND_MODE_MULTIPLY,
    BLEND_MODE_NEGATION,
    BLEND_MODE_NORMAL,
    BLEND_MODE_OVERLAY,
    BLEND_MODE_PHOENIX,
    BLEND_MODE_PINLIGHT,
    BLEND_MODE_REFLECT,
    BLEND_MODE_SCREEN,
    BLEND_MODE_SOFTLIGHT,
    BLEND_MODE_SUBSTRACT,
    BLEND_MODE_SUBTRACT,
    BLEND_MODE_VIVIDLIGHT,
    BLEND_MODE_COUNT
};

struct GLTextureCoords {
    int grid_size;

    int vertexTopLeftX;
    int vertexTopLeftY;
    int vertexTopLeftZ;
    int vertexTopRightX;
    int vertexTopRightY;
    int vertexTopRightZ;
    int vertexBottomLeftX;
    int vertexBottomLeftY;
    int vertexBottomLeftZ;
    int vertexBottomRightX;
    int vertexBottomRightY;
    int vertexBottomRightZ;

    float textureTopLeftX;
    float textureTopLeftY;
    float textureTopLeftQ;
    float textureTopRightX;
    float textureTopRightY;
    float textureTopRightQ;
    float textureBottomRightX;
    float textureBottomRightY;
    float textureBottomRightQ;
    float textureBottomLeftX;
    float textureBottomLeftY;
    float textureBottomLeftQ;

    int blendmode;
    float opacity;
};

const EffectMeta* get_meta_from_name(const QString& input);

qint16 mix_audio_sample(qint16 a, qint16 b);

#include "effectfield.h"
#include "effectrow.h"
#include "effectgizmo.h"

class Effect : public QObject {
    Q_OBJECT
public:
    Effect(Clip* c, const EffectMeta* em);
    ~Effect();
    Clip* parent_clip;
    const EffectMeta* meta;
    int id;
    QString name;
    CollapsibleWidget* container;

    EffectRow* add_row(const QString &name, bool savable = true, bool keyframable = true);
    EffectRow* row(int i);
    int row_count();

    EffectGizmo* add_gizmo(int type);
    EffectGizmo* gizmo(int i);
    int gizmo_count();

    bool is_enabled();
    void set_enabled(bool b);

    virtual void refresh();

    virtual Effect* copy(Clip* c);
    void copy_field_keyframes(Effect *e);

    virtual void load(QXmlStreamReader& stream);
    virtual void custom_load(QXmlStreamReader& stream);
    virtual void save(QXmlStreamWriter& stream);

    void load_from_string(const QByteArray &s);
    QByteArray save_to_string();

    // glsl handling
    bool is_open();
    void open();
    void close();
    bool is_glsl_linked();
    virtual void startEffect();
    virtual void endEffect();

    bool enable_shader;
    bool enable_coords;
    bool enable_superimpose;
    bool enable_image;

    int getIterations();
    void setIterations(int i);

    const char* ffmpeg_filter;

    virtual void process_image(double timecode, uint8_t* input, uint8_t* output, int size);
    virtual void process_shader(double timecode, GLTextureCoords&, int iteration);
    virtual void process_coords(double timecode, GLTextureCoords& coords, int data);
    virtual GLuint process_superimpose(double timecode);
    virtual void process_audio(double timecode_start, double timecode_end, quint8* samples, int nb_bytes, int channel_count);

    virtual void gizmo_draw(double timecode, GLTextureCoords& coords);
    void gizmo_move(EffectGizmo* sender, int x_movement, int y_movement, double timecode, bool done);
    void gizmo_world_to_screen();
    bool are_gizmos_enabled();
public slots:
    void field_changed();
private slots:
    void show_context_menu(const QPoint&);
    void delete_self();
    void move_up();
    void move_down();
    void save_to_file();
    void load_from_file();
protected:
    // glsl effect
    QOpenGLShaderProgram* glslProgram;
    QString vertPath;
    QString fragPath;

    // superimpose effect
    QImage img;
    QOpenGLTexture* texture;

    // enable effect to update constantly
    bool enable_always_update;
private:
    // superimpose effect
    QString script;

    bool isOpen;
    QVector<EffectRow*> rows;
    QVector<EffectGizmo*> gizmos;
    QGridLayout* ui_layout;
    QWidget* ui;
    bool bound;
    int iterations;

    // superimpose functions
    virtual void redraw(double timecode);
    bool valueHasChanged(double timecode);
    QVector<QVariant> cachedValues;
    void delete_texture();
    int get_index_in_clip();
    void validate_meta_path();
};

class EffectInit : public QThread {
public:
    EffectInit();
protected:
    void run();
};

#endif // EFFECT_H