use crate::Suit;
use core::fmt::{self, Write as _};
use core::iter::FusedIterator;
use core::num::NonZero;
use core::ops;
use core::str::FromStr;
use thiserror::Error;
#[derive(Debug, Error, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
#[error("{0} is not a valid rank (2..=14)")]
pub struct InvalidRank(u8);
#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[cfg_attr(feature = "serde", serde(transparent))]
#[repr(transparent)]
pub struct Rank(NonZero<u8>);
impl Rank {
pub const A: Self = Self(NonZero::new(14).unwrap());
pub const K: Self = Self(NonZero::new(13).unwrap());
pub const Q: Self = Self(NonZero::new(12).unwrap());
pub const J: Self = Self(NonZero::new(11).unwrap());
pub const T: Self = Self(NonZero::new(10).unwrap());
#[must_use]
#[inline]
pub const fn new(rank: u8) -> Self {
match Self::try_new(rank) {
Ok(r) => r,
Err(_) => panic!("rank must be in 2..=14"),
}
}
#[inline]
pub const fn try_new(rank: u8) -> Result<Self, InvalidRank> {
match NonZero::new(rank) {
Some(nonzero) if rank >= 2 && rank <= 14 => Ok(Self(nonzero)),
_ => Err(InvalidRank(rank)),
}
}
#[must_use]
#[inline]
pub const fn get(self) -> u8 {
self.0.get()
}
#[must_use]
#[inline]
pub const fn letter(self) -> char {
b"23456789TJQKA"[self.get() as usize - 2] as char
}
}
impl fmt::Display for Rank {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_char(self.letter())
}
}
#[derive(Debug, Error, Clone, Copy, PartialEq, Eq)]
#[error("Invalid rank: expected 2-10, T, J, Q, K, A")]
pub struct ParseRankError;
impl FromStr for Rank {
type Err = ParseRankError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
match s.to_ascii_uppercase().as_str() {
"A" => Ok(Self::A),
"K" => Ok(Self::K),
"Q" => Ok(Self::Q),
"J" => Ok(Self::J),
"T" | "10" => Ok(Self::T),
"9" => Ok(Self::new(9)),
"8" => Ok(Self::new(8)),
"7" => Ok(Self::new(7)),
"6" => Ok(Self::new(6)),
"5" => Ok(Self::new(5)),
"4" => Ok(Self::new(4)),
"3" => Ok(Self::new(3)),
"2" => Ok(Self::new(2)),
_ => Err(ParseRankError),
}
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
#[cfg_attr(
feature = "serde",
derive(serde_with::SerializeDisplay, serde_with::DeserializeFromStr)
)]
pub struct Card {
pub suit: Suit,
pub rank: Rank,
}
impl fmt::Display for Card {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}{}", self.suit, self.rank)
}
}
#[derive(Debug, Error, Clone, Copy, PartialEq, Eq)]
#[non_exhaustive]
pub enum ParseCardError {
#[error("Invalid suit in card: expected one of C, D, H, S, ♣, ♦, ♥, ♠, ♧, ♢, ♡, ♤")]
Suit,
#[error("Invalid rank in card: expected 2-10, T, J, Q, K, A")]
Rank,
}
impl FromStr for Card {
type Err = ParseCardError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
let rank_len = if s.ends_with("10") {
2
} else {
s.chars().next_back().map_or(0, char::len_utf8)
};
let border = s.len().saturating_sub(rank_len);
let (suit, rank) = s.split_at(border);
let suit: Suit = suit.parse().map_err(|_| ParseCardError::Suit)?;
let rank: Rank = rank.parse().map_err(|_| ParseCardError::Rank)?;
Ok(Self { suit, rank })
}
}
#[derive(Debug, Clone, Copy, Default, PartialEq, Eq, Hash)]
#[cfg_attr(
feature = "serde",
derive(serde_with::SerializeDisplay, serde_with::DeserializeFromStr)
)]
#[repr(transparent)]
pub struct Holding(u16);
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct HoldingIter {
rest: u16,
cursor: u8,
}
impl Iterator for HoldingIter {
type Item = Rank;
fn next(&mut self) -> Option<Self::Item> {
if self.rest == 0 {
return None;
}
#[allow(clippy::cast_possible_truncation)]
let pos = 15 - self.rest.leading_zeros() as u8;
self.rest &= !(1u16 << pos);
let rank = self.cursor + pos;
Some(Rank(unsafe { core::num::NonZero::new_unchecked(rank) }))
}
fn size_hint(&self) -> (usize, Option<usize>) {
let count = self.rest.count_ones() as usize;
(count, Some(count))
}
fn count(self) -> usize {
self.rest.count_ones() as usize
}
}
impl DoubleEndedIterator for HoldingIter {
fn next_back(&mut self) -> Option<Self::Item> {
if self.rest == 0 {
return None;
}
#[allow(clippy::cast_possible_truncation)]
let step = self.rest.trailing_zeros() as u8 + 1;
self.rest >>= step;
self.cursor += step;
Some(Rank(unsafe {
core::num::NonZero::new_unchecked(self.cursor - 1)
}))
}
}
impl ExactSizeIterator for HoldingIter {
fn len(&self) -> usize {
self.rest.count_ones() as usize
}
}
impl FusedIterator for HoldingIter {}
impl Holding {
pub const EMPTY: Self = Self(0);
pub const ALL: Self = Self(0x7FFC);
#[must_use]
#[inline]
pub const fn len(self) -> usize {
self.0.count_ones() as usize
}
#[must_use]
#[inline]
pub const fn is_empty(self) -> bool {
self.0 == 0
}
#[must_use]
#[inline]
pub const fn contains(self, rank: Rank) -> bool {
self.0 & 1 << rank.get() != 0
}
#[inline]
pub const fn insert(&mut self, rank: Rank) -> bool {
let insertion = 1 << rank.get() & Self::ALL.0;
let inserted = insertion & !self.0 != 0;
self.0 |= insertion;
inserted
}
#[inline]
pub const fn remove(&mut self, rank: Rank) -> bool {
let removed = self.contains(rank);
self.0 &= !(1 << rank.get());
removed
}
#[inline]
pub const fn toggle(&mut self, rank: Rank) -> bool {
self.0 ^= 1 << rank.get() & Self::ALL.0;
self.contains(rank)
}
#[inline]
pub const fn set(&mut self, rank: Rank, condition: bool) {
let flag = 1 << rank.get();
let mask = (condition as u16).wrapping_neg();
self.0 = (self.0 & !flag) | (mask & flag);
}
#[inline]
#[must_use]
pub const fn iter(self) -> HoldingIter {
HoldingIter {
rest: self.0,
cursor: 0,
}
}
#[must_use]
#[inline]
pub const fn to_bits(self) -> u16 {
self.0
}
#[must_use]
#[inline]
pub const fn from_bits_retain(bits: u16) -> Self {
Self(bits)
}
#[must_use]
#[inline]
pub const fn contains_unknown_bits(self) -> bool {
self.0 & Self::ALL.0 != self.0
}
#[must_use]
#[inline]
pub const fn from_bits(bits: u16) -> Option<Self> {
if bits & Self::ALL.0 == bits {
Some(Self(bits))
} else {
None
}
}
#[must_use]
#[inline]
pub const fn from_bits_truncate(bits: u16) -> Self {
Self(bits & Self::ALL.0)
}
#[must_use]
#[inline]
pub const fn from_rank(rank: Rank) -> Self {
Self(1 << rank.get())
}
}
impl IntoIterator for Holding {
type Item = Rank;
type IntoIter = HoldingIter;
fn into_iter(self) -> HoldingIter {
self.iter()
}
}
impl ops::BitAnd for Holding {
type Output = Self;
#[inline]
fn bitand(self, rhs: Self) -> Self {
Self(self.0 & rhs.0)
}
}
impl ops::BitOr for Holding {
type Output = Self;
#[inline]
fn bitor(self, rhs: Self) -> Self {
Self(self.0 | rhs.0)
}
}
impl ops::BitXor for Holding {
type Output = Self;
#[inline]
fn bitxor(self, rhs: Self) -> Self {
Self(self.0 ^ rhs.0)
}
}
impl ops::Not for Holding {
type Output = Self;
#[inline]
fn not(self) -> Self {
Self::from_bits_truncate(!self.0)
}
}
impl ops::Sub for Holding {
type Output = Self;
#[inline]
fn sub(self, rhs: Self) -> Self {
Self(self.0 & !rhs.0)
}
}
impl ops::BitAndAssign for Holding {
#[inline]
fn bitand_assign(&mut self, rhs: Self) {
*self = *self & rhs;
}
}
impl ops::BitOrAssign for Holding {
#[inline]
fn bitor_assign(&mut self, rhs: Self) {
*self = *self | rhs;
}
}
impl ops::BitXorAssign for Holding {
#[inline]
fn bitxor_assign(&mut self, rhs: Self) {
*self = *self ^ rhs;
}
}
impl ops::SubAssign for Holding {
#[inline]
fn sub_assign(&mut self, rhs: Self) {
*self = *self - rhs;
}
}
impl fmt::Display for Holding {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
for rank in (2u8..15).rev() {
if self.0 & 1 << rank != 0 {
f.write_char(b"23456789TJQKA"[rank as usize - 2] as char)?;
}
}
Ok(())
}
}
#[derive(Debug, Error, Clone, Copy, PartialEq, Eq, Hash)]
#[non_exhaustive]
pub enum ParseHoldingError {
#[error("Ranks are not all valid or in descending order")]
InvalidRanks,
#[error("The same rank appears more than once")]
RepeatedRank,
#[error("A suit contains more than 13 cards")]
TooManyCards,
}
#[derive(Debug, Error, Clone, Copy, PartialEq, Eq, Hash)]
#[non_exhaustive]
pub enum ParseHandError {
#[error(transparent)]
Holding(#[from] ParseHoldingError),
#[error("The hand does not contain 4 suits")]
NotFourSuits,
}
impl FromStr for Holding {
type Err = ParseHoldingError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
if s.len() > 14 {
return Err(ParseHoldingError::TooManyCards);
}
let bytes = s.as_bytes();
let mut i = 0;
let mut prev_rank: u8 = 15;
let mut explicit = Self::EMPTY;
while i < bytes.len() {
let c = bytes[i].to_ascii_uppercase();
let rank: u8 = match c {
b'A' => 14,
b'K' => 13,
b'Q' => 12,
b'J' => 11,
b'T' => 10,
b'1' => {
if bytes.get(i + 1) != Some(&b'0') {
return Err(ParseHoldingError::InvalidRanks);
}
i += 1;
10
}
b'2'..=b'9' => c - b'0',
b'X' => break,
_ => return Err(ParseHoldingError::InvalidRanks),
};
if rank >= prev_rank {
return Err(ParseHoldingError::InvalidRanks);
}
prev_rank = rank;
let r = Rank(unsafe { core::num::NonZero::new_unchecked(rank) });
explicit.insert(r);
i += 1;
}
let spot_count = bytes.len() - i;
if bytes[i..].iter().any(|&b| !b.eq_ignore_ascii_case(&b'x')) {
return Err(ParseHoldingError::InvalidRanks);
}
if spot_count > 13 {
return Err(ParseHoldingError::TooManyCards);
}
let spots = Self::from_bits_truncate((4u16 << spot_count) - 4);
if explicit & spots == Self::EMPTY {
Ok(explicit | spots)
} else {
Err(ParseHoldingError::RepeatedRank)
}
}
}
#[derive(Debug, Clone, Copy, Default, PartialEq, Eq, Hash)]
#[cfg_attr(
feature = "serde",
derive(serde_with::SerializeDisplay, serde_with::DeserializeFromStr)
)]
pub struct Hand([Holding; 4]);
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct HandIter {
suits: [HoldingIter; 4],
fwd: u8,
bwd: u8,
}
impl Iterator for HandIter {
type Item = Card;
fn next(&mut self) -> Option<Self::Item> {
loop {
if self.fwd > 3 {
return None;
}
let suit = Suit::ASC[self.fwd as usize];
if let Some(rank) = self.suits[self.fwd as usize].next() {
return Some(Card { suit, rank });
}
if self.fwd == self.bwd {
self.fwd = 4;
return None;
}
self.fwd -= 1;
}
}
fn size_hint(&self) -> (usize, Option<usize>) {
let count = self.len();
(count, Some(count))
}
fn count(self) -> usize {
self.len()
}
}
impl DoubleEndedIterator for HandIter {
fn next_back(&mut self) -> Option<Self::Item> {
loop {
if self.fwd > 3 {
return None;
}
let suit = Suit::ASC[self.bwd as usize];
if let Some(rank) = self.suits[self.bwd as usize].next_back() {
return Some(Card { suit, rank });
}
if self.fwd == self.bwd {
self.fwd = 4;
return None;
}
self.bwd += 1;
}
}
}
impl ExactSizeIterator for HandIter {
fn len(&self) -> usize {
if self.fwd > 3 {
return 0;
}
(self.bwd as usize..=self.fwd as usize)
.map(|i| self.suits[i].len())
.sum()
}
}
impl FusedIterator for HandIter {}
impl ops::Index<Suit> for Hand {
type Output = Holding;
#[inline]
fn index(&self, suit: Suit) -> &Holding {
&self.0[suit as usize]
}
}
impl ops::IndexMut<Suit> for Hand {
#[inline]
fn index_mut(&mut self, suit: Suit) -> &mut Holding {
&mut self.0[suit as usize]
}
}
impl Hand {
#[must_use]
#[inline]
pub const fn to_bits(self) -> u64 {
unsafe { core::mem::transmute(self.0) }
}
#[must_use]
#[inline]
pub const fn from_bits_retain(bits: u64) -> Self {
unsafe { core::mem::transmute(bits) }
}
#[must_use]
#[inline]
pub const fn contains_unknown_bits(self) -> bool {
self.to_bits() & Self::ALL.to_bits() != self.to_bits()
}
#[must_use]
#[inline]
pub const fn from_bits(bits: u64) -> Option<Self> {
if bits & Self::ALL.to_bits() == bits {
Some(Self::from_bits_retain(bits))
} else {
None
}
}
#[must_use]
#[inline]
pub const fn from_bits_truncate(bits: u64) -> Self {
Self::from_bits_retain(bits & Self::ALL.to_bits())
}
#[must_use]
#[inline]
pub const fn new(clubs: Holding, diamonds: Holding, hearts: Holding, spades: Holding) -> Self {
Self([clubs, diamonds, hearts, spades])
}
pub const EMPTY: Self = Self([Holding::EMPTY; 4]);
pub const ALL: Self = Self([Holding::ALL; 4]);
#[must_use]
#[inline]
pub const fn len(self) -> usize {
self.to_bits().count_ones() as usize
}
#[must_use]
#[inline]
pub const fn is_empty(self) -> bool {
self.to_bits() == 0
}
#[must_use]
#[inline]
pub fn contains(self, card: Card) -> bool {
self[card.suit].contains(card.rank)
}
#[inline]
pub fn insert(&mut self, card: Card) -> bool {
self[card.suit].insert(card.rank)
}
#[inline]
pub fn remove(&mut self, card: Card) -> bool {
self[card.suit].remove(card.rank)
}
#[inline]
pub fn toggle(&mut self, card: Card) -> bool {
self[card.suit].toggle(card.rank)
}
#[inline]
pub fn set(&mut self, card: Card, condition: bool) {
self[card.suit].set(card.rank, condition);
}
#[inline]
#[must_use]
pub const fn iter(self) -> HandIter {
HandIter {
suits: [
self.0[0].iter(),
self.0[1].iter(),
self.0[2].iter(),
self.0[3].iter(),
],
fwd: 3,
bwd: 0,
}
}
}
impl IntoIterator for Hand {
type Item = Card;
type IntoIter = HandIter;
#[inline]
fn into_iter(self) -> HandIter {
self.iter()
}
}
impl FromIterator<Card> for Hand {
fn from_iter<I: IntoIterator<Item = Card>>(iter: I) -> Self {
iter.into_iter().fold(Self::EMPTY, |mut hand, card| {
hand.insert(card);
hand
})
}
}
impl fmt::Display for Hand {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
if self.is_empty() {
return f.write_char('-');
}
self[Suit::Spades].fmt(f)?;
f.write_char('.')?;
self[Suit::Hearts].fmt(f)?;
f.write_char('.')?;
self[Suit::Diamonds].fmt(f)?;
f.write_char('.')?;
self[Suit::Clubs].fmt(f)
}
}
impl FromStr for Hand {
type Err = ParseHandError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
if s.len() > 52 + 4 + 3 {
return Err(ParseHoldingError::TooManyCards.into());
}
if s == "-" {
return Ok(Self::EMPTY);
}
let holdings: Result<Vec<_>, _> = s.split('.').map(Holding::from_str).rev().collect();
Ok(Self(
holdings?
.try_into()
.map_err(|_| ParseHandError::NotFourSuits)?,
))
}
}
impl ops::BitAnd for Hand {
type Output = Self;
#[inline]
fn bitand(self, rhs: Self) -> Self {
Self::from_bits_retain(self.to_bits() & rhs.to_bits())
}
}
impl ops::BitOr for Hand {
type Output = Self;
#[inline]
fn bitor(self, rhs: Self) -> Self {
Self::from_bits_retain(self.to_bits() | rhs.to_bits())
}
}
impl ops::BitXor for Hand {
type Output = Self;
#[inline]
fn bitxor(self, rhs: Self) -> Self {
Self::from_bits_retain(self.to_bits() ^ rhs.to_bits())
}
}
impl ops::Not for Hand {
type Output = Self;
#[inline]
fn not(self) -> Self {
Self::from_bits_truncate(!self.to_bits())
}
}
impl ops::Sub for Hand {
type Output = Self;
#[inline]
fn sub(self, rhs: Self) -> Self {
Self::from_bits_retain(self.to_bits() & !rhs.to_bits())
}
}
impl ops::BitAndAssign for Hand {
#[inline]
fn bitand_assign(&mut self, rhs: Self) {
*self = *self & rhs;
}
}
impl ops::BitOrAssign for Hand {
#[inline]
fn bitor_assign(&mut self, rhs: Self) {
*self = *self | rhs;
}
}
impl ops::BitXorAssign for Hand {
#[inline]
fn bitxor_assign(&mut self, rhs: Self) {
*self = *self ^ rhs;
}
}
impl ops::SubAssign for Hand {
#[inline]
fn sub_assign(&mut self, rhs: Self) {
*self = *self - rhs;
}
}