Subversion Repositories tcp_socket

__author__ = "Jon Dawson"

__copyright__ = "Copyright (C) 2012, Jonathan P Dawson"

__version__ = "0.1"

class Allocator:

    """Maintain a pool of registers, variables and arrays. Keep track of what they are used for."""

    def __init__(self, reuse):

        self.registers = []

        self.all_registers = {}

        self.memory_size_2 = 0

        self.memory_size_4 = 0

        self.reuse = reuse

        self.memory_content_2 = {}

        self.memory_content_4 = {}

    def new_array(self, size, contents, element_size):

        if element_size == 2:

            reg = self.memory_size_2

            self.memory_size_2 += int(size)

            if contents is not None:

                for location, value in enumerate(contents, reg):

                    self.memory_content_2[location] = value

            return reg

        elif element_size == 4:

            reg = self.memory_size_4

            self.memory_size_4 += int(size)

            if contents is not None:

                for location, value in enumerate(contents, reg):

                    self.memory_content_4[location] = value

            return reg

    def regsize(self, reg):

        return self.all_registers[reg][1]

    def new(self, size, name="temporary_register"):

        assert type(size) == int

        reg = 0

        while reg in self.registers or (reg in self.all_registers and self.regsize(reg) != size):

            reg += 1

        self.registers.append(reg)

        self.all_registers[reg] = (name, size)

        return reg

    def free(self, register):

        if register in self.registers and self.reuse:

            self.registers.remove(register)