CommonLua/Ged/Apps/ScreenshotDiffViewer.lua

DefineClass.ScreenshotDiffViewer = {
	__parents = { "GedApp" },
	
	Translate = true,
	Title = "Screenshot Diff Viewer<opt(u(EditorShortcut),' (',')')>",
	AppId = "ScreenshotDiffViewer",
	InitialWidth = 1600,
	InitialHeight = 900,
	
	selected_image_paths = false,
	view_mode = "cycle",
	diff_scale = 2,
}

function ScreenshotDiffViewer:Init(parent, context)
	XAction:new({
		ActionId = "TakeScreenshot",
		ActionToolbar = "main",
		ActionName = "Take Screenshot",
		ActionTranslate = false,
		ActionIcon = "CommonAssets/UI/Ged/new.tga",
		OnAction = function(action, host)
			host:Op("rfnCreateNewScreenshot", "root")
		end,
	}, self)
	XAction:new({
		ActionId = "OpenFolder",
		ActionToolbar = "main",
		ActionName = "Open Folder",
		ActionTranslate = false,
		ActionIcon = "CommonAssets/UI/Ged/explorer.tga",
		OnAction = function(action, host)
			AsyncExec("explorer " .. ConvertToOSPath(g_ScreenshotViewerFolder))
		end,
	}, self)
	XAction:new({
		ActionId = "RefreshList",
		ActionToolbar = "main",
		ActionToolbarSplit = true,
		ActionName = "Refresh List",
		ActionTranslate = false,
		ActionIcon = "CommonAssets/UI/Ged/undo.tga",
		OnAction = function(action, host)
			host:Op("rfnReloadScreenshotItems", "root")
		end,
	}, self)
	XAction:new({
		ActionId = "FitImage",
		ActionToolbar = "main",
		ActionName = "Fit Image",
		ActionTranslate = false,
		ActionIcon = "CommonAssets/UI/Ged/fit.tga",
		OnAction = function(action, host)
			host:SetImageFitScale(nil, true)
		end,
	}, self)
	XAction:new({
		ActionId = "ShowOriginalSize",
		ActionToolbar = "main",
		ActionName = "Original Size",
		ActionTranslate = false,
		ActionIcon = "CommonAssets/UI/Ged/original.tga",
		OnAction = function(action, host)
			host:SetImageFitScale()
		end,
	}, self)
	XAction:new({
		ActionId = "MagnifyImage",
		ActionToolbar = "main",
		ActionToolbarSplit = true,
		ActionName = "200%",
		ActionTranslate = false,
		ActionIcon = "CommonAssets/UI/Ged/x2.tga",
		OnAction = function(action, host)
			host:SetImageFitScale(point(2000, 2000))
		end,
	}, self)
	XAction:new({
		ActionId = "ToggleVStripComp",
		ActionToolbar = "main",
		ActionToolbarSplit = true,
		ActionName = "Toggle Vertical Strips Comparison",
		ActionTranslate = false,
		ActionIcon = "CommonAssets/UI/Ged/collection.tga",
		ActionToggle = true,
		ActionToggled = function (self, host)
			return host.view_mode == "vstrips"
		end,
		OnAction = function(action, host)
			if action:ActionToggled(host) then
				host:SetViewMode("cycle")
			else
				host:SetViewMode("vstrips")
			end
		end,
	}, self)--]]
	XAction:new({
		ActionId = "ToggleCycleDiff",
		ActionToolbar = "main",
		ActionToolbarSplit = true,
		ActionName = "Toggle Screenshot Difference",
		ActionTranslate = false,
		ActionIcon = "CommonAssets/UI/Ged/view.tga",
		ActionToggle = true,
		ActionToggled = function (self, host)
			return host.view_mode == "diff"
		end,
		OnAction = function(action, host)
			if action:ActionToggled(host) then
				host:SetViewMode("cycle")
			else
				host:SetViewMode("diff")
			end
		end,
	}, self)
	XAction:new({
		ActionId = "SetDiff2",
		ActionToolbar = "main",
		ActionName = "2x Difference Scale",
		ActionTranslate = false,
		ActionIcon = "CommonAssets/UI/Ged/2x.tga",
		ActionToggle = true,
		ActionToggled = function (self, host)
			return host.diff_scale == 2
		end,
		OnAction = function(action, host)
			host:SetDiffScale(2)
		end,
	}, self)
	XAction:new({
		ActionId = "SetDiff2",
		ActionToolbar = "main",
		ActionName = "4x Difference Scale",
		ActionTranslate = false,
		ActionIcon = "CommonAssets/UI/Ged/4x.tga",
		ActionToggle = true,
		ActionToggled = function (self, host)
			return host.diff_scale == 4
		end,
		OnAction = function(action, host)
			host:SetDiffScale(4)
		end,
	}, self)
	XAction:new({
		ActionId = "SetDiff2",
		ActionToolbar = "main",
		ActionName = "16x Difference Scale",
		ActionTranslate = false,
		ActionIcon = "CommonAssets/UI/Ged/16x.tga",
		ActionToggle = true,
		ActionToggled = function (self, host)
			return host.diff_scale == 16
		end,
		OnAction = function(action, host)
			host:SetDiffScale(16)
		end,
	}, self)
	
	self.LayoutHSpacing = 0
	GedListPanel:new({
		Id = "idScreenshots",
		Title = "Screenshots",
		TitleFormatFunc = "GedFormatPresets",
		Format = "<display_name>",
		SelectionBind = "SelectedObject,SelectedPreset",
		MultipleSelection = true,
		SearchHistory = 20,
		PersistentSearch = true,
	}, self, "root")
	XPanelSizer:new({
	}, self)
	local win = XWindow:new({
	}, self)
	XImage:new({
		Id = "idImageFit",
		ImageFit = "smallest",
		FoldWhenHidden = true,
	}, win)
	local container = XWindow:new({
		Id = "idScrollContainer",
		FoldWhenHidden = true,
	}, win)
	container:SetVisible(false)
	local area = XScrollArea:new({
		Id = "idScrollArea",
		IdNode = false,
		HScroll = "idHScroll",
		VScroll = "idScroll",
	}, container)
	XImage:new({
		Id = "idImage",
		ImageFit = "none",
	}, area)
	XSleekScroll:new({
		Id = "idHScroll",
		Target = "idScrollArea",
		Dock = "bottom",
		Margins = box(0, 2, 7, 0),
		Horizontal = true,
		AutoHide = true,
		FoldWhenHidden = true,
	}, container)
	XSleekScroll:new({
		Id = "idScroll",
		Target = "idScrollArea",
		Dock = "right",
		Margins = box(2, 0, 0, 0),
		Horizontal = false,
		AutoHide = true,
		FoldWhenHidden = true,
	}, container)
