preview_dialog.h

#pragma once
#include "resource.h"
#include "helpers/DarkMode.h"
#include "icon_map.h"
#include "CGdiPlusBitmap.h"
#include "foo_discogs.h"
#include "tag_writer.h"

#include "preview_ILO.h"
#include "tristate_skip_artwork.h"

class CPreviewTagsDialog : public MyCDialogImpl<CPreviewTagsDialog>,
	public CDialogResize<CPreviewTagsDialog>, public CMessageFilter,
	public ILOD_preview {

public:

	CListControlOwnerData* ilo_get_uilist() override {
		return dynamic_cast<CListControlOwnerData*>(&m_uilist);
	}

	CPreviewTagsDialog(HWND p_parent, TagWriter_ptr tag_writer)
		: m_tag_writer(tag_writer), m_results_list(NULL), m_uilist(this), m_tristate(this), conf(CONF), m_preview_bitmap(NULL) {

		conf.SetName("PreviewDlg");
		g_discogs->preview_tags_dialog = this;
	}

	~CPreviewTagsDialog();

	enum {
		IDD = IDD_DIALOG_PREVIEW_TAGS,
	};

	enum {
		ID_PREVIEW_CMD_BACK = 1,
		ID_PREVIEW_CMD_SKIP_ARTWORK,
		ID_PREVIEW_CMD_SHOW_STATS,
		ID_PREVIEW_CMD_RESULTS_VIEW,
		ID_PREVIEW_CMD_DIFF_VIEW,
		ID_PREVIEW_CMD_ORI_VIEW,

		ID_PREVIEW_CMD_EDIT_RESULT_TAG,
		ID_PREVIEW_CMD_EDIT_RESULT_TAGINP,
		ID_PREVIEW_CMD_WRITE_TAGS,
		ID_PREVIEW_CMD_WRITE_TAGS_MASK,
		ID_PREVIEW_CMD_WRITE_TAGS_MASK_FORCE_WU,
		ID_PREVIEW_CMD_COPY,
		ID_SELECT_ALL,
		ID_INVERT_SELECTION,

		ID_URL_RELEASE,
		ID_URL_MASTER_RELEASE,
		ID_URL_ARTIST,
		ID_URL_ARTISTS,
	};

	enum {
		FLG_TAGMAP_DLG_ATTACHED = 1 << 0,
		FLG_TAGMAP_DLG_OPENED = 1 << 1,
	};

	virtual BOOL PreTranslateMessage(MSG* pMsg) override {
		return ::IsDialogMessage(m_hWnd, pMsg);
	}

	void load_column_layout();
	bool build_current_cfg();
	void pushcfg();
	void showtitle();

#pragma warning( push )
#pragma warning( disable : 26454 )

	MY_BEGIN_MSG_MAP(CPreviewTagsDialog)
		
		MESSAGE_HANDLER(WM_INITDIALOG, OnInitDialog)
		MESSAGE_HANDLER(WM_CONTEXTMENU, OnContextMenu)
		MESSAGE_HANDLER(WM_DESTROY, OnDestroy)
		
		COMMAND_ID_HANDLER(IDCANCEL, OnCancel)
		COMMAND_ID_HANDLER(IDC_BTN_BACK, OnButtonBack)
		COMMAND_ID_HANDLER(IDC_BTN_TAG_MAPPINGS, OnButtonEditTagMappings)
		COMMAND_ID_HANDLER(IDC_BTN_WRITE_TAGS, OnButtonWriteTags)
		
		COMMAND_ID_HANDLER(IDC_CHK_REPLACE_ANV, OnCheckReplaceANVs)
		COMMAND_ID_HANDLER(IDC_CHK_PREV_DLG_SHOW_STATS, OnCheckPreviewShowStats)
		COMMAND_ID_HANDLER(IDC_CHK_SKIP_ARTWORK, OnCheckSkipArtwork)
		COMMAND_ID_HANDLER(IDC_CHK_BIND_TAGS_DLG, OnCheckAttachMappingPanel)

		COMMAND_ID_HANDLER(IDC_VIEW_NORMAL, OnChangePreviewMode)
		COMMAND_ID_HANDLER(IDC_VIEW_DIFFERENCE, OnChangePreviewMode)
		COMMAND_ID_HANDLER(IDC_VIEW_ORIGINAL, OnChangePreviewMode)
		COMMAND_ID_HANDLER(IDC_VIEW_DEBUG, OnChangePreviewMode)

		NOTIFY_HANDLER_EX(IDC_PREVIEW_LIST, NM_DBLCLK, OnListDoubleClick)
		NOTIFY_HANDLER_EX(IDC_PREVIEW_LIST, LVN_KEYDOWN, OnListKeyDown)
		NOTIFY_HANDLER(IDC_PREVIEW_LIST, NM_CUSTOMDRAW, OnCustomDraw)
		CHAIN_MSG_MAP(CDialogResize<CPreviewTagsDialog>)

	MY_END_MSG_MAP()

#pragma warning(pop)

	BEGIN_DLGRESIZE_MAP(CPreviewTagsDialog)
		DLGRESIZE_CONTROL(IDC_ALBUM_ART, DLSZ_MOVE_X)
		DLGRESIZE_CONTROL(IDC_CHK_REPLACE_ANV, DLSZ_MOVE_X)
		DLGRESIZE_CONTROL(IDC_CHK_SKIP_ARTWORK, DLSZ_MOVE_X)
		DLGRESIZE_CONTROL(IDC_CHK_PREV_DLG_SHOW_STATS, DLSZ_MOVE_X)
		DLGRESIZE_CONTROL(IDC_STATIC_VIEW_GROUP, DLSZ_MOVE_X)
		DLGRESIZE_CONTROL(IDC_VIEW_NORMAL, DLSZ_MOVE_X)
		DLGRESIZE_CONTROL(IDC_VIEW_DIFFERENCE, DLSZ_MOVE_X)
		DLGRESIZE_CONTROL(IDC_VIEW_ORIGINAL, DLSZ_MOVE_X)
		DLGRESIZE_CONTROL(IDC_VIEW_DEBUG, DLSZ_MOVE_X)
		DLGRESIZE_CONTROL(IDC_BTN_TAG_MAPPINGS, DLSZ_MOVE_X)
		DLGRESIZE_CONTROL(IDC_BTN_BACK, DLSZ_MOVE_X)
		DLGRESIZE_CONTROL(IDC_BTN_WRITE_TAGS, DLSZ_MOVE_X)
		DLGRESIZE_CONTROL(IDC_STATIC_MATCH_TRACKING_REL_NAME, DLSZ_MOVE_Y)
		DLGRESIZE_CONTROL(IDC_STATIC_MATCH_TRACKING_REL_NAME, DLSZ_SIZE_X)
		DLGRESIZE_CONTROL(IDC_PREVIEW_LIST, DLSZ_SIZE_X | DLSZ_SIZE_Y)
	END_DLGRESIZE_MAP()

	void DlgResize_UpdateLayout(int cxWidth, int cyHeight) {
		CDialogResize<CPreviewTagsDialog>::DlgResize_UpdateLayout(cxWidth, cyHeight);
	}

	LRESULT OnInitDialog(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& /*bHandled*/);
	LRESULT OnContextMenu(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& /*bHandled*/);
	LRESULT OnButtonWriteTags(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/);
	LRESULT OnButtonBack(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/);
	LRESULT OnDestroy(UINT /*uMsg*/, WPARAM /*wParam*/, LPARAM /*lParam*/, BOOL& /*bHandled*/);
	LRESULT OnCancel(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/);
	LRESULT OnCheckReplaceANVs(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/);
	LRESULT OnCheckPreviewShowStats(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/);
	LRESULT OnCheckSkipArtwork(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/);
	LRESULT OnCheckAttachMappingPanel(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/);
	LRESULT OnButtonEditTagMappings(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/);
	LRESULT OnChangePreviewMode(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/);

	LRESULT OnListDoubleClick(LPNMHDR lParam);
	LRESULT OnListKeyDown(LPNMHDR lParam);
	LRESULT OnEdit(UINT uMsg, WPARAM wParam, LPARAM lParam);
	LRESULT OnCustomDraw(int idCtrl, LPNMHDR pnmh, BOOL& bHandled);

	void replace_tag_result(size_t item, tag_result_ptr result);
	void refresh_ui_tag_results(bool computestat, pfc::bit_array_bittable tag_mask = {});
	void spawn_generate_tag_mappings();
	void cb_refresh_ui_tag_results(pfc::bit_array_bittable tag_mask);
	void enable(bool v) override { enable(v, true); };
	void enable(bool v, bool change_focus);
	bool check_write_tags_status();
	void destroy_all();
	void go_back();

private:

	void reset_default_columns(bool breset, bool bshowstats);
	void fix_sorted_icol_map(bool reset, bool bshowstats);

	void compute_stats(pfc::bit_array_bittable tag_mask = {});
	void compute_stats_track_map(pfc::bit_array_bittable tag_mask = {});

	void reset_stats() {
		m_vstats.clear();
	}

	void reset_tag_result_stats();

	void set_preview_mode(PreView mode);
	PreView get_preview_mode();

	bool init_other_controls_and_results();
	void GlobalReplace_ANV(bool state);

	bool delete_selection();
	bool context_menu_show(HWND wnd, size_t isel, LPARAM lParamPos);
	bool context_menu_switch(HWND wnd, POINT point, int cmd, bit_array_bittable selmask, std::vector<std::pair<std::string, std::string>>vartists);


private:

	HWND m_results_list;
	CListControlOwnerData m_uilist;
	CTristate m_tristate;

	fb2k::CDarkModeHooks m_dark;

	HBITMAP m_preview_bitmap;
	foo_conf conf;

	std::vector<preview_stats> m_vstats;
	pfc::bit_array_bittable m_vstats_mask;

	std::vector<std::pair<int, int>> m_vcol_data_subitems;
	bool m_cfg_bshow_stats = false;

	TagWriter_ptr m_tag_writer;
	size_t m_tw_index = 0;
	size_t m_tw_skip = 0;

	//access context menu
	friend CPreviewList;
};