relative_data.py

from copy import deepcopy


class RelativeData:

	def __init__(self, reference_id=0):
		self.reference_id = reference_id
		self.data_size = 4
		self.data = {}

	def globalize_id(self, player_id):
		return (player_id + self.reference_id) % self.data_size

	def localize_id(self, player_id):
		return (player_id - self.reference_id) % self.data_size

	def make_copy(self, reference_id=None):
		copy = deepcopy(self)
		if reference_id is not None:
			copy.reference_id = reference_id
		return copy

	def __getitem__(self, i):
		if not isinstance(i, int):
			raise ValueError("Expected int, got " + str(type(i)))
		if i >= self.data_size or i < 0:
			raise KeyError("Key must be in range 0 to " + str(self.data_size-1))
		global_id = self.globalize_id(i)
		if not global_id in self.data:
			return None
		return self.data[global_id]

	def __contains__(self, i):
		if not isinstance(i, int):
			raise ValueError("Expected int, got " + str(type(i)))
		if i >= self.data_size or i < 0:
			return False
		global_id = self.globalize_id(i)
		return global_id in self.data

	def __len__(self):
		return len(self.data)

	def __iter__(self):
		return iter(self.data)