From 98f0de6c56917010f370fa11b29633a523b424ed Mon Sep 17 00:00:00 2001
From: syuilo <4439005+syuilo@users.noreply.github.com>
Date: Thu, 29 May 2025 09:36:47 +0900
Subject: [PATCH] wip

---
 locales/index.d.ts                            |  12 ++
 locales/ja-JP.yml                             |   3 +
 .../MkImageEffectorDialog.Layer.vue           |   3 +-
 .../src/components/MkImageEffectorDialog.vue  |  18 ++-
 .../src/components/MkUploaderDialog.vue       |   2 +
 .../components/MkWatermarkEditorDialog.vue    |   3 +
 .../pages/settings/drive.WatermarkItem.vue    |   2 +
 .../utility/image-effector/ImageEffector.ts   |  31 ++----
 .../src/utility/image-effector/fxs.ts         |  25 +++++
 .../utility/image-effector/fxs/colorClamp.ts  |  60 ++++++++++
 .../image-effector/fxs/colorClampAdvanced.ts  | 104 ++++++++++++++++++
 .../utility/image-effector/fxs/grayscale.ts   |  40 +++++++
 12 files changed, 277 insertions(+), 26 deletions(-)
 create mode 100644 packages/frontend/src/utility/image-effector/fxs.ts
 create mode 100644 packages/frontend/src/utility/image-effector/fxs/colorClamp.ts
 create mode 100644 packages/frontend/src/utility/image-effector/fxs/colorClampAdvanced.ts
 create mode 100644 packages/frontend/src/utility/image-effector/fxs/grayscale.ts

diff --git a/locales/index.d.ts b/locales/index.d.ts
index ff2019e871..d08af02dd9 100644
--- a/locales/index.d.ts
+++ b/locales/index.d.ts
@@ -12092,6 +12092,18 @@ export interface Locale extends ILocale {
              * 色の反転
              */
             "invert": string;
+            /**
+             * 白黒
+             */
+            "grayscale": string;
+            /**
+             * 色の圧縮
+             */
+            "colorClamp": string;
+            /**
+             * 色の圧縮(高度)
+             */
+            "colorClampAdvanced": string;
         };
     };
 }
diff --git a/locales/ja-JP.yml b/locales/ja-JP.yml
index ad62d74c66..42e496a8ce 100644
--- a/locales/ja-JP.yml
+++ b/locales/ja-JP.yml
@@ -3241,3 +3241,6 @@ _imageEffector:
     glitch: "グリッチ"
     mirror: "ミラー"
     invert: "色の反転"
+    grayscale: "白黒"
+    colorClamp: "色の圧縮"
+    colorClampAdvanced: "色の圧縮(高度)"
diff --git a/packages/frontend/src/components/MkImageEffectorDialog.Layer.vue b/packages/frontend/src/components/MkImageEffectorDialog.Layer.vue
index bd38248d29..5e3981669c 100644
--- a/packages/frontend/src/components/MkImageEffectorDialog.Layer.vue
+++ b/packages/frontend/src/components/MkImageEffectorDialog.Layer.vue
@@ -37,7 +37,7 @@ import { ref, useTemplateRef, watch, onMounted, onUnmounted } from 'vue';
 import { v4 as uuid } from 'uuid';
 import type { ImageEffectorLayer } from '@/utility/image-effector/ImageEffector.js';
 import { i18n } from '@/i18n.js';
-import { FXS, ImageEffector } from '@/utility/image-effector/ImageEffector.js';
+import { ImageEffector } from '@/utility/image-effector/ImageEffector.js';
 import MkFolder from '@/components/MkFolder.vue';
 import MkButton from '@/components/MkButton.vue';
 import MkInput from '@/components/MkInput.vue';
@@ -50,6 +50,7 @@ import * as os from '@/os.js';
 import { selectFile } from '@/utility/drive.js';
 import { misskeyApi } from '@/utility/misskey-api.js';
 import { prefer } from '@/preferences.js';
