Ejemplo n.º 1
0
    def update_arrows(self):
        self.cache_arrows = []
        if self.is_curve and self.style[1]:
            stroke = self.style[1]
            arrs = stroke[9]
            if not arrs:
                return
            for path in self.paths:
                if path[-1] == sk2const.CURVE_CLOSED:
                    self.cache_arrows.append([])
                    continue
                tr = libgeom.multiply_trafo
                coef = self.cache_line_width
                end = start = None
                stroke_trafo = [coef, 0.0, 0.0, coef, 0.0, 0.0]

                def get_vector(path, trafo):
                    p0 = p1 = path[0]
                    is_cp = libgeom.is_curve_point
                    t = 0.001
                    for point in path[1]:
                        p0 = point[0] if is_cp(point) else point
                        if not p0 == p1:
                            break
                        elif is_cp(point) and p1 != point[2]:
                            p0 = libgeom.split_bezier_curve(p1, point, t)[0][2]
                            break
                    return [
                        libgeom.apply_trafo_to_point(p, trafo)
                        for p in (p0, p1)
                    ]

                # end arrow
                if isinstance(arrs[0], int):
                    p0, p1 = get_vector(path, self.trafo)
                    angle = libgeom.get_point_angle(p1, p0)
                    end_trafo = tr(stroke_trafo, libgeom.trafo_rotate(angle))
                    trafo = [1.0, 0.0, 0.0, 1.0, p1[0], p1[1]]
                    end_trafo = tr(end_trafo, trafo)
                    end = arrows.get_arrow_cpath(arrs[0], end_trafo)
                # start arrow
                if isinstance(arrs[1], int):
                    p0, p1 = get_vector(libgeom.reverse_path(path), self.trafo)
                    angle = libgeom.get_point_angle(p1, p0)
                    start_trafo = tr(stroke_trafo, libgeom.trafo_rotate(angle))
                    trafo = [1.0, 0.0, 0.0, 1.0, p1[0], p1[1]]
                    start_trafo = tr(start_trafo, trafo)
                    start = arrows.get_arrow_cpath(arrs[1], start_trafo)
                self.cache_arrows.append([end, start])
Ejemplo n.º 2
0
 def update_arrows(self):
     self.cache_arrows = []
     if self.is_curve and self.style[1]:
         stroke = self.style[1]
         arrs = stroke[9]
         if not arrs:
             return
         for path in self.paths:
             if path[-1] == sk2const.CURVE_CLOSED:
                 self.cache_arrows.append([])
                 continue
             tr = libgeom.multiply_trafo
             coef = self.cache_line_width
             end = start = None
             # end arrow
             if isinstance(arrs[0], int):
                 end_trafo = [coef, 0.0, 0.0, coef, 0.0, 0.0]
                 p1 = libgeom.apply_trafo_to_point(path[0], self.trafo)
                 p0 = libgeom.apply_trafo_to_point(path[1][0], self.trafo)
                 if libgeom.is_curve_point(p0):
                     p0 = p0[0]
                 angle = libgeom.get_point_angle(p1, p0)
                 end_trafo = tr(end_trafo, libgeom.trafo_rotate(angle))
                 trafo = [1.0, 0.0, 0.0, 1.0, p1[0], p1[1]]
                 end_trafo = tr(end_trafo, trafo)
                 end = arrows.get_arrow_cpath(arrs[0], end_trafo)
             #start arrow
             if isinstance(arrs[1], int):
                 start_trafo = [coef, 0.0, 0.0, coef, 0.0, 0.0]
                 p1 = libgeom.apply_trafo_to_point(path[1][-1], self.trafo)
                 if len(path[1]) == 1:
                     p0 = libgeom.apply_trafo_to_point(path[0], self.trafo)
                 else:
                     if libgeom.is_curve_point(p1):
                         p0 = p1[1]
                         p1 = p1[2]
                     else:
                         p0 = libgeom.bezier_base_point(path[1][-2])
                         p0 = libgeom.apply_trafo_to_point(p0, self.trafo)
                 angle = libgeom.get_point_angle(p1, p0)
                 start_trafo = tr(start_trafo, libgeom.trafo_rotate(angle))
                 trafo = [1.0, 0.0, 0.0, 1.0, p1[0], p1[1]]
                 start_trafo = tr(start_trafo, trafo)
                 start = arrows.get_arrow_cpath(arrs[1], start_trafo)
             self.cache_arrows.append([end, start])
