from pypy.jit.codegen import model
from pypy.rpython.lltypesystem import lltype, rffi, llmemory
# Wrapper Classes:
# The opcodes(assemble.py) differ from the type of
# the operand(eg. Register, Immediate...). 
# The to_string method is used to choose the right
# method inside the assembler

class Location(object):
    pass

class Register64(Location):
    def __init__(self, reg):
        self.reg = reg
        self.throw_away = False
        self.contains_genConst = False
        
    def dont_spill(self, throw_away):
        self.throw_away = throw_away # this value is not needed anymore
    
    def contains_Const(self):
        self.contains_genConst = True
        
class Register8(Location):
    def __init__(self, reg):
        self.reg = reg
        self.throw_away = False
        self.contains_genConst = False
 
    def dont_spill(self, throw_away):
        self.throw_away = throw_away # this value is not needed anymore
        
    def contains_Const(self):
        self.contains_genConst = True
        
class Stack64(Location):
    def __init__(self, offset):
        self.offset = offset

class IntVar(model.GenVar):
    def __init__(self, location):
        self.location = location
        self.is_const = False
        assert isinstance(location, Location) 
    
    def to_string(self):
        #return self.location.to_string()
        if isinstance(self.location, Stack64): 
            return "_STACK"
        if isinstance(self.location, Register8): 
            return "_8REG"
        elif isinstance(self.location, Register64): 
            return "_QWREG"

class Immediate8(model.GenConst):
    def __init__(self, value):
        self.value = value
        
    def to_string(self):
        return "_IMM8"
    
class Immediate32(model.GenConst):
    def __init__(self, value):
        self.value = value
        
    def to_string(self):
        return "_IMM32"
    
# TODO: understand GenConst
class Immediate64(model.GenConst):
    def __init__(self, value):
        self.value = value
        
    def to_string(self):
        return "_IMM64"