+import { FXS } from '@/utility/image-effector/fxs.js';
 
 const layer = defineModel<ImageEffectorLayer>('layer', { required: true });
 const fx = FXS.find((fx) => fx.id === layer.value.fxId);
diff --git a/packages/frontend/src/components/MkImageEffectorDialog.vue b/packages/frontend/src/components/MkImageEffectorDialog.vue
index a7c593f1b6..f240e0f94c 100644
--- a/packages/frontend/src/components/MkImageEffectorDialog.vue
+++ b/packages/frontend/src/components/MkImageEffectorDialog.vue
@@ -23,6 +23,8 @@ SPDX-License-Identifier: AGPL-3.0-only
 				<div :class="$style.previewContainer">
 					<div class="_acrylic" :class="$style.previewTitle">{{ i18n.ts.preview }}</div>
 					<div class="_acrylic" :class="$style.previewControls">
+						<button class="_button" :class="[$style.previewControlsButton, !enabled ? $style.active : null]" @click="enabled = false">Before</button>
+						<button class="_button" :class="[$style.previewControlsButton, enabled ? $style.active : null]" @click="enabled = true">After</button>
 					</div>
 				</div>
 			</div>
@@ -49,7 +51,7 @@ import { v4 as uuid } from 'uuid';
 import type { WatermarkPreset } from '@/utility/watermark.js';
 import type { ImageEffectorLayer } from '@/utility/image-effector/ImageEffector.js';
 import { i18n } from '@/i18n.js';
-import { FXS, ImageEffector } from '@/utility/image-effector/ImageEffector.js';
+import { ImageEffector } from '@/utility/image-effector/ImageEffector.js';
 import MkModalWindow from '@/components/MkModalWindow.vue';
 import MkSelect from '@/components/MkSelect.vue';
 import MkButton from '@/components/MkButton.vue';
@@ -57,6 +59,7 @@ import MkInput from '@/components/MkInput.vue';
 import XLayer from '@/components/MkImageEffectorDialog.Layer.vue';
 import * as os from '@/os.js';
 import { deepClone } from '@/utility/clone.js';
+import { FXS } from '@/utility/image-effector/fxs.js';
 
 const props = defineProps<{
 	image: HTMLImageElement;
@@ -114,6 +117,7 @@ onMounted(async () => {
 		height: props.image.height,
 		layers: layers,
 		originalImage: props.image,
+		fxs: FXS,
 	});
 
 	await renderer!.bakeTextures();
@@ -135,6 +139,18 @@ function save() {
 		dialog.value?.close();
 	}, 'image/png');
 }
+
+const enabled = ref(true);
+watch(enabled, () => {
+	if (renderer != null) {
+		if (enabled.value) {
+			renderer.updateLayers(layers);
+		} else {
+			renderer.updateLayers([]);
+		}
+		renderer.render();
+	}
+});
 </script>
 
 <style module>
diff --git a/packages/frontend/src/components/MkUploaderDialog.vue b/packages/frontend/src/components/MkUploaderDialog.vue
index 444cd0619f..48782e0e2e 100644
--- a/packages/frontend/src/components/MkUploaderDialog.vue
+++ b/packages/frontend/src/components/MkUploaderDialog.vue
@@ -98,6 +98,7 @@ import * as os from '@/os.js';
 import { ensureSignin } from '@/i.js';
 import { ImageEffector } from '@/utility/image-effector/ImageEffector.js';
 import { makeImageEffectorLayers } from '@/utility/watermark.js';
+import { FX_watermarkPlacement } from '@/utility/image-effector/fxs/watermarkPlacement.js';
 
 const $i = ensureSignin();
 
@@ -527,6 +528,7 @@ async function preprocess(item: (typeof items)['value'][number]): Promise<void>
 			height: img.height,
 			layers: makeImageEffectorLayers(preset.layers),
 			originalImage: img,
+			fxs: [FX_watermarkPlacement],
 		});
 
 		await renderer.bakeTextures();
