Add locate

fwcd · fwcd · commit e8d3bcfdf2be · 2024-12-23T21:49:56.000+01:00
diff --git a/day21/src/day21.scala b/day21/src/day21.scala
@@ -27,6 +27,10 @@ val PAD_LAYOUTS = Map(
   )),
 )
 
+extension (ptype: PadType) {
+  def locate(c: Char) = PAD_LAYOUTS(ptype).find(_._2 == c).get._1
+}
+
 val DIRECTIONS = Map(
   ('<', Vec2(-1,  0)),
   ('>', Vec2( 1,  0)),
@@ -50,7 +54,7 @@ case class Pad(ptype: PadType, pos: Vec2) {
 }
 
 object Pad {
-  def apply(ptype: PadType): Pad = Pad(ptype, PAD_LAYOUTS(ptype).find(_._2 == 'A').get._1)
+  def apply(ptype: PadType): Pad = Pad(ptype, ptype.locate('A'))
 }
 
 case class State(pads: List[Pad] = List(), output: String = "") {

Original file line number	Diff line number	Diff line change
`@@ -27,6 +27,10 @@ val PAD_LAYOUTS = Map(`
`27`	`27`	`)),`
`28`	`28`	`)`
`29`	`29`
	`30`	`+extension (ptype: PadType) {`
	`31`	`+ def locate(c: Char) = PAD_LAYOUTS(ptype).find(_._2 == c).get._1`
	`32`	`+}`
	`33`	`+`
`30`	`34`	`val DIRECTIONS = Map(`
`31`	`35`	`('<', Vec2(-1, 0)),`
`32`	`36`	`('>', Vec2( 1, 0)),`
`@@ -50,7 +54,7 @@ case class Pad(ptype: PadType, pos: Vec2) {`
`50`	`54`	`}`
`51`	`55`
`52`	`56`	`object Pad {`
`53`		`- def apply(ptype: PadType): Pad = Pad(ptype, PAD_LAYOUTS(ptype).find(_._2 == 'A').get._1)`
	`57`	`+ def apply(ptype: PadType): Pad = Pad(ptype, ptype.locate('A'))`
`54`	`58`	`}`
`55`	`59`
`56`	`60`	`case class State(pads: List[Pad] = List(), output: String = "") {`