From 16519df445b8d85abfb265e3b5efabefc6edd6a5 Mon Sep 17 00:00:00 2001
From: cupcakearmy <hi@nicco.io>
Date: Thu, 2 Dec 2021 11:56:55 +0100
Subject: [PATCH] day 2

---
 2021/02/README.md      | 11 +++++++
 2021/02/python/main.py | 72 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 83 insertions(+)
 create mode 100644 2021/02/README.md
 create mode 100644 2021/02/python/main.py

diff --git a/2021/02/README.md b/2021/02/README.md
new file mode 100644
index 0000000..64ef9e3
--- /dev/null
+++ b/2021/02/README.md
@@ -0,0 +1,11 @@
+# 02
+
+Description
+
+<details>
+  <summary>Solutions</summary>
+  <ol>
+    <li>1670340</li>
+    <li>1954293920</li>
+  </ol>
+</details>
diff --git a/2021/02/python/main.py b/2021/02/python/main.py
new file mode 100644
index 0000000..642ace6
--- /dev/null
+++ b/2021/02/python/main.py
@@ -0,0 +1,72 @@
+#!/usr/bin/env python
+
+from os.path import join, dirname
+
+# Day 01
+
+# Common
+
+
+def read_input(filename):
+    data = join(dirname(__file__), '..', filename)
+    with open(data) as f:
+        return f.read()
+
+
+# 1
+
+class submarine:
+    def __init__(self, with_aim=False):
+        self.distance = 0
+        self.depth = 0
+        self.aim = 0
+        self.with_aim = with_aim
+
+    def execute(self, instruction: str):
+        op, x = instruction.split(' ')
+        x = int(x)
+        if op == 'forward':
+            if self.with_aim:
+                self.distance += x
+                self.depth += x * self.aim
+            else:
+                self.distance += x
+        elif op == 'up':
+            if self.with_aim:
+                self.aim -= x
+            else:
+                self.depth -= x
+        elif op == 'down':
+            if self.with_aim:
+                self.aim += x
+            else:
+                self.depth += x
+
+    def execute_instructions(self, instructions: str):
+        for instruction in instructions.strip().split('\n'):
+            self.execute(instruction)
+
+    def get_current_position(self):
+        return self.distance * self.depth
+
+
+test = read_input('test.txt')
+sub = submarine()
+sub.execute_instructions(test)
+print(f"Test: {sub.get_current_position()}")
+
+data = read_input('input.txt')
+sub = submarine()
+sub.execute_instructions(data)
+print(f"Part 1: {sub.get_current_position()}")
+
+# 2
+
+
+sub = submarine(with_aim=True)
+sub.execute_instructions(test)
+print(f"Test: {sub.get_current_position()}")
+
+sub = submarine(with_aim=True)
+sub.execute_instructions(data)
+print(f"Part 2: {sub.get_current_position()}")