// -----------------------------------------------------------------------------------------
// x264guiEx/x265guiEx/svtAV1guiEx/ffmpegOut/QSVEnc/NVEnc/VCEEnc by rigaya
// -----------------------------------------------------------------------------------------
// The MIT License
//
// Copyright (c) 2010-2022 rigaya
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.
//
// --------------------------------------------------------------------------------------------

#ifndef _AUO_CONF_H_
#define _AUO_CONF_H_

#define WIN32_LEAN_AND_MEAN
#define NOMINMAX
#include <Windows.h>
#include "auo.h"
#include "auo_mes.h"
#include "auo_options.h"
#include "rgy_tchar.h"
#include "rgy_util.h"
#include <json_fwd.hpp>

const int CONF_INITIALIZED = 1;

enum {
    TMP_DIR_OUTPUT = 0,
    TMP_DIR_SYSTEM = 1,
    TMP_DIR_CUSTOM = 2,
};

enum : DWORD {
    RUN_BAT_NONE           = 0x00,
    RUN_BAT_BEFORE_PROCESS = 0x01,
    RUN_BAT_AFTER_PROCESS  = 0x02,
    RUN_BAT_BEFORE_AUDIO   = 0x04,
    RUN_BAT_AFTER_AUDIO    = 0x08,
};

static inline int get_run_bat_idx(DWORD flag) {
    DWORD ret;
    _BitScanForward(&ret, flag);
    return (int)ret;
}

#if ENCODER_X264
static const char *const CONF_NAME_OLD_1 = "x264guiEx ConfigFile";
static const char *const CONF_NAME_OLD_2 = "x264guiEx ConfigFile v2";
static const char *const CONF_NAME_OLD_3 = "x264guiEx ConfigFile v3";
static const char *const CONF_NAME_JSON  = "x264guiEx ConfigFile v3 json";
static const char *const CONF_NAME       = CONF_NAME_OLD_3;
#elif ENCODER_X265
static const char *const CONF_NAME_OLD1  = "x265guiEx ConfigFile";
static const char *const CONF_NAME_OLD2  = "x264/x265guiEx ConfigFile";
static const char *const CONF_NAME_OLD3  = "x265guiEx ConfigFile v2";
static const char *const CONF_NAME_OLD4  = "x265guiEx ConfigFile v3";
static const char *const CONF_NAME_OLD5  = "x265guiEx ConfigFile v4";
static const char *const CONF_NAME_OLD6  = "x265guiEx ConfigFile v5";
static const char *const CONF_NAME_JSON  = "x264guiEx ConfigFile v5 json";
static const char *const CONF_NAME       = CONF_NAME_OLD6;
#elif ENCODER_SVTAV1
static const char *const CONF_NAME_OLD_1 = "svtAV1guiEx ConfigFile v1";
static const char *const CONF_NAME_OLD_2 = "svtAV1guiEx ConfigFile v2";
static const char *const CONF_NAME_JSON  = "svtAV1guiEx ConfigFile v2 json";
static const char *const CONF_NAME       = CONF_NAME_OLD_2;
#elif ENCODER_FFMPEG
static const char *const CONF_NAME_OLD_1 = "ffmpegOut ConfigFile";
static const char *const CONF_NAME_OLD_2 = "ffmpegOut ConfigFile v2";
static const char *const CONF_NAME_JSON  = "ffmpegOut ConfigFile v2 json";
static const char *const CONF_NAME       = CONF_NAME_OLD_2;
#elif ENCODER_QSV
#include "qsv_util.h"
#include "qsv_prm.h"

static const char *const CONF_NAME_OLD_1 = "QSVEnc ConfigFile";
static const char *const CONF_NAME_OLD_2 = "QSVEnc ConfigFile v2";
static const char *const CONF_NAME_OLD_3 = "QSVEnc ConfigFile v3";
static const char *const CONF_NAME_OLD_4 = "QSVEnc ConfigFile v4";
static const char *const CONF_NAME_OLD_5 = "QSVEnc ConfigFile v5";
static const char *const CONF_NAME_OLD_6 = "QSVEnc ConfigFile v6";
static const char *const CONF_NAME_OLD_7 = "QSVEnc ConfigFile v7";
static const char *const CONF_NAME_JSON  = "QSVEnc ConfigFile v7 json";
static const char *const CONF_NAME       = CONF_NAME_OLD_7;
#elif ENCODER_NVENC
#include "NVEncParam.h"

