use crate::Exceptions;
use crate::common::Result;
use crate::qrcode::cpp_port::Type;
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 fn get_terminator_bit_length(version: &Version) -> u8 {
(if version.isMicro() {
version.getVersionNumber() * 2 + 1
} else {
4 - u32::from(version.isRMQR())
}) as u8
}
pub fn get_codec_mode_bits_length(version: &Version) -> u8 {
(if version.isMicro() {
version.getVersionNumber() - 1
} else {
4 - u32::from(version.isRMQR())
}) as u8
}
pub fn CodecModeForBits(bits: u32, qr_type: Option<Type>) -> Result<Self> {
let qr_type = qr_type.unwrap_or(Type::Model2);
let bits = bits as usize;
if qr_type == Type::Micro {
const BITS2MODE: [Mode; 4] =
[Mode::NUMERIC, Mode::ALPHANUMERIC, Mode::BYTE, Mode::KANJI];
if bits < (BITS2MODE.len()) {
return Ok(BITS2MODE[bits]);
}
} else if qr_type == Type::RectMicro {
const BITS2MODE: [Mode; 8] = [
Mode::TERMINATOR,
Mode::NUMERIC,
Mode::ALPHANUMERIC,
Mode::BYTE,
Mode::KANJI,
Mode::FNC1_FIRST_POSITION,
Mode::FNC1_SECOND_POSITION,
Mode::ECI,
];
if bits < (BITS2MODE.len()) {
return Ok(BITS2MODE[bits]);
}
} else if (0x00..=0x05).contains(&bits) || (0x07..=0x09).contains(&bits) || bits == 0x0d {
return Mode::try_from(bits as u32);
}
Err(Exceptions::format_with("Invalid codec mode"))
}
pub fn CharacterCountBits(&self, version: &Version) -> u32 {
let number = version.getVersionNumber() as usize;
if version.isMicro() {
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,
}
}
if version.isRMQR() {
const NUMERIC: [u32; 32] = [
4, 5, 6, 7, 7, 5, 6, 7, 7, 8, 4, 6, 7, 7, 8, 8, 5, 6, 7, 7, 8, 8, 7, 7, 8, 8, 9, 7,
8, 8, 8, 9,
];
const ALPHANUM: [u32; 32] = [
3, 5, 5, 6, 6, 5, 5, 6, 6, 7, 4, 5, 6, 6, 7, 7, 5, 6, 6, 7, 7, 8, 6, 7, 7, 7, 8, 6,
7, 7, 8, 8,
];
const BYTE: [u32; 32] = [
3, 4, 5, 5, 6, 4, 5, 5, 6, 6, 3, 5, 5, 6, 6, 7, 4, 5, 6, 6, 7, 7, 6, 6, 7, 7, 7, 6,
6, 7, 7, 8,
];
const KANJI: [u32; 32] = [
2, 3, 4, 5, 5, 3, 4, 5, 5, 6, 2, 4, 5, 5, 6, 6, 3, 5, 5, 6, 6, 7, 5, 5, 6, 6, 7, 5,
6, 6, 6, 7,
];
match self {
Mode::NUMERIC => return NUMERIC[number - 1],
Mode::ALPHANUMERIC => return ALPHANUM[number - 1],
Mode::BYTE => return BYTE[number - 1],
Mode::KANJI => return KANJI[number - 1],
_ => 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)
}
}