commit 5c49642bd117524f799206a9f800d5009d1b1164 · drubery.com/icfp2007

dna2rna/src/main.rs

···

       1
       1
        
       mod dna;

     

       2
       2
        
       mod parser;

     

       3
       3
       +
       mod pattern;

     

       3
       4
        
       

     

       4
       5
        
       use crate::dna::*;

     

       5
       6

+67

dna2rna/src/parser.rs

···

       1
       1
        
       use crate::Base;

     

       2
       2
        
       use crate::DnaRef;

     

       3
       3
       +
       use crate::pattern::Pattern;

     

       4
       4
       +
       use crate::pattern::PatternItem;

     

       3
       5
        
       

     

       4
       6
        
       struct Parser<'a> {

     

       5
       7
        
           buf: &'a DnaRef,

     
···

       65
       67
        
                   }

     

       66
       68
        
               }

     

       67
       69
        
           }

     

       70
       70
       +
       

     

       71
       71
       +
           pub fn pattern(&mut self) -> Option<Pattern> {

     

       72
       72
       +
               let mut ret = Vec::new();

     

       73
       73
       +
               let mut level = 0;

     

       74
       74
       +
               loop {

     

       75
       75
       +
                   if self.next_is(&[Base::C]) {

     

       76
       76
       +
                       self.index += 1;

     

       77
       77
       +
                       ret.push(PatternItem::Base(Base::I));

     

       78
       78
       +
                   } else if self.next_is(&[Base::F]) {

     

       79
       79
       +
                       self.index += 1;

     

       80
       80
       +
                       ret.push(PatternItem::Base(Base::C));

     

       81
       81
       +
                   } else if self.next_is(&[Base::P]) {

     

       82
       82
       +
                       self.index += 1;

     

       83
       83
       +
                       ret.push(PatternItem::Base(Base::F));

     

       84
       84
       +
                   } else if self.next_is(&[Base::I, Base::C]) {

     

       85
       85
       +
                       self.index += 2;

     

       86
       86
       +
                       ret.push(PatternItem::Base(Base::P));

     

       87
       87
       +
                   } else if self.next_is(&[Base::I, Base::P]) {

     

       88
       88
       +
                       self.index += 2;

     

       89
       89
       +
                       let n = self.nat()?;

     

       90
       90
       +
                       ret.push(PatternItem::Skip(n));

     

       91
       91
       +
                   } else if self.next_is(&[Base::I, Base::F]) {

     

       92
       92
       +
                       self.index += 3;

     

       93
       93
       +
                       let s = self.consts();

     

       94
       94
       +
                       ret.push(PatternItem::Search(s));

     

       95
       95
       +
                   } else if self.next_is(&[Base::I, Base::I, Base::P]) {

     

       96
       96
       +
                       self.index += 3;

     

       97
       97
       +
                       level += 1;

     

       98
       98
       +
                       ret.push(PatternItem::Open);

     

       99
       99
       +
                   } else if self.next_is(&[Base::I, Base::I, Base::C])

     

       100
       100
       +
                       || self.next_is(&[Base::I, Base::I, Base::F])

     

       101
       101
       +
                   {

     

       102
       102
       +
                       self.index += 3;

     

       103
       103
       +
                       if level == 0 {

     

       104
       104
       +
                           return Some(ret);

     

       105
       105
       +
                       }

     

       106
       106
       +
                       level -= 1;

     

       107
       107
       +
                       ret.push(PatternItem::Close);

     

       108
       108
       +
                   } else if self.next_is(&[Base::I, Base::I, Base::I]) {

     

       109
       109
       +
                       self.index += 10;

     

       110
       110
       +
                       // TODO: RNA output

     

       111
       111
       +
                   } else {

     

       112
       112
       +
                       return None;

     

       113
       113
       +
                   }

     

       114
       114
       +
               }

     

       115
       115
       +
           }

     

       68
       116
        
       }

     

       69
       117
        
       

     

       70
       118
        
       #[cfg(test)]

     
···

       105
       153
        
               let mut parser = Parser::new(&dna);

     

       106
       154
        
               assert_eq!(parser.consts(), &[Base::I, Base::C, Base::F, Base::P]);

     

       107
       155
        
               assert_eq!(parser.index, 5);

     

       156
       156
       +
           }

     

       157
       157
       +
       

     

       158
       158
       +
           #[test]

     

       159
       159
       +
           fn test_pattern() {

     

       160
       160
       +
               let dna = DnaRef::from_string("CIIC");

     

       161
       161
       +
               let mut parser = Parser::new(&dna);

     

       162
       162
       +
               assert_eq!(parser.pattern(), Some(vec![PatternItem::Base(Base::I)]));

     

       163
       163
       +
       

     

       164
       164
       +
               let dna = DnaRef::from_string("IIPIPICPIICICIIF");

     

       165
       165
       +
               let mut parser = Parser::new(&dna);

     

       166
       166
       +
               assert_eq!(

     

       167
       167
       +
                   parser.pattern(),

     

       168
       168
       +
                   Some(vec![

     

       169
       169
       +
                       PatternItem::Open,

     

       170
       170
       +
                       PatternItem::Skip(2),

     

       171
       171
       +
                       PatternItem::Close,

     

       172
       172
       +
                       PatternItem::Base(Base::P)

     

       173
       173
       +
                   ])

     

       174
       174
       +
               );

     

       108
       175
        
           }

     

       109
       176
        
       }

+10 -7

dna2rna/src/pattern.rs

···

       1
       1
       -
       use crate::Base; 

     

       1
       1
       +
       use crate::Base;

     

       2
       2
        
       

     

       3
       3
       +
       #[derive(Debug, PartialEq)]

     

       3
       4
        
       pub enum PatternItem {

     

       4
       4
       -
         Base(char),

     

       5
       5
       -
         Skip(usize),

     

       6
       6
       -
         Search(Vec<Base>),

     

       7
       7
       -
         Open,

     

       8
       8
       -
         Close

     

       9
       9
       -
       }
     

       5
       5
       +
           Base(Base),

     

       6
       6
       +
           Skip(usize),

     

       7
       7
       +
           Search(Vec<Base>),

     

       8
       8
       +
           Open,

     

       9
       9
       +
           Close,

     

       10
       10
       +
       }

     

       11
       11
       +
       

     

       12
       12
       +
       pub type Pattern = Vec<PatternItem>;