diff --git a/packages/frontend/src/components/MkWatermarkEditorDialog.vue b/packages/frontend/src/components/MkWatermarkEditorDialog.vue
index e7659405d3..5434b80b66 100644
--- a/packages/frontend/src/components/MkWatermarkEditorDialog.vue
+++ b/packages/frontend/src/components/MkWatermarkEditorDialog.vue
@@ -59,6 +59,7 @@ import XLayer from '@/components/MkWatermarkEditorDialog.Layer.vue';
 import * as os from '@/os.js';
 import { deepClone } from '@/utility/clone.js';
 import { ensureSignin } from '@/i.js';
+import { FX_watermarkPlacement } from '@/utility/image-effector/fxs/watermarkPlacement.js';
 
 const $i = ensureSignin();
 
@@ -160,6 +161,7 @@ async function initRenderer() {
 			height: 1000,
 			layers: makeImageEffectorLayers(preset.layers),
 			originalImage: sampleImage_3_2,
+			fxs: [FX_watermarkPlacement],
 		});
 	} else if (sampleImageType.value === '2_3') {
 		renderer = new ImageEffector({
@@ -168,6 +170,7 @@ async function initRenderer() {
 			height: 1500,
 			layers: makeImageEffectorLayers(preset.layers),
 			originalImage: sampleImage_2_3,
+			fxs: [FX_watermarkPlacement],
 		});
 	}
 
diff --git a/packages/frontend/src/pages/settings/drive.WatermarkItem.vue b/packages/frontend/src/pages/settings/drive.WatermarkItem.vue
index 85881cb7c6..7c8b2c3f79 100644
--- a/packages/frontend/src/pages/settings/drive.WatermarkItem.vue
+++ b/packages/frontend/src/pages/settings/drive.WatermarkItem.vue
@@ -30,6 +30,7 @@ import { i18n } from '@/i18n.js';
 import { deepClone } from '@/utility/clone.js';
 import MkFolder from '@/components/MkFolder.vue';
 import { ImageEffector } from '@/utility/image-effector/ImageEffector.js';
+import { FX_watermarkPlacement } from '@/utility/image-effector/fxs/watermarkPlacement.js';
 
 const props = defineProps<{
 	preset: WatermarkPreset;
@@ -78,6 +79,7 @@ onMounted(() => {
 				height: 1000,
 				layers: makeImageEffectorLayers(props.preset.layers),
 				originalImage: sampleImage,
+				fxs: [FX_watermarkPlacement],
 			});
 
 			await renderer.bakeTextures();
diff --git a/packages/frontend/src/utility/image-effector/ImageEffector.ts b/packages/frontend/src/utility/image-effector/ImageEffector.ts
index d80f921097..d36cd263e7 100644
--- a/packages/frontend/src/utility/image-effector/ImageEffector.ts
+++ b/packages/frontend/src/utility/image-effector/ImageEffector.ts
@@ -4,11 +4,6 @@
  */
 
 import { getProxiedImageUrl } from '../media-proxy.js';
-import { FX_chromaticAberration } from './fxs/chromaticAberration.js';
-import { FX_glitch } from './fxs/glitch.js';
-import { FX_invert } from './fxs/invert.js';
-import { FX_mirror } from './fxs/mirror.js';
-import { FX_watermarkPlacement } from './fxs/watermarkPlacement.js';
 
 type ParamTypeToPrimitive = {
 	'number': number;
@@ -49,31 +44,16 @@ export type ImageEffectorFx<ID extends string, P extends ImageEffectorFxParamDef
 	}) => void;
 };
 
