Refactor dragon curve + add to output

4 files changed, 20 insertions, 14 deletions

diff --git a/2-dragon/main.ua b/2-dragon/main.ua
index 86ece69..be47b59 100644
--- a/2-dragon/main.ua
+++ b/2-dragon/main.ua
@@ -1,18 +1,22 @@
-W ← 200
-H ← 200
-S ← 4
-I ← 2
+W ← 600 # width
+H ← 600 # height
+S ← 4   # size of segment
+I ← 12  # iterations
 
-P ← ℂ:
+StartLine ← [(ℂ: (-(÷2 S) ÷2 W) (÷2 H)) (ℂ: (+(÷2 S) ÷2 W) (÷2 H))]
+Rotate    ← +× ⊙⤚- ℂ: ∿+η ⊙∿.
+Dragon    ← ⍥(⊂ Rotate η ⊣. ⊙(↘1) ⇌.)I StartLine
 
-PStart ← [(P (-(÷2 S) ÷2 W) (÷2 H)) (P (+(÷2 S) ÷2 W) (÷2 H))]
+# Center the image
+MinX           ← /↧ ◌°ℂ Dragon
+MaxX           ← /↥ ◌°ℂ Dragon
+MinY           ← /↧ ◌:°ℂ Dragon
+MaxY           ← /↥ ◌:°ℂ Dragon
+CenteredDragon ← +ℂ: (-MinX ÷2 +MinX -MaxX W) (-MinY ÷2 +MinY -MaxY H) Dragon
 
-RotP       ← +× ⊙⤚- P ∿+η ⊙∿.
-Dragon     ← ⍥(⊂ RotP η ⊣. ⇌.)I PStart
-IsOnLine   ← <0.01 ⌵- /+ ⊙(⌵-⊃⊢⊣)◡(⌵-) ⊟
-IsOnDragon ← ⊙◌ ≥1 /+ ≡(IsOnLine⊃⊢⊣) ⧈∘ 2 Dragon
+IsOnLine   ← ◌: <0.01 ⌵- /+ ⊙(⌵-⊃⊢⊣)◡(⌵-)
+IsOnDragon ← ≥1 /+ ≡(⨬(0◌)IsOnLine ◡(≤S⌵-⊢)) ⧈∘ 2 CenteredDragon
 
-ToPix ← ⨬(0_0_0)(255_255_255) IsOnDragon
-
-Img ← ∵ToPix (⍉ ⊞P ⇡W⇡H)
+ToPix ← ⨬(0_0_0)(⊂: 0.3_0.05 +0.1 ÷W ◌°ℂ) IsOnDragon.
+Img   ← ∵ToPix (⍉ ⊞(ℂ:) ⇡W⇡H)
 &fwa "./2-dragon/output.png" img "png" Img
diff --git a/2-dragon/output-12.png b/2-dragon/output-12.png
new file mode 100644
index 0000000..aa7b485
--- /dev/null
+++ b/2-dragon/output-12.png
diff --git a/2-dragon/output.png b/2-dragon/output.png
index 6c02658..aa7b485 100644
--- a/2-dragon/output.png
+++ b/2-dragon/output.png
diff --git a/README.md b/README.md
index b6e41f1..df9bade 100644
--- a/README.md
+++ b/README.md
@@ -1,4 +1,6 @@
 # Playing with [uiua](https://www.uiua.org/)
 
-![mandelbrot fractal](./1-mandelbrot/output-1080.png)
+| | |
+|---|---|
+| ![mandelbrot fractal](./1-mandelbrot/output-1080.png) | ![dragon curve](./2-dragon/output-12.png) |