Instruction counter

Author: AsynchronousAI

compiler/main.go

@@ -20,15 +20,19 @@ type Options struct {
 	LogIR      bool
 }
 
+func countInstructionTotal(commands []AssemblyCommand) int {
+	total := 0
+	for _, cmd := range commands {
+		if cmd.Type == Instruction && cmd.Name != "" {
+			total++
+		}
+	}
+	return total
+}
+
 func Compile(executable *os.File, options Options) []byte {
 	/* prepare */
-	var writer = &OutputWriter{
-		Buffer:                   []byte(""),
-		MemoryDevelopmentPointer: 0,
-		MaxPC:                    1,
-		Options:                  options,
-		MemoryMap:                make(map[string]int),
-	}
+	writer := newOutputWriter(options)
 
 	elf, err := elf.NewFile(executable)
 	if err != nil {
@@ -46,11 +50,18 @@ func Compile(executable *os.File, options Options) []byte {
 		writer.Commands = ParseFromElf(elf)
 	}
 
+	writer.InstructionTotal = countInstructionTotal(writer.Commands)
+
 	/* compilation */
 	BeforeCompilation(writer)
 	for _, command := range writer.Commands {
 		CompileInstruction(writer, command)
+		if command.Type == Instruction && command.Name != "" {
+			writer.InstructionProcessed++
+		}
+		writer.updateProgress()
 	}
+	writer.finishProgress()
 	if writer.Options.LogIR {
 		dumpIRAsJSON(writer)
 	}

compiler/writer.go

@@ -29,6 +29,18 @@ type OutputWriter struct {
 	MemoryMap                map[string]int /* map static data keys to addresses */
 	Options                  Options        /* user specified options */
 	InlineCounter            int
+	InstructionTotal         int
+	InstructionProcessed     int
+}
+
+func newOutputWriter(options Options) *OutputWriter {
+	return &OutputWriter{
+		Buffer:                   []byte(""),
+		MemoryDevelopmentPointer: 0,
+		MaxPC:                    1,
+		Options:                  options,
+		MemoryMap:                make(map[string]int),
+	}
 }
 
 func (w *OutputWriter) nextInlineTemp() string {
@@ -44,3 +56,17 @@ func WriteIndentedString(writer *OutputWriter, format string, args ...any) {
 	indent := strings.Repeat("\t", writer.Depth)
 	writer.Buffer = append(writer.Buffer, indent+fmt.Sprintf(format, args...)...)
 }
+
+func (w *OutputWriter) updateProgress() {
+	if w.InstructionTotal == 0 {
+		return
+	}
+	fmt.Printf("\rInstruction %d/%d", w.InstructionProcessed, w.InstructionTotal)
+}
+
+func (w *OutputWriter) finishProgress() {
+	if w.InstructionTotal == 0 {
+		return
+	}
+	fmt.Print("\n")
+}