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

use premiere::{self as pr};
use std::slice;

pub mod backends;
pub mod metrics;
pub mod render_properties;
pub mod scheduling;
pub mod shaders;

#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum GPUFramework {
	Metal,
	Cuda,
	OpenCL,
	Other(u32),
}

impl GPUFramework {
	pub fn from_premiere(v: u32) -> Self {
		match v {
			0 => Self::Cuda,
			1 => Self::OpenCL,
			2 => Self::Metal,
			_ => Self::Other(v),
		}
	}
}

#[inline]
fn frames_as_slice<'a>(frames: *const pr::sys::PPixHand, frame_count: usize) -> Result<&'a [pr::sys::PPixHand], pr::Error> {
	if frames.is_null() || frame_count == 0 {
		return Err(pr::Error::Fail);
	}

	Ok(unsafe { slice::from_raw_parts(frames, frame_count) })
}

fn gpu_bytes_per_pixels(pixel_format: pr::PixelFormat) -> i32 {
	match pixel_format {
		pr::PixelFormat::GpuBgra4444_32f => 16,
		pr::PixelFormat::GpuBgra4444_16f => 8,
		_ => panic!("Unsupported pixel format"),
	}
}

pub mod buffer {
	pub use imp::*;

	#[cfg(gpu_backend = "metal")]
	mod imp {
		pub use crate::gpu::backends::metal::buffer::*;
	}

	#[cfg(gpu_backend = "cuda")]
	mod imp {
		pub use crate::gpu::backends::cuda::buffer::*;
	}

	#[cfg(gpu_backend = "opencl")]
	mod imp {
		unimplemented!("OpenCL backend not yet implemented");
	}

	#[cfg(not(any(gpu_backend = "metal", gpu_backend = "cuda", gpu_backend = "opencl", gpu_backend = "other")))]
	mod imp {
		compile_error!("Unsupported gpu_backend");
	}
}

pub mod pipeline {
	pub use imp::*;

	#[cfg(gpu_backend = "metal")]
	mod imp {
		pub use crate::gpu::backends::metal::pipeline::*;
	}

	#[cfg(gpu_backend = "cuda")]
	mod imp {
		pub use crate::gpu::backends::cuda::pipeline::*;
	}

	#[cfg(gpu_backend = "opencl")]
	mod imp {
		unimplemented!("OpenCL backend not yet implemented");
	}

	#[cfg(not(any(gpu_backend = "metal", gpu_backend = "cuda", gpu_backend = "opencl", gpu_backend = "other")))]
	mod imp {
		compile_error!("Unsupported gpu_backend");
	}
}

pub mod fence {
	pub use imp::*;

	#[cfg(gpu_backend = "metal")]
	mod imp {
		pub use crate::gpu::backends::metal::fence::*;
	}

	#[cfg(gpu_backend = "cuda")]
	mod imp {
		pub use crate::gpu::backends::cuda::fence::*;
	}

	#[cfg(gpu_backend = "opencl")]
	mod imp {
		unimplemented!("OpenCL backend not yet implemented");
	}

	#[cfg(not(any(gpu_backend = "metal", gpu_backend = "cuda", gpu_backend = "opencl", gpu_backend = "other")))]
	mod imp {
		compile_error!("Unsupported gpu_backend");
	}
}