use std::fs;
use std::io::{self, BufRead};

fn probe(cycle: &i32, register: &i32) -> Option<i32> {
    let probes = vec![20, 60, 100, 140, 180, 220];

    if probes.contains(cycle) {
        Some(cycle * register)
    } else {
        None
    }
}

fn draw_pixel(cycle: &i32, register: &i32) {
    let pixel = (cycle - 1) % 40;
    if pixel == 0 {
        println!("");
    }
    let mark = if (register - 1) <= pixel && pixel <= (register + 1) {
        "#"
    } else {
        "."
    };
    print!("{}", mark);
}
fn main() {
    let file = fs::File::open("input").unwrap();
    let lines = io::BufReader::new(file).lines();
    let mut register = 1;
    let mut cycle = 1;
    let mut measures = Vec::<i32>::new();

    for line in lines {
        draw_pixel(&cycle, &register);
        cycle += 1;
        let step1 = probe(&cycle, &register);
        let inst = line.expect("valid instruction");
        let inst = inst.split(" ").collect::<Vec<&str>>();
        let step2 = if inst[0] == "addx" {
            draw_pixel(&cycle, &register);
            cycle += 1;
            register += inst[1].parse::<i32>().unwrap();
            probe(&cycle, &register)
        } else {
            None
        };

        match [step1, step2] {
            [Some(x), _] => measures.push(x),
            [None, Some(y)] => measures.push(y),
            _ => (),
        }
    }
    println!("\n{:?} {}", measures, measures.iter().sum::<i32>());
}