jean_io 0.1.0

I/O library feature for jean
Documentation 
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

use std::collections::HashMap;
use super::{entry::Entry, lexer::FastaLexer, token::Token};
use anyhow::Result;
use jean_core::sequence::Sequence;

#[derive(Debug, Clone)]
pub struct FastaParser;

#[allow(dead_code)]
impl FastaParser {
  pub fn parse_str(str: &str) -> Result<HashMap<String, Entry>> {
    let mut entries = HashMap::new();

    let lexed = FastaLexer::lex(str);

    lexed
      .windows(3)
      .map(|c| (&c[0], &c[1], &c[2]))
      .for_each(|(_0, _1, _2)| match (_0.0, _1.0, _2.0) {
        (Token::Id, Token::Metadata, Token::Sequence) => {
          entries.insert(
            _0.1.clone(),
            Entry {
              metadata: _1.1.clone(),
              sequence: Sequence::try_from(_2.1.as_str()).unwrap(),
            },
          );
        }
        (Token::Id, Token::Sequence, _) => {
          entries.insert(
            _0.1.clone(),
            Entry {
              metadata: "".to_string(),
              sequence: Sequence::try_from(_1.1.as_str()).unwrap(),
            },
          );
        }
        _ => (),
      });

    Ok(entries)
  }
}

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

  #[test]
  fn test_1() {
    let fasta = "; Some comment!\n>seq1\nACAGCTAAG\nCGACT ACGACTG\n\n>seq2\nACGACTAGCTT\n; Some comment!\nAGTAGT";
    let parsed = FastaParser::parse_str(fasta);
    assert!(parsed.is_ok());
  }
}