static const char *const CONF_NAME_OLD_1 = "NVEnc ConfigFile";
static const char *const CONF_NAME_OLD_2 = "NVEnc ConfigFile v2";
static const char *const CONF_NAME_OLD_3 = "NVEnc ConfigFile v3";
static const char *const CONF_NAME_OLD_4 = "NVEnc ConfigFile v4";
static const char *const CONF_NAME_OLD_5 = "NVEnc ConfigFile v5";
static const char *const CONF_NAME_JSON  = "NVEnc ConfigFile v5 json";
static const char *const CONF_NAME       = CONF_NAME_OLD_5;
#elif ENCODER_VCEENC
#include "vce_param.h"

static const char *const CONF_NAME_OLD_1 = "VCEEnc ConfigFile v3";
static const char *const CONF_NAME_OLD_2 = "VCEEnc ConfigFile v4";
static const char *const CONF_NAME_JSON  = "VCEEnc ConfigFile v4 json";
static const char *const CONF_NAME       = CONF_NAME_OLD_2;
#else
static_assert(false);
#endif

const int CONF_NAME_BLOCK_LEN            = 32;
const int CONF_BLOCK_MAX                 = 32;
const int CONF_BLOCK_COUNT               = 5; //最大 CONF_BLOCK_MAXまで
const int CONF_HEAD_SIZE                 = (3 + CONF_BLOCK_MAX) * sizeof(int) + CONF_BLOCK_MAX * sizeof(size_t) + CONF_NAME_BLOCK_LEN;

static const wchar_t *const CONF_LAST_OUT   = L"前回出力.stg";

typedef struct {
    WCHAR *text;
    AuoMes mes;
    DWORD value;
} PRIORITY_CLASS;

const DWORD AVIUTLSYNC_PRIORITY_CLASS = 0;

const PRIORITY_CLASS priority_table[] = {
    {L"AviutlSync",       AUO_CONF_PRIORITY_AVIUTLSYNC, AVIUTLSYNC_PRIORITY_CLASS   },
    {L"higher",           AUO_CONF_PRIORITY_HIGHER,     HIGH_PRIORITY_CLASS         },
    {L"high",             AUO_CONF_PRIORITY_HIGH,       ABOVE_NORMAL_PRIORITY_CLASS },
    {L"normal",           AUO_CONF_PRIORITY_NORMAL,     NORMAL_PRIORITY_CLASS       },
    {L"low",              AUO_CONF_PRIORITY_LOW,        BELOW_NORMAL_PRIORITY_CLASS },
    {L"lower",            AUO_CONF_PRIORITY_LOWER,      IDLE_PRIORITY_CLASS         },
    {L"",                 AUO_MES_UNKNOWN,              NORMAL_PRIORITY_CLASS       },
    {L"realtime(非推奨)", AUO_CONF_PRIORITY_REALTIME,   REALTIME_PRIORITY_CLASS     },
    {NULL,                AUO_MES_UNKNOWN, 0                           }
};

enum {
    CONF_ERROR_NONE = 0,
    CONF_ERROR_FILE_OPEN,
    CONF_ERROR_BLOCK_SIZE,
    CONF_ERROR_INVALID_FILENAME,
};

const int CMDEX_MAX_LEN = 2048;    //追加コマンドラインの最大長

enum {
    AMPLIMIT_FILE_SIZE     = 0x01, //自動マルチパス時、ファイルサイズのチェックを行う
    AMPLIMIT_BITRATE_UPPER = 0x02, //自動マルチパス時、ビットレート上限のチェックを行う
    AMPLIMIT_BITRATE_LOWER = 0x04, //自動マルチパス時、ビットレート下限のチェックを行う
};

enum {
    CHECK_KEYFRAME_NONE    = 0x00,
    CHECK_KEYFRAME_AVIUTL  = 0x01, //Aviutlのキーフレームフラグをチェックする
    CHECK_KEYFRAME_CHAPTER = 0x02, //チャプターの位置にキーフレームを設定する
};

enum {
    AUDIO_DELAY_CUT_NONE         = 0, //音声エンコード遅延の削除を行わない
    AUDIO_DELAY_CUT_DELETE_AUDIO = 1, //音声エンコード遅延の削除を音声の先頭を削除することで行う
    AUDIO_DELAY_CUT_ADD_VIDEO    = 2, //音声エンコード遅延の削除を映像を先頭に追加することで行う
    AUDIO_DELAY_CUT_EDTS         = 3, //音声エンコード遅延の削除をedtsを用いて行う
};

