/*
 * Acid Cam v2 - Qt/OpenCV Edition
 * written by Jared Bruni ( http://lostsidedead.com )
 * (C) 2017 GPL
 */

#ifndef __MAIN_WINDOW_H__
#define __MAIN_WINDOW_H__

#include "qtheaders.h"
#include "new_dialog.h"
#include "display_window.h"
#include "playback_thread.h"
#include "search_box.h"
#include "goto_window.h"
#include "chroma_window.h"
#include "user_define.h"
#include "dl-man.h"
#include "image_window.h"
#include "options_window.h"
#include "controller.h"
#include "color_range.h"
#include "slitscan_win.h"
#include "ffmpeg_write.h"

class SearchWindow;
class ChromaWindow;

class AC_MainWindow : public QMainWindow {
    Q_OBJECT
public:
    AC_MainWindow(QWidget *parent = 0);
    ~AC_MainWindow();
    void Log(const QString &s);
    bool startCamera(int res, int dev, const QString &outdir, bool record, int type,
                     bool useFFmpeg = false, FFmpegCodec codec = FFmpegCodec::LIBX264, int crf = 23);
    bool startVideo(const QString &filename, const QString &outdir, bool record, bool png_record, int type,
                    bool useFFmpeg = false, FFmpegCodec codec = FFmpegCodec::LIBX264, int crf = 23, bool muxAudio = true);
    QListWidget /**filters,*/ *custom_filters;
    QPushButton *btn_add, *btn_remove, *btn_moveup, *btn_movedown,*btn_load, *btn_save, *btn_sub, *btn_clr;
    QTextEdit *log_text;
    QCheckBox *chk_negate, *use_settings;
    QComboBox *combo_rgb;
    QSlider *slide_r, *slide_g, *slide_b, *slide_bright, *slide_gamma, *slide_saturation;
    QProgressBar *progress_bar;
    QComboBox *color_maps, *filters, *menu_cat;
    QMenu *file_menu, *controls_menu, *help_menu, *options, *movement, *speed_menu;
    QAction *file_exit, *file_new_capture, *file_new_video;
    QAction *controls_snapshot, *controls_pause, *controls_step, *controls_stop, *controls_setimage,*controls_setkey,*controls_showvideo, *clear_images, *reset_filters;
    QAction *help_about;
    QAction *open_search;
    QAction *in_out_increase;
    QAction *in_out;
    QAction *out_reset;
    QAction *speed_action_items[7];
    QMenu *image_menu;
    QAction *flip1, *flip2, *flip3, *noflip;
    QAction *clear_sub;
    QAction *clear_image;
    QAction *repeat_v;
    QAction *fade_on;
    QAction *select_key;
    QAction *set_newnames;
    QAction *show_fullscreen;
    QAction *show_image_window;
    QAction *show_options_window;
    QAction *show_control_window;
    QAction *select_random_filter;
    QAction *select_next_filter, *select_prev_filter;
    QAction *cycle_custom;
    QAction *show_glDisplay;
    QAction *show_range;
    QAction *show_slit;
    QTimer *joy_timer;
    double speed_actions[7];
    QRadioButton *filter_single, *filter_custom;
    void updateList();
    void setSubFilter(const QString &num);
    void setFrameIndex(int i);
    void resetMenu();
    void setSpeedIndex(int index);
    void setOptionString(std::string op, std::string value);
    void setProcMode(int index);
    bool checkAdd(QString str);
    
    Playback *playback;
    
public slots:
    void showRange();
    void chk_Joystick();
    void addClicked();
    void rmvClicked();
    void upClicked();
    void setSub();
    void downClicked();
    void file_Exit();
    void file_NewVideo();
    void file_NewCamera();
    void controls_Stop();
    void controls_Snap();
    void controls_Pause();
    void controls_Step();
    void controls_SetImage();
    void controls_ShowVideo();
    void controls_SetKey();
    void controls_Reset();
    void controls_ShowDisp2();
    void help_About();
    void updateFrame(QImage img);
    void stopRecording();
    void resetIndex();
    void chk_Clicked();
    void cb_SetIndex(int index);
    void frameInc();
    void slideChanged(int pos);
    void colorChanged(int pos);
    void colorMapChanged(int pos);
    void comboFilterChanged(int pos);
    void setFilterSingle();
    void setFilterCustom();
    void openSearch();
    void movementOption1();
    void movementOption2();
    void movementOption3();
    void speed1();
    void speed2();
    void speed3();
    void speed4();
    void speed5();
    void speed6();
    void speed7();
    void showGLDisplay();
    void flip1_action();
    void flip2_action();
    void flip3_action();
    void noflip_action();
    void clear_subfilter();
    void clear_img();
    void repeat_vid();
    void setFade();
    void openColorWindow();
    void menuFilterChanged(int index);
    void show_Favorites();
    void load_CustomFile();
    void save_CustomFile();
    void showFull();
    void showImageWindow();
    void showPrefWindow();
    void setRandomFilterValue();
    void setCustomCycle_Menu();
    void next_filter();
    void prev_filter();
    void showSlit();
    void onFFmpegFinished(QString tempFile, QString sourceFile, QString outputFile);
    void customSearchChanged(const QString &text);
    void addSearchResult();
     
private:
    void createControls();
    void createMenu();
    DisplayWindow *disp;
    DisplayWindow *disp2;
    CaptureCamera *cap_camera;
    CaptureVideo *cap_video;
    SearchWindow *search_box;
    ChromaWindow *chroma_window;
    DefineWindow *define_window;
    ImageWindow  *image_window;
    GotoWindow *goto_window;
    OptionsWindow *pref_window;
    SlitScanWindow *slit_win;
    ColorRangeWindow *color_range_window;
    cv::VideoCapture capture_camera, capture_video;
    cv::VideoWriter writer;
    unsigned long video_frames;
    double video_fps;
    bool paused, recording, step_frame;
    QString video_file_name;
    QString output_directory;
    bool take_snapshot;
    unsigned long file_pos, frame_index;
    VideoMode programMode;
    void buildVector(std::vector<FilterValue> &v);
    bool loading;
    int speed_index;
    cv::ocl::Context context;
#ifndef DISABLE_JOYSTICK
    Controller controller;
#endif
    QSettings *settings;
    std::vector<std::string> draw_strings;
    QLineEdit *custom_search_text;
    QComboBox *custom_search_combo;
    QPushButton *custom_search_add;
    
};

extern const char *filer_names[];
extern std::unordered_map<std::string, FilterValue> filter_map;
void generate_map();

#endif