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

   1 import curses
   2
   3
   4 def set_window_geometries():
   5
   6     def set_window_size():
   7         win["size"], win["start"] = [0, 0], [0, 0]
   8         win["size"][0] = win["config"][0]
   9         if (win["config"][0] == 0):
  10             win["size"][0] = screen_size[0] - sep_size
  11         elif (win["config"][0] < 0):
  12             win["size"][0] = screen_size[0] + win["config"][0] - sep_size
  13         win["size"][1] = win["config"][1]
  14         if (win["config"][1] == 0):
  15             win["size"][1] = screen_size[1]
  16         elif (win["config"][1] < 0):
  17             win["size"][1] = screen_size[1] + win["config"][1]
  18
  19     def place_window():
  20         win_i = windows.index(win)
  21
  22         # If win is first window, it goes into the top left corner.
  23         win["start"][0] = 0 + sep_size
  24         win["start"][1] = 0
  25         if (win_i > 0):
  26
  27             # If not, get win's closest predecessor starting a new stack on the
  28             # screen top, fit win's top left corner that predecessor's top
  29             # right corner.
  30             win_top = None
  31             for i in range(win_i - 1, -1, -1):
  32                 win_top = windows[i]
  33                 if (win_top["start"][0] == 0 + sep_size):
  34                     break
  35             win["start"][1] = win_top["start"][1] + win_top["size"][1] \
  36                 + sep_size
  37
  38             # If enough space is found below win's predecessor, fit win's top
  39             # left corner to that predecessor's bottom left corner.
  40             win_prev = windows[win_i - 1]
  41             next_free_y = win_prev["start"][0] + win_prev["size"][0] + sep_size
  42             if (win["size"][1] <= win_prev["size"][1] and
  43                     win["size"][0] <= screen_size[0] - next_free_y):
  44                 win["start"][1] = win_prev["start"][1]
  45                 win["start"][0] = next_free_y
  46
  47             # If that fails, try to fit win's top left corner to the top right
  48             # corner of its closest predecessor win_test that 1) is below
  49             # win_top (win's closest predecessor starting a new stack on the
  50             # screen top) 2) and has enough space open to its right between its
  51             # right edge and the lower edge of a window win_high located
  52             # directly above win_test to fit win there (without growing further
  53             # to the right than win_high does or surpassing the lower edge of
  54             # the screen).
  55             else:
  56                 win_test = win_prev
  57                 win_high = None
  58                 while (win_test != win_top):
  59                     for i in range(win_i - 2, -1, -1):
  60                         win_high = windows[i]
  61                         if win_test["start"][0] > win_high["start"][0]:
  62                             break
  63                     next_free_y = win_high["start"][0] + win_high["size"][0] \
  64                         + sep_size
  65                     first_free_x = win_test["start"][1] + win_test["size"][1] \
  66                         + sep_size
  67                     last_free_x = win_high["start"][1] + win_high["size"][1]
  68                     if (win["size"][0] <= screen_size[0] - next_free_y and
  69                             win["size"][1] <= last_free_x - first_free_x):
  70                         win["start"][1] = first_free_x
  71                         win["start"][0] = next_free_y
  72                         break
  73                     win_test = win_high
  74
  75     for win in windows:
  76         set_window_size()
  77         place_window()
  78
  79
  80 def draw_screen():
  81
  82     def draw_window_border_lines():
  83         for win in windows:
  84             for k in range(2):
  85                 j = win["start"][int(k == 0)] - sep_size
  86                 if (j >= 0 and j < screen_size[int(k == 0)]):
  87                     start = win["start"][k]
  88                     end = win["start"][k] + win["size"][k]
  89                     start = start if start >= 0 else 0
  90                     end = end if end < screen_size[k] else screen_size[k]
  91                     if k:
  92                         [stdscr.addch(j, i, '-') for i in range(start, end)]
  93                     else:
  94                         [stdscr.addch(i, j, '|') for i in range(start, end)]
  95
  96     def draw_window_border_corners():
  97         for win in windows:
  98             up = win["start"][0] - sep_size
  99             down = win["start"][0] + win["size"][0]
 100             left = win["start"][1] - sep_size
 101             right = win["start"][1] + win["size"][1]
 102             if (up >= 0 and up < screen_size[0]):
 103                 if (left >= 0 and left < screen_size[1]):
 104                     stdscr.addch(up, left, '+')
 105                 if (right >= 0 and right < screen_size[1]):
 106                     stdscr.addch(up, right, '+')
 107             if (down >= 0 and down < screen_size[0]):
 108                 if (left >= 0 and left < screen_size[1]):
 109                     stdscr.addch(down, left, '+')
 110                 if (right >= 0 and right < screen_size[1]):
 111                     stdscr.addch(down, right, '+')
 112
 113     def draw_window_contents():
 114         for win in windows:
 115             offset, size, winmap = win["func"]()
 116             stop = [0, 0]
 117             for i in range(2):
 118                 stop[i] = win["size"][i] + offset[i]
 119                 stop[i] = stop[i] if stop[i] < size[i] else size[i]
 120             for y in range(offset[0], stop[0]):
 121                 for x in range(offset[1], stop[1]):
 122                     cell = winmap[y * size[1] + x]
 123                     y_in_screen = win["start"][0] + (y - offset[0])
 124                     x_in_screen = win["start"][1] + (x - offset[1])
 125                     if (y_in_screen < screen_size[0]
 126                             and x_in_screen < screen_size[1]):
 127                         stdscr.addch(y_in_screen, x_in_screen, cell)
 128
 129     draw_window_border_lines()
 130     draw_window_border_corners()
 131     draw_window_contents()
 132     stdscr.refresh()
 133
 134
 135 def main(stdscr):
 136     curses.noecho()
 137     curses.curs_set(False)
 138     # stdscr.keypad(True)
 139     set_window_geometries()
 140     while True:
 141         draw_screen()
 142         stdscr.getch()
 143
 144
 145 def foo():
 146     winmap = ['.', 'o', '.', 'o', 'O', 'o', '.', 'o', '.', 'x', 'y', 'x']
 147     size = [4, 3]
 148     offset = [0, 0]
 149     return offset, size, winmap
 150
 151
 152 windows = [
 153     {"config": [1, 33], "func": foo},
 154     {"config": [-7, 33], "func": foo},
 155     {"config": [4, 16], "func": foo},
 156     {"config": [4, 16], "func": foo},
 157     {"config": [0, -34], "func": foo}
 158 ]
 159
 160 sep_size = 1  # Width of inter-window borders and title bars.
 161 stdscr = curses.initscr()
 162 screen_size = stdscr.getmaxyx()
 163
 164 curses.wrapper(main)