mozjpeg/
lib.rs

1#![doc = include_str!("../README.md")]
2#![allow(unused_attributes)]
3#![allow(unused_imports)]
4#![allow(clippy::manual_range_contains)]
5
6use mozjpeg_sys as ffi;
7
8pub use crate::colorspace::ColorSpace;
9pub use crate::colorspace::ColorSpaceExt;
10pub use crate::component::CompInfo;
11pub use crate::component::CompInfoExt;
12pub use crate::compress::Compress;
13pub use crate::compress::ScanMode;
14pub use crate::decompress::{DctMethod, Format};
15pub use crate::decompress::{Decompress, ALL_MARKERS, NO_MARKERS};
16pub use crate::density::{PixelDensity, PixelDensityUnit};
17use crate::ffi::boolean;
18use crate::ffi::jpeg_common_struct;
19use crate::ffi::jpeg_compress_struct;
20pub use crate::ffi::DCTSIZE;
21use crate::ffi::JDIMENSION;
22pub use crate::ffi::JPEG_LIB_VERSION;
23use crate::ffi::J_BOOLEAN_PARAM;
24use crate::ffi::J_INT_PARAM;
25pub use crate::marker::Marker;
26
27use libc::free;
28use std::cmp::min;
29use std::mem;
30use std::os::raw::{c_int, c_uchar, c_ulong, c_void};
31use std::ptr;
32use std::slice;
33
34mod colorspace;
35mod component;
36pub mod compress;
37pub mod decompress;
38mod density;
39mod errormgr;
40mod marker;
41/// Quantization table presets from MozJPEG
42pub mod qtable;
43mod readsrc;
44mod writedst;
45
46#[test]
47fn recompress() {
48    use crate::colorspace::{ColorSpace, ColorSpaceExt};
49    use std::fs::File;
50    use std::io::{Read, Write};
51
52    let dinfo = Decompress::new_path("tests/test.jpg").unwrap();
53
54    assert_eq!(1.0, dinfo.gamma());
55    assert_eq!(ColorSpace::JCS_YCbCr, dinfo.color_space());
56    assert_eq!(dinfo.components().len(), dinfo.color_space().num_components());
57
58    let samp_factors = dinfo.components().iter().map(|c| c.v_samp_factor).collect::<Vec<_>>();
59
60    assert_eq!((45, 30), dinfo.size());
61
62    let mut dinfo = dinfo.raw().unwrap();
63
64    let mut bitmaps = [&mut Vec::new(), &mut Vec::new(), &mut Vec::new()];
65    dinfo.read_raw_data(&mut bitmaps);
66
67    dinfo.finish().unwrap();
68
69    fn write_jpeg(bitmaps: &[&mut Vec<u8>; 3], samp_factors: &Vec<i32>, scale: (f32, f32)) -> Vec<u8> {
70        let mut cinfo = Compress::new(ColorSpace::JCS_YCbCr);
71
72        cinfo.set_size(45, 30);
73
74        #[allow(deprecated)] {
75            cinfo.set_gamma(1.0);
76        }
77
78        cinfo.set_raw_data_in(true);
79
80        cinfo.set_quality(100.);
81
82        cinfo.set_luma_qtable(&qtable::AnnexK_Luma.scaled(99. * scale.0, 90. * scale.1));
83        cinfo.set_chroma_qtable(&qtable::AnnexK_Chroma.scaled(100. * scale.0, 60. * scale.1));
84
85        for (c, samp) in cinfo.components_mut().iter_mut().zip(samp_factors) {
86            c.v_samp_factor = *samp;
87            c.h_samp_factor = *samp;
88        }
89
90        let mut cinfo = cinfo.start_compress(Vec::new()).unwrap();
91
92        assert!(cinfo.write_raw_data(&bitmaps.iter().map(|c| &c[..]).collect::<Vec<_>>()));
93
94        cinfo.finish().unwrap()
95    }
96
97    let data1 = &write_jpeg(&bitmaps, &samp_factors, (1., 1.));
98    let data1_len = data1.len();
99    let data2 = &write_jpeg(&bitmaps, &samp_factors, (0.5, 0.5));
100    let data2_len = data2.len();
101
102    File::create("testout-r1.jpg").unwrap().write_all(data1).unwrap();
103    File::create("testout-r2.jpg").unwrap().write_all(data2).unwrap();
104
105    assert!(data1_len > data2_len);
106}
107
108#[cold]
109fn fail(cinfo: &mut jpeg_common_struct, code: c_int) -> ! {
110    unsafe {
111        let err = &mut *cinfo.err;
112        err.msg_code = code;
113        if let Some(e) = err.error_exit {
114            (e)(cinfo); // should have been defined as !
115        }
116        std::process::abort();
117    }
118}
119
120fn warn(cinfo: &mut jpeg_common_struct, code: c_int) {
121    unsafe {
122        let err = &mut *cinfo.err;
123        err.msg_code = code;
124        if let Some(e) = err.emit_message {
125            (e)(cinfo, -1);
126        }
127    }
128}
mozjpeg/lib.rs

mozjpeg/
lib.rs
