Dump screen of Coot with white background, black chicken wire, and ball-stick representation | Codetter（こーどったー）

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
# This script works on Coot.
# Save screen image with white background, ball-stick representation, and black mesh for Gakusei Jikken.
#
# Ver. 2 of http://codetter.com/?p=400
# by yam_cpp, 2011-10-31
#
def save_png_gakusei(arg):
        def get_displayed_maps():
                displayed_maps = []
                for i in range(graphics_n_molecules()):
                        if map_is_displayed(i):
                                print i, "is displayed map."
                                displayed_maps.append(i)
                return displayed_maps
        # get_displayed_maps()
        def change_representation(ballstick):
                # http://biop.ox.ac.uk/coot/doc/coot/Ball-and-Stick-Representation.html
                # (make-ball-and-stick imol atom-selection bond-thickness sphere-size draw-spheres-flag)
                for i in range(graphics_n_molecules()):
                        if mol_is_displayed(i):
                                if  ballstick:
                                        make_ball_and_stick(i, "*", 0.2, 0.3, 1)
                                else:
                                        clear_ball_and_stick(i)
        # get_displayed_mols()
        # Assuming only one map is displayed
        displayed_map = get_displayed_maps()[0]
        # save map colour to restore later
        save_previous_map_colour(displayed_map)
        # change map & background colour and set ball-stick representation
        set_map_colour(displayed_map, 0, 0, 0)
        brighten_maps()
        set_background_colour(255,255,255)
        change_representation(True)
        # save image
        screendump_image(time.strftime("coot-%Y%m%d_%H%M%S.png"))
        # restore
        set_background_colour(0,0,0)
        darken_maps()
        restore_previous_map_colour(displayed_map)
        change_representation(False)
# save_png_gakusei()
add_simple_coot_menu_menuitem(coot_menubar_menu("Extensions"), "Save Image", save_png_gakusei)
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX