Web   ·   Wiki   ·   Activities   ·   Blog   ·   Lists   ·   Chat   ·   Meeting   ·   Bugs   ·   Git   ·   Translate   ·   Archive   ·   People   ·   Donate
summaryrefslogtreecommitdiffstats
path: root/functions.py
blob: 040e3b2346023d6d63ed852ed794089061f35d3e (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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
"""
This is a Gtk TreeView to list and modify the plotted functions
"""
# Copyright (C) 2012 S. Daniel Francis <francis@sugarlabs.org>
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the
# Free Software Foundation, Inc., 59 Temple Place - Suite 330,
# Boston, MA 02111-1307, USA.

import logging
logger = logging.getLogger('functions')
import gobject
import gtk
from sugar.graphics import style
from sugar.graphics.xocolor import XoColor
from sugar.graphics.icon import CellRendererIcon


def color2string(color):
    color_string = ["#"]
    color_string.append("%02x" % (color.red / 256))
    color_string.append("%02x" % (color.green / 256))
    color_string.append("%02x" % (color.blue / 256))
    return "".join(color_string)


class FunctionsList(gtk.TreeView):
    __gsignals__ = {'list-updated': (gobject.SIGNAL_RUN_LAST,
                                     gobject.TYPE_NONE,
                                     (gobject.TYPE_PYOBJECT,)),
                    'function-selected': (gobject.SIGNAL_RUN_LAST,
                                          gobject.TYPE_NONE,
                                          (gobject.TYPE_PYOBJECT,))}

    def __init__(self):
        gtk.TreeView.__init__(self)
        self.set_rules_hint(True)
        self.grab_focus()
        self.model = gtk.ListStore(object, str, str, str, bool)
        self.set_model(self.model)
        column = gtk.TreeViewColumn()
        color_renderer = CellRendererIcon(self)
        color_renderer.set_fixed_size(int(style.SMALL_ICON_SIZE * 4), -1)
        color_renderer.props.size = style.SMALL_ICON_SIZE * 1.25
        color_renderer.set_icon_name('color-preview')
        color_renderer.props.stroke_color = style.COLOR_BUTTON_GREY.get_svg()
        color_renderer.props.fill_color = style.COLOR_TRANSPARENT.get_svg()
        color_renderer.props.prelit_stroke_color = "#666666"
        color_renderer.props.prelit_fill_color = "#FFFFFF"
        column.pack_start(color_renderer, False)
        column.add_attribute(color_renderer, 'xo-color', 0)
        name_renderer = gtk.CellRendererText()
        column.pack_start(name_renderer, False)
        column.add_attribute(name_renderer, 'text', 1)
        text_renderer = gtk.CellRendererText()
        text_renderer.set_property('editable', True)
        text_renderer.connect('edited', self._function_changed)
        column.pack_start(text_renderer, True)
        column.add_attribute(text_renderer, 'text', 2)
        self.set_headers_visible(False)
        self.append_column(column)
        self.evaluating_column = gtk.TreeViewColumn()
        y_renderer = gtk.CellRendererText()
        y_renderer.set_property('text', ' = ')
        self.evaluating_column.pack_start(y_renderer, False)
        evaluation_cell = gtk.CellRendererText()
        self.evaluating_column.pack_start(evaluation_cell, False)
        self.evaluating_column.add_attribute(evaluation_cell, 'text', 3)
        self.append_column(self.evaluating_column)
        self.evaluating_column.set_visible(False)
        self.selection = self.get_selection()
        self.selection.set_select_function(self.update_color)
        self.updating_color = False

    def evaluate(self, safe_dict):
        for func in self.model:
            if safe_dict['x'] == None:
                self.evaluating_column.set_visible(False)
                return
            result = eval(func[2].replace('^', '**'),
                          {'__builtins__': {}}, safe_dict)
            func[3] = str(result)
        self.evaluating_column.set_visible(True)

    def set_current_line_color(self, color):
        if not self.updating_color:
            rows = self.selection.get_selected_rows()
            self.model[rows[1][0]][0] = XoColor(color2string(color) + "," +\
                                                           color2string(color))

    def update_color(self, info):
        path = info[0]
        for i in self.model:
            i[-1] = False
        color = self.model[path][0]
        self.updating_color = True
        self.model[path][-1] = True
        self.emit('function-selected', color.get_fill_color())
        self.updating_color = False
        return True

    def get_list(self):
        funcs = []
        for func in self.model:
            row = [gtk.gdk.color_parse(func[0].get_fill_color()),
                    func[1], func[2], func[-1]]
            funcs.append(row)
        logger.debug(str(funcs))
        return funcs

    def _function_changed(self, widget, path, new_text):
        self.model[path][2] = new_text
        for i in self.model:
            i[-1] = False
        self.model[path][-1] = True
        self.emit('list-updated', self.get_list())

    def append_function(self, color, expression='sin(x)'):
        self.selection.select_iter(self.model.append([XoColor(color + "," +\
                                                              color),
                                              "y%d = " % (len(self.model) + 1),
                                                      expression, None, True]))

    def remove_function(self):
        rows = self.selection.get_selected_rows()
        _iter = self.model.get_iter(rows[1][0])
        self.model.remove(_iter)
        count = 0
        for i in self.model:
            count += 1
            i[1] = 'y%d =' % count