Fixed incorrect number of history entries

src/asm/CanSkip.go

@@ -2,8 +2,12 @@ package asm
 
 import "git.akyoto.dev/cli/q/src/cpu"
 
-// unnecessary returns true if the register/register operation can be skipped.
-func (a *Assembler) unnecessary(mnemonic Mnemonic, left cpu.Register, right cpu.Register) bool {
+// CanSkip returns true if the register/register operation can be skipped.
+func (a *Assembler) CanSkip(mnemonic Mnemonic, left cpu.Register, right cpu.Register) bool {
+	if mnemonic == MOVE && left == right {
+		return true
+	}
+
 	if len(a.Instructions) == 0 {
 		return false
 	}

src/asm/CanSkipReturn.go

@@ -0,0 +1,16 @@
+package asm
+
+// CanSkipReturn returns true if the return operation can be skipped.
+func (a *Assembler) CanSkipReturn() bool {
+	if len(a.Instructions) == 0 {
+		return false
+	}
+
+	lastMnemonic := a.Instructions[len(a.Instructions)-1].Mnemonic
+
+	if lastMnemonic == RETURN || lastMnemonic == JUMP {
+		return true
+	}
+
+	return false
+}

src/asm/Instructions.go

@@ -32,14 +32,6 @@ func (a *Assembler) DLLCall(name string) {
 
 // Return returns back to the caller.
 func (a *Assembler) Return() {
-	if len(a.Instructions) > 0 {
-		lastMnemonic := a.Instructions[len(a.Instructions)-1].Mnemonic
-
-		if lastMnemonic == RETURN || lastMnemonic == JUMP {
-			return
-		}
-	}
-
 	a.Instructions = append(a.Instructions, Instruction{Mnemonic: RETURN})
 }
 

src/asm/RegisterRegister.go

@@ -19,10 +19,6 @@ func (data *RegisterRegister) String() string {
 
 // RegisterRegister adds an instruction using two registers.
 func (a *Assembler) RegisterRegister(mnemonic Mnemonic, left cpu.Register, right cpu.Register) {
-	if a.unnecessary(mnemonic, left, right) {
-		return
-	}
-
 	a.Instructions = append(a.Instructions, Instruction{
 		Mnemonic: mnemonic,
 		Data: &RegisterRegister{

src/core/CompileReturn.go

@@ -8,9 +8,8 @@ import (
 
 // CompileReturn compiles a return instruction.
 func (f *Function) CompileReturn(node *ast.Return) error {
-	defer f.Return()
-
 	if len(node.Values) == 0 {
+		f.Return()
 		return nil
 	}
 
@@ -27,12 +26,13 @@ func (f *Function) CompileReturn(node *ast.Return) error {
 
 		if !types.Is(typ, f.Output[i].Type) {
 			if f.Package == "mem" && f.Name == "alloc" {
-				return nil
+				continue
 			}
 
 			return errors.New(&errors.TypeMismatch{Encountered: typ.Name(), Expected: f.Output[i].Type.Name(), ParameterName: "", IsReturn: true}, f.File, node.Values[i].Token.Position)
 		}
 	}
 
+	f.Return()
 	return nil
 }

src/register/MemoryRegister.go

@@ -7,5 +7,10 @@ import (
 
 func (f *Machine) MemoryRegister(mnemonic asm.Mnemonic, a asm.Memory, b cpu.Register) {
 	f.Assembler.MemoryRegister(mnemonic, a, b)
+
+	if mnemonic == asm.LOAD {
+		f.UseRegister(b)
+	}
+
 	f.postInstruction()
 }

src/register/RegisterNumber.go

@@ -27,7 +27,6 @@ func (f *Machine) RegisterNumber(mnemonic asm.Mnemonic, a cpu.Register, b int) {
 	f.Assembler.RegisterNumber(asm.MOVE, tmp, b)
 	f.UseRegister(tmp)
 	f.postInstruction()
-	f.Assembler.RegisterRegister(mnemonic, a, tmp)
-	f.postInstruction()
+	f.RegisterRegister(mnemonic, a, tmp)
 	f.FreeRegister(tmp)
 }

src/register/RegisterRegister.go

@@ -6,7 +6,7 @@ import (
 )
 
 func (f *Machine) RegisterRegister(mnemonic asm.Mnemonic, a cpu.Register, b cpu.Register) {
-	if mnemonic == asm.MOVE && a == b {
+	if f.Assembler.CanSkip(mnemonic, a, b) {
 		return
 	}
 

src/register/Return.go

@@ -1,6 +1,13 @@
 package register
 
 func (f *Machine) Return() {
+	if f.Assembler.CanSkipReturn() {
+		return
+	}
+
 	f.Assembler.Return()
+	scope := f.CurrentScope()
+	scope.Reserved = 0
+	scope.Used = 0
 	f.postInstruction()
 }