pypog/core/Grid.py

'''
Created on 7 nov. 2016
    Game Grid
@author: olinox
'''
from core import geometry
from core.Cell import Cell
from core.geometry import gline, gzone, gtriangle, grectangle
from core.pathfinder import pathfinder


class Grid(object):
    def __init__(self, grid_shape, width, height):
        self._grid_shape = None
        self.grid_shape = grid_shape
        
        self._width = 0
        self.width = width
        self._height = 0
        self.height = height
        
        self._cells = {}
        
    # properties
    @property
    def grid_shape(self):
        return self._grid_shape
    
    @grid_shape.setter
    def grid_shape(self, grid_shape):
        if not grid_shape in geometry.GRID_GEOMETRIES:
            raise ValueError("'grid_shape' has to be a value from GRID_GEOMETRIES")
        self._grid_shape = grid_shape
        
    @property
    def width(self):
        return self._width
    
    @width.setter
    def width(self, width):
        if not isinstance(width, int) or not width > 0:
            raise ValueError("'width' has to be a strictly positive integer")
        self._width = width
        
    @property
    def height(self):
        return self._height
    
    @height.setter
    def height(self, height):
        if not isinstance(height, int) or not height > 0:
            raise ValueError("'width' has to be a strictly positive integer")
        self._height = height    
    
    def cell(self, coord):
        # temporary
        try:
            return self._cells[coord]
        except KeyError:
            x, y = coord
            cell = Cell(self._geometry, x, y)
            self._cells[coord] = cell
            return cell
    
    # methods
    def cases_number(self):
        return self.height * self.width
    
    def in_grid(self, x, y):
        """return True if the coordinates are in the grid"""
        return (x > 0 and x <= self._width and y > 0 and y <= self._height)
    
    def line(self, x1, y1, x2, y2):
        return gline.line2d(self.grid_shape, x1, y1, x2, y2)
    
    def line3d(self, x1, y1, z1, x2, y2, z2):
        return gline.line3d(self.grid_shape, x1, y1, z1, x2, y2, z2)
    
    def zone(self, x, y, radius):
        return gzone.zone(self.grid_shape, x, y, radius)
    
    def triangle(self, xa, ya, xh, yh, iAngle):
        return gtriangle.triangle(self.grid_shape, xa, ya, xh, yh, iAngle)
    
    def triangle3d(self, xa, ya, za, xh, yh, zh, iAngle):
        return gtriangle.triangle3d(self.grid_shape, xa, ya, za, xh, yh, zh, iAngle)

    def rect(self, x1, y1, x2, y2):
        return grectangle.rect(x1, y1, x2, y2)
    
    def hollow_rect(self, x1, y1, x2, y2):
        return grectangle.hollow_rect(x1, y1, x2, y2)

    def path(self, x1, y1, x2, y2):
        return pathfinder.path( self, (x1, y1), (x2,y2) )

    
    
    
    
class HexGrid(Grid):
    def __init__(self, width, height):
        Grid.__init__(self, 5, width, height)

class SquareGrid(Grid):
    def __init__(self, width, height):
        Grid.__init__(self, 4, width, height)