1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105

//! Pos files are ASCII position files.

use std::fmt::Debug;
use std::fs::File;
use std::io::{BufRead, BufReader};
use std::path::Path;

use Result;
use point::Point;
use source::Source;
use units::Radians;

/// A pos reader.
#[derive(Debug)]
pub struct Reader<R: BufRead> {
    reader: R,
}

impl Reader<BufReader<File>> {
    /// Creates a new reader from a path.
    ///
    /// # Examples
    ///
    /// ```
    /// use pos::pos::Reader;
    /// let reader = Reader::from_path("data/0916_2014_ie.pos").unwrap();
    /// ```
    pub fn from_path<P: AsRef<Path>>(path: P) -> Result<Reader<BufReader<File>>> {
        let mut reader = BufReader::new(try!(File::open(path)));
        let ref mut header: String = String::new();
        let _ = try!(reader.read_line(header));
        Ok(Reader { reader: reader })
    }
}

impl<R: BufRead> Reader<R> {
    /// Reads a point from the file.
    ///
    /// # Examples
    ///
    /// ```
    /// use pos::pos::Reader;
    /// let mut reader = Reader::from_path("data/0916_2014_ie.pos").unwrap();
    /// let point = reader.read_point().unwrap();
    /// ```
    pub fn read_point(&mut self) -> Result<Option<Point>> {
        let mut line = String::new();
        let _ = try!(self.reader.read_line(&mut line));
        let values: Vec<_> = line.split_whitespace().map(|s| s.clone()).collect();
        if values.is_empty() {
            return Ok(None);
        }
        Ok(Some(Point {
            time: try!(values[0].parse()),
            latitude: Radians::from_degrees(try!(values[1].parse())),
            longitude: Radians::from_degrees(try!(values[2].parse())),
            altitude: try!(values[3].parse()),
            roll: Radians::from_degrees(try!(values[4].parse())),
            pitch: Radians::from_degrees(try!(values[5].parse())),
            yaw: Radians::from_degrees(try!(values[6].parse())),
            ..Default::default()
        }))
    }
}

impl<R: BufRead> IntoIterator for Reader<R> {
    type Item = Point;
    type IntoIter = ReaderIterator<R>;
    fn into_iter(self) -> Self::IntoIter {
        ReaderIterator { reader: self }
    }
}

/// An iterator over a pos reader.
#[derive(Debug)]
pub struct ReaderIterator<R: BufRead> {
    reader: Reader<R>,
}

impl<R: BufRead> Iterator for ReaderIterator<R> {
    type Item = Point;
    fn next(&mut self) -> Option<Self::Item> {
        self.reader.read_point().unwrap()
    }
}

impl<R: Debug + BufRead> Source for Reader<R> {
    fn source(&mut self) -> Result<Option<Point>> {
        self.read_point()
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn point_count() {
        let points: Vec<_> = Reader::from_path("data/0916_2014_ie.pos")
                                 .unwrap()
                                 .into_iter()
                                 .collect();
        assert_eq!(722800, points.len());
    }
}