Simplified Equality
This commit is contained in:
53
vm/vm.go
53
vm/vm.go
@@ -659,39 +659,18 @@ func (vm *VM) executeComparison(op code.Opcode) error {
|
||||
right := vm.pop()
|
||||
left := vm.pop()
|
||||
|
||||
if left.Type() == object.INTEGER_OBJ && right.Type() == object.INTEGER_OBJ {
|
||||
return vm.executeIntegerComparison(op, left, right)
|
||||
}
|
||||
|
||||
if left.Type() == object.STRING_OBJ && right.Type() == object.STRING_OBJ {
|
||||
return vm.executeStringComparison(op, left, right)
|
||||
}
|
||||
|
||||
switch op {
|
||||
case code.OpEqual:
|
||||
return vm.push(nativeBoolToBooleanObject(right == left))
|
||||
return vm.push(nativeBoolToBooleanObject(left.(object.Comparable).Compare(right) == 0))
|
||||
case code.OpNotEqual:
|
||||
return vm.push(nativeBoolToBooleanObject(right != left))
|
||||
default:
|
||||
return fmt.Errorf("unknown operator: %d (%s %s)", op, left.Type(), right.Type())
|
||||
}
|
||||
}
|
||||
|
||||
func (vm *VM) executeIntegerComparison(op code.Opcode, left, right object.Object) error {
|
||||
leftValue := left.(*object.Integer).Value
|
||||
rightValue := right.(*object.Integer).Value
|
||||
|
||||
switch op {
|
||||
case code.OpEqual:
|
||||
return vm.push(nativeBoolToBooleanObject(rightValue == leftValue))
|
||||
case code.OpNotEqual:
|
||||
return vm.push(nativeBoolToBooleanObject(rightValue != leftValue))
|
||||
case code.OpGreaterThan:
|
||||
return vm.push(nativeBoolToBooleanObject(leftValue > rightValue))
|
||||
return vm.push(nativeBoolToBooleanObject(left.(object.Comparable).Compare(right) != 0))
|
||||
case code.OpGreaterThanEqual:
|
||||
return vm.push(nativeBoolToBooleanObject(leftValue >= rightValue))
|
||||
return vm.push(nativeBoolToBooleanObject(left.(object.Comparable).Compare(right) > -1))
|
||||
case code.OpGreaterThan:
|
||||
return vm.push(nativeBoolToBooleanObject(left.(object.Comparable).Compare(right) == 1))
|
||||
default:
|
||||
return fmt.Errorf("unknown operator: %d", op)
|
||||
return fmt.Errorf("unknown operator: %d (%s %s)",
|
||||
op, left.Type(), right.Type())
|
||||
}
|
||||
}
|
||||
|
||||
@@ -826,24 +805,6 @@ func (vm *VM) executeStringIndex(str, index object.Object) error {
|
||||
)
|
||||
}
|
||||
|
||||
func (vm *VM) executeStringComparison(op code.Opcode, left, right object.Object) error {
|
||||
leftValue := left.(*object.String).Value
|
||||
rightValue := right.(*object.String).Value
|
||||
|
||||
switch op {
|
||||
case code.OpEqual:
|
||||
return vm.push(nativeBoolToBooleanObject(rightValue == leftValue))
|
||||
case code.OpNotEqual:
|
||||
return vm.push(nativeBoolToBooleanObject(rightValue != leftValue))
|
||||
case code.OpGreaterThan:
|
||||
return vm.push(nativeBoolToBooleanObject(leftValue > rightValue))
|
||||
case code.OpGreaterThanEqual:
|
||||
return vm.push(nativeBoolToBooleanObject(leftValue >= rightValue))
|
||||
default:
|
||||
return fmt.Errorf("unknown operator: %d", op)
|
||||
}
|
||||
}
|
||||
|
||||
func nativeBoolToBooleanObject(input bool) *object.Boolean {
|
||||
if input {
|
||||
return True
|
||||
|
||||
Reference in New Issue
Block a user