#include <iostream>
#include <utility>
#include <tuple>
#include <map>

#include <intcode.hpp>
using namespace intcode;

namespace {

using Coord = std::pair<ValueType, ValueType>;

auto Compute(Machine machine, ValueType start)
  -> std::map<Coord, ValueType> {
  
  Coord here {0,0};
  std::map<Coord, ValueType> paint {{here,start}};
  ValueType dx = 1;
  ValueType dy = 0;

  for (;;) {
    auto effect = machine.Step();

    if (std::holds_alternative<Halt>(effect)) {
      return paint;
    }

    std::get<Input>(effect).pos = paint[here];
    auto color = std::get<Output>(machine.Step()).val;
    auto dir = std::get<Output>(machine.Step()).val;
    paint[here] = color;

    std::swap(dx, dy);
    if (dir) {  
      dx = -dx;
    } else {
      dy = -dy;
    }
    here.first += dx;
    here.second += dy;
  }
}

auto Draw(std::map<Coord, ValueType> image) {
  ValueType minX = 1000, maxX = -1000, minY = 1000, maxY = -1000;
}

} // namespace

auto main() -> int {
  auto machine = Machine{ParseStream(std::cin)};

  std::cout << "Part 1: " << Compute(machine, 0).size() << std::endl;
  
  Draw(Compute(std::move(machine), 1));
}