Ejemplo n.º 3
0
 def translate_ellipse(self, obj, cfg):
     cx = obj.center_x
     cy = obj.center_y
     rx = obj.radius_x
     ry = obj.radius_y
     props = dict(
         rect=[cx - rx, cy - ry, 2.0 * rx, 2.0 * ry],
         style=self.get_style(obj),
     )
     new_obj = sk2_model.Circle(cfg, **props)
     trafo_rotate = libgeom.trafo_rotate(-obj.angle, cx, cy)
     trafo = libgeom.multiply_trafo(new_obj.trafo, trafo_rotate)
     new_obj.trafo = libgeom.multiply_trafo(trafo, self.trafo)
     return new_obj
Ejemplo n.º 4
0
    def translate_text(self, obj, cfg):
        trafo_rotate = libgeom.trafo_rotate(obj.angle)
        base_point = libgeom.apply_trafo_to_point([obj.x, obj.y], self.trafo)
        base_point[1] -= obj.font_size
        trafo = [72.0 / 90.0, 0.0, 0.0, 72.0 / 90.0] + base_point
        trafo = libgeom.multiply_trafo(trafo_rotate, trafo)
        text_style = self.get_text_style(obj)
        props = dict(point=[0.0, obj.font_size],
                     style=text_style,
                     text=obj.string,
                     trafo=trafo)
        new_obj = sk2_model.Text(cfg, **props)

        txt_length = len(obj.string)
        tags = []
        new_obj.markup = [[tags, (0, txt_length)]]
        return new_obj
Ejemplo n.º 5
0
    def _calc_bottom_left_rotate_trafo(self, event):
        is_centering = not event.is_shift()
        is_constraining = event.is_ctrl()

        start_point = self.canvas.win_to_doc(self.start)
        end_point = self.canvas.win_to_doc(self.end)
        bbox = self.presenter.selection.bbox

        if is_centering:
            bbox_center = libgeom.bbox_center(bbox)
            center_offset = self.selection.center_offset
            center = libgeom.add_points(bbox_center, center_offset)
        else:
            center = libgeom.bbox_points(bbox)[3]

        a1 = libgeom.get_point_angle(start_point, center)
        a2 = libgeom.get_point_angle(end_point, center)
        angle = a2 - a1
        if is_constraining:
            step = math.radians(config.rotation_step)
            angle = (angle + step / 2.0) // step * step

        return libgeom.trafo_rotate(angle, center[0], center[1])
