xutf 0.1.0

//! Visible width benchmark.
//!
//! Run default ratio bench: `cargo bench --bench width`
//! Run Divan bench: `DIVAN=1 cargo bench --bench width`

use divan::Bencher;
use rayon::prelude::*;
use unicode_width::UnicodeWidthStr;
use xutf::{Utf16, width, width_str};

mod common;
use common::{build_input, measure, print_ratio_table};

common::bench_main!();

const TARGET: usize = 1 << 20;

fn run_ratio_table() {
	let inputs = [
		("ascii", build_input("The quick brown fox jumps over 13 lazy dogs. ", TARGET)),
		("cjk", build_input("漢字界面 日本語 カタカナ 한글 ", TARGET)),
		("emoji", build_input("😀 ✅ 🚀 ⚠️ ℹ️ ❤️ 0️⃣ 👨‍👩‍👧 👋🏽 🇺🇸 ", TARGET)),
		("mixed", build_input("status: ✅ café 漢字 e\u{301} 🇯🇵 box ├─┤ ", TARGET)),
	];
	let utf16: Vec<Vec<u16>> = inputs
		.iter()
		.map(|(_, input)| input.encode_utf16().collect())
		.collect();

	let rows: Vec<_> = inputs
		.par_iter()
		.enumerate()
		.map(|(index, (name, input))| {
			let units = &utf16[index];
			(*name, vec![
				measure(input.len(), || width_str(input)),
				measure(input.len(), || UnicodeWidthStr::width(input.as_str())),
				measure(input.len(), || width::<Utf16<false>>(units)),
				measure(input.len(), || {
					let decoded: String = char::decode_utf16(units.iter().copied())
						.map(|result| result.unwrap_or(char::REPLACEMENT_CHARACTER))
						.collect();
					UnicodeWidthStr::width(decoded.as_str())
				}),
			])
		})
		.collect();

	print_ratio_table(
		"Visible width",
		&["xutf", "unicode-width", "xutf utf16", "utf16 via String"],
		&rows,
	);
}

#[divan::bench_group]
mod ascii {
	use super::*;

	fn input() -> String {
		build_input("The quick brown fox jumps over 13 lazy dogs. ", TARGET)
	}

	#[divan::bench]
	fn xutf(bencher: Bencher) {
		let s = input();
		bencher
			.counter(divan::counter::BytesCount::of_slice(s.as_bytes()))
			.bench_local(|| width_str(&s));
	}

	#[divan::bench]
	fn unicode_width(bencher: Bencher) {
		let s = input();
		bencher
			.counter(divan::counter::BytesCount::of_slice(s.as_bytes()))
			.bench_local(|| UnicodeWidthStr::width(s.as_str()));
	}

	#[divan::bench]
	fn xutf_utf16(bencher: Bencher) {
		let s = input();
		let utf16: Vec<u16> = s.encode_utf16().collect();
		bencher
			.counter(divan::counter::BytesCount::of_slice(s.as_bytes()))
			.bench_local(|| width::<Utf16<false>>(&utf16));
	}

	#[divan::bench]
	fn utf16_via_string(bencher: Bencher) {
		let s = input();
		let utf16: Vec<u16> = s.encode_utf16().collect();
		bencher
			.counter(divan::counter::BytesCount::of_slice(s.as_bytes()))
			.bench_local(|| {
				let decoded: String = char::decode_utf16(utf16.iter().copied())
					.map(|result| result.unwrap_or(char::REPLACEMENT_CHARACTER))
					.collect();
				UnicodeWidthStr::width(decoded.as_str())
			});
	}
}

#[divan::bench_group]
mod cjk {
	use super::*;

	fn input() -> String {
		build_input("漢字界面 日本語 カタカナ 한글 ", TARGET)
	}

	#[divan::bench]
	fn xutf(bencher: Bencher) {
		let s = input();
		bencher
			.counter(divan::counter::BytesCount::of_slice(s.as_bytes()))
			.bench_local(|| width_str(&s));
	}

	#[divan::bench]
	fn unicode_width(bencher: Bencher) {
		let s = input();
		bencher
			.counter(divan::counter::BytesCount::of_slice(s.as_bytes()))
			.bench_local(|| UnicodeWidthStr::width(s.as_str()));
	}

