dynamo-llm 1.5.1

Dynamo LLM Library
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

// SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
// SPDX-License-Identifier: Apache-2.0

use anyhow::Result;
use image::ImageFormat;

use super::{
    BackendAvailability, BackendDecline, DecodedImage, ImageDecodeBackend, ImageDecodeOutcome,
    ImageDecodeRequest, PixelFormat,
};
use crate::preprocessor::media::jpeg_turbo;

pub(super) struct TurboJpegBackend;

impl ImageDecodeBackend for TurboJpegBackend {
    fn name(&self) -> &'static str {
        "libjpeg_turbo"
    }

    fn availability(&self) -> BackendAvailability {
        if jpeg_turbo::available() {
            BackendAvailability::Available
        } else {
            BackendAvailability::Unavailable
        }
    }

    fn supports(&self, format: ImageFormat) -> bool {
        format == ImageFormat::Jpeg
    }

    fn try_decode(&self, request: ImageDecodeRequest<'_>) -> Result<ImageDecodeOutcome> {
        if !self.supports(request.format) {
            return Ok(ImageDecodeOutcome::NotHandled(
                BackendDecline::UnsupportedFormat(request.format),
            ));
        }
        if self.availability() == BackendAvailability::Unavailable {
            return Ok(ImageDecodeOutcome::NotHandled(BackendDecline::Unavailable));
        }

        let Some(jpeg) = jpeg_turbo::decode_jpeg(
            request.bytes,
            request.limits.max_image_width,
            request.limits.max_image_height,
            request.limits.max_alloc,
        )?
        else {
            return Ok(ImageDecodeOutcome::NotHandled(BackendDecline::DecodeFailed));
        };

        let pixel_format = match jpeg.channels {
            1 => PixelFormat::L8,
            3 => PixelFormat::Rgb8,
            other => anyhow::bail!("Unsupported TurboJPEG channel count {other}"),
        };
        Ok(ImageDecodeOutcome::Decoded(DecodedImage::new(
            request.format,
            jpeg.width,
            jpeg.height,
            pixel_format,
            jpeg.data,
            request.limits,
        )?))
    }
}