agb/agb-image-converter/src/lib.rs

use std::fs::File;
use std::io::BufWriter;
use std::path::PathBuf;

use typed_builder::TypedBuilder;

mod colour;
mod image_loader;
mod palette16;
mod rust_generator;

use image_loader::Image;

pub use colour::Colour;

#[derive(Debug, Clone, Copy)]
pub enum TileSize {
    Tile8,
    Tile16,
}

impl TileSize {
    fn to_size(self) -> usize {
        match self {
            TileSize::Tile8 => 8,
            TileSize::Tile16 => 16,
        }
    }
}

#[derive(TypedBuilder)]
pub struct ImageConverterConfig {
    #[builder(default, setter(strip_option))]
    transparent_colour: Option<Colour>,
    tile_size: TileSize,
    input_image: PathBuf,
    output_file: PathBuf,

    #[builder(default, setter(strip_option))]
    crate_prefix: Option<String>,
}

pub fn convert_image(settings: ImageConverterConfig) {
    let image = Image::load_from_file(&settings.input_image);

    let tile_size = settings.tile_size.to_size();
    if image.width % tile_size != 0 || image.height % tile_size != 0 {
        panic!("Image size not a multiple of tile size");
    }

    let optimiser = optimiser_for_image(&image, tile_size);
    let optimisation_results = optimiser.optimise_palettes(settings.transparent_colour);

    let output_file = File::create(&settings.output_file).expect("Failed to create file");
    let mut writer = BufWriter::new(output_file);

    rust_generator::generate_code(
        &mut writer,
        &optimisation_results,
        &image,
        settings.tile_size,
        settings.crate_prefix.unwrap_or("agb".to_owned()),
    )
    .expect("Failed to write data");
}

fn optimiser_for_image(image: &Image, tile_size: usize) -> palette16::Palette16Optimiser {
    let tiles_x = image.width / tile_size;
    let tiles_y = image.height / tile_size;

    let mut palette_optimiser = palette16::Palette16Optimiser::new();

    for y in 0..tiles_y {
        for x in 0..tiles_x {
            let mut palette = palette16::Palette16::new();

            for j in 0..tile_size {
                for i in 0..tile_size {
                    let colour = image.colour(x * tile_size + i, y * tile_size + j);

                    palette.add_colour(colour);
                }
            }

            palette_optimiser.add_palette(palette);
        }
    }

    palette_optimiser
}