gistfile1.txt

$ rustc hong.rs
hong.rs:34:20: 34:27 error: `cursors` does not live long enough
hong.rs:34         for cur in cursors.iter() {
                              ^~~~~~~
hong.rs:26:68: 47:2 note: reference must be valid for the lifetime 'a as defined on the block at 26:67...
hong.rs:26     (dict: &'a SuffixTree<E>, mut iter: Iter) -> Vec<Match<'a, E>> {
hong.rs:27 
hong.rs:28     let mut offset: uint = 0;
hong.rs:29     let mut cursors: Vec<Cursor<E>> = Vec::new();
hong.rs:30     let mut matches = Vec::new();
hong.rs:31     for ch in iter {
           ...
hong.rs:26:68: 47:2 note: ...but borrowed value is only valid for the block at 26:67
hong.rs:26     (dict: &'a SuffixTree<E>, mut iter: Iter) -> Vec<Match<'a, E>> {
hong.rs:27 
hong.rs:28     let mut offset: uint = 0;
hong.rs:29     let mut cursors: Vec<Cursor<E>> = Vec::new();
hong.rs:30     let mut matches = Vec::new();
hong.rs:31     for ch in iter {

hello.rs

extern crate collections;

use std::io::{BufferedReader, File};
use suffix_tree::{SuffixTree, Cursor};

pub fn main() {
    let mut dict: SuffixTree<char> = SuffixTree::new();

    // read in dictionary
    let args = std::os::args();
    let dict_path = Path::new(args[0].clone());
    let mut dict_reader = BufferedReader::new(File::open(&dict_path));
    for i in dict_reader.lines() {
        let t: Vec<char> = i.unwrap().as_slice().trim().chars().collect();
        dict.insert(t);
    }
}

struct Match<'a, E: 'a> {
    start: uint,
    end: uint,
    seq: &'a SuffixTree<E>
}

fn find_matches<'a, E: Copy+Ord, Iter: Iterator<E>>
    (dict: &'a SuffixTree<E>, mut iter: Iter) -> Vec<Match<'a, E>> {

    let mut offset: uint = 0;
    let mut cursors: Vec<Cursor<E>> = Vec::new();
    let mut matches = Vec::new();
    for ch in iter {
        cursors.push(Cursor::new(dict));
        cursors = cursors.into_iter().filter_map(|cur| cur.go(ch)).collect();
        for cur in cursors.iter() {
            if cur.is_terminal() {
                // we have a hit
                matches.push(Match{
                    start: offset - cur.path.len(),
                    end: offset,
                    seq: &**cur,
                });
            }
        }
        offset += 1; 
    }
    matches
}

pub mod suffix_tree {
    use collections::treemap::TreeMap;

    pub struct SuffixTree<E> {
        suffixes: TreeMap<E, SuffixTree<E>>,
        terminal: bool,
    }

    impl<E: Ord + Copy> SuffixTree<E> {
        pub fn new() -> SuffixTree<E> {
            SuffixTree {
                suffixes: TreeMap::new(),
                terminal: false,
            }
        }

        pub fn is_terminal(&self) -> bool {
            self.terminal
        }

        pub fn insert(&mut self, el: Vec<E>) {
            unsafe {
                let mut tree: *mut SuffixTree<E> = self;
                for i in el.into_iter() {
                    let new = match (*tree).suffixes.find_mut(&i) {
                        Some(next) => next as *mut SuffixTree<E>,
                        None => {
                            (*tree).suffixes.insert(i, SuffixTree::new());
                            (*tree).suffixes.find_mut(&i).unwrap() as *mut SuffixTree<E>
                        }
                    };
                    tree = new;
                }
                (*tree).terminal = true;
            }
        }
    }

    pub struct Cursor<'a, E: 'a> {
        cursor: &'a SuffixTree<E>,
        pub path: Vec<E>,
    }

    impl<'a, E: Ord> Cursor<'a, E> {
        pub fn new(array: &'a SuffixTree<E>) -> Cursor<'a, E> {
            Cursor {
                cursor: array,
                path: Vec::new(),
            }
        }

        pub fn go(mut self, el: E) -> Option<Cursor<'a, E>> {
            match self.cursor.suffixes.find(&el) {
                Some(next) => {
                    self.cursor = next;
                    self.path.push(el);
                    Some(self)
                }
                None => None
            }
        }
    }

    impl<'a, E> Deref<SuffixTree<E>> for Cursor<'a, E> {
        fn deref(&self) -> &SuffixTree<E> {
            self.cursor
        }
    }
}