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 new(self, size, name="temporary_register"):

    assert type(size) == int

    reg = 0

    while reg in self.registers:

      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)