-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
2 changed files
with
271 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,269 @@ | ||
use core::fmt::Display; | ||
|
||
#[derive(Clone, Debug, PartialEq, Eq, Hash)] | ||
pub struct IndexedString { | ||
chars: Vec<char>, | ||
offsets: Vec<usize>, | ||
string: String, | ||
} | ||
|
||
impl IndexedString { | ||
pub fn from_str(s: &str) -> Self { | ||
let chars: Vec<char> = s.chars().collect(); | ||
let offsets: Vec<usize> = s.char_indices().map(|(i, _)| i).collect(); | ||
IndexedString { | ||
chars, | ||
offsets, | ||
string: s.to_string(), | ||
} | ||
} | ||
|
||
pub fn from_chars(chars: impl Iterator<Item = char>) -> Self { | ||
let chars: Vec<char> = chars.collect(); | ||
let offsets: Vec<usize> = chars.iter().enumerate().map(|(i, _)| i).collect(); | ||
let string: String = chars.iter().collect(); | ||
IndexedString { | ||
chars, | ||
offsets, | ||
string, | ||
} | ||
} | ||
|
||
pub fn char_at(&self, index: usize) -> Option<char> { | ||
self.chars.get(index).copied() | ||
} | ||
|
||
pub fn chars(&self) -> &Vec<char> { | ||
&self.chars | ||
} | ||
|
||
pub fn as_str(&self) -> &str { | ||
&self.string | ||
} | ||
|
||
pub fn len(&self) -> usize { | ||
self.chars.len() | ||
} | ||
|
||
pub fn byte_len(&self) -> usize { | ||
self.string.len() | ||
} | ||
|
||
pub fn to_string(&self) -> String { | ||
self.string.clone() | ||
} | ||
|
||
pub fn is_empty(&self) -> bool { | ||
self.chars.is_empty() | ||
} | ||
|
||
pub fn contains(&self, st: &str) -> bool { | ||
self.string.contains(st) | ||
} | ||
|
||
pub fn slice(&self, range: core::ops::Range<usize>) -> IndexedSlice { | ||
IndexedSlice { | ||
source: self, | ||
start: range.start, | ||
end: range.end, | ||
} | ||
} | ||
} | ||
|
||
impl core::ops::Index<usize> for IndexedString { | ||
type Output = char; | ||
|
||
fn index(&self, index: usize) -> &Self::Output { | ||
&self.chars[index] | ||
} | ||
} | ||
|
||
impl core::ops::Index<core::ops::Range<usize>> for IndexedString { | ||
type Output = str; | ||
|
||
fn index(&self, range: core::ops::Range<usize>) -> &Self::Output { | ||
if range.start >= self.offsets.len() | ||
|| range.end > self.offsets.len() | ||
|| range.start > range.end | ||
{ | ||
return ""; | ||
} | ||
|
||
let start_byte = self.offsets[range.start]; | ||
let end_byte = if range.end == self.offsets.len() { | ||
self.string.len() | ||
} else { | ||
self.offsets[range.end] | ||
}; | ||
|
||
&self.string[start_byte..end_byte] | ||
} | ||
} | ||
|
||
impl core::ops::Index<core::ops::RangeFrom<usize>> for IndexedString { | ||
type Output = str; | ||
|
||
fn index(&self, range: core::ops::RangeFrom<usize>) -> &Self::Output { | ||
if range.start >= self.offsets.len() { | ||
return ""; | ||
} | ||
|
||
let start_byte = self.offsets[range.start]; | ||
&self.string[start_byte..] | ||
} | ||
} | ||
|
||
impl core::ops::Index<core::ops::RangeTo<usize>> for IndexedString { | ||
type Output = str; | ||
|
||
fn index(&self, range: core::ops::RangeTo<usize>) -> &Self::Output { | ||
if range.end >= self.offsets.len() { | ||
return ""; | ||
} | ||
|
||
let end_byte = self.offsets[range.end]; | ||
&self.string[..end_byte] | ||
} | ||
} | ||
|
||
impl Display for IndexedString { | ||
fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result { | ||
write!(f, "{}", self.string) | ||
} | ||
} | ||
|
||
impl<S: AsRef<str>> PartialEq<S> for IndexedString { | ||
fn eq(&self, other: &S) -> bool { | ||
self.string == other.as_ref() | ||
} | ||
} | ||
|
||
impl<S: AsRef<str>> From<S> for IndexedString { | ||
fn from(s: S) -> Self { | ||
IndexedString::from_str(s.as_ref()) | ||
} | ||
} | ||
|
||
pub struct IndexedSlice<'a> { | ||
source: &'a IndexedString, | ||
start: usize, | ||
end: usize, | ||
} | ||
|
||
impl<'a> IndexedSlice<'a> { | ||
pub fn as_str(&self) -> &str { | ||
if self.start >= self.source.offsets.len() | ||
|| self.end > self.source.offsets.len() | ||
|| self.start > self.end | ||
{ | ||
return ""; | ||
} | ||
|
||
let start_byte = self.source.offsets[self.start]; | ||
let end_byte = if self.end == self.source.offsets.len() { | ||
self.source.string.len() | ||
} else { | ||
self.source.offsets[self.end] | ||
}; | ||
|
||
&self.source.string[start_byte..end_byte] | ||
} | ||
|
||
pub fn len(&self) -> usize { | ||
self.end - self.start | ||
} | ||
|
||
pub fn byte_len(&self) -> usize { | ||
self.source.offsets[self.end] - self.source.offsets[self.start] | ||
} | ||
} | ||
|
||
#[test] | ||
fn test_indexed_string_equality() { | ||
let indexed_string = IndexedString::from_str("h₳ello"); | ||
assert_eq!(indexed_string, "h₳ello"); | ||
assert_eq!(indexed_string.as_str(), "h₳ello"); | ||
assert_eq!(indexed_string.to_string(), "h₳ello"); | ||
} | ||
|
||
#[test] | ||
fn test_from_chars() { | ||
let indexed_string = IndexedString::from_chars("h₳ello".chars()); | ||
assert_eq!(indexed_string, "h₳ello"); | ||
assert_eq!(indexed_string.as_str(), "h₳ello"); | ||
assert_eq!(indexed_string.to_string(), "h₳ello"); | ||
} | ||
|
||
#[test] | ||
fn test_indexing() { | ||
let indexed_string = IndexedString::from_str("h₳ello"); | ||
assert_eq!(indexed_string[0], 'h'); | ||
assert_eq!(&indexed_string[1..4], "₳el"); | ||
assert_eq!(&indexed_string[4..], "lo"); | ||
} | ||
|
||
#[test] | ||
fn test_empty_string() { | ||
let indexed_string: IndexedString = (&String::from("")).into(); | ||
assert_eq!(indexed_string.as_str(), ""); | ||
assert!(indexed_string.char_at(0).is_none()); | ||
} | ||
|
||
#[test] | ||
fn test_single_character() { | ||
let indexed_string: IndexedString = String::from("a").into(); | ||
assert_eq!(indexed_string[0], 'a'); | ||
assert_eq!(indexed_string.as_str(), "a"); | ||
assert_eq!(indexed_string.len(), 1); | ||
} | ||
|
||
#[test] | ||
fn test_multibyte_characters() { | ||
let indexed_string: IndexedString = "😊🌍".into(); | ||
assert_eq!(indexed_string[0], '😊'); | ||
assert_eq!(indexed_string[1], '🌍'); | ||
assert_eq!(&indexed_string[0..1], "😊"); | ||
assert_eq!(indexed_string.len(), 2); | ||
} | ||
|
||
#[test] | ||
fn test_out_of_bounds_indexing() { | ||
let indexed_string = IndexedString::from_str("test"); | ||
assert!(indexed_string.char_at(10).is_none()); | ||
} | ||
|
||
#[test] | ||
fn test_reverse_range() { | ||
let indexed_string = IndexedString::from_str("hello"); | ||
assert_eq!(&indexed_string[3..1], ""); | ||
} | ||
|
||
#[test] | ||
fn test_full_range() { | ||
let indexed_string = IndexedString::from_str("hello"); | ||
assert_eq!(&indexed_string[0..5], "hello"); | ||
} | ||
|
||
#[test] | ||
fn test_adjacent_ranges() { | ||
let indexed_string = IndexedString::from_str("hello world"); | ||
assert_eq!(&indexed_string[0..5], "hello"); | ||
assert_eq!(&indexed_string[5..6], " "); | ||
assert_eq!(&indexed_string[6..11], "world"); | ||
} | ||
|
||
#[test] | ||
fn test_non_ascii_ranges() { | ||
let indexed_string = IndexedString::from_str("🎉🌍🚀"); | ||
assert_eq!(&indexed_string[0..1], "🎉"); | ||
assert_eq!(&indexed_string[1..3], "🌍🚀"); | ||
} | ||
|
||
#[test] | ||
fn test_edge_case_ranges() { | ||
let indexed_string = IndexedString::from_str("abc"); | ||
assert_eq!(&indexed_string[0..0], ""); | ||
assert_eq!(&indexed_string[0..1], "a"); | ||
assert_eq!(&indexed_string[2..3], "c"); | ||
assert_eq!(&indexed_string[3..3], ""); | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters