Web   ·   Wiki   ·   Activities   ·   Blog   ·   Lists   ·   Chat   ·   Meeting   ·   Bugs   ·   Git   ·   Translate   ·   Archive   ·   People   ·   Donate
summaryrefslogtreecommitdiffstats
path: root/pysamples/forward_push.py
blob: ca527db709c11f8283d56575170fdb4ac78de655 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#Copyright (c) 2012-2013, Walter Bender

# Usage: Import this code into a Python (user-definable) block; when
# this code is run, a new block will be added to the Turtle Palette.
# This block will be named 'name', the value of the argument block
# docked to the Python block.
#
# The new block will cause the turtle to draw a line of the
# length. But before drawing the line, it pushes the rgb values of the
# destination to the FILO.


def myblock(tw, name):
    ''' '''

    def _prim_forward_push(tw, line_length):
        try:  # make sure line_length is a number
            line_length = float(line_length)
        except ValueError:
            return
        penstatus = tw.turtles.get_active_turtle().get_pen_status()
        tw.turtles.get_active_turtle().set_pen_state(False)
        tw.turtles.get_active_turtle().forward(line_length)
        r, g, b, a = tw.turtles.get_active_turtle().get_pixel()
        tw.lc.heap.append(b)
        tw.lc.heap.append(g)
        tw.lc.heap.append(r)
        tw.turtles.get_active_turtle().forward(-line_length)
        tw.turtles.get_active_turtle().set_pen_state(penstatus)
        tw.turtles.get_active_turtle().forward(line_length)
        return

    from TurtleArt.tapalette import make_palette, palette_name_to_index
    from TurtleArt.talogo import primitive_dictionary
    from gettext import gettext as _

    # Choose a palette for the new block.
    palette = make_palette('turtle')

    primitive_dictionary['forwardpush'] = _prim_forward_push

    # Create a new block prototype.
    palette.add_block('forwardpush',
                      style='basic-style-1arg',
                      label=name,
                      default=100,
                      prim_name='forwardpush',
                      help_string=_('push destination rgb value to heap'))

    # Add its primitive to the LogoCode dictionary.
    tw.lc.def_prim('forwardpush', 1,
                   lambda self, x: primitive_dictionary['forwardpush'](tw, x))

    # Regenerate the palette, which will now include the new block.
    tw.show_toolbar_palette(palette_name_to_index('turtle'),
                            regenerate=True)