-export const FXS = [
-	FX_watermarkPlacement,
-	FX_chromaticAberration,
-	FX_glitch,
-	FX_mirror,
-	FX_invert,
-] as const satisfies ImageEffectorFx<string, any>[];
-
-export type ImageEffectorLayerOf<
-	FXID extends (typeof FXS)[number]['id'],
-	FX extends { params: ImageEffectorFxParamDefs } = Extract<(typeof FXS)[number], { id: FXID }>,
-> = {
+export type ImageEffectorLayer = {
 	id: string;
-	fxId: FXID;
-	params: {
-		[key in keyof FX['params']]: ParamTypeToPrimitive[FX['params'][key]['type']];
-	};
+	fxId: string;
+	params: Record<string, any>;
 
 	// for watermarkPlacement fx
 	imageUrl?: string | null;
 	text?: string | null;
 };
 
-export type ImageEffectorLayer = ImageEffectorLayerOf<(typeof FXS)[number]['id'], Extract<(typeof FXS)[number], { id: (typeof FXS)[number]['id'] }>>;
-
 export class ImageEffector {
 	private canvas: HTMLCanvasElement | null = null;
 	private gl: WebGL2RenderingContext | null = null;
@@ -89,6 +69,7 @@ export class ImageEffector {
 	private shaderCache: Map<string, WebGLProgram> = new Map();
 	private perLayerResultTextures: Map<string, WebGLTexture> = new Map();
 	private perLayerResultFrameBuffers: Map<string, WebGLFramebuffer> = new Map();
+	private fxs: ImageEffectorFx<string, any>[];
 
 	constructor(options: {
 		canvas: HTMLCanvasElement;
@@ -96,6 +77,7 @@ export class ImageEffector {
 		height: number;
 		originalImage: ImageData | ImageBitmap | HTMLImageElement | HTMLCanvasElement;
 		layers: ImageEffectorLayer[];
+		fxs: ImageEffectorFx<string, any>[];
 	}) {
 		this.canvas = options.canvas;
 		this.canvas.width = options.width;
@@ -104,6 +86,7 @@ export class ImageEffector {
 		this.renderHeight = options.height;
 		this.originalImage = options.originalImage;
 		this.layers = options.layers;
+		this.fxs = options.fxs;
 		this.texturesKey = this.calcTexturesKey();
 
 		this.gl = this.canvas.getContext('webgl2', {
@@ -328,7 +311,7 @@ export class ImageEffector {
 			throw new Error('gl is not initialized');
 		}
 
-		const fx = FXS.find(fx => fx.id === layer.fxId);
+		const fx = this.fxs.find(fx => fx.id === layer.fxId);
 		if (fx == null) return;
 
 		const watermark = layer.fxId === 'watermarkPlacement' ? this.bakedTexturesForWatermarkFx.get(layer.id) : undefined;
diff --git a/packages/frontend/src/utility/image-effector/fxs.ts b/packages/frontend/src/utility/image-effector/fxs.ts
new file mode 100644
index 0000000000..07d455d960
--- /dev/null
+++ b/packages/frontend/src/utility/image-effector/fxs.ts
@@ -0,0 +1,25 @@
+/*
+ * SPDX-FileCopyrightText: syuilo and misskey-project
+ * SPDX-License-Identifier: AGPL-3.0-only
+ */
+
+import { FX_chromaticAberration } from './fxs/chromaticAberration.js';
+import { FX_colorClamp } from './fxs/colorClamp.js';
+import { FX_colorClampAdvanced } from './fxs/colorClampAdvanced.js';
+import { FX_glitch } from './fxs/glitch.js';
+import { FX_grayscale } from './fxs/grayscale.js';
+import { FX_invert } from './fxs/invert.js';
+import { FX_mirror } from './fxs/mirror.js';
+import { FX_watermarkPlacement } from './fxs/watermarkPlacement.js';
+import type { ImageEffectorFx } from './ImageEffector.js';
+
+export const FXS = [
+	FX_watermarkPlacement,
+	FX_chromaticAberration,
+	FX_glitch,
+	FX_mirror,
+	FX_invert,
+	FX_grayscale,
+	FX_colorClamp,
+	FX_colorClampAdvanced,
+] as const satisfies ImageEffectorFx<string, any>[];
diff --git a/packages/frontend/src/utility/image-effector/fxs/colorClamp.ts b/packages/frontend/src/utility/image-effector/fxs/colorClamp.ts
new file mode 100644
index 0000000000..5cc13d7900
--- /dev/null
+++ b/packages/frontend/src/utility/image-effector/fxs/colorClamp.ts
@@ -0,0 +1,60 @@
+/*
+ * SPDX-FileCopyrightText: syuilo and misskey-project
+ * SPDX-License-Identifier: AGPL-3.0-only
+ */
+
+import { defineImageEffectorFx } from '../ImageEffector.js';
+import { i18n } from '@/i18n.js';
+
+const shader = `#version 300 es
+precision mediump float;
+
+in vec2 in_uv;
+uniform sampler2D u_texture;
+uniform vec2 u_resolution;
+uniform float u_max;
+uniform float u_min;
+out vec4 out_color;
+
+void main() {
+	vec4 in_color = texture(u_texture, in_uv);
+	float r = min(max(in_color.r, u_min), u_max);
+	float g = min(max(in_color.g, u_min), u_max);
+	float b = min(max(in_color.b, u_min), u_max);
+	out_color = vec4(r, g, b, in_color.a);
+}
+`;
+
+export const FX_colorClamp = defineImageEffectorFx({
+	id: 'colorClamp' as const,
+	name: i18n.ts._imageEffector._fxs.colorClamp,
+	shader,
+	params: {
+		max: {
+			type: 'number' as const,
+			default: 1.0,
+			min: 0.0,
+			max: 1.0,
+			step: 0.01,
+		},
+		min: {
+			type: 'number' as const,
+			default: -1.0,
+			min: -1.0,
+			max: 0.0,
+			step: 0.01,
+		},
+	},
+	main: ({ gl, program, params, preTexture }) => {
+		gl.activeTexture(gl.TEXTURE0);
+		gl.bindTexture(gl.TEXTURE_2D, preTexture);
+		const u_texture = gl.getUniformLocation(program, 'u_texture');
+		gl.uniform1i(u_texture, 0);
+
+		const u_max = gl.getUniformLocation(program, 'u_max');
+		gl.uniform1f(u_max, params.max);
+
+		const u_min = gl.getUniformLocation(program, 'u_min');
+		gl.uniform1f(u_min, 1.0 + params.min);
+	},
+});
diff --git a/packages/frontend/src/utility/image-effector/fxs/colorClampAdvanced.ts b/packages/frontend/src/utility/image-effector/fxs/colorClampAdvanced.ts
new file mode 100644
index 0000000000..80b5b9f37e
--- /dev/null
+++ b/packages/frontend/src/utility/image-effector/fxs/colorClampAdvanced.ts
@@ -0,0 +1,104 @@
+/*
+ * SPDX-FileCopyrightText: syuilo and misskey-project
+ * SPDX-License-Identifier: AGPL-3.0-only
+ */
+
+import { defineImageEffectorFx } from '../ImageEffector.js';
+import { i18n } from '@/i18n.js';
+
+const shader = `#version 300 es
+precision mediump float;
+
+in vec2 in_uv;
+uniform sampler2D u_texture;
+uniform vec2 u_resolution;
+uniform float u_rMax;
+uniform float u_rMin;
+uniform float u_gMax;
+uniform float u_gMin;
+uniform float u_bMax;
+uniform float u_bMin;
+out vec4 out_color;
+
+void main() {
+	vec4 in_color = texture(u_texture, in_uv);
+	float r = min(max(in_color.r, u_rMin), u_rMax);
+	float g = min(max(in_color.g, u_gMin), u_gMax);
+	float b = min(max(in_color.b, u_bMin), u_bMax);
+	out_color = vec4(r, g, b, in_color.a);
+}
+`;
+
+export const FX_colorClampAdvanced = defineImageEffectorFx({
+	id: 'colorClampAdvanced' as const,
+	name: i18n.ts._imageEffector._fxs.colorClampAdvanced,
+	shader,
+	params: {
+		rMax: {
+			type: 'number' as const,
+			default: 1.0,
+			min: 0.0,
+			max: 1.0,
+			step: 0.01,
+		},
+		rMin: {
+			type: 'number' as const,
+			default: -1.0,
+			min: -1.0,
+			max: 0.0,
+			step: 0.01,
+		},
+		gMax: {
+			type: 'number' as const,
+			default: 1.0,
+			min: 0.0,
+			max: 1.0,
+			step: 0.01,
+		},
+		gMin: {
+			type: 'number' as const,
+			default: -1.0,
+			min: -1.0,
+			max: 0.0,
+			step: 0.01,
+		},
+		bMax: {
+			type: 'number' as const,
+			default: 1.0,
+			min: 0.0,
+			max: 1.0,
+			step: 0.01,
+		},
+		bMin: {
+			type: 'number' as const,
+			default: -1.0,
+			min: -1.0,
+			max: 0.0,
+			step: 0.01,
+		},
+	},
+	main: ({ gl, program, params, preTexture }) => {
+		gl.activeTexture(gl.TEXTURE0);
+		gl.bindTexture(gl.TEXTURE_2D, preTexture);
+		const u_texture = gl.getUniformLocation(program, 'u_texture');
+		gl.uniform1i(u_texture, 0);
+
+		const u_rMax = gl.getUniformLocation(program, 'u_rMax');
+		gl.uniform1f(u_rMax, params.rMax);
+
+		const u_rMin = gl.getUniformLocation(program, 'u_rMin');
+		gl.uniform1f(u_rMin, 1.0 + params.rMin);
+
+		const u_gMax = gl.getUniformLocation(program, 'u_gMax');
+		gl.uniform1f(u_gMax, params.gMax);
+
+		const u_gMin = gl.getUniformLocation(program, 'u_gMin');
+		gl.uniform1f(u_gMin, 1.0 + params.gMin);
+
+		const u_bMax = gl.getUniformLocation(program, 'u_bMax');
+		gl.uniform1f(u_bMax, params.bMax);
+
+		const u_bMin = gl.getUniformLocation(program, 'u_bMin');
+		gl.uniform1f(u_bMin, 1.0 + params.bMin);
+	},
+});
diff --git a/packages/frontend/src/utility/image-effector/fxs/grayscale.ts b/packages/frontend/src/utility/image-effector/fxs/grayscale.ts
new file mode 100644
index 0000000000..b398712781
--- /dev/null
+++ b/packages/frontend/src/utility/image-effector/fxs/grayscale.ts
@@ -0,0 +1,40 @@
+/*
+ * SPDX-FileCopyrightText: syuilo and misskey-project
+ * SPDX-License-Identifier: AGPL-3.0-only
+ */
+
+import { defineImageEffectorFx } from '../ImageEffector.js';
+import { i18n } from '@/i18n.js';
+
+const shader = `#version 300 es
+precision mediump float;
+
+in vec2 in_uv;
+uniform sampler2D u_texture;
+uniform vec2 u_resolution;
+out vec4 out_color;
+
+float getBrightness(vec4 color) {
+	return (color.r + color.g + color.b) / 3.0;
+}
+
+void main() {
+	vec4 in_color = texture(u_texture, in_uv);
+	float brightness = getBrightness(in_color);
+	out_color = vec4(brightness, brightness, brightness, in_color.a);
+}
+`;
+
+export const FX_grayscale = defineImageEffectorFx({
+	id: 'grayscale' as const,
+	name: i18n.ts._imageEffector._fxs.grayscale,
+	shader,
+	params: {
+	},
+	main: ({ gl, program, params, preTexture }) => {
+		gl.activeTexture(gl.TEXTURE0);
+		gl.bindTexture(gl.TEXTURE_2D, preTexture);
+		const u_texture = gl.getUniformLocation(program, 'u_texture');
+		gl.uniform1i(u_texture, 0);
+	},
+});