end

function ScreenshotDiffViewer:SetViewMode(mode)
	self.view_mode = mode
	self:UpdateShownImages()
end

function ScreenshotDiffViewer:SetDiffScale(scale)
	self.diff_scale = scale
	self:UpdateShownImages()
end

function ScreenshotDiffViewer:UpdateShownImages()
	self:DeleteThread("cycle_thread")
	local images = self.selected_image_paths
	if #(images or "") > 1 then
		if self.view_mode == "cycle" then
			self:CreateThread("cycle_thread", function(self)
				local idx = 1
				local count = #images
				while true do
					self:SetImage(images[idx])
					Sleep(500)
					idx = idx % count + 1
				end
			end, self)
		elseif self.view_mode == "diff" then
			self:ShowDiffImage()
		elseif self.view_mode == "vstrips" then
			self:ShowVStripsComparison()
		end
	elseif #(images or "") > 0 then
		self:SetImage(images[1])
	end
end

function ScreenshotDiffViewer:ShowVStripsComparison()
	local images = self.selected_image_paths
	local hash = ""
	for i, img in ipairs(images) do hash = hash..xxhash(img) end
	local result_path = string.format("%s%d%s.png", g_ScreenshotViewerCacheFolder, #images, hash)
	if not io.exists(result_path) then
		AsyncCreatePath(g_ScreenshotViewerCacheFolder)
		CreateRealTimeThread(function()
			local err = self.connection:Call("rfnOp", self:GetState(), "rfnCompareScreenshotItemsVstrips", "root", images, result_path)
			if not err and io.exists(result_path) then
				self:SetImage(result_path)
			end
		end, self, images, result_path)
		return
	end
	self:SetImage(result_path)
end

function ScreenshotDiffViewer:ShowDiffImage()
	local images = self.selected_image_paths
	local diff_path = string.format("%s%d_%d_%d.png", g_ScreenshotViewerCacheFolder, xxhash(images[1]), xxhash(images[2]), self.diff_scale)
	if not io.exists(diff_path) then
		AsyncCreatePath(g_ScreenshotViewerCacheFolder)
		CreateRealTimeThread(function()
			local err = self.connection:Call("rfnOp", self:GetState(), "rfnCompareScreenshotItemsDiff", "root", images[1], images[2], diff_path, self.diff_scale)
			if not err and io.exists(diff_path) then
				self:SetImage(diff_path)
			end
		end, self, images, diff_path)
		return
	end
	self:SetImage(diff_path)
end

function ScreenshotDiffViewer:SetImage(image)
	self.idImage:SetImage(image)
	self.idImageFit:SetImage(image)
end

function ScreenshotDiffViewer:SetImageFitScale(scale, fit)
	scale = scale or point(1000, 1000)
	self.idImage:SetScaleModifier(scale)
	self.idScrollContainer:SetVisible(not fit)
	self.idImageFit:SetVisible(fit)
end

function ScreenshotDiffViewer:rfnSetSelectedFilePath(file_path, selected)
	self.selected_image_paths = self.selected_image_paths or {}
	local t = self.selected_image_paths
	if selected then
		table.insert_unique(t, file_path)
	else
		table.remove_value(t, file_path)
	end
	table.sort(t)
	self:UpdateShownImages()
end