Ejemplo n.º 6
0
def parse_svg_path_cmds(pathcmds):
    index = 0
    last = None
    last_index = 0
    cmds = []
    pathcmds = re.sub('  *', ' ', pathcmds)
    for item in pathcmds:
        if item in 'MmZzLlHhVvCcSsQqTtAa':
            if last:
                coords = parse_svg_coords(pathcmds[last_index + 1:index])
                cmds.append((last, coords))
            last = item
            last_index = index
        index += 1

    coords = parse_svg_coords(pathcmds[last_index + 1:index])
    cmds.append([last, coords])

    paths = []
    path = []
    cpoint = []
    rel_flag = False
    last_cmd = 'M'
    last_quad = None

    for cmd in cmds:
        if cmd[0] in 'Mm':
            if path: paths.append(path)
            path = deepcopy(PATH_STUB)
            rel_flag = cmd[0] == 'm'
            points = [cmd[1][i:i + 2] for i in range(0, len(cmd[1]), 2)]
            for point in points:
                if cpoint and rel_flag:
                    point = add_points(base_point(cpoint), point)
                if not path[0]:
                    path[0] = point
                else:
                    path[1].append(point)
                cpoint = point
        elif cmd[0] in 'Zz':
            p0 = [] + base_point(cpoint)
            p1 = [] + path[0]
            if not libgeom.is_equal_points(p0, p1, 8):
                path[1].append([] + path[0])
            path[2] = sk2const.CURVE_CLOSED
            cpoint = [] + path[0]
        elif cmd[0] in 'Cc':
            rel_flag = cmd[0] == 'c'
            points = [cmd[1][i:i + 2] for i in range(0, len(cmd[1]), 2)]
            points = [points[i:i + 3] for i in range(0, len(points), 3)]
            for point in points:
                if rel_flag:
                    point = [
                        add_points(base_point(cpoint), point[0]),
                        add_points(base_point(cpoint), point[1]),
                        add_points(base_point(cpoint), point[2])
                    ]
                qpoint = [] + point
                qpoint.append(sk2const.NODE_CUSP)
                path[1].append(qpoint)
                cpoint = point
        elif cmd[0] in 'Ll':
            rel_flag = cmd[0] == 'l'
            points = [cmd[1][i:i + 2] for i in range(0, len(cmd[1]), 2)]
            for point in points:
                if rel_flag:
                    point = add_points(base_point(cpoint), point)
                path[1].append(point)
                cpoint = point
        elif cmd[0] in 'Hh':
            rel_flag = cmd[0] == 'h'
            for x in cmd[1]:
                dx, y = base_point(cpoint)
                if rel_flag:
                    point = [x + dx, y]
                else:
                    point = [x, y]
                path[1].append(point)
                cpoint = point
        elif cmd[0] in 'Vv':
            rel_flag = cmd[0] == 'v'
            for y in cmd[1]:
                x, dy = base_point(cpoint)
                if rel_flag:
                    point = [x, y + dy]
                else:
                    point = [x, y]
                path[1].append(point)
                cpoint = point
        elif cmd[0] in 'Ss':
            rel_flag = cmd[0] == 's'
            points = [cmd[1][i:i + 2] for i in range(0, len(cmd[1]), 2)]
            points = [points[i:i + 2] for i in range(0, len(points), 2)]
            for point in points:
                q = cpoint
                p = cpoint
                if len(cpoint) > 2:
                    q = cpoint[1]
                    p = cpoint[2]
                p1 = sub_points(add_points(p, p), q)
                if rel_flag:
                    p2 = add_points(base_point(cpoint), point[0])
                    p3 = add_points(base_point(cpoint), point[1])
                else:
                    p2, p3 = point
                point = [p1, p2, p3]
                qpoint = [] + point
                qpoint.append(sk2const.NODE_CUSP)
                path[1].append(qpoint)
                cpoint = point

        elif cmd[0] in 'Qq':
            rel_flag = cmd[0] == 'q'
            groups = [cmd[1][i:i + 4] for i in range(0, len(cmd[1]), 4)]
            for vals in groups:
                p = base_point(cpoint)
                if rel_flag:
                    q = add_points(p, [vals[0], vals[1]])
                    p3 = add_points(p, [vals[2], vals[3]])
                else:
                    q = [vals[0], vals[1]]
                    p3 = [vals[2], vals[3]]
                p1 = add_points(mult_point(p, F13), mult_point(q, F23))
                p2 = add_points(mult_point(p3, F13), mult_point(q, F23))

                point = [p1, p2, p3]
                qpoint = [] + point
                qpoint.append(sk2const.NODE_CUSP)
                path[1].append(qpoint)
                cpoint = point
                last_quad = q

        elif cmd[0] in 'Tt':
            rel_flag = cmd[0] == 't'
            groups = [cmd[1][i:i + 2] for i in range(0, len(cmd[1]), 2)]
            if last_cmd not in 'QqTt' or last_quad is None:
                last_quad = base_point(cpoint)
            for vals in groups:
                p = base_point(cpoint)
                q = sub_points(mult_point(p, 2.0), last_quad)
                if rel_flag:
                    p3 = add_points(p, [vals[0], vals[1]])
                else:
                    p3 = [vals[0], vals[1]]
                p1 = add_points(mult_point(p, F13), mult_point(q, F23))
                p2 = add_points(mult_point(p3, F13), mult_point(q, F23))

                point = [p1, p2, p3]
                qpoint = [] + point
                qpoint.append(sk2const.NODE_CUSP)
                path[1].append(qpoint)
                cpoint = point
                last_quad = q

        elif cmd[0] in 'Aa':
            rel_flag = cmd[0] == 'a'
            arcs = [cmd[1][i:i + 7] for i in range(0, len(cmd[1]), 7)]

            for arc in arcs:
                cpoint = base_point(cpoint)
                rev_flag = False
                rx, ry, xrot, large_arc_flag, sweep_flag, x, y = arc
                rx = abs(rx)
                ry = abs(ry)
                if rel_flag:
                    x += cpoint[0]
                    y += cpoint[1]
                if cpoint == [x, y]: continue
                if not rx or not ry:
                    path[1].append([x, y])
                    continue

                vector = [[] + cpoint, [x, y]]
                if sweep_flag:
                    vector = [[x, y], [] + cpoint]
                    rev_flag = True
                cpoint = [x, y]

                dir_tr = libgeom.trafo_rotate_grad(-xrot)

                if rx > ry:
                    tr = [1.0, 0.0, 0.0, rx / ry, 0.0, 0.0]
                    r = rx
                else:
                    tr = [ry / rx, 0.0, 0.0, 1.0, 0.0, 0.0]
                    r = ry

                dir_tr = libgeom.multiply_trafo(dir_tr, tr)
                vector = libgeom.apply_trafo_to_points(vector, dir_tr)

                l = libgeom.distance(*vector)

                if l > 2.0 * r: r = l / 2.0

                mp = libgeom.midpoint(*vector)

                tr0 = libgeom.trafo_rotate(math.pi / 2.0, mp[0], mp[1])
                pvector = libgeom.apply_trafo_to_points(vector, tr0)

                k = math.sqrt(r * r - l * l / 4.0)
                if large_arc_flag:
                    center = libgeom.midpoint(mp, pvector[1], 2.0 * k / l)
                else:
                    center = libgeom.midpoint(mp, pvector[0], 2.0 * k / l)

                angle1 = libgeom.get_point_angle(vector[0], center)
                angle2 = libgeom.get_point_angle(vector[1], center)

                da = angle2 - angle1
                start = angle1
                end = angle2
                if large_arc_flag:
                    if -math.pi >= da or da <= math.pi:
                        start = angle2
                        end = angle1
                        rev_flag = not rev_flag
                else:
                    if -math.pi <= da or da >= math.pi:
                        start = angle2
                        end = angle1
                        rev_flag = not rev_flag

                pth = libgeom.get_circle_paths(start, end, sk2const.ARC_ARC)[0]

                if rev_flag:
                    pth = libgeom.reverse_path(pth)

                points = pth[1]
                for point in points:
                    if len(point) == 3:
                        point.append(sk2const.NODE_CUSP)

                tr0 = [1.0, 0.0, 0.0, 1.0, -0.5, -0.5]
                points = libgeom.apply_trafo_to_points(points, tr0)

                tr1 = [2.0 * r, 0.0, 0.0, 2.0 * r, 0.0, 0.0]
                points = libgeom.apply_trafo_to_points(points, tr1)

                tr2 = [1.0, 0.0, 0.0, 1.0, center[0], center[1]]
                points = libgeom.apply_trafo_to_points(points, tr2)

                tr3 = libgeom.invert_trafo(dir_tr)
                points = libgeom.apply_trafo_to_points(points, tr3)

                for point in points:
                    path[1].append(point)

        last_cmd = cmd[0]

    if path: paths.append(path)
    return paths