item_collection.py

from fileinput import close
from solid import *
from solid.utils import *

# import graphviz

import logging
import sys

from switch import Switch
from support import Support
from support_cutout import SupportCutout
from cell import Cell


class ItemCollection:
    def __init__(self, rotation = 0.0):
        
        self.logger = logging.getLogger().getChild(__name__)

        self.collection = {}

        self.rotation = rotation

        self.get_collection_bounds_call_count = 0
        self.get_moved_union_call_count = 0

        # self.dot_recurse = graphviz.Digraph()
        # self.dot = graphviz.Digraph()


    def get_collection_dict(self, rx = 0.0, ry = 0.0):
        return self.collection[rx][ry]
    
    def add_item(self, x_offset, y_offset, cell: Cell, rx = 0.0, ry = 0.0):
        if rx not in self.collection.keys():
            self.collection[rx] = {}

        if ry not in self.collection[rx].keys():
            self.collection[rx][ry] = {}

        if x_offset not in self.collection[rx][ry].keys():
            self.collection[rx][ry][x_offset] = {}

        if rx != 0.0 or ry != 0.0:
            self.logger.debug('Adding item to collection with rx = {}, ry = {}'.format(rx, ry))

        self.collection[rx][ry][x_offset][y_offset] = cell


    def get_item(self, x_offset, y_offset, rx = 0.0, ry = 0.0) -> Cell:
        return self.collection[rx][ry][x_offset][y_offset]

    def get_item_with_value(self, value):
        for rx in self.get_rx_list():
            for ry in self.get_ry_list_in_rx(rx):
                for x in self.get_x_list_in_rx_ry(rx, ry):
                    for y in self.get_y_list_in_rx_ry_x(x, rx, ry):
                        current_switch = self.get_item(x, y, rx, ry)

                        if current_switch.cell_value == value:
                            return current_switch

    def get_moved_item(self, x_offset, y_offset, rx = 0.0, ry = 0.0) -> Cell:
        return self.collection[rx][ry][x_offset][y_offset].get_moved()

    def collection_has_keys(self, rx = 0.0, ry = None, x = None, y = None):
        if rx not in self.collection.keys():
            self.logger.debug('Collection has no rx = {}'.format(rx))
            return False
        if ry is not None and ry not in self.collection[rx].keys():
            self.logger.debug('Collection has no ry = {}'.format(ry))
            return False
        if x is not None and x not in self.collection[rx][ry].keys():
            self.logger.debug('Collection has no x = {}'.format(x))
            return False
        if y is not None and y not in self.collection[rx][ry][x].keys():
            self.logger.debug('Collection has no y = {}'.format(y))
            return False
        
        return True

    def get_rx_list(self):
        return self.collection.keys()

    def get_ry_list_in_rx(self, rx):
        if self.collection_has_keys(rx):
            return self.collection[rx].keys()
        else:
            return []

    def get_x_list_in_rx_ry(self, rx = 0.0, ry = 0.0):
        if self.collection_has_keys(rx, ry):
            return self.collection[rx][ry].keys()
        else:
            return []
    def get_y_list_in_rx_ry_x(self, x, rx = 0.0, ry = 0.0):
        if self.collection_has_keys(rx, ry, x):
            return self.collection[rx][ry][x].keys()
        else:
            return []

    def get_x_list(self, rx = 0.0, ry = 0.0):
        return self.get_x_list_in_rx_ry(rx, ry)


    def get_sorted_x_list(self, rx = 0.0, ry = 0.0):
        return sorted(self.get_x_list(rx, ry))

    def get_y_list_in_x(self, x, rx = 0.0, ry = 0.0):
        return self.get_y_list_in_rx_ry_x(x, rx, ry)

    def get_sorted_y_list_in_x(self, x, rx = 0.0, ry = 0.0):
        return sorted(self.get_y_list_in_x(x, rx, ry))

    def get_min_x(self, rx = 0.0, ry = 0.0):
        return min(self.get_x_list(rx, ry))

    def get_max_x(self, rx = 0.0, ry = 0.0):
        return max(self.get_x_list(rx, ry))
        
    def get_min_y(self, rx = 0.0, ry = 0.0):
        min_y = 0
        
        for x in self.get_x_list(rx, ry):
            col_min_y = min(self.get_y_list_in_rx_ry_x(x, rx, ry))

            if col_min_y < min_y:
                min_y = col_min_y    
        return min_y

    def get_collection_bounds(self, rx = 0.0, ry = 0.0) -> float:
        

        min_y = 1000.0
        max_y = -1000.0
        max_x = -1000.0
        min_x = 1000.0
        
        for x in self.get_x_list(rx, ry):
            for y in self.get_y_list_in_x(x, rx, ry):
                cell = self.get_item(x, y, rx, ry)
                start_x = cell.get_start_x()
                end_x = cell.get_end_x()
                start_y = cell.get_start_y()
                end_y = cell.get_end_y()

                # self.logger.debug('end_x: %f, end_y: %f', end_x, end_y)
                if start_x < min_x:
                    min_x = start_x

                if end_x > max_x:
                    max_x = end_x

                if start_y > max_y:
                    max_y = start_y

                if end_y < min_y:
                    min_y = end_y

        self.logger.debug('min_x: %f, max_x: %f, max_y: %f, min_y: %f', min_x, max_x, max_y, min_y)
        return (min_x, max_x, max_y, min_y)


    def get_moved_union(self, rx = 0.0, ry = 0.0):
        solid = union()

        for x in self.get_x_list_in_rx_ry(rx, ry):
            for y in self.get_y_list_in_rx_ry_x(x, rx, ry):
                temp_solid = self.get_moved_item(x, y, rx, ry)
                solid += temp_solid
        return solid

    
    def set_collection_neighbors(self, neighbor_group = 'local'):
        

        self.logger.debug('Set %s neighbors', neighbor_group)
        for rx in self.get_rx_list():
            for ry in self.get_ry_list_in_rx(rx):
                for x in self.get_x_list_in_rx_ry(rx, ry):
                    for y in self.get_y_list_in_rx_ry_x(x, rx, ry):
                        current_switch: Switch
                        current_switch = self.get_item(x, y, rx, ry)
                        all_neighbors_set = current_switch.get_all_neighbors_set(neighbor_group = neighbor_group)
                        if all_neighbors_set == False:
                            # self.logger.debug('set_collection_neighbors call set_item_neighbor for switch %s', str(current_switch))
                            self.set_item_neighbor(current_switch, neighbor_group = neighbor_group)

    
    
    def set_item_neighbor(self, item: Switch, neighbor_group = 'local', tabs = ''):
        # self.logger.debug('%sSet neighbors for switch %s', tabs, str(item))
        
        x_min = item.x
        x_max = item.x + item.w
        y_min = item.y - item.h
        y_max = item.y

        neighbor_list_dict = {
            'right': [],
            'left': [],
            'top': [],
            'bottom': []
        }

        neighbor_oposite_dict = {
            'right': 'left',
            'left': 'right',
            'top': 'bottom',
            'bottom': 'top'
        }


        for sib_rx in self.get_rx_list():
            for sib_ry in self.get_ry_list_in_rx(sib_rx):
                for sib_x in self.get_x_list_in_rx_ry(sib_rx, sib_ry):
                    for sib_y in self.get_y_list_in_rx_ry_x(sib_x, sib_rx, sib_ry):
                        sibling_switch = self.get_item(sib_x, sib_y, sib_rx, sib_ry)

                        sib_x_min = sibling_switch.x
                        sib_x_max = sibling_switch.x + sibling_switch.w
                        sib_y_min = sibling_switch.y - sibling_switch.h
                        sib_y_max = sibling_switch.y

                        # check right neighbor
                        if sib_y_max > y_min and sib_y_min < y_max and sib_x_min >= x_max:
                            neighbor_list_dict['right'].append(sibling_switch)
                            # right_neighbor_offset = sib_x_min - x_max
                        # check left neighbor
                        if sib_y_max > y_min and sib_y_min < y_max and sib_x_max <= x_min:
                            neighbor_list_dict['left'].append(sibling_switch)
                            # left_neighbor_offset = x_min - sib_x_max
                        # check top neighbor
                        if sib_x_max > x_min and sib_x_min < x_max and sib_y_min >= y_max:
                            neighbor_list_dict['top'].append(sibling_switch)
                            # top_neighbor_offset = sib_y_min - y_max
                        # check bottom neighbor
                        if sib_x_max > x_min and sib_x_min < x_max and sib_y_max <= y_min:
                            neighbor_list_dict['bottom'].append(sibling_switch)
                            # bottom_neighbor_offset = y_min - sib_y_max
        

        for direction in neighbor_list_dict.keys():
            # self.logger.debug('direction: %s', direction)
            offset = 0.0
            
            if len(neighbor_list_dict[direction]) > 0: 
                closest_neighbor: Switch = None

                if direction == 'right':
                    closest_neighbor = min(neighbor_list_dict[direction], key=lambda item: item.x)
                    offset = closest_neighbor.x_min - x_max
                    perp_offset = closest_neighbor.y - item.y
                elif direction == 'left':
                    closest_neighbor = max(neighbor_list_dict[direction], key=lambda item: item.x)
                    offset = x_min - closest_neighbor.x_max
                    perp_offset = closest_neighbor.y - item.y
                elif direction == 'top':
                    closest_neighbor = min(neighbor_list_dict[direction], key=lambda item: item.y)
                    offset = closest_neighbor.y_min - y_max
                    perp_offset = closest_neighbor.x - item.x
                elif direction == 'bottom':
                    closest_neighbor = max(neighbor_list_dict[direction], key=lambda item: item.y)
                    offset = y_min - closest_neighbor.y_max
                    perp_offset = closest_neighbor.x - item.x

                item.set_neighbor(neighbor = closest_neighbor, neighbor_name = direction, offset = offset, neighbor_group = neighbor_group, perp_offset = perp_offset)
                closest_neighbor.set_neighbor(neighbor = item, neighbor_name = Switch.NEIGHBOR_OPOSITE_DICT[direction], offset = offset, neighbor_group = neighbor_group, perp_offset = perp_offset)
                closest_neighbor.update_all_neighbors_set(neighbor_group = neighbor_group)
            else:
                # self.logger.debug('set switch %s no neighbor %s', str(item), direction)
                closest_neighbor = None
                item.set_neighbor(neighbor_name = direction, has_neighbor = False, neighbor_group = neighbor_group)

        item.update_all_neighbors_set(neighbor_group = neighbor_group)

        all_neighbors_set = item.get_all_neighbors_set(neighbor_group = neighbor_group)

        # pos = '%f,%f!' % (item.center_x, item.center_y)
        # self.dot_recurse.node(item.cell_value, pos = pos)

        if all_neighbors_set == True:
            for direction in neighbor_list_dict.keys():
                neighbor: Switch
                neighbor = item.get_neighbor(direction, neighbor_group = neighbor_group)

                if neighbor is not None:
                    neighbor_all_neighbors_set = neighbor.get_all_neighbors_set(neighbor_group = neighbor_group)

                    if neighbor_all_neighbors_set == False:
                        # self.logger.debug('\t\tset neighbors for neighbor switch %s', str(neighbor))
                        # pos = '%f,%f!' % (neighbor.center_x, neighbor.center_y)
                        # self.dot_recurse.node(neighbor.cell_value, pos = pos)
                        # self.dot_recurse.edge(item.cell_value, neighbor.cell_value)

                        self.set_item_neighbor(neighbor, neighbor_group = neighbor_group, tabs = tabs + ' ')



    def draw_rotated_items(self, rx = 0.0, ry = 0.0):
        solid = union()

        for x in self.get_x_list_in_rx_ry(rx, ry):
            for y in self.get_y_list_in_rx_ry_x(x, rx, ry):
                cell = self.get_item(x, y, rx, ry)

                poly_points = cell.get_rotation_info_points()
                poly_path = [[0, 1, 2, 3]]

                rotated_polygon = polygon(poly_points, poly_path)

                solid += rotated_polygon
                
        return solid



    def render_graph(self, output_filename):
        

        # self.dot.render(output_filename, engine = 'neato')

        filename = output_filename.name.replace('.gv', '_recurse.gv')
        path = output_filename.parent
        self.logger.debug('type(output_filename): %s', str(type(output_filename)))

        # self.dot_recurse.render(path / filename, engine = 'neato')


    def neighbor_check(self, neighbor_group = 'local', output_filename = ''):
        
        
        # self.dot = graphviz.Digraph(comment='Keyboard')
        for rx in self.get_rx_list():
            for ry in self.get_ry_list_in_rx(rx):
                for x in self.get_x_list_in_rx_ry(rx, ry):
                    for y in self.get_y_list_in_rx_ry_x(x, rx, ry):
                        item: Switch
                        item = self.get_item(x, y, rx, ry)

                        item_cell_value = item.cell_value

                        # pos = '%f,%f!' % (item.center_x, item.center_y)
                        # self.dot.node(item_cell_value, pos = pos)

                        for direction in item.get_neighbor_direction_list():
                            reverse_direction = Switch.NEIGHBOR_OPOSITE_DICT[direction]
                            neighbor: Switch = item.get_neighbor(direction, neighbor_group = neighbor_group)

                            if neighbor is not None:
                                neighbor_cell_value = neighbor.cell_value
                                # pos = '%f,%f!' % (neighbor.center_x, neighbor.center_y)
                                # self.dot.node(neighbor_cell_value, pos = pos)
                                # self.dot.edge(item_cell_value, neighbor_cell_value)

                                reverse_neighbor: Switch = neighbor.get_neighbor(reverse_direction, neighbor_group = neighbor_group)

                                reverse_neighbor_cell_value = reverse_neighbor.cell_value

                                if item_cell_value != reverse_neighbor_cell_value:
                                    self.logger.debug('Cell "%s" %s neighbor "%s" reverse neighbor %s has has different value %s', item_cell_value, direction, neighbor_cell_value, reverse_direction, reverse_neighbor_cell_value)
        
        # self.dot.render(output_filename, engine = 'neato')

    def has_global_neighbor_section(self, neighbor_name = ''):
        for rx in self.get_rx_list():
            for ry in self.get_ry_list_in_rx(rx):
                for x in self.get_x_list_in_rx_ry(rx, ry):
                    for y in self.get_y_list_in_rx_ry_x(x, rx, ry):
                        item: Switch
                        item = self.get_item(x, y, rx, ry)

                        local_neighbor = item.get_neighbor(neighbor_name, neighbor_group = 'local')
                        global_neighbor = item.get_neighbor(neighbor_name, neighbor_group = 'global')

                        if local_neighbor is None and global_neighbor is not None:
                            return True


    def has_global_right_neighbor_section(self):
        return self.has_global_neighbor_section(neighbor_name = 'right')
    
    def has_global_left_neighbor_section(self):
        return self.has_global_neighbor_section(neighbor_name = 'left')