352 lines
12 KiB
Python
Executable File
352 lines
12 KiB
Python
Executable File
#!/usr/bin/python3
|
|
"""
|
|
Calculator for ToaruOS
|
|
"""
|
|
import math
|
|
import sys
|
|
|
|
import cairo
|
|
|
|
import yutani
|
|
import text_region
|
|
import toaru_fonts
|
|
|
|
import ast
|
|
import operator as op
|
|
|
|
operators = {ast.Add: op.add, ast.Sub: op.sub, ast.Mult: op.mul,
|
|
ast.Div: op.truediv, ast.Pow: op.pow, ast.BitXor: op.xor,
|
|
ast.USub: op.neg}
|
|
|
|
def eval_expr(expr):
|
|
"""
|
|
>>> eval_expr('2^6')
|
|
4
|
|
>>> eval_expr('2**6')
|
|
64
|
|
>>> eval_expr('1 + 2*3**(4^5) / (6 + -7)')
|
|
-5.0
|
|
"""
|
|
return eval_(ast.parse(expr, mode='eval').body)
|
|
|
|
def eval_(node):
|
|
if isinstance(node, ast.Num): # <number>
|
|
return node.n
|
|
elif isinstance(node, ast.BinOp): # <left> <operator> <right>
|
|
return operators[type(node.op)](eval_(node.left), eval_(node.right))
|
|
elif isinstance(node, ast.UnaryOp): # <operator> <operand> e.g., -1
|
|
return operators[type(node.op)](eval_(node.operand))
|
|
else:
|
|
raise TypeError("invalid operation")
|
|
|
|
def rounded_rectangle(ctx,x,y,w,h,r):
|
|
degrees = math.pi / 180
|
|
ctx.new_sub_path()
|
|
|
|
ctx.arc(x + w - r, y + r, r, -90 * degrees, 0 * degrees)
|
|
ctx.arc(x + w - r, y + h - r, r, 0 * degrees, 90 * degrees)
|
|
ctx.arc(x + r, y + h - r, r, 90 * degrees, 180 * degrees)
|
|
ctx.arc(x + r, y + r, r, 180 * degrees, 270 * degrees)
|
|
ctx.close_path()
|
|
|
|
def draw_button(ctx,x,y,w,h,hilight):
|
|
"""Theme definition for drawing a button."""
|
|
ctx.save()
|
|
|
|
ctx.set_line_cap(cairo.LINE_CAP_ROUND)
|
|
ctx.set_line_join(cairo.LINE_JOIN_ROUND)
|
|
|
|
if hilight == 2:
|
|
rounded_rectangle(ctx, 2 + x, 2 + y, w - 4, h - 4, 2.0)
|
|
ctx.set_source_rgba(134/255,173/255,201/255,1.0)
|
|
ctx.set_line_width(2)
|
|
ctx.stroke()
|
|
|
|
rounded_rectangle(ctx, 2 + x, 2 + y, w - 4, h - 4, 2.0)
|
|
ctx.set_source_rgba(202/255,211/255,232/255,1.0)
|
|
ctx.fill()
|
|
else:
|
|
rounded_rectangle(ctx, 2 + x, 2 + y, w - 4, h - 4, 2.0)
|
|
ctx.set_source_rgba(44/255,71/255,91/255,29/255)
|
|
ctx.set_line_width(4)
|
|
ctx.stroke()
|
|
|
|
rounded_rectangle(ctx, 2 + x, 2 + y, w - 4, h - 4, 2.0)
|
|
ctx.set_source_rgba(158/255,169/255,177/255,1.0)
|
|
ctx.set_line_width(2)
|
|
ctx.stroke()
|
|
|
|
if hilight == 1:
|
|
pat = cairo.LinearGradient(2+x,2+y,2+x,2+y+h-4)
|
|
pat.add_color_stop_rgba(0,1,1,1,1)
|
|
pat.add_color_stop_rgba(1,229/255,229/255,246/255,1)
|
|
rounded_rectangle(ctx,2+x,2+y,w-4,h-4,2.0)
|
|
ctx.set_source(pat)
|
|
ctx.fill()
|
|
|
|
pat = cairo.LinearGradient(3+x,3+y,3+x,3+y+h-4)
|
|
pat.add_color_stop_rgba(0,252/255,252/255,254/255,1)
|
|
pat.add_color_stop_rgba(1,212/255,223/255,251/255,1)
|
|
rounded_rectangle(ctx,3+x,3+y,w-5,h-5,2.0)
|
|
ctx.set_source(pat)
|
|
ctx.fill()
|
|
|
|
else:
|
|
pat = cairo.LinearGradient(2+x,2+y,2+x,2+y+h-4)
|
|
pat.add_color_stop_rgba(0,1,1,1,1)
|
|
pat.add_color_stop_rgba(1,241/255,241/255,244/255,1)
|
|
rounded_rectangle(ctx,2+x,2+y,w-4,h-4,2.0)
|
|
ctx.set_source(pat)
|
|
ctx.fill()
|
|
|
|
pat = cairo.LinearGradient(3+x,3+y,3+x,3+y+h-4)
|
|
pat.add_color_stop_rgba(0,252/255,252/255,254/255,1)
|
|
pat.add_color_stop_rgba(1,223/255,225/255,230/255,1)
|
|
rounded_rectangle(ctx,3+x,3+y,w-5,h-5,2.0)
|
|
ctx.set_source(pat)
|
|
ctx.fill()
|
|
|
|
ctx.restore()
|
|
|
|
class Button(object):
|
|
|
|
def __init__(self, text, callback):
|
|
self.text = text
|
|
self.callback = callback
|
|
self.hilight = 0
|
|
self.x, self.y, self.width, self.height = 0,0,0,0
|
|
|
|
def draw(self, window, ctx, x, y, w, h):
|
|
self.x, self.y, self.width, self.height = x, y, w, h
|
|
draw_button(ctx,x,y,w,h,self.hilight)
|
|
|
|
x_, y_ = ctx.user_to_device(x,y)
|
|
tr = text_region.TextRegion(int(x_),int(y_),w,h)
|
|
tr.set_alignment(2)
|
|
tr.set_valignment(2)
|
|
tr.set_text(self.text)
|
|
tr.draw(window)
|
|
|
|
def focus_enter(self):
|
|
self.hilight = 1
|
|
|
|
def focus_leave(self):
|
|
self.hilight = 0
|
|
|
|
class CalculatorWindow(yutani.Window):
|
|
|
|
base_width = 200
|
|
base_height = 200
|
|
|
|
def __init__(self, decorator):
|
|
super(CalculatorWindow, self).__init__(self.base_width + decorator.width(), self.base_height + decorator.height(), title="Calculator", icon="calculator", doublebuffer=True)
|
|
self.move(100,100)
|
|
self.decorator = decorator
|
|
|
|
def add_string(button):
|
|
self.add_string(button.text)
|
|
|
|
def clear(button):
|
|
self.clear_text()
|
|
|
|
def calculate(button):
|
|
self.calculate()
|
|
|
|
self.buttons = [
|
|
[Button("C",clear), None, Button("(",add_string), Button(")",add_string)],
|
|
[Button("7",add_string), Button("8",add_string), Button("9",add_string), Button("/",add_string)],
|
|
[Button("4",add_string), Button("5",add_string), Button("6",add_string), Button("*",add_string)],
|
|
[Button("1",add_string), Button("2",add_string), Button("3",add_string), Button("-",add_string)],
|
|
[Button("0",add_string), Button(".",add_string), Button("=",calculate), Button("+",add_string)],
|
|
]
|
|
|
|
self.tr = text_region.TextRegion(self.decorator.left_width()+5,self.decorator.top_height(),self.base_width-10,40)
|
|
self.tr.set_font(toaru_fonts.Font(toaru_fonts.FONT_MONOSPACE,18))
|
|
self.tr.set_text("")
|
|
self.tr.set_alignment(1)
|
|
self.tr.set_valignment(2)
|
|
self.tr.set_one_line()
|
|
self.tr.set_ellipsis()
|
|
|
|
self.error = False
|
|
|
|
self.hover_widget = None
|
|
self.down_button = None
|
|
|
|
def calculate(self):
|
|
if self.error or len(self.tr.text) == 0:
|
|
self.tr.set_text("0")
|
|
self.error = False
|
|
try:
|
|
self.tr.set_text(str(eval_expr(self.tr.text)))
|
|
except Exception as e:
|
|
error = str(e)
|
|
if "(" in error:
|
|
error = error[:error.find("(")-1]
|
|
self.tr.set_richtext(f"<i><color 0xFF0000>{e.__class__.__name__}</color>: {error}</i>")
|
|
self.error = True
|
|
self.draw()
|
|
self.flip()
|
|
|
|
def add_string(self, text):
|
|
if self.error:
|
|
self.tr.text = ""
|
|
self.error = False
|
|
self.tr.set_text(self.tr.text + text)
|
|
self.draw()
|
|
self.flip()
|
|
|
|
def clear_text(self):
|
|
self.error = False
|
|
self.tr.set_text("")
|
|
self.draw()
|
|
self.flip()
|
|
|
|
def clear_last(self):
|
|
if self.error:
|
|
self.error = False
|
|
self.tr.set_text("")
|
|
if len(self.tr.text):
|
|
self.tr.set_text(self.tr.text[:-1])
|
|
self.draw()
|
|
self.flip()
|
|
|
|
def draw(self):
|
|
surface = self.get_cairo_surface()
|
|
|
|
WIDTH, HEIGHT = self.width - self.decorator.width(), self.height - self.decorator.height()
|
|
|
|
ctx = cairo.Context(surface)
|
|
ctx.translate(self.decorator.left_width(), self.decorator.top_height())
|
|
ctx.rectangle(0,0,WIDTH,HEIGHT)
|
|
ctx.set_source_rgb(204/255,204/255,204/255)
|
|
ctx.fill()
|
|
|
|
ctx.rectangle(0,5,WIDTH,self.tr.height-10)
|
|
ctx.set_source_rgb(1,1,1)
|
|
ctx.fill()
|
|
self.tr.resize(WIDTH-10, self.tr.height)
|
|
self.tr.draw(self)
|
|
|
|
offset_x = 0
|
|
offset_y = self.tr.height
|
|
button_height = int((HEIGHT - self.tr.height) / len(self.buttons))
|
|
for row in self.buttons:
|
|
button_width = int(WIDTH / len(row))
|
|
for button in row:
|
|
if button:
|
|
button.draw(self,ctx,offset_x,offset_y,button_width,button_height)
|
|
offset_x += button_width
|
|
offset_x = 0
|
|
offset_y += button_height
|
|
|
|
|
|
|
|
self.decorator.render(self)
|
|
|
|
def finish_resize(self, msg):
|
|
"""Accept a resize."""
|
|
self.resize_accept(msg.width, msg.height)
|
|
self.reinit()
|
|
self.int_width = msg.width - self.decorator.width()
|
|
self.int_height = msg.height - self.decorator.height()
|
|
self.draw()
|
|
self.resize_done()
|
|
self.flip()
|
|
|
|
def mouse_event(self, msg):
|
|
if d.handle_event(msg) == yutani.Decor.EVENT_CLOSE:
|
|
window.close()
|
|
sys.exit(0)
|
|
x,y = msg.new_x - self.decorator.left_width(), msg.new_y - self.decorator.top_height()
|
|
w,h = self.width - self.decorator.width(), self.height - self.decorator.height()
|
|
|
|
redraw = False
|
|
if self.down_button:
|
|
if msg.command == yutani.MouseEvent.RAISE or msg.command == yutani.MouseEvent.CLICK:
|
|
if not (msg.buttons & yutani.MouseButton.BUTTON_LEFT):
|
|
if x >= self.down_button.x and \
|
|
x < self.down_button.x + self.down_button.width and \
|
|
y >= self.down_button.y and \
|
|
y < self.down_button.y + self.down_button.height:
|
|
self.down_button.focus_enter()
|
|
self.down_button.callback(self.down_button)
|
|
self.down_button = None
|
|
redraw = True
|
|
else:
|
|
self.down_button.focus_leave()
|
|
self.down_button = None
|
|
redraw = True
|
|
|
|
else:
|
|
if y > self.tr.height and y < h and x >= 0 and x < w:
|
|
row = int((y - self.tr.height) / (self.height - self.decorator.height() - self.tr.height) * len(self.buttons))
|
|
col = int(x / (self.width - self.decorator.width()) * len(self.buttons[row]))
|
|
button = self.buttons[row][col]
|
|
if button != self.hover_widget:
|
|
if button:
|
|
button.focus_enter()
|
|
redraw = True
|
|
if self.hover_widget:
|
|
self.hover_widget.focus_leave()
|
|
redraw = True
|
|
self.hover_widget = button
|
|
|
|
if msg.command == yutani.MouseEvent.DOWN:
|
|
if button:
|
|
button.hilight = 2
|
|
self.down_button = button
|
|
redraw = True
|
|
else:
|
|
if self.hover_widget:
|
|
self.hover_widget.focus_leave()
|
|
redraw = True
|
|
self.hover_widget = None
|
|
|
|
if redraw:
|
|
self.draw()
|
|
self.flip()
|
|
|
|
def keyboard_event(self, msg):
|
|
if msg.event.action != 0x01:
|
|
return # Ignore anything that isn't a key down.
|
|
if msg.event.key in b"0123456789.+-/*()":
|
|
self.add_string(msg.event.key.decode('utf-8'))
|
|
if msg.event.key == b"\n":
|
|
self.calculate()
|
|
if msg.event.key == b"c":
|
|
self.clear_text()
|
|
if msg.event.keycode == 8:
|
|
self.clear_last()
|
|
if msg.event.key == b"q":
|
|
self.close()
|
|
sys.exit(0)
|
|
|
|
if __name__ == '__main__':
|
|
yutani.Yutani()
|
|
d = yutani.Decor()
|
|
|
|
window = CalculatorWindow(d)
|
|
window.draw()
|
|
window.flip()
|
|
|
|
while 1:
|
|
# Poll for events.
|
|
msg = yutani.yutani_ctx.poll()
|
|
if msg.type == yutani.Message.MSG_SESSION_END:
|
|
window.close()
|
|
break
|
|
elif msg.type == yutani.Message.MSG_KEY_EVENT:
|
|
if msg.wid == window.wid:
|
|
window.keyboard_event(msg)
|
|
elif msg.type == yutani.Message.MSG_WINDOW_FOCUS_CHANGE:
|
|
if msg.wid == window.wid:
|
|
window.focused = msg.focused
|
|
window.draw()
|
|
window.flip()
|
|
elif msg.type == yutani.Message.MSG_RESIZE_OFFER:
|
|
window.finish_resize(msg)
|
|
elif msg.type == yutani.Message.MSG_WINDOW_MOUSE_EVENT:
|
|
if msg.wid == window.wid:
|
|
window.mouse_event(msg)
|