sarpro 0.3.2

A high-performance Sentinel-1 Synthetic Aperture Radar (SAR) GRD product to image processor.
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

use gdal::Dataset;
use gdal::DriverManager;
use gdal::raster::{Buffer, ColorInterpretation};
use std::path::Path;

pub fn write_tiff_u8(
    output: &Path,
    cols: usize,
    rows: usize,
    data: &[u8],
) -> Result<Dataset, Box<dyn std::error::Error>> {
    let driver = DriverManager::get_driver_by_name("GTiff")?;
    let ds = driver.create_with_band_type::<u8, _>(output, cols, rows, 1)?;
    let mut buf = Buffer::new((cols, rows), data.to_vec());
    let mut band = ds.rasterband(1)?;
    band.write((0, 0), (cols, rows), &mut buf)?;
    Ok(ds)
}

pub fn write_tiff_u16(
    output: &Path,
    cols: usize,
    rows: usize,
    data: &[u16],
) -> Result<Dataset, Box<dyn std::error::Error>> {
    let driver = DriverManager::get_driver_by_name("GTiff")?;
    let ds = driver.create_with_band_type::<u16, _>(output, cols, rows, 1)?;
    let mut buf = Buffer::new((cols, rows), data.to_vec());
    let mut band = ds.rasterband(1)?;
    band.write((0, 0), (cols, rows), &mut buf)?;
    Ok(ds)
}

pub fn write_tiff_multiband_u8(
    output: &Path,
    cols: usize,
    rows: usize,
    band1: &[u8],
    band2: &[u8],
) -> Result<Dataset, Box<dyn std::error::Error>> {
    let driver = DriverManager::get_driver_by_name("GTiff")?;
    let ds = driver.create_with_band_type::<u8, _>(output, cols, rows, 2)?;

    let mut band1_handle = ds.rasterband(1)?;
    band1_handle.set_color_interpretation(ColorInterpretation::GrayIndex)?;
    let mut buf1 = Buffer::new((cols, rows), band1.to_vec());
    band1_handle.write((0, 0), (cols, rows), &mut buf1)?;

    let mut band2_handle = ds.rasterband(2)?;
    band2_handle.set_color_interpretation(ColorInterpretation::GrayIndex)?;
    let mut buf2 = Buffer::new((cols, rows), band2.to_vec());
    band2_handle.write((0, 0), (cols, rows), &mut buf2)?;

    Ok(ds)
}

pub fn write_tiff_multiband_u16(
    output: &Path,
    cols: usize,
    rows: usize,
    band1: &[u16],
    band2: &[u16],
) -> Result<Dataset, Box<dyn std::error::Error>> {
    let driver = DriverManager::get_driver_by_name("GTiff")?;
    let ds = driver.create_with_band_type::<u16, _>(output, cols, rows, 2)?;

    let mut band1_handle = ds.rasterband(1)?;
    band1_handle.set_color_interpretation(ColorInterpretation::GrayIndex)?;
    let mut buf1 = Buffer::new((cols, rows), band1.to_vec());
    band1_handle.write((0, 0), (cols, rows), &mut buf1)?;

    let mut band2_handle = ds.rasterband(2)?;
    band2_handle.set_color_interpretation(ColorInterpretation::GrayIndex)?;
    let mut buf2 = Buffer::new((cols, rows), band2.to_vec());
    band2_handle.write((0, 0), (cols, rows), &mut buf2)?;

    Ok(ds)
}