diff options
| author | Daniel Schadt <kingdread@gmx.de> | 2023-01-12 22:35:44 +0100 | 
|---|---|---|
| committer | Daniel Schadt <kingdread@gmx.de> | 2023-01-12 22:35:44 +0100 | 
| commit | def44c6e969f1027da82e4130a82db7e4916ba86 (patch) | |
| tree | 79fb1209b24261dff93e656f471d599c51b356ef /src | |
| parent | 4042105e6d6a711dfec353b9cd9932fd9bf9c8af (diff) | |
| download | hittekaart-def44c6e969f1027da82e4130a82db7e4916ba86.tar.gz hittekaart-def44c6e969f1027da82e4130a82db7e4916ba86.tar.bz2 hittekaart-def44c6e969f1027da82e4130a82db7e4916ba86.zip  | |
clean up some unused code
Some of it was written because it fit the API, but we didn't end up
using it in main.rs.
Diffstat (limited to 'src')
| -rw-r--r-- | src/layer.rs | 116 | ||||
| -rw-r--r-- | src/renderer.rs | 40 | 
2 files changed, 19 insertions, 137 deletions
diff --git a/src/layer.rs b/src/layer.rs index 74c36e0..a8057a0 100644 --- a/src/layer.rs +++ b/src/layer.rs @@ -3,60 +3,36 @@  //! This supports OSM-style "tiled" images, but not all of the tiles have to be present. If a tile  //! is not present, a default pixel is returned. The tile is allocated with the first call to a  //! mutating operation. -use std::{ -    fs::{self, File}, -    io::BufWriter, -    path::Path, -}; +use std::{fs::File, io::BufWriter, path::Path}; -use color_eyre::eyre::{bail, Result}; +use color_eyre::eyre::Result;  use fnv::FnvHashMap;  use image::{      codecs::png::{CompressionType, FilterType, PngEncoder}, -    ColorType, ImageBuffer, ImageEncoder, Pixel, Rgba, RgbaImage, +    ColorType, ImageBuffer, ImageEncoder, Pixel, RgbaImage,  };  use num_traits::Zero;  pub const TILE_HEIGHT: u64 = 256;  pub const TILE_WIDTH: u64 = 256; +type TileIndex = (u64, u64); +  /// Main "lazy image buffer" struct.  #[derive(Debug, Clone)]  pub struct TileLayer<P: Pixel> { -    tiles: FnvHashMap<(u64, u64), ImageBuffer<P, Vec<P::Subpixel>>>, +    tiles: FnvHashMap<TileIndex, ImageBuffer<P, Vec<P::Subpixel>>>,      default_pixel: P, -    width: u64, -    height: u64,  }  impl<P: Pixel> TileLayer<P> { -    pub fn from_pixel(width: u64, height: u64, pixel: P) -> Self { +    pub fn from_pixel(pixel: P) -> Self {          TileLayer {              tiles: Default::default(),              default_pixel: pixel, -            width, -            height,          }      } -    pub fn width(&self) -> u64 { -        self.width -    } - -    pub fn height(&self) -> u64 { -        self.height -    } - -    fn index(&self, x: u64, y: u64) -> ((u64, u64), (u32, u32)) { -        ( -            (x / TILE_WIDTH, y / TILE_HEIGHT), -            ( -                (x % TILE_WIDTH).try_into().unwrap(), -                (y % TILE_HEIGHT).try_into().unwrap(), -            ), -        ) -    } -      pub fn enumerate_tiles(          &self,      ) -> impl Iterator<Item = (u64, u64, &ImageBuffer<P, Vec<P::Subpixel>>)> { @@ -69,45 +45,6 @@ impl<P: Pixel> TileLayer<P> {          })      } -    pub fn tile_for_mut(&mut self, x: u64, y: u64) -> &mut ImageBuffer<P, Vec<P::Subpixel>> { -        let ((tx, ty), _) = self.index(x, y); -        self.tile_mut(tx, ty) -    } - -    pub fn get_pixel_checked(&self, x: u64, y: u64) -> Option<&P> { -        if x >= self.width || y >= self.height { -            return None; -        } - -        let (outer_idx, (inner_x, inner_y)) = self.index(x, y); -        self.tiles -            .get(&outer_idx) -            .map(|tile| tile.get_pixel(inner_x, inner_y)) -            .or(Some(&self.default_pixel)) -    } - -    pub fn get_pixel(&self, x: u64, y: u64) -> &P { -        // This is kinda cheating, but we care about the API for now, not the speed. We can -        // optimize this later. -        self.get_pixel_checked(x, y).unwrap() -    } - -    pub fn get_pixel_mut_checked(&mut self, x: u64, y: u64) -> Option<&mut P> { -        if x >= self.width || y >= self.height { -            return None; -        } - -        let ((outer_x, outer_y), (inner_x, inner_y)) = self.index(x, y); -        Some( -            self.tile_mut(outer_x, outer_y) -                .get_pixel_mut(inner_x, inner_y), -        ) -    } - -    pub fn get_pixel_mut(&mut self, x: u64, y: u64) -> &mut P { -        self.get_pixel_mut_checked(x, y).unwrap() -    } -      /// Enumerate all pixels that are explicitely set in this layer.      pub fn enumerate_pixels(&self) -> impl Iterator<Item = (u64, u64, &P)> {          self.tiles.iter().flat_map(|((tx, ty), tile)| { @@ -121,27 +58,10 @@ impl<P: Pixel> TileLayer<P> {          })      } -    /// Mutably enumerate all pixels that are explicitely set in this layer. -    pub fn enumerate_pixels_mut(&mut self) -> impl Iterator<Item = (u64, u64, &mut P)> { -        self.tiles.iter_mut().flat_map(|((tx, ty), tile)| { -            tile.enumerate_pixels_mut().map(move |(x, y, p)| { -                ( -                    u64::from(x) + tx * TILE_WIDTH, -                    u64::from(y) + ty * TILE_HEIGHT, -                    p, -                ) -            }) -        }) -    } -      pub fn pixels(&self) -> impl Iterator<Item = &P> {          self.enumerate_pixels().map(|x| x.2)      } -    pub fn pixels_mut(&mut self) -> impl Iterator<Item = &mut P> { -        self.enumerate_pixels_mut().map(|x| x.2) -    } -      pub fn tile_count(&self) -> usize {          self.tiles.len()      } @@ -184,32 +104,12 @@ impl<P: Pixel> TileLayer<P> {      }  } -impl TileLayer<Rgba<u8>> { -    pub fn save_to_directory<S: AsRef<Path>>(&self, path: S) -> Result<()> { -        let path = path.as_ref(); - -        for ((x, y), tile) in self.tiles.iter() { -            let folder = path.join(&x.to_string()); -            let metadata = folder.metadata(); -            match metadata { -                Err(_) => fs::create_dir(&folder)?, -                Ok(m) if !m.is_dir() => bail!("Output path is not a directory"), -                _ => {} -            } -            let file = folder.join(&format!("{y}.png")); -            compress_png(tile, file)?; -        } - -        Ok(()) -    } -} -  pub fn compress_png<P: AsRef<Path>>(image: &RgbaImage, path: P) -> Result<()> {      let outstream = BufWriter::new(File::create(path)?);      let encoder =          PngEncoder::new_with_quality(outstream, CompressionType::Best, FilterType::Adaptive); -    encoder.write_image(&image, image.width(), image.height(), ColorType::Rgba8)?; +    encoder.write_image(image, image.width(), image.height(), ColorType::Rgba8)?;      Ok(())  } diff --git a/src/renderer.rs b/src/renderer.rs index 6e17304..cece8f3 100644 --- a/src/renderer.rs +++ b/src/renderer.rs @@ -1,9 +1,8 @@ -use std::{fs, mem, path::Path}; +use std::{fs, path::Path};  use color_eyre::eyre::{bail, Result}; -use image::{ImageBuffer, Luma, Pixel, Rgba, RgbaImage}; +use image::{ImageBuffer, Luma, Pixel, RgbaImage};  use nalgebra::{vector, Vector2}; -use num_traits::identities::Zero;  use super::{      gpx::Coordinates, @@ -12,8 +11,6 @@ use super::{  pub type HeatCounter = TileLayer<Luma<u8>>; -pub type HeatMap = TileLayer<Rgba<u8>>; -  fn render_circle<P: Pixel>(layer: &mut TileLayer<P>, center: (u64, u64), radius: u64, pixel: P) {      let topleft = (center.0 - radius, center.1 - radius);      let rad_32: u32 = radius.try_into().unwrap(); @@ -59,14 +56,14 @@ fn render_line<P: Pixel>(      fn unsigned_add(a: Vector2<u64>, b: Vector2<i32>) -> Vector2<u64> {          let x = if b[0] < 0 { -            a[0] - b[0].abs() as u64 +            a[0] - u64::from(b[0].unsigned_abs())          } else { -            a[0] + b[0] as u64 +            a[0] + u64::try_from(b[0]).unwrap()          };          let y = if b[1] < 0 { -            a[1] - b[1].abs() as u64 +            a[1] - u64::from(b[1].unsigned_abs())          } else { -            a[1] + b[1] as u64 +            a[1] + u64::try_from(b[1]).unwrap()          };          vector![x, y]      } @@ -130,19 +127,6 @@ fn colorize_tile(tile: &ImageBuffer<Luma<u8>, Vec<u8>>, max: u32) -> RgbaImage {      result  } -pub fn colorize_heatcounter(layer: &HeatCounter) -> HeatMap { -    let max = layer.pixels().map(|l| l.0[0]).max().unwrap_or_default(); -    let mut result = TileLayer::from_pixel(layer.width(), layer.height(), [0, 0, 0, 0].into()); -    if max == 0 { -        return result; -    } -    for (tile_x, tile_y, tile) in layer.enumerate_tiles() { -        let colorized = colorize_tile(&tile, max.into()); -        *result.tile_mut(tile_x, tile_y) = colorized; -    } -    result -} -  /// Lazily colorizes a [`HeatCounter`] by colorizing it tile-by-tile and saving a tile before  /// rendering the next one.  /// @@ -159,15 +143,15 @@ pub fn lazy_colorization<P: AsRef<Path>, F: FnMut(usize)>(      }      for (tile_x, tile_y, tile) in layer.enumerate_tiles() { -        let colorized = colorize_tile(&tile, max.into()); -        let folder = base_dir.join(&tile_x.to_string()); +        let colorized = colorize_tile(tile, max.into()); +        let folder = base_dir.join(tile_x.to_string());          let metadata = folder.metadata();          match metadata {              Err(_) => fs::create_dir(&folder)?,              Ok(m) if !m.is_dir() => bail!("Output path is not a directory"),              _ => {}          } -        let file = folder.join(&format!("{tile_y}.png")); +        let file = folder.join(format!("{tile_y}.png"));          layer::compress_png(&colorized, file)?;          progress_callback(1);      } @@ -180,12 +164,10 @@ pub fn render_heatcounter<F: FnMut(usize)>(      tracks: &[Vec<Coordinates>],      mut progress_callback: F,  ) -> HeatCounter { -    let size = 256 * 2u64.pow(zoom); - -    let mut heatcounter = TileLayer::from_pixel(size, size, [0].into()); +    let mut heatcounter = TileLayer::from_pixel([0].into());      for track in tracks { -        let mut layer = TileLayer::from_pixel(size, size, [0].into()); +        let mut layer = TileLayer::from_pixel([0].into());          let points = track              .iter()  | 
