use crate::{
common::{BitArray, Result},
Exceptions,
};
use super::{BinaryShiftToken, SimpleToken};
#[derive(Debug, PartialEq, Eq, Clone)]
pub enum TokenType {
Simple(SimpleToken),
BinaryShift(BinaryShiftToken),
Empty,
}
impl TokenType {
pub fn appendTo(&self, bit_array: &mut BitArray, text: &[u8]) -> Result<()> {
match self {
TokenType::Simple(a) => a.appendTo(bit_array, text),
TokenType::BinaryShift(a) => a.appendTo(bit_array, text),
TokenType::Empty => Err(Exceptions::illegal_state_with(
"cannot appendTo on Empty final item",
)),
}
}
}
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Token {
tokens: Vec<TokenType>,
}
impl Token {
pub fn new() -> Self {
Self {
tokens: Vec::new(),
}
}
pub fn add(&mut self, value: i32, bit_count: u32) {
self.tokens
.push(TokenType::Simple(SimpleToken::new(value, bit_count)));
}
pub fn addBinaryShift(&mut self, start: u32, byte_count: u32) {
self.tokens
.push(TokenType::BinaryShift(BinaryShiftToken::new(
start, byte_count,
)));
}
}
pub struct TokenIter {
src: Vec<TokenType>,
}
impl Iterator for TokenIter {
type Item = TokenType;
fn next(&mut self) -> Option<Self::Item> {
self.src.pop()
}
}
impl IntoIterator for Token {
type Item = TokenType;
type IntoIter = TokenIter;
fn into_iter(self) -> Self::IntoIter {
TokenIter {
src: self.tokens,
}
}
}
impl Default for Token {
fn default() -> Self {
Self::new()
}
}