#if ENCODER_SVTAV1
typedef struct CONF_ENC_PRM {
    TCHAR cmd[MAX_CMD_LEN];
    int sar_x;
    int sar_y;
} CONF_ENC_PRM;
#elif ENCODER_FFMPEG
typedef struct CONF_ENC {
    BOOL    use_highbit_depth;
    int     output_csp;
    int     pass;
    BOOL    use_auto_npass;
    int     auto_npass;
    int     bitrate;
    BOOL    interlaced;
    BOOL    audio_input;
    TCHAR   outext[MAX_APPENDIX_LEN];   //出力拡張子
    TCHAR   incmd[256];                 //入力オプション
} CONF_ENC;
#elif ENCODER_QSV || ENCODER_NVENC || ENCODER_VCEENC
typedef struct CONF_ENC {
    TCHAR cmd[MAX_CMD_LEN];
    RGY_CODEC codec_rgy;
    BOOL resize_enable;
    int resize_width;
    int resize_height;
} CONF_ENC;
#endif

typedef struct CONF_VIDEO {
    BOOL   afs;                      //自動フィールドシフトの使用
    BOOL   afs_bitrate_correction;   //afs & 2pass時、ドロップ数に応じてビットレートを補正
    BOOL   auo_tcfile_out;           //auo側でタイムコードを出力する
    DWORD  check_keyframe;           //キーフレームチェックを行う (CHECK_KEYFRAME_xxx)
    int    priority;                 //x264のCPU優先度(インデックス)
    TCHAR stats[MAX_PATH_LEN];     //x264用ステータスファイルの場所
    TCHAR tcfile_in[MAX_PATH_LEN]; //x264 tcfile-in用タイムコードファイルの場所
    TCHAR cqmfile[MAX_PATH_LEN];   //x264 cqmfileの場所
    TCHAR cmdex[CMDEX_MAX_LEN];    //追加コマンドライン
    DWORD  amp_check;                //自動マルチパス時のチェックの種類(AMPLIMIT_FILE_SIZE/AMPLIMIT_BITRATE)
    double amp_limit_file_size;      //自動マルチパス時のファイルサイズ制限(MB)
    double amp_limit_bitrate_upper;  //自動マルチパス時のビットレート上限(kbps)
    double amp_limit_bitrate_lower;  //自動マルチパス時のビットレート下限(kbps)
    BOOL   input_as_lw48;            //LW48モード
    TCHAR  parallel_div_info[64];    //プロセス並列モード時に使用する情報
    TCHAR  analysis_file[MAX_PATH_LEN]; //x265用ステータスファイルの場所
    BOOL   sync_process_affinity;     //プロセスアフィニティをAviutlと同期
} CONF_VIDEO; //動画用設定(x264以外)

typedef struct {
    int  encoder;             //使用する音声エンコーダ
    int  enc_mode;            //使用する音声エンコーダの設定
    int  bitrate;             //ビットレート指定モード
    BOOL use_2pass;           //音声2passエンコードを行う
    BOOL use_wav;             //パイプを使用せず、wavを出力してエンコードを行う
    BOOL faw_check;           //FAWCheckを行う
    int  priority;            //音声エンコーダのCPU優先度(インデックス)
    BOOL minimized;           //音声エンコーダを最小化で実行
    int  aud_temp_dir;        //音声専用一時フォルダ
    int  audio_encode_timing; //音声を先にエンコード
    int  delay_cut;           //エンコード遅延の削除
} CONF_AUDIO_BASE; //音声用設定

typedef struct CONF_AUDIO {
    CONF_AUDIO_BASE ext;
    CONF_AUDIO_BASE in;
    BOOL use_internal;
} CONF_AUDIO; //音声用設定

typedef struct CONF_MUX {
    BOOL disable_mp4ext;  //mp4出力時、外部muxerを使用する
    BOOL disable_mkvext;  //mkv出力時、外部muxerを使用する
    int  mp4_mode;        //mp4 外部muxer用追加コマンドの設定
    int  mkv_mode;        //mkv 外部muxer用追加コマンドの設定
    BOOL minimized;       //muxを最小化で実行
    int  priority;        //mux優先度(インデックス)
    int  mp4_temp_dir;    //mp4box用一時ディレクトリ
    BOOL apple_mode;      //Apple用モード(mp4系専用)
    BOOL use_internal;    //内蔵muxerの使用
    int  internal_mode;   //内蔵muxer用のオプション
} CONF_MUX; //muxer用設定

