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

//! # JFIF Dump
//!
//! A crate for reading the content of a JFIF file without decoding JPEG image data.
//!
//! ## Example: Print image dimensions
//!
//! ```no_run
//! # use jfifdump::JfifError;
//! # fn main() -> Result<(), JfifError> {
//!
//! use jfifdump::{Reader, SegmentKind};
//! use std::fs::File;
//! use std::io::BufReader;
//!
//! let file = File::open("some.jpeg")?;
//!
//! let mut reader = Reader::new(BufReader::new(file))?;
//!
//! loop {
//!     match reader.next_segment()?.kind {
//!         SegmentKind::Eoi => break,
//!         SegmentKind::Frame(frame) => {
//!             println!("{}x{}", frame.dimension_x, frame.dimension_y);
//!             break;
//!         }
//!         _ => {
//!             // Ignore other segments
//!         }
//!     }
//! }
//!
//! # Ok(())
//! }
//! ```

#![allow(clippy::uninlined_format_args)]

use std::io::{ErrorKind, Read};

pub use error::JfifError;
pub use handler::Handler;
pub use reader::{
    App0Jfif, Dac, Dht, Dqt, Frame, FrameComponent, Reader, Rst, Scan, ScanComponent, Segment,
    SegmentKind,
};
pub use text::TextFormat;

#[cfg(feature = "json")]
pub use crate::json::JsonFormat;

mod error;
mod handler;
#[cfg(feature = "json")]
mod json;
mod reader;
mod text;

/// Read JFIF input and call handler for all segments
pub fn read<H: Handler, R: Read>(input: R, handler: &mut H) -> Result<(), JfifError> {
    let mut reader = Reader::new(input)?;

    loop {
        let segment = match reader.next_segment() {
            Ok(segment) => segment,
            Err(JfifError::IoError(ioerror)) => {
                return if ioerror.kind() == ErrorKind::UnexpectedEof {
                    Ok(())
                } else {
                    Err(JfifError::IoError(ioerror))
                }
            }
            Err(err) => return Err(err),
        };

        match segment.kind {
            SegmentKind::Soi => handler.handle_soi(segment.position, segment.length),
            SegmentKind::Eoi => handler.handle_eoi(segment.position, segment.length),
            SegmentKind::App { nr, data } => {
                handler.handle_app(segment.position, segment.length, nr, &data)
            }
            SegmentKind::App0Jfif(jfif) => {
                handler.handle_app0_jfif(segment.position, segment.length, &jfif)
            }
            SegmentKind::Dqt(tables) => {
                handler.handle_dqt(segment.position, segment.length, &tables)
            }
            SegmentKind::Dht(tables) => {
                handler.handle_dht(segment.position, segment.length, &tables)
            }
            SegmentKind::Dac(dac) => handler.handle_dac(segment.position, segment.length, &dac),
            SegmentKind::Frame(frame) => {
                handler.handle_frame(segment.position, segment.length, &frame)
            }
            SegmentKind::Scan(scan) => handler.handle_scan(segment.position, segment.length, &scan),
            SegmentKind::Dri(restart) => {
                handler.handle_dri(segment.position, segment.length, restart)
            }
            SegmentKind::Rst(rst) => handler.handle_rst(segment.position, segment.length, &rst),
            SegmentKind::Comment(data) => {
                handler.handle_comment(segment.position, segment.length, &data)
            }
            SegmentKind::Unknown { marker, data } => {
                handler.handle_unknown(segment.position, segment.length, marker, &data)
            }
        };
    }
}