initial commit

This commit is contained in:
2019-12-28 16:50:04 +01:00
commit 2188ade9c7
6 changed files with 341 additions and 0 deletions

2
.gitignore vendored Normal file
View File

@@ -0,0 +1,2 @@
/target
**/*.rs.bk

15
Cargo.toml Normal file
View File

@@ -0,0 +1,15 @@
[package]
name = "levtree"
version = "0.1.0"
authors = ["Walter Oggioni <oggioni.walter@gmail.com>"]
[dependencies]
[[bin]]
name = "linked_list"
path = "src/linked_list.rs"
[[bin]]
name = "levtree"
#path = "src/test.rs"
path = "src/main.rs"

145
src/levtree.rs Normal file
View File

@@ -0,0 +1,145 @@
use std::collections::HashMap;
struct Node {
key: char,
row: Vec<u32>,
prev: usize,
next: usize,
child: usize,
parent: usize,
processed: bool
}
impl Node {
fn new(key: char) -> Node {
Node {
key,
row: Vec::new(),
prev: 0,
next: 0,
child: 0,
parent: 0,
processed: false,
}
}
}
pub struct Result {
word: std::rc::Rc<String>,
distance: usize,
}
impl PartialOrd for Result {
fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
Some(self.distance.cmp(&other.distance))
}
}
impl PartialEq for Result {
fn eq(&self, other: &Self) -> bool {
self.distance == other.distance
}
}
impl Eq for Result {}
impl Ord for Result {
fn cmp(&self, other: &Self) -> std::cmp::Ordering {
self.distance.cmp(&other.distance)
}
}
//struct Standing {
// size: usize,
// results: Vec<Result>,
//}
//
//impl Standing {
// pub fn new(size: usize) -> Standing {
// Standing {
// size,
// results: BTreeSet::new(),
// }
// }
//
// pub fn addResult(&mut self, res: Result) {
// self.results.push(res)
// }
//}
pub struct Tree {
maxsize: usize,
allocated: bool,
torealloc: bool,
nodes : Vec<Node>,
checker: fn(c1: char, c2: char) -> bool,
distance_calculator: fn() -> usize,
words : HashMap<usize, String>
}
impl Tree {
pub fn new(words: Vec<String>) -> Tree {
let mut result = Tree {
maxsize: words.len(),
allocated: false,
torealloc: false,
nodes : vec!(Node::new('\0')),
checker: |c1: char, c2: char| c1 == c2,
distance_calculator: || -> usize { 0 },
words : HashMap::new()
};
result.build_tree(words);
result
}
fn root(&mut self) -> &mut Node {
&mut self.nodes[0]
}
fn add_child(&mut self, parent_index : usize, key : char) -> usize {
let mut new_child = Node::new(key);
new_child.parent = parent_index;
let mut child = self.nodes[parent_index].child;
let mut prev = child;
while child != 0 {
prev = child;
child = self.nodes[child].next
}
new_child.prev = prev;
self.nodes.push(new_child);
let new_child_index = self.nodes.len();
if prev != 0 {
self.nodes[prev].next = new_child_index;
}
if self.nodes[parent_index].child == 0 {
self.nodes[parent_index].child = new_child_index
}
self.nodes.len() - 1
}
fn build_tree(&mut self, words : Vec<String>) {
for word in words {
let mut nindex = 0;
for c in word.chars() {
if self.nodes[nindex].child == 0 {
nindex = self.add_child(nindex, c);
} else {
nindex = self.nodes[nindex].child;
loop {
if self.nodes[nindex].key == c {
break
} else {
if self.nodes[nindex].next == 0 {
nindex = self.add_child(nindex, c);
break
} else {
nindex = self.nodes[nindex].next;
}
}
}
}
}
let leaf = self.add_child(nindex, '\0');
self.words.insert(leaf, word);
}
}
}

153
src/linked_list.rs Normal file
View File

@@ -0,0 +1,153 @@
use std::mem;
use std::ptr;
pub struct List<T> {
list_head: Option<Box<Node<T>>>,
list_tail: Rawlink<Node<T>>,
}
struct Rawlink<T> { p: *mut T }
impl<T> Copy for Rawlink<T> {}
impl<T> Clone for Rawlink<T> {
fn clone(&self) -> Self { Rawlink { p: self.p } }
}
pub struct Node<T> {
next: Option<Box<Node<T>>>,
prev: Rawlink<Node<T>>,
value: T,
}
impl<T> List<T> {
pub fn is_empty(&self) -> bool {
self.list_head.is_none()
}
pub fn len(&self) -> usize {
let mut node = &self.list_head;
let mut i = 0;
loop {
match *node {
Some(ref n) => {
i+=1;
node=&n.next;
}
None => {
return i;
}
}
}
}
/// Create an empty DList
pub fn new() -> List<T> {
List{list_head: None, list_tail: Rawlink::none()}
}
pub fn push_front(&mut self, elt: T) {
self.push_front_node(Box::new(Node::new(elt)))
}
pub fn push_front_node(&mut self, mut new_head: Box<Node<T>>) {
match self.list_head {
None => {
self.list_tail = Rawlink::some(&mut new_head);
new_head.prev = Rawlink::none();
self.list_head = Some(new_head);
}
Some(ref mut head) => {
new_head.prev = Rawlink::none();
head.prev = Rawlink::some(&mut new_head);
mem::swap(head, &mut new_head);
head.next = Some(new_head);
}
}
}
/// Provide a forward iterator
#[inline]
pub fn iter<'a>(&'a self) -> ListIterator<'a, T> {
ListIterator{nelem: self.len(), head: &self.list_head, tail: self.list_tail}
}
}
impl<T> Node<T> {
fn new(v: T) -> Node<T> {
Node{value: v, next: None, prev: Rawlink::none()}
}
}
/// Rawlink is a type like Option<T> but for holding a raw pointer
impl<T> Rawlink<T> {
/// Like Option::None for Rawlink
fn none() -> Rawlink<T> {
Rawlink{p: ptr::null_mut()}
}
/// Like Option::Some for Rawlink
fn some(n: &mut T) -> Rawlink<T> {
Rawlink{p: n as *mut T}
}
/// Convert the `Rawlink` into an Option value
fn resolve_immut<'a>(&self) -> Option<&'a T> {
unsafe { self.p.as_ref() }
}
/// Convert the `Rawlink` into an Option value
fn resolve<'a>(&mut self) -> Option<&'a mut T> {
unsafe { self.p.as_mut() }
}
/// Return the `Rawlink` and replace with `Rawlink::none()`
fn take(&mut self) -> Rawlink<T> {
mem::replace(self, Rawlink::none())
}
}
pub struct ListIterator<'a, T: 'a> {
head: &'a Option<Box<Node<T>>>,
tail: Rawlink<Node<T>>,
nelem: usize,
}
impl<'a, A> Iterator for ListIterator<'a, A> {
type Item = &'a A;
#[inline]
fn next(&mut self) -> Option<&'a A> {
if self.nelem == 0 {
return None;
}
self.head.as_ref().map(|head| {
self.nelem -= 1;
self.head = &head.next;
&head.value
})
}
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
(self.nelem, Some(self.nelem))
}
}
impl<'a, A> DoubleEndedIterator for ListIterator<'a, A> {
#[inline]
fn next_back(&mut self) -> Option<&'a A> {
if self.nelem == 0 {
return None;
}
let tmp = self.tail.resolve_immut();
tmp.as_ref().map(|prev| {
self.nelem -= 1;
self.tail = prev.prev;
&prev.value
})
}
}
fn main() {
}

20
src/main.rs Normal file
View File

@@ -0,0 +1,20 @@
mod levtree;
use std::fs::File;
//use std::io::prelude::*;
use std::io::BufReader;
use std::io::BufRead;
fn main() {
//let mut words : Vec<String> = Vec::new();
//let infile : File = try!(File::open("/usr/share/dict/cracklib-small"));
let filename = "/usr/share/dict/cracklib-small";
let infile : File = File::open(filename).expect(
&format!("Error opening {}:", &filename)
);
let reader = BufReader::new(infile);
let tree = levtree::Tree::new(
reader.lines()
.map(|line| line.unwrap())
.collect());
}

6
src/test.rs Normal file
View File

@@ -0,0 +1,6 @@
fn main(){
let mut ints = vec!(1,2,3,4,5,6);
ints[4] = 2;
}