simplify print logic.

tristanmorgan · Aug 22, 2024 · 126fd0b · 126fd0b
1 parent ad80fdc
commit 126fd0b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/parser/print.go b/parser/print.go
@@ -66,15 +66,13 @@ func Print(program []Instruction, writer *bufio.Writer) {
 	endLoop := NewInstruction(']')
 	lastInst := NewInstruction('!')
 	for _, inst := range program {
-		if inst.operator == opMulVal ||
-			inst.operator == opDupVal {
-			lastInst = inst
-			continue
-		}
 		if inst.SameOp(endLoop) {
 			depth--
 		}
-		fmt.Fprintln(writer, strings.Repeat("\t", depth), instPrint(inst, lastInst))
+		printout := instPrint(inst, lastInst)
+		if printout != "" {
+			fmt.Fprintln(writer, strings.Repeat("\t", depth), printout)
+		}
 		if inst.SameOp(startLoop) {
 			depth++
 		}

diff --git a/parser/print_test.go b/parser/print_test.go
@@ -31,7 +31,7 @@ func TestPrint(t *testing.T) {
 	outputBuf := bufio.NewWriter(&buf)
 	Print(program, outputBuf)
 	got := buf.String()
-	want := " \n >>>>>\n [-]\n +++++\n [->>+<<]\n [\n\t ,\n\t [<<<]\n\t ++\n\t [->>++<<]\n\t [->+>+<<]\n ]\n >>\n .\n [->>-<<]\n"
+	want := " >>>>>\n [-]\n +++++\n [->>+<<]\n [\n\t ,\n\t [<<<]\n\t ++\n\t [->>++<<]\n\t [->+>+<<]\n ]\n >>\n .\n [->>-<<]\n"
 
 	if !reflect.DeepEqual(got, want) {
 		t.Errorf("got %v want %v", got, want)