gtool_general.py

from geo_object import Point, Line, Circle
from gtool import GTool
from tools import Tool
from tool_step import ToolStep
from collections import defaultdict
from parse import Parser, type_to_c

"""
GToolDict investigates a dictionary of parsed (and primitive) tools,
selects the geometrical ones, and is capable of creating the appropriate
general tools that can be reached from GUI from the input entry in the toolbar.
"""
class GToolDict:
    def __init__(self, tool_dict):
        """
        a "prefix" here is a tuple of types that were selected so far,
        prefix_d is a dictionary from prefixes to the appropriate actions
        appropriate action can be either
         * a tool (if the tool already have anough data), or
         * a set of types -- the available next objects that can be selected
        """
        self.name_to_prefixes = defaultdict(dict)
        for (name, arg_types), tool in tool_dict.items():
            if arg_types is None: continue
            if name.startswith("prim__"): continue
            if not all(
                issubclass(arg, (Point, Line, Circle))
                for arg in arg_types
            ): continue
            if len(arg_types) == 0: continue
            prefix_d = self.name_to_prefixes[name]
            prefix_d[arg_types] = tool
            for i,t in enumerate(arg_types):
                prefix = tuple(arg_types[:i])
                type_set = prefix_d.setdefault(prefix, set())
                if isinstance(type_set, set): type_set.add(t)

    def make_tool(self, name):
        if name not in self.name_to_prefixes: return None
        return GToolGeneral(self.name_to_prefixes[name], name)

class GToolGeneral(GTool):

    def __init__(self, prefix_d, name):
        self.name = name
        GTool.__init__(self)
        types_to_selector = {
            frozenset((Point, Line, Circle)): self.select_pcl,
            frozenset((Point, Line)):   self.select_pl,
            frozenset((Point, Circle)): self.select_pc,
            frozenset((Line, Circle)):  self.select_cl,
            frozenset((Point,)):  self.select_point,
            frozenset((Line,)):   self.select_line,
            frozenset((Circle,)): self.select_circle,
        }
        self.prefix_d = dict()
        for prefix,tool_or_types in prefix_d.items():
            if isinstance(tool_or_types, Tool):
                self.prefix_d[prefix] = tool_or_types
            else:
                types = frozenset(tool_or_types)
                selector = types_to_selector[types]
                self.prefix_d[prefix] = selector

    def enter(self, viewport):
        GTool.enter(self, viewport)
        # print self
        name = None
        ttypes = []
        for itype, tool in self.prefix_d.items():
            if isinstance(tool, Tool):
                name = tool.name
                otype = self.tools[name,itype].out_types
                output = ' '.join(type_to_c[t] for t in otype)
                args = ' '.join(type_to_c[t] for t in itype)
                ttypes.append("{} -> {}".format(args, output))
        print("Tool: {} : {}".format(name, ', or '.join(ttypes)))

    def update_basic(self, coor, args = (), types = ()):

        selector = self.prefix_d[types]
        obj,objn = selector(coor)
        if obj is not None:
            args = args+(obj,)
            types = types+(type(objn),)
            tool = self.prefix_d[types]
            if isinstance(tool, Tool):
                self.confirm = self.run_tool, tool, *args
            else: self.confirm_next = self.update_basic, args, types