imsz 0.4.1

Get width and height from an image file reading as few bytes as possible.
Documentation 
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

#!/usr/bin/env python3

from typing import List, Tuple

import os

from os.path import dirname, abspath, join as joinpath, splitext

base_dir = dirname(abspath(__file__))
files = os.listdir(joinpath(base_dir, "testdata"))

buf: List[str] = [
"""\
fn get_testdata(fname: &str) -> std::path::PathBuf {
    let mut path = std::path::PathBuf::from(env!("CARGO_MANIFEST_DIR"));
    path.push("testdata");
    path.push(fname);
    return path;
}
"""
]

funcs: List[Tuple[str, str]] = []

format_map = {
    '.exr': 'OpenEXR',
    '.jp2': 'JP2K',
}

format_names = {
    'JP2K': 'JPEG 2000',
    'WEBP': 'WebP',
}

for fname in files:
    ident = fname.replace('.', '_')
    path = f"testdata/{fname}"

    leaf, ext = splitext(fname)
    format = format_map.get(ext) or ext[1:].upper()
    format_name = format_names.get(format, format)

    ident = '_'.join([ ext[1:], *leaf.split('_')[1:] ])

    funcs.append((ident, f"""
#[test]
fn {ident}() {{
    let info = imsz::imsz_from_path(get_testdata("{fname}"));
    match info {{
        Ok(info) => {{
            assert_eq!(info.format, imsz::ImFormat::{format});
            assert_eq!(info.format.name(), "{format_name}");
            assert_eq!(info.width,  32);
            assert_eq!(info.height, 16);
        }}
        Err(error) => {{
            assert!(false, "{{}}", error);
        }}
    }}
}}
"""))

funcs.sort()
buf.extend(code for _, code in funcs)

with open(joinpath(base_dir, "tests", "ok_files.rs"), "w") as fp:
    fp.write('\n'.join(buf))