Add lines of code plot

LiquidFun · LiquidFun · commit fdd3cd9d7391 · 2021-12-23T20:19:12.000+02:00
diff --git a/Media/2021loc.png b/Media/2021loc.png
diff --git a/README.md b/README.md
@@ -5,3 +5,7 @@ All solutions expect the input via stdin, which can be easily achieved by piping
 Programs are initialized with the `init-day.sh` script. I.e. typing `init-day.sh 10` initializes the 10th day by creating a folder named `10`, downloading the input test case with the `session.cookie`, copying the `dummy.py` file and opening the solution file.
 
 * 2021: There is a solution for each problem in Julia. Sometimes there is a Python or APL solution as well. I'm trying out Julia for the first time, mostly focusing on short and elegant code. 
+
+## Lines of code
+
+![](./Media/2021loc.png)
diff --git a/loc-plot.py b/loc-plot.py
@@ -0,0 +1,59 @@
+#!/usr/bin/python3
+
+import argparse
+from pathlib import Path
+from datetime import datetime
+from collections import Counter
+
+from matplotlib import pyplot as plt
+
+directory = Path(__file__).parent
+
+def get_languages_used(path: Path):
+    """ Returns (count, language) tuples, e.g.: [(1, '.apl'), (3, '.py'), (24, '.jl')] """
+    ignore = [".in", ".run", ".sh", ".ans", "", ".cookie"]
+    occ = Counter(file.suffix for file in path.glob("**/*") if file.suffix not in ignore)
+    return sorted(zip(occ.values(), occ.keys()))
+
+def make_plot_for_year(year: int, lang="auto"):
+    year_dir = directory / str(year)
+    if lang == "auto":
+        lang = get_languages_used(year_dir)[-1][1]
+    if lang[0] != '.':
+        lang = f".{lang}"
+
+    loc = []
+    for day_dir in filter(lambda p: p.is_dir(), year_dir.iterdir()):
+        names = [f"{day_dir.name}", f"{int(day_dir.name)}"]
+        for name in names:
+            if (filepath := day_dir / f"{name}{lang}").exists():
+                with open(filepath, "r") as file:
+                    lines = [*filter(lambda line: line.strip() != "", file.read().strip().split("\n"))]
+                    loc.append(len(lines))
+                    break
+    xticks = [*range(1, len(loc)+1)]
+    bars = plt.bar(xticks, loc)
+    for bar, line_count in zip(bars, loc):
+        plt.text(bar.get_x() + .45, line_count + 1, line_count, ha="center")
+    plt.title(f"Lines of code for each '{lang}' solution in Advent of Code {year}")
+    plt.xticks(xticks)
+    plt.yticks(range(0, 130, 10))
+    plt.ylabel("Lines of code")
+    plt.xlabel("Day")
+    # plt.show()
+    plt.savefig(directory / "Media" / f"{year}loc.png")
+    
+
+def main():
+    date = datetime.now()
+    year = date.year - (date.month <= 11)
+    parser = argparse.ArgumentParser("Make fancy plot (not really)")
+    parser.add_argument("-y", "--year", default=year)
+    parser.add_argument("-l", "--lang", default="auto")
+    parsed = parser.parse_args()
+
+    make_plot_for_year(parsed.year, parsed.lang)
+
+
+if __name__ == "__main__":
+    main()
