#!/usr/bin/env python3 -tt
"""
"""
[docs]class Grid(list):
def __init__(self, num_rows=0, num_cols=0, value=None):
self.num_rows = num_rows
self.num_cols = num_cols
# Elements are stored in row-major order
self.elems = [[value for __ in range(num_cols)] for _ in range(num_rows)]
[docs] def empty(self):
return self.num_rows == 0 or self.num_cols == 0
[docs] def fill(self, value):
self.elems = [[value for __ in range(self.num_cols)] for _ in range(self.num_rows)]
[docs] def get(self, row, col):
return self.elems[row][col]
[docs] def set(self, row, col, value):
self.elems[row][col] = value
[docs] def in_bounds(self, row, col):
pass
[docs] def resize(self, num_rows, num_cols, retain=False):
if not retain:
self.__init__(num_rows, num_cols)
else:
old = self.elems
self.elems = [[None for __ in range(self.num_cols)] for _ in range(self.num_rows)]
for oldrow, newrow in zip(old, self.elems):
for pos, elems in enumerate(zip(oldelems, newelems)):
oldelem, newelem = elems
if oldelem != newelem:
newrow[pos] = oldelem
# Use this when you need to copy
[docs] @classmethod
def from_grid(cls, grid):
new = cls(grid.num_rows, grid.num_cols)
new.elems = grid.elems
return new
@property
def height(self):
return num_rows
@property
def width(self):
return num_cols
# Magic methods to make a grid look like a list of lists
def __len__(self):
return len(self.elems)
# Note - setitem and delitem don't make sense in this case
def __getitem__(self, key):
return self.elems[key]
def __iter__(self):
return iter(self.elems)
def __reversed__(self):
return reversed(self.elems)
def __contains__(self, item):
return item in self.elems
def __repr__(self):
return repr(self.elems)