use wcloud::{Tokenizer, WordCloud, WordCloudSize, Word};
use nanorand::{Rng, WyRand};
use palette::{Pixel, Srgb, Hsl, IntoColor};
use image::{ImageFormat, Rgba};

use std::time::Instant;

fn main() {
    let text = include_str!("computer_font_wikipedia.txt");

    let tokenizer = Tokenizer::default()
        .with_max_words(1000)
        .with_repeat(true);

    let wordcloud = WordCloud::default()
        .with_tokenizer(tokenizer)
        .with_rng_seed(0)
        .with_font_from_path("examples/custom_fonts/DroidSansMono.ttf".into());

    let mask_buf = include_bytes!("mask.png");
    let mask_image = image::load_from_memory_with_format(mask_buf, ImageFormat::Png)
        .expect("Unable to load mask from memory")
        .to_luma8();

    let mask = WordCloudSize::FromMask(mask_image);

    let color_func = |_word: &Word, rng: &mut WyRand| {
        let lightness = rng.generate_range(40..100);

        let col = Hsl::new(0.0, 0.0, lightness as f32 / 100.0);
        let rgb: Srgb = col.into_color();

        let raw: [u8; 3] = rgb.into_format()
            .into_raw();

        Rgba([raw[0], raw[1], raw[2], 1])
    };

    let now = Instant::now();
    let wordcloud_image = wordcloud.generate_from_text_with_color_func(text, mask, 1.0, color_func);

    println!("Generated in {}ms", now.elapsed().as_millis());

    wordcloud_image.save("examples/custom_fonts/cloud.png")
        .expect("Unable to save image");
}