prgpu 0.1.12

GPU-accelerated rendering utilities for Adobe Premiere Pro and After Effects plugins
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124

//! Reusable blend-mode popup parameter.
//!
//! Mirrors the `BLEND_*` constants in `vekl/color/blend/dispatch.slang` byte-for-byte.
//!
//! ```ignore
//! use prgpu::ui::{add_blend_mode_param, BlendMode};
//!
//! // setup:
//! add_blend_mode_param(params, Params::TintBlendMode, "Tint Blend Mode", BlendMode::Multiply)?;
//!
//! // kernel_params:
//! tint_blend_mode: u32 = [popup(TintBlendMode)];
//!
//! // shader:
//! float3 tinted = BlendApply(params.tintBlendMode, base.rgb, tint.rgb);
//! ```
//!
//! For any variant, `BlendMode as u32`, `BLEND_MODE_OPTIONS[k]`, the slang
//! `BLEND_*` constant, and the kernel `u32` are byte-equal. The `popup(V)`
//! extractor handles the AE-vs-Premiere host conversion internally.
//!
//! "No blend / pass-through" is a separate strength slider, not a mode: skip
//! `BlendApply` at strength = 0 or `lerp(base, BlendApply(...), strength)` to fade.

use after_effects::{self as ae, ParamFlag, Parameters};

use crate::params::SetupParams;

/// Blend-mode discriminant. The integer value equals the 0-based popup index
/// delivered to the kernel and the `BLEND_*` constants in `vekl/color/blend/dispatch.slang`.
#[repr(u32)]
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub enum BlendMode {
	Add = 0,
	Multiply = 1,
	Screen = 2,
	ColorBurn = 3,
	ColorDodge = 4,
	DarkerColor = 5,
	Overlay = 6,
	Difference = 7,
	Subtract = 8,
	Divide = 9,
	Hue = 10,
	Saturation = 11,
	Color = 12,
	Luminosity = 13,
}

impl BlendMode {
	pub const COUNT: usize = 14;

	pub fn as_u32(self) -> u32 {
		self as u32
	}

	/// Popup value for `PopupDef::set_default` / `set_value`. AE PF popups are 1-based at storage, so we add 1.
	pub fn to_popup_value(self) -> i32 {
		(self as i32) + 1
	}

	/// Decode an AE 1-based `PopupDef.value()`. Out-of-range clamps to `Add`.
	pub fn from_popup_value(v: i32) -> Self {
		match v {
			1 => Self::Add,
			2 => Self::Multiply,
			3 => Self::Screen,
			4 => Self::ColorBurn,
			5 => Self::ColorDodge,
			6 => Self::DarkerColor,
			7 => Self::Overlay,
			8 => Self::Difference,
			9 => Self::Subtract,
			10 => Self::Divide,
			11 => Self::Hue,
			12 => Self::Saturation,
			13 => Self::Color,
			14 => Self::Luminosity,
			_ => Self::Add,
		}
	}
}

/// Popup option list. Order matches `BlendMode` discriminants 0-based; `BLEND_MODE_OPTIONS[BlendMode::Multiply as usize] == "Multiply"`.
pub const BLEND_MODE_OPTIONS: &[&str] = &[
	"Add",
	"Multiply",
	"Screen",
	"Color Burn",
	"Color Dodge",
	"Darker Color",
	"Overlay",
	"Difference",
	"Subtract",
	"Divide",
	"Hue",
	"Saturation",
	"Color",
	"Luminosity",
];

/// Add a blend-mode popup to `Parameters<P>` with `default` selected.
pub fn add_blend_mode_param<P: SetupParams>(
	params: &mut Parameters<'_, P>,
	id: P,
	label: &str,
	default: BlendMode,
) -> Result<(), ae::Error> {
	let default_popup = default.to_popup_value();

	params.add_customized(
		id,
		label,
		ae::PopupDef::setup(|f| {
			f.set_default(default_popup);
			f.set_value(default_popup);
			f.set_options(BLEND_MODE_OPTIONS);
		}),
		|p| {
			p.set_flag(ParamFlag::SUPERVISE, true);
			-1
		},
	)
}