typedef struct CONF_OTHER {
    BOOL  disable_guicmd;         //GUIによるコマンドライン生成を停止(CLIモード)
    int   temp_dir;               //一時ディレクトリ
    BOOL  out_audio_only;         //音声のみ出力
    TCHAR notes[128];           //メモ
    DWORD run_bat;                //バッチファイルを実行するかどうか (RUN_BAT_xxx)
    DWORD dont_wait_bat_fin;      //バッチファイルの処理終了待機をするかどうか (RUN_BAT_xxx)
    union {
        TCHAR batfiles[4][MAX_PATH_LEN];     //バッチファイルのパス
        struct {
            TCHAR before_process[MAX_PATH_LEN]; //エンコ前バッチファイルのパス
            TCHAR after_process[MAX_PATH_LEN];  //エンコ後バッチファイルのパス
            TCHAR before_audio[MAX_PATH_LEN];   //音声エンコ前バッチファイルのパス
            TCHAR after_audio[MAX_PATH_LEN];    //音声エンコ後バッチファイルのパス
        } batfile;
    };
} CONF_OTHER;

struct CONF_GUIEX_HEADER {
    char        conf_name[CONF_NAME_BLOCK_LEN];  //保存時に使用
    int         size_all;                        //保存時: CONF_GUIEXの全サイズ / 設定中、エンコ中: CONF_INITIALIZED
    int         head_size;                       //ヘッダ部分の全サイズ
    int         block_count;                     //ヘッダ部を除いた設定のブロック数
    int         block_size[CONF_BLOCK_MAX];      //各ブロックのサイズ
    uint32_t    block_head_p[CONF_BLOCK_MAX];    //各ブロックのポインタ位置
};

#pragma pack(push, 1)

typedef struct CONF_OTHER_OLD {
    int   temp_dir;               //一時ディレクトリ
    BOOL  out_audio_only;         //音声のみ出力
    char  notes[128];             //メモ
    DWORD run_bat;                //バッチファイルを実行するかどうか (RUN_BAT_xxx)
    DWORD dont_wait_bat_fin;      //バッチファイルの処理終了待機をするかどうか (RUN_BAT_xxx)
    union {
        char batfiles[4][512];        //バッチファイルのパス
        struct {
            char before_process[512]; //エンコ前バッチファイルのパス
            char after_process[512];  //エンコ後バッチファイルのパス
            char before_audio[512];   //音声エンコ前バッチファイルのパス
            char after_audio[512];    //音声エンコ後バッチファイルのパス
        } batfile;
    };
} CONF_OTHER_OLD;

#if ENCODER_QSV || ENCODER_NVENC || ENCODER_VCEENC
typedef struct CONF_ENC_OLD {
    RGY_CODEC codec_rgy;
    int reserved[128];
#if ENCODER_QSV
    char reserved3[1024];
#endif
    char cmd[3072];
    char cmdex[512];
    char reserved2[512];
} CONF_ENC_OLD;

typedef struct CONF_VIDEO_OLD {
    BOOL afs;                      //自動フィールドシフトの使用
    BOOL auo_tcfile_out;           //auo側でタイムコードを出力する
#if ENCODER_QSV
    int  reserved[2];
#endif
    BOOL resize_enable;
    int resize_width;
    int resize_height;
} CONF_VIDEO_OLD;

typedef struct CONF_MUX_OLD {
    BOOL disable_mp4ext;  //mp4出力時、外部muxerを使用する
    BOOL disable_mkvext;  //mkv出力時、外部muxerを使用する
    int  mp4_mode;        //mp4 外部muxer用追加コマンドの設定
    int  mkv_mode;        //mkv 外部muxer用追加コマンドの設定
    BOOL minimized;       //muxを最小化で実行
    int  priority;        //mux優先度(インデックス)
    int  mp4_temp_dir;    //mp4box用一時ディレクトリ
    BOOL apple_mode;      //Apple用モード(mp4系専用)
    BOOL unused;
    int  unused2;
    BOOL use_internal;    //内蔵muxerの使用
    int  internal_mode;   //内蔵muxer用のオプション
} CONF_MUX_OLD; //muxer用設定


typedef struct CONF_GUIEX_OLD {
    CONF_GUIEX_HEADER header;
    CONF_ENC_OLD     enc;                            //x264についての設定
    CONF_VIDEO_OLD   vid;                            //その他動画についての設定
    CONF_AUDIO       aud;                         //音声についての設定
    CONF_MUX_OLD     mux;                            //muxについての設定
    CONF_OTHER_OLD   oth;                            //その他の設定
} CONF_GUIEX_OLD;
static_assert(sizeof(CONF_ENC_OLD) == (5636-(ENCODER_QSV?0:1024)));
static_assert(sizeof(CONF_GUIEX_OLD)==(8296-(ENCODER_QSV?0:1032)));
#endif
#pragma pack(pop)

