Infrastructure/immich
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-03-11 02:23:09 -05:00
Code Issues Activity

libplacebo for nvenc

Browse source 
update dockerfile
This commit is contained in:
mertalev 2024-05-11 12:04:58 -04:00
parent f420befc15
commit a2e8b657e6
No known key found for this signature in database
GPG key ID: 13C97EF14A338984
1 changed files with 18 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
server/src/utils/media.ts
View file

@ -440,7 +440,7 @@ export class NVENCConfig extends BaseHWConfig {
} }
getBaseInputOptions() { getBaseInputOptions() {
return ['-init_hw_device cuda=cuda:0', '-filter_hw_device cuda']; return ['-hwaccel cuda', '-hwaccel_output_format cuda'];
} }
getBaseOutputOptions(target: TranscodeTarget, videoStream: VideoStreamInfo, audioStream?: AudioStreamInfo) { getBaseOutputOptions(target: TranscodeTarget, videoStream: VideoStreamInfo, audioStream?: AudioStreamInfo) {
@ -461,13 +461,27 @@ export class NVENCConfig extends BaseHWConfig {
return options; return options;
} }
getToneMapping() {
const colors = this.getColors();
// https://stackoverflow.com/a/65542002
return [
'hwupload=derive_device=vulkan',
`libplacebo=tonemapping=${this.config.tonemap}:colorspace=${colors.matrix}:color_primaries=${colors.primaries}:color_trc=${colors.transfer}:format=yuv420p:preset=high_quality:downscaler=lanczos`,
'hwupload=derive_device=cuda',
];
}
getFilterOptions(videoStream: VideoStreamInfo) { getFilterOptions(videoStream: VideoStreamInfo) {
const options = this.shouldToneMap(videoStream) ? this.getToneMapping() : []; const options = [];
options.push('format=nv12', 'hwupload_cuda');
if (this.shouldScale(videoStream)) { if (this.shouldScale(videoStream)) {
options.push(`scale_cuda=${this.getScaling(videoStream)}`); options.push(`scale_cuda=${this.getScaling(videoStream)}`);
} }
if (this.shouldToneMap(videoStream)) {
options.push(...this.getToneMapping());
}
return options; return options;
} }
@ -501,7 +515,7 @@ export class NVENCConfig extends BaseHWConfig {
} }
getThreadOptions() { getThreadOptions() {
return []; return [`-threads ${this.config.threads <= 0 ? 1 : this.config.threads}`];
} }
getRefs() { getRefs() {