day14part1 working

input14

@@ -1,6 +1,56 @@
-10 ORE => 10 A
+18 FHWM => 1 XGQNZ
-1 ORE => 1 B
+4 FVPWN, 9 CQGW => 7 QGHT
-7 A, 1 B => 1 C
+22 KZMS, 1 DMCJL => 8 TWGCK
-7 A, 1 C => 1 D
+1 LMGQN, 1 DSWDM, 1 GKGZ => 1 TGPH
-7 A, 1 D => 1 E
+22 WCSW => 1 LVTG
-7 A, 1 E => 1 FUEL
+13 JSWR => 4 GKGZ
+162 ORE => 3 FVPWN
+59 CQGW, 15 MSNG, 6 XGKRF, 10 LJRQ, 1 HRKGV, 15 RKVC => 1 FUEL
+5 DMCJL => 1 QBLH
+2 XDRJ, 2 RKVC => 8 CTCNL
+1 QXHX => 5 GFPSK
+22 QGHT, 6 GFPSK, 5 DHTPL => 3 CSDR
+4 QGHT, 2 HFXD => 4 XDRJ
+10 WQCGV, 1 JSWR, 21 RHTLN => 7 VTPC
+11 CQGW, 1 FVPWN => 3 HFXD
+5 VTPC => 2 NCXW
+8 LDZVS => 6 DQLH
+117 ORE => 2 KWZNB
+3 TGPH, 1 JPFQ, 2 WHWLK, 5 RKVC, 16 DQLH => 9 LJRQ
+14 KWZNB, 2 CQGW => 8 MLPK
+6 LDZVS => 2 JSWR
+1 RKVC, 8 HCGT, 9 DHTPL => 6 FHWM
+3 DHTPL, 1 HWSR, 36 LDZVS => 6 DSWDM
+5 WHWLK, 1 LJHWT, 8 HSTHS => 7 VMPX
+22 ZJCDZ, 3 WQCGV => 5 DHTPL
+10 LJHWT, 32 GFPSK, 2 RHTLN => 4 HFRMP
+2 FKVD, 3 TWGCK, 1 HWSR => 1 RNLZW
+2 CSDR, 3 DQLH, 2 HSTHS => 9 JPFQ
+1 JSWR, 1 PCWS, 1 HFRMP => 3 XGKRF
+2 QGHT, 9 LVTG, 3 QBLH => 7 RHTLN
+10 LJHWT, 4 CTCNL => 8 QXHX
+16 MLPK, 1 HFXD => 9 ZJCDZ
+6 QGHT => 9 WCSW
+4 HWSR, 4 MLPK, 1 KZMS => 3 BGZHQ
+12 MLPK => 8 RKVC
+1 HWSR, 1 VNWFS => 7 BGFJ
+7 FHWM, 11 CTDF, 1 LDZVS => 2 VNWFS
+4 CTDF => 4 HSTHS
+2 ZJCDZ => 6 LJHWT
+1 VMPX, 1 NCXW, 1 HSTHS, 41 XGKRF, 30 HLNG, 1 GKGZ => 7 HRKGV
+1 XGQNZ, 10 PCWS, 3 BGFJ => 8 FKVD
+1 GFPSK, 1 DMCJL, 1 LVTG => 5 XDTZB
+3 WCSW => 5 KZMS
+6 TWGCK, 1 QXHX, 4 BGFJ => 2 LMGQN
+1 WCSW => 7 LDZVS
+1 XDTZB, 9 VNWFS => 3 WHWLK
+3 HFXD, 4 WCSW, 1 MLPK => 5 WQCGV
+2 BGFJ, 1 HSTHS, 22 MDCB, 10 HWSR, 6 RNLZW, 8 GKGZ => 5 MSNG
+4 QGHT, 1 FKVD => 7 MDCB
+9 MLPK, 3 LJHWT => 7 DMCJL
+121 ORE => 2 CQGW
+9 DHTPL, 2 BGZHQ => 8 CTDF
+2 JSWR, 30 RHTLN => 7 HLNG
+2 QBLH => 7 PCWS
+14 LVTG => 8 HWSR
+7 DMCJL => 1 HCGT

src/day14/main.rs

