raw/
raw.rs

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
/*
 * SPDX-FileCopyrightText: 2022 Purism, SPC <https://puri.sm>
 * SPDX-FileCopyrightText: 2024 DorotaC
 *
 * SPDX-License-Identifier: MPL-2.0 OR LGPL-2.1-or-later
 */

/*! Bayer raw processor */


use clap::Parser;
use crispy::egl::DmabufImage;
use crispy::egl::headless;
use crispy::glium;
use crispyconv::{draw, get_bgra_tiff_writer, read_format, parse_quirks, OutInfo, PixOrder, ImageInfo, InputInfo};
use crispyconv::processing::render_bayer;
use std::path::PathBuf;

/// Processes Bayer data into RGB
#[derive(Parser)]
#[clap(about)]
struct Args {
    /// Path to Bayer data
    input: PathBuf,
    /// Interpret file as raw data.
    /// Format: width_px,height_px,bpp
    #[clap(long = "raw", value_parser = OutInfo::parse_str)]
    format: Option<OutInfo>,
    /// Subpixel order
    #[clap(value_enum)]
    order: PixOrder,
    /// TIFF output
    output: PathBuf,
    /// Overrides the default set of hardware peculiarities.
    /// Defaults to most compatible.
    #[clap(long="quirks")]
    quirks: Option<String>,
}

fn main() {
    let args = Args::parse();
    let (info, reader) = read_format(args.format, args.input.as_path()).unwrap();
    let out_info = match info {
        ImageInfo::Output(out) => out,
        ImageInfo::Input(InputInfo {dimensions:(width_px, height_px), bpp}) => OutInfo { width_px, height_px, bpp},
    };
    let quirks = parse_quirks(args.quirks.as_deref()).unwrap();
    let OutInfo { width_px, height_px, bpp } = out_info;
    let dimensions = (width_px, height_px);
    
    draw(
        headless::ContextRef::new(),
        reader,
        bpp,
        quirks,
        dimensions,
        dimensions,
        |egl: &headless::ContextRef, source: &DmabufImage, target: &mut glium::framebuffer::SimpleFrameBuffer, gpubuf_bpp, quirks, dimensions | render_bayer(egl, source, target, gpubuf_bpp, quirks, dimensions, args.order),
        &mut get_bgra_tiff_writer(args.output.to_path_buf(), dimensions)
    ).unwrap();
}