plomlompom.com Git - plomrogue/blob - client_prototype.py

   1 import curses
   2 import signal
   3 import os
   4
   5
   6 def set_window_geometries():
   7
   8     def set_window_size():
   9         win["size"], win["start"] = [0, 0], [0, 0]
  10         win["size"][0] = win["config"][0]
  11         if (win["config"][0] == 0):
  12             win["size"][0] = screen_size[0] - sep_size
  13         elif (win["config"][0] < 0):
  14             win["size"][0] = screen_size[0] + win["config"][0] - sep_size
  15         win["size"][1] = win["config"][1]
  16         if (win["config"][1] == 0):
  17             win["size"][1] = screen_size[1]
  18         elif (win["config"][1] < 0):
  19             win["size"][1] = screen_size[1] + win["config"][1]
  20
  21     def place_window():
  22         win_i = windows.index(win)
  23
  24         # If win is first window, it goes into the top left corner.
  25         win["start"][0] = 0 + sep_size
  26         win["start"][1] = 0
  27         if (win_i > 0):
  28
  29             # If not, get win's closest predecessor starting a new stack on the
  30             # screen top,fit win's top left to that win_top's top right corner.
  31             win_top = None
  32             for i in range(win_i - 1, -1, -1):
  33                 win_top = windows[i]
  34                 if (win_top["start"][0] == 0 + sep_size):
  35                     break
  36             win["start"][1] = win_top["start"][1] + win_top["size"][1] \
  37                 + sep_size
  38
  39             # If enough space is found below win's predecessor, fit win's top
  40             # left corner to that predecessor's bottom left corner.
  41             win_prev = windows[win_i - 1]
  42             next_free_y = win_prev["start"][0] + win_prev["size"][0] + sep_size
  43             if (win["size"][1] <= win_prev["size"][1] and
  44                     win["size"][0] <= screen_size[0] - next_free_y):
  45                 win["start"][1] = win_prev["start"][1]
  46                 win["start"][0] = next_free_y
  47
  48             # If that fails, try to fit win's top left corner to the top right
  49             # corner of its closest predecessor win_test 1) below win_top (see
  50             # above) 2) and with enough space open to its right between its
  51             # right edge and the lower edge of a win_high located directly
  52             # above win_test to fit win there (without growing further to the
  53             # right than win_high does or surpassing the screen's lower edge).
  54             else:
  55                 win_test = win_prev
  56                 win_high = None
  57                 while (win_test != win_top):
  58                     for i in range(win_i - 2, -1, -1):
  59                         win_high = windows[i]
  60                         if win_test["start"][0] > win_high["start"][0]:
  61                             break
  62                     next_free_y = win_high["start"][0] + win_high["size"][0] \
  63                         + sep_size
  64                     first_free_x = win_test["start"][1] + win_test["size"][1] \
  65                         + sep_size
  66                     last_free_x = win_high["start"][1] + win_high["size"][1]
  67                     if (win["size"][0] <= screen_size[0] - next_free_y and
  68                             win["size"][1] <= last_free_x - first_free_x):
  69                         win["start"][1] = first_free_x
  70                         win["start"][0] = next_free_y
  71                         break
  72                     win_test = win_high
  73
  74     global screen_size, stdscr
  75     curses.endwin()
  76     stdscr = curses.initscr()
  77     screen_size = stdscr.getmaxyx()
  78     for win in windows:
  79         set_window_size()
  80         place_window()
  81
  82
  83 def draw_screen():
  84
  85     def draw_window_border_lines():
  86         for win in windows:
  87             for k in range(2):
  88                 j = win["start"][int(k == 0)] - sep_size
  89                 if (j >= 0 and j < screen_size[int(k == 0)]):
  90                     start = win["start"][k]
  91                     # start = start if start >= 0 else 0
  92                     end = win["start"][k] + win["size"][k]
  93                     end = end if end < screen_size[k] else screen_size[k]
  94                     if k:
  95                         [stdscr.addch(j, i, '-') for i in range(start, end)]
  96                     else:
  97                         [stdscr.addch(i, j, '|') for i in range(start, end)]
  98
  99     def draw_window_border_corners():
 100         for win in windows:
 101             up = win["start"][0] - sep_size
 102             down = win["start"][0] + win["size"][0]
 103             left = win["start"][1] - sep_size
 104             right = win["start"][1] + win["size"][1]
 105             if (up >= 0 and up < screen_size[0]):
 106                 if (left >= 0 and left < screen_size[1]):
 107                     stdscr.addch(up, left, '+')
 108                 if (right >= 0 and right < screen_size[1]):
 109                     stdscr.addch(up, right, '+')
 110             if (down >= 0 and down < screen_size[0]):
 111                 if (left >= 0 and left < screen_size[1]):
 112                     stdscr.addch(down, left, '+')
 113                 if (right >= 0 and right < screen_size[1]):
 114                     stdscr.addch(down, right, '+')
 115
 116     def draw_window_contents():
 117         def draw_winmap():
 118             stop = [0, 0]
 119             for i in range(2):
 120                 stop[i] = win["size"][i] + offset[i]
 121                 stop[i] = stop[i] if stop[i] < size[i] else size[i]
 122             for y in range(offset[0], stop[0]):
 123                 for x in range(offset[1], stop[1]):
 124                     cell = winmap[y * size[1] + x]
 125                     y_in_screen = win["start"][0] + (y - offset[0])
 126                     x_in_screen = win["start"][1] + (x - offset[1])
 127                     if (y_in_screen < screen_size[0]
 128                             and x_in_screen < screen_size[1]):
 129                         stdscr.addch(y_in_screen, x_in_screen, cell)
 130         def draw_scroll_hints():
 131             def draw_scroll_string(n_lines_outside):
 132                 hint = ' ' + str(n_lines_outside + 1) + ' more ' + unit + ' '
 133                 if len(hint) <= win["size"][ni]:
 134                     non_hint_space = win["size"][ni] - len(hint)
 135                     hint_offset = int(non_hint_space / 2)
 136                     for j in range(win["size"][ni] - non_hint_space):
 137                         pos_2 = win["start"][ni] + hint_offset + j
 138                         x, y = (pos_2, pos_1) if ni else (pos_1, pos_2)
 139                         stdscr.addch(y, x, hint[j], curses.A_REVERSE)
 140             def draw_scroll_arrows(ar1, ar2):
 141                 for j in range(win["size"][ni]):
 142                     pos_2 = win["start"][ni] + j
 143                     x, y = (pos_2, pos_1) if ni else (pos_1, pos_2)
 144                     stdscr.addch(y, x, ar1 if ni else ar2, curses.A_REVERSE)
 145             for i in range(2):
 146                 ni = int(i == 0)
 147                 unit = 'rows' if ni else 'columns'
 148                 if (offset[i] > 0):
 149                     pos_1 = win["start"][i]
 150                     draw_scroll_arrows('^', '<')
 151                     draw_scroll_string(offset[i])
 152                 if (size[i] > offset[i] + win["size"][i]):
 153                     pos_1 = win["start"][i] + win["size"][i] - 1
 154                     draw_scroll_arrows('v', '>')
 155                     draw_scroll_string(size[i] - offset[i] - win["size"][i])
 156         for win in windows:
 157             offset, size, winmap = win["func"]()
 158             draw_winmap()
 159             draw_scroll_hints()
 160
 161     stdscr.clear()
 162     draw_window_border_lines()
 163     draw_window_border_corners()
 164     draw_window_contents()
 165     stdscr.refresh()
 166
 167
 168 def cursed_main(stdscr):
 169     curses.noecho()
 170     curses.curs_set(False)
 171     # stdscr.keypad(True)
 172     signal.signal(signal.SIGWINCH,
 173         lambda ignore_1, ignore_2: set_window_geometries())
 174     set_window_geometries()
 175     while True:
 176         draw_screen()
 177         char = stdscr.getch()
 178         if (char >= 0):
 179             if chr(char) in commands:
 180                 commands[chr(char)]()
 181
 182
 183 def foo():
 184     winmap = ['.', 'o', '.', 'o', 'O', 'o', '.', 'o', '.', 'x', 'y', 'x']
 185     size = [4, 3]
 186     offset = [0, 0]
 187     return offset, size, winmap
 188
 189
 190 def command_quit():
 191     io["file_out"].write("QUIT\n")
 192     io["file_out"].flush()
 193     raise SystemExit("Received QUIT command, forwarded to server, leaving.")
 194
 195
 196 windows = [
 197     {"config": [1, 33], "func": foo},
 198     {"config": [-7, 33], "func": foo},
 199     {"config": [4, 16], "func": foo},
 200     {"config": [4, 16], "func": foo},
 201     {"config": [0, -34], "func": foo}
 202 ]
 203 io = {
 204     "path_out": "server/in"
 205 }
 206 commands = {
 207     "Q": command_quit
 208 }
 209 sep_size = 1  # Width of inter-window borders and title bars.
 210 stdscr = None
 211 screen_size = [0,0]
 212
 213
 214 try:
 215     if (not os.access(io["path_out"], os.F_OK)):
 216         msg = "No server input file found at " + io["path_out"] + "."
 217         raise SystemExit(msg)
 218     io["file_out"] = open(io["path_out"], "a")
 219     curses.wrapper(cursed_main)
 220 except SystemExit as exit:
 221     print("ABORTING: " + exit.args[0])
 222 except:
 223     print("SOMETHING WENT WRONG IN UNEXPECTED WAYS")
 224     raise
 225 finally:
 226     if "file_out" in io:
 227         io["file_out"].close()