@@ -1,3 +1,4 @@
+use std::borrow::{Borrow, BorrowMut};
 use std::collections::HashMap;
 use std::convert::*;
 use std::fs::File;
@@ -38,23 +39,43 @@ enum NanoFactoryBuildState {
 
 impl NanoFactory {
     fn generate_fuel(&mut self) -> usize {
-        return self.lookup_resource("FUEL");
+        let storage = self.storage.borrow_mut();
+        let pipeline = self.pipeline.borrow();
+        return NanoFactory::lookup_resource(storage, pipeline, "FUEL");
     }
 
-    fn lookup_resource(&mut self, resource: &str) -> usize {
+    fn lookup_resource(
+        storage: &mut HashMap<Resource, usize>,
+        pipeline: &Pipeline,
+        resource: &str,
+    ) -> usize {
         if resource == "ORE" {
             return 1;
         }
-        if let Some(count) = self.storage.get_mut(resource.into()) {
+
-            *count -= 1
+        // use from storage
+        if let Some(count) = storage.get_mut(resource.into()) {
+            if *count > 0 {
+                *count -= 1;
+                return 0;
+            }
         }
-        let pipeline = self.pipeline.as_mut();
+
-        let required = self
+        // cloned to still allow writable acces to self
-            .pipeline
+        let required = pipeline
             .get(resource.into())
             .expect("unable to find resource");
+
+        // initialize storage with zero
+        if let None = storage.get(resource.into()) {
+            storage.insert(resource.into(), 0);
+        }
+
+        // build resource and and to storage
+        let stored = storage.get_mut(resource.into()).unwrap();
+        *stored += required.out_count - 1;
         required.requirements.iter().fold(0, |mut acc, c| {
-            acc += self.lookup_resource(c);
+            acc += NanoFactory::lookup_resource(storage, pipeline, c);
             acc
         })
     }
@@ -74,6 +95,7 @@ where
             let mut cur_requirements = Vec::new();
             loop {
                 match state {
+                    // numeric input
                     NanoFactoryBuildState::InputCount | NanoFactoryBuildState::OutputCount => {
                         while let Some(c) = next_if_in_range(&mut line, 48..=57) {
                             count_var = count_var * 10 + (c as usize) - 48;
@@ -89,6 +111,7 @@ where
                             _ => unreachable!(),
                         }
                     }
+                    // input resouce name
                     NanoFactoryBuildState::InputResource => {
                         let mut r = Resource::new();
                         while let Some(c) = next_if_in_range(&mut line, 65..=90) {
@@ -112,6 +135,7 @@ where
                             ),
                         }
                     }
+                    // output resource name
                     NanoFactoryBuildState::OutputResource => {
                         let mut r = Resource::new();
                         while let Some(c) = next_if_in_range(&mut line, 65..=90) {
@@ -122,6 +146,7 @@ where
                         state = NanoFactoryBuildState::InputCount;
                         return;
                     }
+                    // arrow between input and output ' => '
                     NanoFactoryBuildState::OutputArrow => {
                         assert_eq!(line.next().unwrap(), b'=');
                         assert_eq!(line.next().unwrap(), b'>');
@@ -131,7 +156,6 @@ where
                 };
             }
         });
-        println! {"{:?}", f};
         f
     }
 }
@@ -160,5 +184,125 @@ fn main() {
     let bufreader = BufReader::new(f);
     let mut factory = NanoFactory::from(bufreader);
     let ores = factory.generate_fuel();
-    println!("ores: {}", ores)
+    println!("ores: {}", ores);
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example1() {
+        let bufreader = BufReader::new(
+            "10 ORE => 10 A
+1 ORE => 1 B
+7 A, 1 B => 1 C
+7 A, 1 C => 1 D
+7 A, 1 D => 1 E
+7 A, 1 E => 1 FUEL
+"
+            .trim()
+            .as_bytes(),
+        );
+        let mut factory = NanoFactory::from(bufreader);
+        let ores = factory.generate_fuel();
+        assert_eq!(ores, 31)
+    }
+
+    #[test]
+    fn example2() {
+        let bufreader = BufReader::new(
+            "
+9 ORE => 2 A
+8 ORE => 3 B
+7 ORE => 5 C
+3 A, 4 B => 1 AB
+5 B, 7 C => 1 BC
+4 C, 1 A => 1 CA
+2 AB, 3 BC, 4 CA => 1 FUEL
+"
+            .trim()
+            .as_bytes(),
+        );
+        let mut factory = NanoFactory::from(bufreader);
+        let ores = factory.generate_fuel();
+        assert_eq!(ores, 165)
+    }
+
+    #[test]
+    fn example3() {
+        let bufreader = BufReader::new(
+            "
+157 ORE => 5 NZVS
+165 ORE => 6 DCFZ
+44 XJWVT, 5 KHKGT, 1 QDVJ, 29 NZVS, 9 GPVTF, 48 HKGWZ => 1 FUEL
+12 HKGWZ, 1 GPVTF, 8 PSHF => 9 QDVJ
+179 ORE => 7 PSHF
+177 ORE => 5 HKGWZ
+7 DCFZ, 7 PSHF => 2 XJWVT
+165 ORE => 2 GPVTF
+3 DCFZ, 7 NZVS, 5 HKGWZ, 10 PSHF => 8 KHKGT
+"
+            .trim()
+            .as_bytes(),
+        );
+        let mut factory = NanoFactory::from(bufreader);
+        let ores = factory.generate_fuel();
+        assert_eq!(ores, 13312)
+    }
+
+    #[test]
+    fn example4() {
+        let bufreader = BufReader::new(
+            "
+2 VPVL, 7 FWMGM, 2 CXFTF, 11 MNCFX => 1 STKFG
+17 NVRVD, 3 JNWZP => 8 VPVL
+53 STKFG, 6 MNCFX, 46 VJHF, 81 HVMC, 68 CXFTF, 25 GNMV => 1 FUEL
+22 VJHF, 37 MNCFX => 5 FWMGM
+139 ORE => 4 NVRVD
+144 ORE => 7 JNWZP
+5 MNCFX, 7 RFSQX, 2 FWMGM, 2 VPVL, 19 CXFTF => 3 HVMC
+5 VJHF, 7 MNCFX, 9 VPVL, 37 CXFTF => 6 GNMV
+145 ORE => 6 MNCFX
+1 NVRVD => 8 CXFTF
+1 VJHF, 6 MNCFX => 4 RFSQX
+176 ORE => 6 VJHF
+"
+            .trim()
+            .as_bytes(),
+        );
+        let mut factory = NanoFactory::from(bufreader);
+        let ores = factory.generate_fuel();
+        assert_eq!(ores, 180697)
+    }
+
+    #[test]
+    fn example5() {
+        let bufreader = BufReader::new(
+            "
+171 ORE => 8 CNZTR
+7 ZLQW, 3 BMBT, 9 XCVML, 26 XMNCP, 1 WPTQ, 2 MZWV, 1 RJRHP => 4 PLWSL
+114 ORE => 4 BHXH
+14 VRPVC => 6 BMBT
+6 BHXH, 18 KTJDG, 12 WPTQ, 7 PLWSL, 31 FHTLT, 37 ZDVW => 1 FUEL
+6 WPTQ, 2 BMBT, 8 ZLQW, 18 KTJDG, 1 XMNCP, 6 MZWV, 1 RJRHP => 6 FHTLT
+15 XDBXC, 2 LTCX, 1 VRPVC => 6 ZLQW
+13 WPTQ, 10 LTCX, 3 RJRHP, 14 XMNCP, 2 MZWV, 1 ZLQW => 1 ZDVW
+5 BMBT => 4 WPTQ
+189 ORE => 9 KTJDG
+1 MZWV, 17 XDBXC, 3 XCVML => 2 XMNCP
+12 VRPVC, 27 CNZTR => 2 XDBXC
+15 KTJDG, 12 BHXH => 5 XCVML
+3 BHXH, 2 VRPVC => 7 MZWV
+121 ORE => 7 VRPVC
+7 XCVML => 6 RJRHP
+5 BHXH, 4 VRPVC => 5 LTCX
+"
+            .trim()
+            .as_bytes(),
+        );
+        let mut factory = NanoFactory::from(bufreader);
+        let ores = factory.generate_fuel();
+        assert_eq!(ores, 2210736)
+    }
 }