use ahash::AHashMap;
use std::collections::hash_map::{Entry, Keys};
use std::default::Default;
use std::fmt;
use std::hash::Hash;
use std::iter::{FromIterator, IntoIterator};
use crate::Commute;
#[derive(Clone)]
pub struct Frequencies<T> {
data: AHashMap<T, u64>,
}
impl<T: fmt::Debug + Eq + Hash> fmt::Debug for Frequencies<T> {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "{:?}", self.data)
}
}
impl<T: Eq + Hash> Frequencies<T> {
#[must_use]
pub fn new() -> Frequencies<T> {
Default::default()
}
#[inline]
pub fn add(&mut self, v: T) {
match self.data.entry(v) {
Entry::Vacant(count) => {
count.insert(1);
}
Entry::Occupied(mut count) => {
*count.get_mut() += 1;
}
}
}
#[inline]
#[must_use]
pub fn count(&self, v: &T) -> u64 {
self.data.get(v).copied().unwrap_or(0)
}
#[inline]
#[must_use]
pub fn cardinality(&self) -> u64 {
self.len() as u64
}
#[inline]
#[must_use]
pub fn mode(&self) -> Option<&T> {
let counts = self.most_frequent();
if counts.is_empty() {
return None;
}
if counts.len() >= 2 && counts[0].1 == counts[1].1 {
None
} else {
Some(counts[0].0)
}
}
#[inline]
#[must_use]
pub fn most_frequent(&self) -> Vec<(&T, u64)> {
let mut counts: Vec<_> = self.data.iter().map(|(k, &v)| (k, v)).collect();
counts.sort_unstable_by(|&(_, c1), &(_, c2)| c2.cmp(&c1));
counts
}
#[inline]
#[must_use]
pub fn least_frequent(&self) -> Vec<(&T, u64)> {
let mut counts: Vec<_> = self.data.iter().map(|(k, &v)| (k, v)).collect();
counts.sort_unstable_by(|&(_, c1), &(_, c2)| c1.cmp(&c2));
counts
}
#[must_use]
pub fn len(&self) -> usize {
self.data.len()
}
#[must_use]
pub fn is_empty(&self) -> bool {
self.data.is_empty()
}
#[must_use]
pub fn unique_values(&self) -> UniqueValues<T> {
UniqueValues {
data_keys: self.data.keys(),
}
}
}
impl<T: Eq + Hash> Commute for Frequencies<T> {
#[inline]
fn merge(&mut self, v: Frequencies<T>) {
for (k, v2) in v.data {
match self.data.entry(k) {
Entry::Vacant(v1) => {
v1.insert(v2);
}
Entry::Occupied(mut v1) => {
*v1.get_mut() += v2;
}
}
}
}
}
impl<T: Eq + Hash> Default for Frequencies<T> {
#[inline]
fn default() -> Frequencies<T> {
Frequencies {
data: AHashMap::with_capacity(100_000),
}
}
}
impl<T: Eq + Hash> FromIterator<T> for Frequencies<T> {
#[inline]
fn from_iter<I: IntoIterator<Item = T>>(it: I) -> Frequencies<T> {
let mut v = Frequencies::new();
v.extend(it);
v
}
}
impl<T: Eq + Hash> Extend<T> for Frequencies<T> {
#[inline]
fn extend<I: IntoIterator<Item = T>>(&mut self, it: I) {
for sample in it {
self.add(sample);
}
}
}
pub struct UniqueValues<'a, K> {
data_keys: Keys<'a, K, u64>,
}
impl<'a, K> Iterator for UniqueValues<'a, K> {
type Item = &'a K;
fn next(&mut self) -> Option<Self::Item> {
self.data_keys.next()
}
}
#[cfg(test)]
mod test {
use super::Frequencies;
use std::iter::FromIterator;
#[test]
fn ranked() {
let mut counts = Frequencies::new();
counts.extend(vec![1usize, 1, 2, 2, 2, 2, 2, 3, 4, 4, 4].into_iter());
assert_eq!(counts.most_frequent()[0], (&2, 5));
assert_eq!(counts.least_frequent()[0], (&3, 1));
}
#[test]
fn unique_values() {
let freqs = Frequencies::from_iter(vec![8, 6, 5, 1, 1, 2, 2, 2, 3, 4, 7, 4, 4]);
let mut unique: Vec<isize> = freqs.unique_values().copied().collect();
unique.sort_unstable();
assert_eq!(unique, vec![1, 2, 3, 4, 5, 6, 7, 8]);
}
}