rasterfakers 0.1.0

A Rust library and CLI tool to generate fake GeoTIFF files for testing and fixtures.
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
105
106
107
108

use clap::Parser;
use rasterfakers::geotiff::{FakeGeoTiff, FakeGeoTiffConfig};
use std::error::Error;

/// A CLI tool to generate fake GeoTIFF files for testing and fixtures.
#[derive(Parser)]
#[command(name = "RasterFakers")]
#[command(version = "0.1.0")]
#[command(about = "Generates fake GeoTIFF files", long_about = None)]
struct CliArgs {
    /// Output file path
    #[arg(short = 'o', long)]
    output: String,

    /// Width of the GeoTIFF in pixels
    #[arg(short = 'w', long, default_value_t = 256)]
    width: usize,

    /// Height of the GeoTIFF in pixels - short is 'e' to avoid conflict with 'help'
    #[arg(short = 'e', long, default_value_t = 256)]
    height: usize,

    /// Number of bands
    #[arg(short = 'b', long, default_value_t = 1)]
    bands: usize,

    /// Data type (u8, u16, i16, u32, i32, f32, f64)
    #[arg(short = 't', long, default_value = "f64")]
    data_type: String,

    /// Projection (e.g., EPSG:4326)
    #[arg(short = 'p', long, default_value = "EPSG:4326")]
    projection: String,

    /// Pixel resolution as two comma-separated values (e.g., "0.25,0.25")
    #[arg(short = 'r', long, default_value = "1.0,1.0")]
    pixel_resolution: String,

    /// Upper-left corner coordinates as two comma-separated values (e.g., "30.0,10.0")
    #[arg(short = 'c', long, default_value = "0.0,0.0")]
    upper_left_corner: String,
}

fn main() -> Result<(), Box<dyn Error>> {
    // Parse command-line arguments
    let args = CliArgs::parse();

    // Parse pixel resolution
    let pixel_resolution = parse_tuple(&args.pixel_resolution)?;
    // Parse upper-left corner
    let upper_left_corner = parse_tuple(&args.upper_left_corner)?;

    // Create the configuration
    let config = FakeGeoTiffConfig {
        width: args.width,
        height: args.height,
        bands: args.bands,
        projection: Some(args.projection),
        pixel_resolution: Some(pixel_resolution),
        upper_left_corner: Some(upper_left_corner),
        output_path: Some(args.output.clone()),
        ..Default::default()
    };

    // Determine data type and generate the GeoTIFF
    match args.data_type.as_str() {
        "u8" => {
            FakeGeoTiff::<u8>::from_config(config).write_to_file()?;
        }
        "u16" => {
            FakeGeoTiff::<u16>::from_config(config).write_to_file()?;
        }
        "i16" => {
            FakeGeoTiff::<i16>::from_config(config).write_to_file()?;
        }
        "u32" => {
            FakeGeoTiff::<u32>::from_config(config).write_to_file()?;
        }
        "i32" => {
            FakeGeoTiff::<i32>::from_config(config).write_to_file()?;
        }
        "f32" => {
            FakeGeoTiff::<f32>::from_config(config).write_to_file()?;
        }
        "f64" => {
            FakeGeoTiff::<f64>::from_config(config).write_to_file()?;
        }
        _ => {
            eprintln!("Unsupported data type: {}", args.data_type);
            std::process::exit(1);
        }
    }

    println!("GeoTIFF generated successfully at {}", args.output);

    Ok(())
}

// Helper function to parse comma-separated values into a tuple of f64
fn parse_tuple(s: &str) -> Result<(f64, f64), Box<dyn Error>> {
    let parts: Vec<&str> = s.split(',').collect();
    if parts.len() != 2 {
        return Err(format!("Expected two comma-separated values, got '{}'", s).into());
    }
    let first = parts[0].trim().parse::<f64>()?;
    let second = parts[1].trim().parse::<f64>()?;
    Ok((first, second))
}