use crate::common::Result;
use crate::Exceptions;
use super::Version;
#[derive(Debug, PartialEq, Eq, Copy, Clone)]
pub enum Mode {
TERMINATOR, NUMERIC, ALPHANUMERIC, STRUCTURED_APPEND, BYTE, ECI, KANJI, FNC1_FIRST_POSITION, FNC1_SECOND_POSITION,
HANZI, }
impl Mode {
pub fn forBits(bits: u8) -> Result<Self> {
match bits {
0x0 => Ok(Self::TERMINATOR),
0x1 => Ok(Self::NUMERIC),
0x2 => Ok(Self::ALPHANUMERIC),
0x3 => Ok(Self::STRUCTURED_APPEND),
0x4 => Ok(Self::BYTE),
0x5 => Ok(Self::FNC1_FIRST_POSITION),
0x7 => Ok(Self::ECI),
0x8 => Ok(Self::KANJI),
0x9 => Ok(Self::FNC1_SECOND_POSITION),
0xD =>
{
Ok(Self::HANZI)
}
_ => Err(Exceptions::illegal_argument_with(format!(
"{bits} is not valid"
))),
}
}
pub fn getCharacterCountBits(&self, version: &Version) -> u8 {
let number = version.getVersionNumber();
let offset = if number <= 9 {
0
} else if number <= 26 {
1
} else {
2
};
self.get_character_counts()[offset]
}
fn get_character_counts(&self) -> &[u8] {
match self {
Mode::TERMINATOR => &[0, 0, 0],
Mode::NUMERIC => &[10, 12, 14],
Mode::ALPHANUMERIC => &[9, 11, 13],
Mode::STRUCTURED_APPEND => &[0, 0, 0],
Mode::BYTE => &[8, 16, 16],
Mode::ECI => &[0, 0, 0],
Mode::KANJI => &[8, 10, 12],
Mode::FNC1_FIRST_POSITION => &[0, 0, 0],
Mode::FNC1_SECOND_POSITION => &[0, 0, 0],
Mode::HANZI => &[8, 10, 12],
}
}
pub fn getBits(&self) -> u8 {
match self {
Mode::TERMINATOR => 0x00,
Mode::NUMERIC => 0x01,
Mode::ALPHANUMERIC => 0x02,
Mode::STRUCTURED_APPEND => 0x03,
Mode::BYTE => 0x04,
Mode::ECI => 0x07,
Mode::KANJI => 0x08,
Mode::FNC1_FIRST_POSITION => 0x05,
Mode::FNC1_SECOND_POSITION => 0x09,
Mode::HANZI => 0x0D,
}
}
pub const fn get_terminator_bit_length(version: &Version) -> u8 {
(if version.isMicroQRCode() {
version.getVersionNumber() * 2 + 1
} else {
4
}) as u8
}
pub const fn get_codec_mode_bits_length(version: &Version) -> u8 {
(if version.isMicroQRCode() {
version.getVersionNumber() - 1
} else {
4
}) as u8
}
pub fn CodecModeForBits(bits: u32, isMicro: Option<bool>) -> Result<Self> {
let isMicro = isMicro.unwrap_or(false);
const BITS_2_MODE_LEN: usize = 4;
if !isMicro {
if (0x00..=0x05).contains(&bits) || (0x07..=0x09).contains(&bits) || bits == 0x0d {
return Mode::try_from(bits);
}
} else {
const BITS_2_MODE: [Mode; BITS_2_MODE_LEN] =
[Mode::NUMERIC, Mode::ALPHANUMERIC, Mode::BYTE, Mode::KANJI];
if (bits as usize) < BITS_2_MODE_LEN {
return Ok(BITS_2_MODE[bits as usize]);
}
}
Err(Exceptions::format_with("Invalid codec mode"))
}
pub fn CharacterCountBits(&self, version: &Version) -> u32 {
let number = version.getVersionNumber() as usize;
if version.isMicroQRCode() {
match self {
Mode::NUMERIC=> return [3, 4, 5, 6][number - 1],
Mode::ALPHANUMERIC=> return [3, 4, 5][number - 2],
Mode::BYTE=> return [4, 5][number - 3],
Mode::KANJI | Mode::HANZI=> return [3, 4][number - 3],
_=> return 0,
}
}
let i = if number <= 9 {
0
} else if number <= 26 {
1
} else {
2
};
match self {
Mode::NUMERIC=> [10, 12, 14][i],
Mode::ALPHANUMERIC=> [9, 11, 13][i],
Mode::BYTE=> [8, 16, 16][i],
Mode::KANJI| Mode::HANZI=> [8, 10, 12][i],
_=> 0,
}
}
}
impl From<Mode> for u8 {
fn from(value: Mode) -> Self {
value.getBits()
}
}
impl TryFrom<u8> for Mode {
type Error = Exceptions;
fn try_from(value: u8) -> Result<Self, Self::Error> {
Self::forBits(value)
}
}
impl TryFrom<u32> for Mode {
type Error = Exceptions;
fn try_from(value: u32) -> Result<Self, Self::Error> {
Self::forBits(value as u8)
}
}