summaryrefslogtreecommitdiff
path: root/asciifarm/client/display/display.py
blob: 1d5f542b6fd59d961213052d8b436d07e7345365 (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
143
144
145
146
147
148
149
150
151
152
153
154

import curses

from .field import Field
from .info import Info
from .health import Health
from .inventory import Inventory
from .screen import Screen
from .colours import Colours
from .messages import Messages
from .switcher import Switcher
from .textinput import TextInput
from .widget import Widget

from asciifarm.common.utils import get


SIDEWIDTH = 20

ALPHABET = "!\"#$%&'()*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`abcdefghijklmnopqrstuvwxyz{|}~"

class Display:
    
    def __init__(self, stdscr, charMap, colours=False):
        
        if colours and curses.has_colors and curses.COLORS > 1:
            self.colours = Colours()
        else:
            self.colours = None
        self.characters = {}
        
        def parseSprite(sprite):
            if isinstance(sprite, str):
                return (sprite, None, None)
            char = get(sprite, 0, " ")
            fg = get(sprite, 1)
            bg = get(sprite, 2)
            return (char, fg, bg)
        for name, sprite in charMap["mapping"].items():
            vals = parseSprite(sprite)
            if vals:
                self.characters[name] = vals
        
        for name, colours in charMap.get("writable", {}).items():
            fg = get(colours, 0)
            bg = get(colours, 1)
            for i in range(min(len(ALPHABET), len(charMap.get("alphabet", [])))):
                self.characters[name + '-' + ALPHABET[i]] = (charMap["alphabet"][i], fg, bg)
        
        self.defaultChar = parseSprite(charMap.get("default", "?"))
        screen = Screen(self, stdscr, self.colours)
        self.screen = screen
        
        self.widgets = {}
        
        self.addWidget(Field((1, 1), charMap.get("charwidth", 1), self.colours), "field")
        self.addWidget(Info(), "info")
        self.addWidget(Health(
                    charMap.get("healthfull", ("@",7, 2)),
                    charMap.get("healthempty", ("-",7, 1)),
                    self.colours),
            "health")
        self.addWidget(Inventory("Inventory"), "inventory")
        self.addWidget(Inventory("Ground"), "ground")
        self.addWidget(Inventory("Equipment"), "equipment")
        self.addWidget(Messages(), "msg")
        self.addWidget(TextInput(), "textinput")
        
        
        switcher = Switcher([self.widgets["ground"], self.widgets["inventory"]], 1)
        self.addWidget(switcher, "switch")
        
        self.forced = False
        self.update()
    
    def addWidget(self, w, name, winname=None):
        if not winname:
            winname = name
        widget = Widget(w)
        self.widgets[name] = widget
        widget.setWin(winname, self.screen)
    
    def getWidget(self, name):
        if name in self.widgets:
            return self.widgets[name].getImpl()
        else:
            return None
    
    def resizeField(self, size):
        self.getWidget("field").resize(*size)
    
    def drawFieldCells(self, cells):
        field = self.getWidget("field")
        for cell in cells:
            (x, y), spriteNames = cell
            sprites = [self.getChar(spriteName) for spriteName in spriteNames]
            if not len(sprites):
                sprites = [self.getChar(" ")]
            field.changeCell(x, y, sprites)
        
    
    def setFieldCenter(self, pos):
        self.getWidget("field").setCenter(pos)
    
    def setHealth(self, health, maxHealth):
        self.getWidget("health").setHealth(health, maxHealth)
        
    
    def showInfo(self, infostring):
        self.getWidget("info").showString(infostring)
            
    
    def setInventory(self, items):
        self.getWidget("inventory").setInventory(items)
        
    
    def setEquipment(self, slots):
        self.getWidget("equipment").setInventory(
            sorted([
                slot + ": " + (item if item else "")
                for slot, item in slots.items()
            ])
        )
    
    def setGround(self, items):
        self.getWidget("ground").setInventory(items)
        
    
    def addMessage(self, message):
        self.getWidget("msg").addMessage(message)
    
    def scrollBack(self, amount, relative=True):
        self.getWidget("msg").scroll(amount, relative)
    
    def getChar(self, sprite):
        """This returns the character belonging to some spritename. This does not read a character"""
        return self.characters.get(sprite, self.defaultChar)
    
    def setInputString(self, string, cursor):
        self.getWidget("textinput").setText(string, cursor)
    
    def update(self):
        changed = False
        for widget in self.widgets.values():
            if self.forced or widget.isChanged():
                widget.update()
                changed = True
        if changed:
            self.screen.update()
        self.forced = False
    
    def forceUpdate(self):
        self.forced = True