	#[divan::bench]
	fn xutf_utf16(bencher: Bencher) {
		let s = input();
		let utf16: Vec<u16> = s.encode_utf16().collect();
		bencher
			.counter(divan::counter::BytesCount::of_slice(s.as_bytes()))
			.bench_local(|| width::<Utf16<false>>(&utf16));
	}

	#[divan::bench]
	fn utf16_via_string(bencher: Bencher) {
		let s = input();
		let utf16: Vec<u16> = s.encode_utf16().collect();
		bencher
			.counter(divan::counter::BytesCount::of_slice(s.as_bytes()))
			.bench_local(|| {
				let decoded: String = char::decode_utf16(utf16.iter().copied())
					.map(|result| result.unwrap_or(char::REPLACEMENT_CHARACTER))
					.collect();
				UnicodeWidthStr::width(decoded.as_str())
			});
	}
}

#[divan::bench_group]
mod emoji {
	use super::*;

	fn input() -> String {
		build_input("😀 ✅ 🚀 ⚠️ ℹ️ ❤️ 0️⃣ 👨‍👩‍👧 👋🏽 🇺🇸 ", TARGET)
	}

	#[divan::bench]
	fn xutf(bencher: Bencher) {
		let s = input();
		bencher
			.counter(divan::counter::BytesCount::of_slice(s.as_bytes()))
			.bench_local(|| width_str(&s));
	}

	#[divan::bench]
	fn unicode_width(bencher: Bencher) {
		let s = input();
		bencher
			.counter(divan::counter::BytesCount::of_slice(s.as_bytes()))
			.bench_local(|| UnicodeWidthStr::width(s.as_str()));
	}

	#[divan::bench]
	fn xutf_utf16(bencher: Bencher) {
		let s = input();
		let utf16: Vec<u16> = s.encode_utf16().collect();
		bencher
			.counter(divan::counter::BytesCount::of_slice(s.as_bytes()))
			.bench_local(|| width::<Utf16<false>>(&utf16));
	}

	#[divan::bench]
	fn utf16_via_string(bencher: Bencher) {
		let s = input();
		let utf16: Vec<u16> = s.encode_utf16().collect();
		bencher
			.counter(divan::counter::BytesCount::of_slice(s.as_bytes()))
			.bench_local(|| {
				let decoded: String = char::decode_utf16(utf16.iter().copied())
					.map(|result| result.unwrap_or(char::REPLACEMENT_CHARACTER))
					.collect();
				UnicodeWidthStr::width(decoded.as_str())
			});
	}
}

#[divan::bench_group]
mod mixed {
	use super::*;

	fn input() -> String {
		build_input("status: ✅ café 漢字 e\u{301} 🇯🇵 box ├─┤ ", TARGET)
	}

	#[divan::bench]
	fn xutf(bencher: Bencher) {
		let s = input();
		bencher
			.counter(divan::counter::BytesCount::of_slice(s.as_bytes()))
			.bench_local(|| width_str(&s));
	}

	#[divan::bench]
	fn unicode_width(bencher: Bencher) {
		let s = input();
		bencher
			.counter(divan::counter::BytesCount::of_slice(s.as_bytes()))
			.bench_local(|| UnicodeWidthStr::width(s.as_str()));
	}

	#[divan::bench]
	fn xutf_utf16(bencher: Bencher) {
		let s = input();
		let utf16: Vec<u16> = s.encode_utf16().collect();
		bencher
			.counter(divan::counter::BytesCount::of_slice(s.as_bytes()))
			.bench_local(|| width::<Utf16<false>>(&utf16));
	}

	#[divan::bench]
	fn utf16_via_string(bencher: Bencher) {
		let s = input();
		let utf16: Vec<u16> = s.encode_utf16().collect();
		bencher
			.counter(divan::counter::BytesCount::of_slice(s.as_bytes()))
			.bench_local(|| {
				let decoded: String = char::decode_utf16(utf16.iter().copied())
					.map(|result| result.unwrap_or(char::REPLACEMENT_CHARACTER))
					.collect();
				UnicodeWidthStr::width(decoded.as_str())
			});
	}
}