typedef struct CONF_GUIEX {
    CONF_GUIEX_HEADER header;
#if ENCODER_SVTAV1
    CONF_ENC_PRM enc;                             //エンコーダについての設定
#else
    CONF_ENC    enc;                             //エンコーダについての設定
#endif
    CONF_VIDEO  vid;                             //その他動画についての設定
    CONF_AUDIO  aud;                             //音声についての設定
    CONF_MUX    mux;                             //muxについての設定
    CONF_OTHER  oth;                             //その他の設定
} CONF_GUIEX;

class guiEx_config {
private:
    static const uint32_t conf_block_pointer[CONF_BLOCK_COUNT];
    static const int conf_block_data[CONF_BLOCK_COUNT];
#if ENCODER_QSV
    static void *convert_qsvstgv1_to_stgv3(void *_conf, int size);
    static void *convert_qsvstgv2_to_stgv3(void *_conf);
    static void *convert_qsvstgv3_to_stgv4(void *_conf);
    static void *convert_qsvstgv4_to_stgv5(void *_conf);
    static void *convert_qsvstgv5_to_stgv6(void *_conf);
#elif ENCODER_NVENC
    static int  stgv3_block_size();
    static void convert_nvencstg_to_nvencstgv4(CONF_GUIEX_OLD *conf, const void *dat);
    static void convert_nvencstgv2_to_nvencstgv3(void *dat);
    static void convert_nvencstgv2_to_nvencstgv4(CONF_GUIEX_OLD *conf, const void *dat);
    static void convert_nvencstgv3_to_nvencstgv4(CONF_GUIEX_OLD *conf, const void *dat);
#endif
    // ブロック別JSON変換関数
    static void video_to_json(nlohmann::json& j, const CONF_VIDEO& vid);                 //ビデオ設定をJSONに変換
    static void audio_to_json(nlohmann::json& j, const CONF_AUDIO& aud);                 //オーディオ設定をJSONに変換
    static void mux_to_json(nlohmann::json& j, const CONF_MUX& mux);                     //Mux設定をJSONに変換
    static void other_to_json(nlohmann::json& j, const CONF_OTHER& oth);                 //その他設定をJSONに変換
    
    static void json_to_video(const nlohmann::json& j, CONF_VIDEO& vid);                 //JSONからビデオ設定を復元
    static void json_to_audio(const nlohmann::json& j, CONF_AUDIO& aud);                 //JSONからオーディオ設定を復元
    static void json_to_mux(const nlohmann::json& j, CONF_MUX& mux);                     //JSONからMux設定を復元
    static void json_to_other(const nlohmann::json& j, CONF_OTHER& oth);                 //JSONからその他設定を復元
public:
    guiEx_config();
    static void write_conf_header(CONF_GUIEX_HEADER *conf_header);
    static std::string old_conf_to_json(const CONF_GUIEX_OLD *old_conf);
    static int  load_guiEx_conf(CONF_GUIEX *conf, const TCHAR *stg_file);       //設定をstgファイルから読み込み (バイナリ & JSON対応)
    static int  save_guiEx_conf(const CONF_GUIEX *conf, const TCHAR *stg_file); //設定をJSONファイルとして保存
    static int  load_guiEx_conf_legacy(CONF_GUIEX *conf, const TCHAR *stg_file); //旧形式のstgファイルから読み込み

    static std::string conf_to_json(const CONF_GUIEX *conf, int indent);                 //設定をJSON文字列に変換
    static bool json_to_conf(CONF_GUIEX *conf, const std::string &json_str);             //JSON文字列から設定を復元
};

static bool cnf_disable_guicmd(const CONF_OTHER *oth) {
#if ENCODER_QSV || ENCODER_NVENC || ENCODER_VCEENC
    return false;
#else
    return oth->disable_guicmd;
#endif    
}

void init_CONF_GUIEX(CONF_GUIEX *conf, BOOL use_highbit); //初期化し、デフォルトを設定

//出力ファイルの拡張子フィルタを作成
//filterがNULLならauoのOUTPUT_PLUGIN_TABLE用のフィルタを書き換える
void make_file_filter(char *filter, size_t nSize, int default_index);

void overwrite_aviutl_ini_file_filter(int idx);
void overwrite_aviutl_ini_auo_info();

#endif //_AUO_CONF_H_
