https://github.com/mirefek/geo_logic
Tip revision: c8c9b715cae0acfb07585c25659b1333d075cc5b authored by mirefek on 25 July 2021, 19:43:55 UTC
bugfix
bugfix
Tip revision: c8c9b71
knowledge_visualisation.py
from movable_tools import MovableTool
from collections import defaultdict
import itertools
from geo_object import *
"""
KnowledgeVisualisation investigates LogicalCore in a GraphicalEnv
and extract data which should be displayed by Viewport
(which have access to the knowledge_visualisation).
The logical objects are mostly accessed as logical indices (li),
although we also sometimes need GUI indices (gi) used by GraphicalEnv.
"""
### segment distribution
"""
For distances, arcs and angles, we sometimes put them to different
levels to avoid collisions. This segment distribution is done on every line
(distribute_segments), and on every circle / around every point (distribute_segments_cyc)
The input is a list of quadruples of the form (a,b,start,info) meaning that we want to
draw the segment from "a" to "b".
We want to assign the segments as low (non-negative) level as possible so that
no intersecting segments share their level.
"start" is 0 or 1, if 1 the segment is forbiden to be on level 0.
Similarly, we forbid the segment level zero if its inside contains
some of the lev_zero_points given as a potential argument.
The output of the function is an iterator of pairs (info, level)
"""
def distribute_segments(segments, lev_zero_points = ()):
segments.sort(key = lambda x: (x[0], -x[1]))
lev_zero_points = sorted(lev_zero_points, reverse = True)
occupied = [None]
for a,b,start,info in segments:
while lev_zero_points and not eps_smaller(a, lev_zero_points[-1]):
lev_zero_points.pop()
if lev_zero_points and eps_smaller(lev_zero_points[-1], b) and start == 0:
start = 1
for lev,b_ori in enumerate(occupied):
if lev < start: continue
if b_ori is None or not eps_smaller(a, b_ori):
occupied[lev] = b
break
else:
lev = len(occupied)
occupied.append(b)
yield info,lev
# cyclic variant, (a,b,info), a in (0,2), b in (0,4), cyclic mod 2
def distribute_segments_cyc(segments, lev_zero_points = ()):
segments.sort(key = lambda x: (x[0], -x[1]))
lev_zero_points = list(lev_zero_points) + [x+2 for x in lev_zero_points]
lev_zero_points.sort(reverse = True)
occupied = [(None,None)]
for a,b,start,info in segments:
while lev_zero_points and not eps_smaller(a, lev_zero_points[-1]):
lev_zero_points.pop()
if lev_zero_points and eps_smaller(lev_zero_points[-1], b) and start == 0:
start = 1
for lev,(a_ori,b_ori) in enumerate(occupied):
if lev < start: continue
if a_ori is None or (not eps_smaller(a, b_ori) and not eps_smaller(a_ori, b-2)):
if a_ori is None: a_ori = a
occupied[lev] = a_ori,b
break
else:
lev = len(occupied)
occupied.append((a,b))
yield info,lev
"""
In KnowledgeVisualisation, we assign data to numerical objects for two reasons.
* First, we want to detect identical objects to show only one of them and allow
swapping using shift key.
* Second, we want to distribute segments even on a line / around a point
even when we are not logically guarrantied that the segments are on a single line
or the angles are around the same point.
The abstract class for it is NumData, with functions handling the object ambiguity.
"""
class NumData:
def __init__(self, vis, num_obj):
self.vis = vis # backward reference
self.num_obj = num_obj
self.active_candidates = []
self.extra_candidates = []
self.visible = None # li
self.is_active = False
# non-hiden GUI element
def add_active_candidate(self, li, priority):
self.is_active = True
self.active_candidates.append((priority, li))
# extra object, to be drawn as dashed, grey
def add_extra_candidate(self, li, priority):
self.extra_candidates.append((priority, li))
# find the object
def select_visible(self):
if self.is_active:
if self.vis.move_mode:
movables = [
(priority, obj)
for (priority, obj) in self.active_candidates
if self.vis.is_movable(obj)
]
if movables: self.active_candidates = movables
_,self.visible = max(self.active_candidates)
elif self.extra_candidates:
_,self.visible = max(self.extra_candidates)
return self.visible
class NumPointData(NumData):
def __init__(self, vis, point):
NumData.__init__(self, vis, point)
# data for angles
self.angles_ppl = []
self.angles_ll = []
self.exact_lines = defaultdict(set)
self.exact_angles = defaultdict(list)
def add_angle(self, angle_data):
if angle_data.color < 0:
d, = self.vis.logic.get_constr(self.vis.tools.direction_of, (angle_data.l1,))
d,_ = self.vis.logic.angles.equal_to[d]
self.exact_lines[d].update((angle_data.l1, angle_data.l2))
self.exact_angles[d].append(angle_data)
elif angle_data.l1_dir_sgn is None:
self.angles_ll.append(angle_data)
else:
self.angles_ppl.append(angle_data)
def add_exact(self, line, d):
self.exact_lines[d].add(line)
def distribute_angles(self):
if not self.angles_ppl and not self.angles_ll: return
used = set()
segments = []
for angle in self.angles_ppl:
key = (angle.l1, angle.l2, angle.l1_dir_sgn)
if key in used: continue
used.add(key)
angle.find_arc()
segments.append((angle.pos_a, angle.pos_b, 0, angle))
for angle in self.angles_ll:
if (angle.l1, angle.l2, 1) in used or (angle.l1, angle.l2, -1) in used:
continue
angle.find_nicer_l1_dir()
used.add((angle.l1, angle.l2, angle.l1_dir_sgn))
angle.find_arc()
segments.append((angle.pos_a, angle.pos_b, 0, angle))
ang_lev = distribute_segments_cyc(segments)
self.vis.visible_angles.extend(
(ang.p.a, ang.pos_a, ang.pos_b, ang.color, lev)
for ang,lev in ang_lev
)
def distribute_exact_angles(self):
for d, lines in self.exact_lines.items():
lines &= self.vis.visible_lines
if len(lines) < 1: continue
line_pos = sorted(
(vector_direction(self.vis.li_to_num(l).v)%1, l)
for l in lines
)
li_to_pos_index = dict(
(li,i) for i,(pos,li) in enumerate(line_pos)
)
half_circ = len(line_pos)
line_pos = [
pos for (pos, l) in line_pos
] + [
pos+1 for (pos, l) in line_pos
]
full_circ = len(line_pos)
used = [False]*full_circ
def set_used(i1, i2):
i1 = i1 % full_circ
i2 = i1 + (i2-i1) % half_circ
for i in range(i1, i2):
used[i % full_circ] = True
def count_unused(i1, i2):
i1 = i1 % full_circ
i2 = i1 + (i2-i1) % half_circ
res = 0
for i in range(i1, i2):
if not used[i % full_circ]: res += 1
return res
def get_line_dir_indices(l_num, l_dir_sgn, l):
i = li_to_pos_index[l]
if l_dir_sgn is None: return [i, i+half_circ]
pos_real = vector_direction(l_num.v * l_dir_sgn)
if eps_identical((line_pos[i] - pos_real + 1)%2, 1): return [i]
else: return [i+half_circ]
def shorter_seg(a,b):
if (b-a) % full_circ <= half_circ: return a,b
else: return b,a
angle_candidates = [
[
shorter_seg(a,b)
for a in get_line_dir_indices(angle.l1_num, angle.l1_dir_sgn, angle.l1)
for b in get_line_dir_indices(angle.l2_num, angle.l2_dir_sgn, angle.l2)
]
for angle in self.exact_angles[d]
]
angle_candidates.sort(key = len)
#print("candidates", angle_candidates)
for candidates in angle_candidates:
best = min(candidates, key = lambda seg: count_unused(*seg))
set_used(*best)
last_added = None
for i in range(half_circ):
if not used[i] and not used[i+half_circ]:
used[i] = True
last_added = i
if last_added is not None: used[last_added] = False
for i,i_used in enumerate(used):
if not i_used: continue
pos1 = line_pos[i]
pos2 = line_pos[(i+1)%full_circ]
self.vis.visible_exact_angles.append((self.num_obj.a, pos1, pos2))
class NumLineData(NumData):
def __init__(self, vis, line):
NumData.__init__(self, vis, line)
self.dists = []
self.extra_segments = []
def add_dist(self, la,lb, na,nb, col):
self.dists.append((la,lb,na,nb,col))
def distribute_dists(self):
if self.visible is None: points = ()
else: points = self.vis.line_to_points[self.visible]
#print("POINTS", points, "ON", self.visible)
segments = []
for la,lb,na,nb,col in self.dists:
if self.visible is None:
start = 0
elif la in points and lb in points:
start = 0
else: start = 1
pos_a = np.dot(self.num_obj.v, na)
pos_b = np.dot(self.num_obj.v, nb)
if pos_a < pos_b: segments.append((pos_a, pos_b, start, (na,nb,pos_a,pos_b,col)))
else: segments.append((pos_b, pos_a, start, (nb,na,pos_b,pos_a,col)))
point_pos = [np.dot(self.num_obj.v, self.vis.li_to_num(p).a) for p in points]
dists_lev = tuple(distribute_segments(segments, point_pos))
if len(points) <= 1:
if self.is_active: self.colorization = (-np.inf, np.inf, -1),
else: self.colorization = (-np.inf, np.inf, -2),
self.extra_segments = [
(a,b,col,lev)
for (a,b,_,_,col),lev in dists_lev
]
else:
if self.is_active:
active_positions = tuple(
np.dot(self.num_obj.v, self.vis.li_to_num(p).a)
for p in points
)
start_pos = min(active_positions)
end_pos = max(active_positions)
self.colorization = []
cur_pos = None
for (a,b,pos_a,pos_b,col),lev in dists_lev:
if lev == 0:
if cur_pos is None:
if self.is_active:
self.colorization.append((-np.inf, start_pos, -2))
if start_pos < pos_a: self.colorization.append((start_pos, pos_a, -1))
else: self.colorization.append((-np.inf, pos_a, -2))
elif cur_pos < pos_a:
self.colorization.append((cur_pos, pos_a, -2))
self.colorization.append((pos_a,pos_b, col))
cur_pos = pos_b
else: self.extra_segments.append((a,b,col,lev))
if cur_pos is None:
if self.is_active:
self.colorization = [
(-np.inf, start_pos, -2),
(start_pos, end_pos, -1),
(end_pos, np.inf, -2),
]
else: self.colorization = (-np.inf, np.inf, -2),
else:
if self.is_active:
if cur_pos < end_pos:
self.colorization.append((cur_pos,end_pos, -1))
self.colorization.append((end_pos, np.inf, -2))
else: self.colorization.append((cur_pos, np.inf, -2))
self.vis.visible_dists.extend(self.extra_segments)
def find_available_level(self, a, b):
pos_a = np.dot(self.num_obj.v, a)
pos_b = np.dot(self.num_obj.v, b)
blocked = set()
if self.visible is not None: blocked.add(0)
for x,y,col,lev in self.extra_segments:
pos_x = np.dot(self.num_obj.v, x)
pos_y = np.dot(self.num_obj.v, y)
if eps_smaller(pos_x, pos_b) and eps_smaller(pos_a, pos_y):
blocked.add(lev)
lev = 0
while lev in blocked: lev += 1
return lev
class NumCircleData(NumData):
"""
This function technically don't have to be a part of NumCircleData
as we display only arcs that are bound to the displayed circle, but it
fits to the approach to points and lines.
"""
def distribute_arcs(self):
if self.visible is None: return
if self.is_active: circ_color = -1
else: circ_color = -2
points = self.vis.circle_to_points[self.visible]
if len(points) <= 1:
self.colorization = (0,2,circ_color),
return
arcs = self.vis.circle_to_arcs[self.visible]
point_to_pos = dict(
(li, vector_direction(self.vis.li_to_num(li).a
- self.num_obj.c)%2)
for li in points
)
segments = []
for a,b,col in arcs:
pos_a = point_to_pos.get(a, None)
if pos_a is None: continue
pos_b = point_to_pos.get(b, None)
if pos_b is None: continue
if pos_b < pos_a: pos_b += 2
start = 0
segments.append((pos_a, pos_b, start, (pos_a,pos_b,col)))
arc_lev = tuple(distribute_segments_cyc(segments, point_to_pos.values()))
positions = sorted(point_to_pos.values())
out_start = None
out_end = None
if self.is_active:
if eps_bigger(positions[0]+1, positions[-1]):
out_start = positions[-1]
out_end = positions[0]
else:
for pos_a, pos_b in zip(positions, positions[1:]):
if eps_bigger(pos_b, pos_a+1):
out_start = pos_a
out_end = pos_b
break
self.colorization = []
def add_black_arc(pos_a, pos_b):
if eps_identical(pos_a, pos_b): return
#print("ADD_BLACK", pos_a, pos_b)
#print("OUTSIDE", out_start, out_end)
if out_start is not None and eps_bigger((pos_b-pos_a)%2, 1):
#print(" outside")
if not eps_identical(pos_a, out_start): self.colorization.append((pos_a, out_start, -1))
self.colorization.append((out_start, out_end, -2))
if not eps_identical(out_end, pos_b): self.colorization.append((out_end, pos_b, -1))
else:
#print(" inside")
self.colorization.append((pos_a, pos_b, circ_color))
cur_pos = None
for (pos_a, pos_b, col), lev in arc_lev:
if lev == 0:
if cur_pos is not None:
add_black_arc(cur_pos, pos_a)
else: first_pos = pos_a
cur_pos = pos_b
self.colorization.append((pos_a, pos_b, col))
else: self.vis.visible_arcs.append(
(pos_a, pos_b, self.num_obj, col, lev)
)
if cur_pos is None:
if out_start is None: self.colorization = (0,2,circ_color),
else: self.colorization = [
(out_start, out_end, -2),
(out_end, out_start, -1),
]
else: add_black_arc(cur_pos, first_pos)
"""
ArcData and AngleData are not related to NumData, and are places
that could display an angle.
"""
class ArcData:
def __init__(self, vis,p1,p2,c): # arc from p1 to p2 on a circle c
self.vis = vis # backward reference
self.p1 = p1
self.p2 = p2
self.c = c
# activate is called once we decide that the angle is worth showing
# (at least twice in the picture)
def activate(self, color, positive):
if positive: p1,p2 = self.p1, self.p2
else: p1,p2 = self.p2, self.p1
self.vis.circle_to_arcs[self.c].append((p1, p2, color))
# to count how many different places of that angle are in the picture
# (we don't display the angle if only one)
def get_repr(self):
if self.p1 < self.p2: return self.p1,self.p2,self.c
else: return self.p2,self.p1,self.c
class AngleData:
def __init__(self, vis, lines, dir_reqs):
self.vis = vis
self.l1, self.l2 = lines
self.dr1, self.dr2 = dir_reqs
self.positive = True
def require_lines(self): # angle encourage display of extra helper lines
self.vis.line_to_angle_number[self.l1] += 1
self.vis.line_to_angle_number[self.l2] += 1
# activate is called once we decide that we the angle is worth showing
# (at least twice in the picture)
def activate(self, color, positive):
self.color = color
if self.positive != positive:
self.l1, self.l2 = self.l2, self.l1
self.dr1, self.dr2 = self.dr2, self.dr1
self.positive = positive
self.vis.angles.append(self)
self.require_lines()
# to count how many different places of that angle are in the picture
# (we don't display the angle if only one)
def get_repr(self):
return tuple(sorted((self.l1, self.l2)))
# now we know that the angle will be displayed
def add_to_point(self):
self.p = Point(intersection_ll(
self.vis.li_to_num(self.l1),
self.vis.li_to_num(self.l2),
))
self.l1_num = self.vis.li_to_num_data[self.l1].num_obj
self.l2_num = self.vis.li_to_num_data[self.l2].num_obj
# get orientation: self.l1_dir_sgn, self.l2_dir_sgn
self.l1_dir_sgn = None
self.l2_dir_sgn = None
if self.dr1 is not None or self.dr2 is not None:
if self.dr1 is not None:
p1,p2 = self.dr1
if np.dot(self.vis.li_to_num(p2).a - self.vis.li_to_num(p1).a, self.l1_num.v) > 0:
self.l1_dir_sgn = 1
else: self.l1_dir_sgn = -1
if self.dr2 is not None:
p1,p2 = self.dr2
if np.dot(self.vis.li_to_num(p2).a - self.vis.li_to_num(p1).a, self.l2_num.v) > 0:
self.l2_dir_sgn = 1
else: self.l2_dir_sgn = -1
if self.color >= 0:
if (vector_direction(self.l2_num.v) - vector_direction(self.l1_num.v))%2 <= 1:
relative_sgn = 1
else: relative_sgn = -1
if self.l2_dir_sgn is None: self.l2_dir_sgn = relative_sgn * self.l1_dir_sgn
elif self.l1_dir_sgn is None: self.l1_dir_sgn = relative_sgn * self.l2_dir_sgn
elif self.l1_dir_sgn != self.l2_dir_sgn * relative_sgn:
self.l1_dir_sgn = None
self.l2_dir_sgn = None
self.vis.get_num_point(self.p).add_angle(self)
# select l1_dir if it was not forced by previous calculation
def find_nicer_l1_dir(self):
points = self.vis.line_to_points[self.l1]
if not points:
self.l1_dir_sgn = 1
return
positions = [
np.dot(self.vis.li_to_num(p).a, self.l1_num.v)
for p in points
]
min_pos = min(positions)
max_pos = max(positions)
pos = np.dot(self.p.a, self.l1_num.v)
if eps_bigger(pos - min_pos, max_pos - pos): self.l1_dir_sgn = -1
else: self.l1_dir_sgn = 1
# given the computed angle orientation, return it as an interval mod 2
def find_arc(self):
self.pos_a = vector_direction(self.l1_num.v * self.l1_dir_sgn)%2
self.pos_b = vector_direction(self.l2_num.v)%2
if (self.pos_b - self.pos_a)%2 > 1:
if self.pos_b >= 1: self.pos_b -= 1
else: self.pos_b += 1
if self.pos_b < self.pos_a: self.pos_b += 2
### The main class
"""
KnowledgeVisualisation investigates the logical core upon calling
method refresh(). For some changes that do not change the showed
objects, only visual details (selection, label), it is sufficient
to call self.visible_export().
"""
class KnowledgeVisualisation:
def __init__(self, env):
# primary data
self.env = env
self.tools = env.tools
self.logic = None
self.gi_to_li_l = None
# alternative modes
self.move_mode = False
self.show_all_mode = False
self.ambi_select_mode = False
# set to true upon export, or highlight change
self.view_changed = False
# permanent data
self.gi_to_priority = []
self.gi_to_hidden = []
self.gi_label_show = []
self.gi_label_position = []
# label position =
# Point: np.array([x,y])
# Line: position (float 0 to 1), offset (pixels)
# Circle: direction (float mod 2), offset (pixels)
# extra elements to the main construction
self.hl_proposals = ()
self.hl_helpers = ()
# when selection changes, StepList wants to recognize it too
self.update_selected_hook = lambda: None
# static dictionary
self.angle_label_to_larg = {
self.tools.angle_ll : (True, True),
self.tools.angle_ppl : (False, True),
self.tools.angle_lpp : (True, False),
self.tools.angle_pppp : (False, False),
}
########
#
# The __init__ function could end here, all the following
# data are constructed from scratch by every refresh
#
########
## exported data for GTool (selection)
self.selectable_points = []
self.selectable_lines = []
self.selectable_circles = []
## exported data for Vievport (visualisation)
self.visible_points_numer = []
self.active_lines_numer = []
self.extra_lines_numer = []
self.active_circles_numer = []
self.extra_circles_numer = []
self.visible_angles = []
self.visible_arcs = []
self.visible_dists = []
self.visible_exact_angles = []
self.visible_labels = []
self.visible_parallels = []
## data for internal usage
# numerical representation
self.num_points_d = dict()
self.num_lines_d = dict()
self.num_circles_d = dict()
self.num_points = ()
self.num_lines = ()
self.num_circles = ()
# gi <-> li <-> data
self.li_to_gi_first = dict()
self.li_to_gi_last = dict()
self.li_to_num_data = dict()
self.li_to_gi_movable = dict()
# lies_on data
self.line_to_points = defaultdict(list)
self.circle_to_points = defaultdict(list)
# angle data
self.angles = []
self.circle_to_arcs = defaultdict(list)
self.line_to_angle_number = defaultdict(int)
# logical indices which were decided to be shown
self.visible_points = set()
self.visible_lines = set()
self.visible_circles = set()
# update number / status of GUI indices
def add_gis(self, number):
self.gi_to_priority += [2]*number
self.gi_to_hidden += [False]*number
self.gi_label_show += [False]*number
self.gi_label_position += [None]*number
def truncate_gis(self, number):
del self.gi_to_priority[number:]
del self.gi_to_hidden[number:]
del self.gi_label_show[number:]
del self.gi_label_position[number:]
# helper functions for translating indices
# gi -> li -> num -> data
def li_root(self, li):
return self.logic.ufd.obj_to_root(li)
def gi_to_li(self, gi): # graphical index to logic index
li = self.gi_to_li_l[gi]
if li is None: return None
return self.li_root(li)
def li_to_num(self, li): # logic index to numerical object
return self.logic.num_model[li]
def li_to_type(self, li): # logic index to type
return self.logic.obj_types[li]
def gi_to_num(self, gi): # graphical index to numerical object
li = self.gi_to_li_l[gi]
if li is None: return None
return self.li_to_num(self.li_root(li))
def gi_to_type(self, gi): # graphical index to numerical object
li = self.gi_to_li_l[gi]
if li is None: return None
return self.li_to_type(self.li_root(li))
## Assigning NumData to numerical objects
def _get_num_obj(self, d, l, constructor, obj, keys):
for key in keys:
res = d.get(key, None)
if res is not None: return res
value = constructor(self, obj)
l.append(value)
for key in keys: d[key] = value
return value
def get_num_point(self, point):
x0,y0 = map(int, np.floor(point.a / epsilon))
keys = tuple(
(x,y) for x in (x0,x0+1) for y in (y0,y0+1)
)
return self._get_num_obj(
self.num_points_d, self.num_points,
NumPointData, point, keys
)
def get_num_line(self, line):
x0,y0,c0 = map(int, np.floor(line.data / epsilon))
keys = tuple(
(x*s,y*s,c*s)
for x in (x0,x0+1) for y in (y0,y0+1)
for c in (c0,c0+1) for s in (1, -1)
)
return self._get_num_obj(
self.num_lines_d, self.num_lines,
NumLineData, line, keys,
)
def get_num_circle(self, circle):
x0,y0,r0 = map(int, np.floor(circle.data / epsilon))
keys = tuple(
(x,y,r)
for x in (x0,x0+1) for y in (y0,y0+1)
for r in (r0,r0+1)
)
return self._get_num_obj(
self.num_circles_d, self.num_circles,
NumCircleData, circle, keys,
)
def get_num_data(self, obj):
if isinstance(obj, Point): return self.get_num_point(obj)
elif isinstance(obj, Line): return self.get_num_line(obj)
elif isinstance(obj, Circle): return self.get_num_circle(obj)
else: raise Exception("Unexpected type {}".format(type(obj)))
## steps of refresh
def update_rev_links(self):
self.li_to_gi_movable = dict()
self.li_to_gi_first = dict()
self.li_to_gi_last = dict()
self.li_to_num_data = dict()
self.num_points_d = dict()
self.num_lines_d = dict()
self.num_circles_d = dict()
self.num_points = []
self.num_lines = []
self.num_circles = []
for gi,li in enumerate(self.gi_to_li_l):
if self.gi_to_hidden[gi] and not self.show_all_mode: continue
if li is None: continue
if self.li_to_type(li) not in (Point, Line, Circle):
continue
li = self.li_root(li)
self.li_to_gi_last[li] = gi
gi_ori = self.li_to_gi_first.setdefault(li, gi)
if gi is gi_ori:
num = self.li_to_num(li)
num_data = self.get_num_data(num)
self.li_to_num_data[li] = num_data
step = self.env.gi_to_step(gi)
if isinstance(step.tool, MovableTool):
self.li_to_gi_movable.setdefault(li, gi)
for li, gi in self.li_to_gi_last.items():
num_data = self.li_to_num_data[li]
priority = self.gi_to_priority[gi]
num_data.add_active_candidate(li, (priority,gi))
# Determining which objects should be shown (especially among ambiguous ones)
def select_visible_objs(self, objs):
visible_objs = set()
for num_data in objs:
visible = num_data.select_visible()
if visible is not None: visible_objs.add(visible)
return visible_objs
def select_visible_points(self):
self.visible_points = self.select_visible_objs(self.num_points)
def select_visible_lines(self):
self.visible_lines = self.select_visible_objs(self.num_lines)
def select_visible_circles(self):
self.visible_circles = self.select_visible_objs(self.num_circles)
# helper function for extract_knowledge
def get_angle_lines(self, angle_label, args):
res = []
dir_reqs = []
arg_it = iter(args)
for is_larg in self.angle_label_to_larg[angle_label]:
if is_larg:
l = next(arg_it)
if l not in self.li_to_gi_first and len(self.line_to_points[l]) < 2:
return None
dir_req = None
else:
p1 = next(arg_it)
p2 = next(arg_it)
if p1 not in self.li_to_gi_first or p2 not in self.li_to_gi_first:
return None
l, = self.logic.ufd.get(self.tools.line, (p1,p2))
dir_req = (p1,p2)
res.append(l)
dir_reqs.append(dir_req)
return res, dir_reqs
# Go through the logical core database, and save the relevant data
# to internal attributes
def extract_knowledge(self):
self.line_to_points = defaultdict(list)
self.circle_to_points = defaultdict(list)
self.circle_to_arcs = defaultdict(list)
self.angles = []
self.direction_to_line = defaultdict(list)
direction_q_to_line = defaultdict(list)
self.line_to_angle_number = defaultdict(int)
self.visible_parallels = []
lies_on_labels = self.tools.lies_on_l, self.tools.lies_on_c
dist_to_points = defaultdict(list)
angle_to_data = defaultdict(list)
# points on lines and circles
for (label, args), out in self.logic.ufd.data.items():
if label is self.tools.lies_on_l:
p,l = args
if p in self.li_to_gi_first: self.line_to_points[l].append(p)
elif label is self.tools.lies_on_c:
p,c = args
if p in self.li_to_gi_first: self.circle_to_points[c].append(p)
zero_angle = Angle(0)
for (label, args), out in self.logic.ufd.data.items():
if label is self.tools.arc_length: #### arc length
p1,p2,c = args
if p1 not in self.li_to_gi_first or p2 not in self.li_to_gi_first:
continue
a, = out
if self.logic.angles.has_exact_difference(a, self.logic.exact_angle):
continue
angle_to_data[a].append(ArcData(self, p1,p2,c))
elif label is self.tools.dist: #### distance
p1,p2 = args
if p1 >= p2: continue
if p1 not in self.li_to_gi_first or p2 not in self.li_to_gi_first:
continue
d, = out
dist_to_points[d].append((p1,p2))
elif label is self.tools.direction_of: #### exact angles
l, = args
if l not in self.li_to_gi_first: continue
d, = out
self.direction_to_line[d].append(l)
direction_q_to_line[self.logic.angles.equal_to[d][0]].append((l,d))
elif label in self.tools.angle_tools: #### angles
lines_dr = self.get_angle_lines(label, args)
if lines_dr is None: continue
a, = out
num = self.li_to_num(a)
if num.identical_to(zero_angle): continue
lines, dir_req = lines_dr
angle_to_data[a].append(AngleData(self, lines, dir_req))
#### store repeated dists
dist_num = 0
for d, l in sorted(dist_to_points.items(), key = lambda x: x[0]):
if len(l) <= 1: continue
for a,b in l:
if a not in self.visible_points or b not in self.visible_points:
continue
na = self.li_to_num(a).a
nb = self.li_to_num(b).a
line = line_passing_np_points(na, nb)
self.get_num_line(line).add_dist(
a,b, na,nb, dist_num
)
dist_num += 1
#### store repeated angles / arcs
angle_num = 0
used_compl = set()
for a, l in sorted(angle_to_data.items(), key = lambda x: x[0]):
if a in used_compl: continue
if self.logic.angles.has_exact_difference(a, self.logic.exact_angle):
for data in l: data.activate(-1, True)
else:
c = self.li_root(self.logic.angles.get_complement(a))
if c is not None:
used_compl.add(c)
cl = angle_to_data[c]
else: cl = []
l_reprs = set(x.get_repr() for x in l)
cl = [x for x in cl if x.get_repr() not in l_reprs]
if len(l) + len(cl) <= 1: continue
positive = self.li_to_num(a).data % 1 <= 0.5
for data in l: data.activate(angle_num, positive)
for data in cl: data.activate(angle_num, not positive)
angle_num += 1
#### store additional exact angles
for d,lines in direction_q_to_line.items():
for (l1,d1),(l2,d2) in itertools.combinations(lines, 2):
if d1 == d2: continue
p = Point(intersection_ll(
self.li_to_num(l1),
self.li_to_num(l2))
)
point_data = self.get_num_point(p)
point_data.add_exact(l1,d)
point_data.add_exact(l2,d)
#### Filter out hidden points
for line, points in self.line_to_points.items():
points = [p for p in points if p in self.visible_points]
self.line_to_points[line] = points
for circle, points in self.circle_to_points.items():
points = [p for p in points if p in self.visible_points]
self.circle_to_points[circle] = points
# clines to be shown gray and dashed
def find_extra_clines(self):
lines = set(self.line_to_points.keys()) | set(self.line_to_angle_number.keys())
for line in lines:
if line in self.li_to_gi_first: continue
points_number = len(self.line_to_points[line])
angle_number = self.line_to_angle_number[line]
if points_number < 3 and not angle_number: continue
if points_number < 2: continue
num_line = self.li_to_num(line)
num_data = self.get_num_line(num_line)
self.li_to_num_data[line] = num_data
num_data.add_extra_candidate(line, (angle_number, points_number))
#assert(all(self.li_to_type(c) == Circle for c in self.circle_to_points.keys()))
#assert(all(self.li_to_type(c) == Circle for c in self.circle_to_arcs.keys()))
circles = set(self.circle_to_points.keys()) | set(self.circle_to_arcs.keys())
for circle in circles:
if circle in self.li_to_gi_first: continue
points_number = len(self.circle_to_points[circle])
arcs_number = len(self.circle_to_arcs[circle])
if arcs_number < 1 and points_number < 4: continue
num_circle = self.li_to_num(circle)
num_data = self.get_num_circle(num_circle)
self.li_to_num_data[circle] = num_data
num_data.add_extra_candidate(circle, (arcs_number, points_number))
def find_parallels(self):
parallel_num = 0
for d,lines in sorted(self.direction_to_line.items(), key = lambda x: x[0]):
if len(lines) < 2: continue
lines = [l for l in lines if l in self.visible_lines]
if len(lines) < 2: continue
for l in lines:
self.visible_parallels.append((self.li_to_num(l), parallel_num))
parallel_num += 1
def li_is_visible(self, li):
num_data = self.li_to_num_data.get(li, None)
if num_data is None or num_data.visible is None: return False
return num_data.visible == li
# do not show an angle if one of its corresponding lines is not shown
def filter_visible_angles(self):
self.line_to_angle_number = defaultdict(int)
used_angles = []
for a in self.angles:
if a.l1 in self.visible_lines and a.l2 in self.visible_lines:
used_angles.append(a)
a.require_lines()
a.add_to_point()
self.angles = used_angles
# remove redundant lines
for line_data in self.num_lines:
if line_data.visible is None or line_data.is_active: continue
li = line_data.visible
if len(self.line_to_points[li]) < 3 and li not in self.line_to_angle_number:
continue
def distribute_dists(self):
self.visible_dists = []
for line_data in self.num_lines:
line_data.distribute_dists()
def distribute_arcs(self):
self.visible_arcs = []
for circle_data in self.num_circles:
circle_data.distribute_arcs()
def distribute_angles(self):
self.visible_angles = []
self.visible_exact_angles = []
for point_data in self.num_points:
point_data.distribute_angles()
point_data.distribute_exact_angles()
# helper functions for visible_export
def is_ambiguous(self, li):
num_data = self.li_to_num_data[li]
if len(num_data.active_candidates) > 1: return True
else: return False
def is_movable(self, li):
return li in self.li_to_gi_movable
def is_move_mode(self):
return self.move_mode and not self.ambi_select_mode
def obj_color(self, li):
if self.ambi_select_mode and self.is_ambiguous(li): return 1
elif self.move_mode and self.is_movable(li): return 2
else: return 0
def is_selectable(self, li):
if self.ambi_select_mode: return self.is_ambiguous(li)
elif self.is_move_mode(): return self.is_movable(li)
else: return True
def get_label_position(self, gi):
position = self.gi_label_position[gi]
if position is None: # default position
num_obj = self.gi_to_num(gi)
if isinstance(num_obj, Point): position = np.array((0, -20))
elif isinstance(num_obj, Circle): position = (0, 20)
elif isinstance(num_obj, Line): position = (0.1, 20)
else: raise Exception("Unexpected type {}".format(type(num_obj)))
self.gi_label_position[gi] = position
return position
# create data for Viewport and GTool
def visible_export(self):
self.view_changed = True
self.visible_labels = []
def add_label(obj):
gi = self.li_to_gi_first[obj]
if not self.gi_label_show[gi]: return
num_obj = self.li_to_num(obj)
self.visible_labels.append((
self.env.gi_to_name[gi], num_obj, self.get_label_position(gi),
))
point_li_to_vi = dict()
self.visible_points_numer = []
for point in self.visible_points:
point_li_to_vi[point] = len(self.visible_points_numer)
self.visible_points_numer.append((
self.li_to_num(point),
self.obj_color(point),
self.obj_is_selected.get(point, None),
))
add_label(point)
self.active_lines_numer = []
self.extra_lines_numer = []
self.active_circles_numer = []
self.extra_circles_numer = []
self.selectable_lines = []
self.selectable_circles = []
if self.is_move_mode():
li_to_select = self.li_to_gi_movable
else: li_to_select = self.li_to_gi_first
self.selectable_points = [
(li_to_select[li], self.li_to_num(li))
for li in self.visible_points
if self.is_selectable(li)
]
def export_numer(data_list, to_points):
active = []
extra = []
selectable = []
for num_data in data_list:
obj = num_data.visible
if obj is None: continue
points = [
point_li_to_vi[point]
for point in to_points[obj]
]
extras = (
num_data.colorization,
self.obj_is_selected.get(obj, None),
self.obj_color(obj),
points,
)
exported = num_data.num_obj, extras
if num_data.is_active:
add_label(obj)
active.append(exported)
if self.is_selectable(obj):
selectable.append((
li_to_select[obj],
#num_data.num_obj,
self.li_to_num(obj),
))
else: extra.append(exported)
return active, extra, selectable
self.active_lines_numer, self.extra_lines_numer, self.selectable_lines\
= export_numer(self.num_lines, self.line_to_points)
self.active_circles_numer, self.extra_circles_numer, self.selectable_circles\
= export_numer(self.num_circles, self.circle_to_points)
def refresh(self):
self.update_rev_links()
self.select_visible_points()
# lies_on, dist, arc_length
# relevant to visible points / top_level objects
self.extract_knowledge()
self.find_extra_clines()
self.select_visible_lines()
self.select_visible_circles()
self.find_parallels()
self.filter_visible_angles()
self.distribute_dists()
self.distribute_arcs()
self.distribute_angles()
self.visible_export()
## Highlights / selection / noticing change in the view
def update_hl_selected(self, selected):
obj_is_selected = dict()
for obj in selected:
if isinstance(obj, tuple):
data = obj[1:]
obj = obj[0]
else: data = True
if obj >= len(self.gi_to_li_l): continue
li = self.gi_to_li(obj)
if li is None: continue
obj_is_selected[li] = data
# check change
if not self.view_changed:
if self.obj_is_selected == obj_is_selected: return
self.obj_is_selected = obj_is_selected
self.visible_export()
self.update_selected_hook()
def update_hl_proposals(self, proposals):
if not self.view_changed and len(proposals) == len(self.hl_proposals):
if all(
prev_prop.identical_to(prop)
for prev_prop, prop in zip(self.hl_proposals, proposals)
): return
self.view_changed = True
self.hl_proposals = proposals
def update_hl_helpers(self, helpers):
# push segments away from lines
hl_helpers = []
hl_helpers_ori = iter(self.hl_helpers)
def next_ori():
try:
return next(hl_helpers_ori)
except StopIteration:
return None
for helper in helpers:
if isinstance(helper, GeoObject):
if not helper.identical_to(next_ori()): self.view_changed = True
hl_helpers.append(helper)
else:
a,b = helper
if eps_identical(a,b): continue
hl_helper_ori = next_ori()
if not isinstance(hl_helper_ori, tuple): self.view_changed = True
else:
ori_a,ori_b,ori_lev = hl_helper_ori
if not eps_identical(ori_a, a) or not eps_identical(ori_b, b):
self.view_changed = True
line_data = self.get_num_line(line_passing_np_points(a,b))
v = line_data.num_obj.v
if np.dot(v,a) > np.dot(v,b): a,b = b,a
lev = line_data.find_available_level(a,b)
hl_helpers.append((a,b,lev))
if self.view_changed: self.hl_helpers = hl_helpers
## functions for external manipulation
def swap_priorities(self, gi, direction):
li = self.gi_to_li(gi)
candidates = self.li_to_num_data[li].active_candidates
if len(candidates) <= 1:
print("Unambiguous object")
return
candidates = sorted(
( cand for _, cand in candidates ),
key = lambda cand: self.li_to_gi_last[cand],
)
#print(li, candidates)
i = candidates.index(li)
i2 = (i + direction) % len(candidates)
#print("{} -> {}".format(i, i2))
for li in candidates[i2+1:i+1]:
self.gi_to_priority[self.li_to_gi_last[li]] = 1
self.gi_to_priority[self.li_to_gi_last[candidates[i2]]] = 2
#print([self.gi_to_priority[self.li_to_gi_last[x]] for x in candidates])
self.refresh()
def hide(self, gi):
li = self.gi_to_li(gi)
gi_first = self.li_to_gi_first[li]
gi_last = self.li_to_gi_last[li]
print("hide li {} in range {} -- {}".format(li, gi_first, gi_last))
for gi in range(gi_first, gi_last+1):
if self.gi_to_li(gi) == li:
print(" hide gi {}".format(gi))
self.gi_to_hidden[gi] = True
self.refresh()
def set_logic(self, logic, gi_to_li):
self.logic = logic
self.gi_to_li_l = gi_to_li