lensight 0.1.0

A command-line tool for adding camera information and shooting details to photos.
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
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

//! Utility functions module
//!
//! This module contains helper functions for processing files and directories.

use std::path::Path;
use walkdir::WalkDir;

/// Processes all JPEG files in a directory
///
/// # Arguments
/// * `input` - Input directory path
/// * `output` - Output directory path
/// * `info_height` - Height of the information bar in pixels
/// * `force_16_9` - Whether to force 16:9 aspect ratio
/// * `custom_logo_path` - Optional path to a custom logo file
///
/// # Returns
/// * `Result<(), Box<dyn std::error::Error>>` - Ok if successful
///
/// # Errors
/// Returns an error if:
/// - The output directory cannot be created
/// - Any file cannot be processed
pub fn process_directory(
    input: &Path,
    output: &Path,
    info_height: u32,
    force_16_9: bool,
    custom_logo_path: Option<&Path>,
) -> Result<(), Box<dyn std::error::Error>> {
    if !output.exists() {
        std::fs::create_dir_all(output)?;
    }

    let entries: Vec<_> = WalkDir::new(input)
        .into_iter()
        .filter_map(|e| e.ok())
        .filter(|e| e.file_type().is_file())
        .filter(|e| {
            if let Some(ext) = e.path().extension() {
                ext == "jpg" || ext == "jpeg"
            } else {
                false
            }
        })
        .collect();
    let total = entries.len();
    for (idx, entry) in entries.iter().enumerate() {
        let path = entry.path();
        let output_path = output.join(path.file_name().unwrap());
        println!("Processing {}/{}: {}", idx + 1, total, path.display());
        process_single_file(
            path,
            &output_path,
            info_height,
            force_16_9,
            custom_logo_path,
        )?;
    }

    Ok(())
}

/// Processes a single image file
///
/// # Arguments
/// * `input` - Input file path
/// * `output` - Output file path
/// * `info_height` - Height of the information bar in pixels
/// * `force_16_9` - Whether to force 16:9 aspect ratio
/// * `custom_logo_path` - Optional path to a custom logo file
///
/// # Returns
/// * `Result<(), Box<dyn std::error::Error>>` - Ok if successful
///
/// # Errors
/// Returns an error if:
/// - The input file cannot be opened
/// - The image cannot be processed
/// - The output file cannot be saved
pub fn process_single_file(
    input: &Path,
    output: &Path,
    info_height: u32,
    force_16_9: bool,
    custom_logo_path: Option<&Path>,
) -> Result<(), Box<dyn std::error::Error>> {
    let orig_img = image::open(input)?;
    let resources = crate::resource::Resources::new(info_height)?;
    let watermarked = crate::image_processor::add_info_bar(
        orig_img.clone(),
        input,
        info_height,
        &resources,
        custom_logo_path,
    )?;
    let final_img = if force_16_9 {
        crate::image_processor::pad_to_16_9(&orig_img, &watermarked)
    } else {
        watermarked
    };
    final_img.save(output)?;
    Ok(())
}