py/kmsmodeview.py

   1 #!/usr/bin/python3
   2
   3 import urwid
   4 import pykms
   5
   6 def exit_on_q(key):
   7         if key in ('q', 'Q'):
   8                 raise urwid.ExitMainLoop()
   9         elif key == 'a':
  10                 apply_mode()
  11
  12 alarm_handle = None
  13
  14 def recalc_info(l, d):
  15         global alarm_handle
  16
  17         alarm_handle = None
  18
  19         for w in recalc_list:
  20                 w.recalc()
  21
  22 def div_or_zero(n, d):
  23         if d == 0:
  24                 return 0
  25         else:
  26                 return n / d
  27
  28 class MyIntEdit(urwid.IntEdit):
  29         _metaclass_ = urwid.signals.MetaSignals
  30         signals = ['value_change']
  31
  32         def __init__(self, caption, calc=None):
  33                 self._myval = 0
  34                 self._disable_change = False
  35                 self._calc = calc
  36                 self._updlist = None
  37
  38                 super().__init__(caption, 0)
  39
  40         def set_edit_text(self, text):
  41                 global alarm_handle
  42
  43                 super().set_edit_text(text)
  44                 newtext = super().get_edit_text()
  45                 new_val = int(newtext) if newtext != "" else 0
  46                 if new_val != self._myval:
  47                         self._myval = new_val
  48                         if not self._disable_change:
  49                                 urwid.emit_signal(self, 'value_change', self, self._myval)
  50
  51                                 if alarm_handle == None:
  52                                         alarm_handle = loop.set_alarm_in(0, recalc_info)
  53
  54                                 if self._updlist != None:
  55                                         for w in self._updlist:
  56                                                 w.recalc()
  57
  58         def recalc(self):
  59                 self._disable_change = True
  60                 self.set_val(self._calc())
  61                 self._disable_change = False
  62
  63         def set_val(self, val):
  64                 self.set_edit_text(str(int(val)))
  65
  66         def get_val(self):
  67                 return self._myval
  68
  69         def set_updlist(self, list):
  70                 self._updlist = list
  71
  72         def keypress(self, size, key):
  73                 if key == '+':
  74                         self.set_edit_text(str(self.value() + 1))
  75                 elif key == '-':
  76                         self.set_edit_text(str(self.value() - 1))
  77                 else:
  78                         return super().keypress(size, key)
  79
  80 class MyIntText(urwid.Text):
  81         def __init__(self, fmt, calc=None):
  82                 super().__init__("")
  83                 self._fmt = fmt
  84                 self._calc = calc
  85
  86         def recalc(self):
  87                 val = self._calc()
  88                 super().set_text(self._fmt.format(val))
  89
  90 def khz_to_ps(khz):
  91         if khz == 0:
  92                 return 0
  93         else:
  94                 return 1.0 / khz * 1000 * 1000 * 1000
  95
  96 def khz_to_us(khz):
  97         if khz == 0:
  98                 return 0
  99         else:
 100                 return 1.0 / khz * 1000
 101
 102 pclk_khz_widget = MyIntEdit(u"pclk (kHz) ")
 103 pclk_ps_widget = MyIntText(fmt="pclk {:.2f} ps", calc = lambda: khz_to_ps(pclk_khz_widget.get_val()))
 104
 105 pclk_widgets = [pclk_khz_widget, pclk_ps_widget]
 106
 107 pclk_columns = urwid.LineBox(urwid.Columns(pclk_widgets), title = "Pixel clock")
 108
 109 # Horizontal widgets
 110
 111 hdisp_widget = MyIntEdit(u"hdisp ", calc = lambda: hdisp2_widget.get_val())
 112 hfp_widget = MyIntEdit(u"hfp   ", calc = lambda: hss_widget.get_val() - hdisp_widget.get_val())
 113 hsw_widget = MyIntEdit(u"hsw   ", calc = lambda: hse_widget.get_val() - hss_widget.get_val())
 114 hbp_widget = MyIntEdit(u"hbp   ", calc = lambda: htot_widget.get_val() - hse_widget.get_val())
 115
 116 hdisp2_widget = MyIntEdit(u"hdisp ", calc = lambda: hdisp_widget.get_val())
 117 hss_widget = MyIntEdit(u"hss   ",
 118         calc = lambda: hdisp_widget.get_val() + hfp_widget.get_val())
 119 hse_widget = MyIntEdit(u"hse   ",
 120         calc = lambda: hdisp_widget.get_val() + hfp_widget.get_val() + hsw_widget.get_val())
 121 htot_widget = MyIntEdit(u"htot  ",
 122         calc = lambda: hdisp_widget.get_val() + hfp_widget.get_val() + hsw_widget.get_val() + hbp_widget.get_val())
 123
 124 hwidgets1 = [hdisp_widget, hfp_widget, hsw_widget, hbp_widget]
 125 hwidgets2 = [hdisp2_widget, hss_widget, hse_widget, htot_widget]
 126
 127 horiz_pile1 = urwid.Pile(hwidgets1)
 128 horiz_pile2 = urwid.Pile(hwidgets2)
 129
 130 h_columns = urwid.LineBox(urwid.Columns([(15, horiz_pile1), (15, horiz_pile2)]), title = "Horizontal")
 131
 132 # Vertical columns
 133
 134 vdisp_widget = MyIntEdit(u"vdisp ", calc = lambda: vdisp2_widget.get_val())
 135 vfp_widget = MyIntEdit(u"vfp   ", calc = lambda: vss_widget.get_val() - vdisp_widget.get_val())
 136 vsw_widget = MyIntEdit(u"vsw   ", calc = lambda: vse_widget.get_val() - vss_widget.get_val())
 137 vbp_widget = MyIntEdit(u"vbp   ", calc = lambda: vtot_widget.get_val() - vse_widget.get_val())
 138
 139 vdisp2_widget = MyIntEdit(u"vdisp ", calc = lambda: vdisp_widget.get_val())
 140 vss_widget = MyIntEdit(u"vss   ",
 141         calc = lambda: vdisp_widget.get_val() + vfp_widget.get_val())
 142 vse_widget = MyIntEdit(u"vse   ",
 143         calc = lambda: vdisp_widget.get_val() + vfp_widget.get_val() + vsw_widget.get_val())
 144 vtot_widget = MyIntEdit(u"vtot  ",
 145         calc = lambda: vdisp_widget.get_val() + vfp_widget.get_val() + vsw_widget.get_val() + vbp_widget.get_val())
 146
 147 vwidgets1 = [vdisp_widget, vfp_widget, vsw_widget, vbp_widget]
 148 vwidgets2 = [vdisp2_widget, vss_widget, vse_widget, vtot_widget]
 149
 150 vert_pile1 = urwid.Pile(vwidgets1)
 151 vert_pile2 = urwid.Pile(vwidgets2)
 152
 153 v_columns = urwid.LineBox(urwid.Columns([(15, vert_pile1), (15, vert_pile2)]), title = "Vertical")
 154
 155 # Info widgets
 156
 157 line_us_widget = MyIntText(fmt="line {:.2f} us",
 158         calc = lambda: khz_to_us(pclk_khz_widget.get_val()) * htot_widget.get_val())
 159 line_khz_widget = MyIntText(fmt="line {:.2f} kHz",
 160         calc = lambda: div_or_zero(pclk_khz_widget.get_val(), htot_widget.get_val()))
 161
 162 frame_tot_widget = MyIntText(fmt="tot {} pix",
 163         calc = lambda: htot_widget.get_val() * vtot_widget.get_val())
 164 frame_us_widget = MyIntText(fmt="frame {:.2f} ms",
 165         calc = lambda: khz_to_us(pclk_khz_widget.get_val()) * htot_widget.get_val() * vtot_widget.get_val() / 1000)
 166 frame_khz_widget = MyIntText(fmt="frame {:.2f} Hz",
 167         calc = lambda: div_or_zero(pclk_khz_widget.get_val() * 1000,  htot_widget.get_val() * vtot_widget.get_val()))
 168
 169 info_box = urwid.LineBox(urwid.Pile([line_us_widget, line_khz_widget, urwid.Divider(), frame_tot_widget, frame_us_widget, frame_khz_widget]), title = "Info")
 170
 171 # Set update lists
 172
 173 recalc_list = [ pclk_ps_widget, line_us_widget, line_khz_widget, frame_tot_widget, frame_us_widget, frame_khz_widget ]
 174
 175 hdisp_widget.set_updlist([hdisp2_widget, hss_widget, hse_widget, htot_widget])
 176 hfp_widget.set_updlist([hss_widget, hse_widget, htot_widget])
 177 hsw_widget.set_updlist([hse_widget, htot_widget])
 178 hbp_widget.set_updlist([htot_widget])
 179 hdisp2_widget.set_updlist([hdisp_widget, hfp_widget])
 180 hss_widget.set_updlist([hfp_widget, hsw_widget])
 181 hse_widget.set_updlist([hsw_widget, hbp_widget])
 182 htot_widget.set_updlist([hbp_widget])
 183
 184 vdisp_widget.set_updlist([vdisp2_widget, vss_widget, vse_widget, vtot_widget])
 185 vfp_widget.set_updlist([vss_widget, vse_widget, vtot_widget])
 186 vsw_widget.set_updlist([vse_widget, vtot_widget])
 187 vbp_widget.set_updlist([vtot_widget])
 188 vdisp2_widget.set_updlist([vdisp_widget, vfp_widget])
 189 vss_widget.set_updlist([vfp_widget, vsw_widget])
 190 vse_widget.set_updlist([vsw_widget, vbp_widget])
 191 vtot_widget.set_updlist([vbp_widget])
 192
 193 # Flags
 194
 195 fb = None
 196
 197 DRM_MODE_FLAG_PHSYNC = (1<<0)
 198 DRM_MODE_FLAG_NHSYNC = (1<<1)
 199 DRM_MODE_FLAG_PVSYNC = (1<<2)
 200 DRM_MODE_FLAG_NVSYNC = (1<<3)
 201 DRM_MODE_FLAG_INTERLACE = (1<<4)
 202 DRM_MODE_FLAG_DBLCLK = (1<<12)
 203
 204 def mode_is_ilace(mode):
 205         return (mode.flags & DRM_MODE_FLAG_INTERLACE) != 0
 206
 207 def apply_mode():
 208         global fb
 209
 210         mode = pykms.Videomode()
 211         mode.clock = pclk_khz_widget.get_val()
 212
 213         mode.hdisplay = hdisp2_widget.get_val()
 214         mode.hsync_start = hss_widget.get_val()
 215         mode.hsync_end = hse_widget.get_val()
 216         mode.htotal = htot_widget.get_val()
 217
 218         mode.vdisplay = vdisp2_widget.get_val()
 219         mode.vsync_start = vss_widget.get_val()
 220         mode.vsync_end = vse_widget.get_val()
 221         mode.vtotal = vtot_widget.get_val()
 222
 223         if ilace_box.state:
 224                 mode.flags |= DRM_MODE_FLAG_INTERLACE
 225
 226         if dblclk_box.state:
 227                 mode.flags |= DRM_MODE_FLAG_DBLCLK
 228
 229         if hsync_pol.state == True:
 230                 mode.flags |= DRM_MODE_FLAG_PHSYNC
 231         elif hsync_pol.state == False:
 232                 mode.flags |= DRM_MODE_FLAG_NHSYNC
 233
 234         if vsync_pol.state == True:
 235                 mode.flags |= DRM_MODE_FLAG_PVSYNC
 236         elif vsync_pol.state == False:
 237                 mode.flags |= DRM_MODE_FLAG_NVSYNC
 238
 239         fb = pykms.DumbFramebuffer(card, mode.hdisplay, mode.vdisplay, "XR24");
 240         pykms.draw_test_pattern(fb);
 241
 242         crtc.set_mode(conn, fb, mode)
 243
 244 def read_mode(mode):
 245         pclk_khz_widget.set_val(mode.clock)
 246         hdisp2_widget.set_val(mode.hdisplay)
 247         hss_widget.set_val(mode.hsync_start)
 248         hse_widget.set_val(mode.hsync_end)
 249         htot_widget.set_val(mode.htotal)
 250
 251         vdisp2_widget.set_val(mode.vdisplay)
 252         vss_widget.set_val(mode.vsync_start)
 253         vse_widget.set_val(mode.vsync_end)
 254         vtot_widget.set_val(mode.vtotal)
 255
 256         ilace_box.set_state(mode_is_ilace(mode))
 257         dblclk_box.set_state((mode.flags & DRM_MODE_FLAG_DBLCLK) != 0)
 258
 259         sync = 'mixed'
 260         if (mode.flags & DRM_MODE_FLAG_PHSYNC) != 0:
 261                 sync = True
 262         elif (mode.flags & DRM_MODE_FLAG_NHSYNC) != 0:
 263                 sync = False
 264         hsync_pol.set_state(sync)
 265
 266         sync = 'mixed'
 267         if (mode.flags & DRM_MODE_FLAG_PVSYNC) != 0:
 268                 sync = True
 269         elif (mode.flags & DRM_MODE_FLAG_NVSYNC) != 0:
 270                 sync = False
 271         vsync_pol.set_state(sync)
 272
 273 def apply_press(w):
 274         apply_mode()
 275
 276 ilace_box = urwid.CheckBox('interlace')
 277 hsync_pol = urwid.CheckBox('hsync positive', has_mixed=True)
 278 vsync_pol = urwid.CheckBox('vsync positive', has_mixed=True)
 279 dblclk_box = urwid.CheckBox('double clock')
 280
 281 flags_pile = urwid.LineBox(urwid.Pile([ilace_box, hsync_pol, vsync_pol, dblclk_box]), title = "Flags")
 282
 283 apply_button = urwid.LineBox(urwid.Padding(urwid.Button('apply', on_press=apply_press)))
 284
 285 # Main
 286
 287 def mode_press(w, mode):
 288         read_mode(mode)
 289
 290 def mode_to_str(mode):
 291         return "{}@{}{}".format(mode.name, mode.vrefresh, "i" if mode_is_ilace(mode) else "")
 292
 293 mode_buttons = []
 294
 295 card = pykms.Card()
 296 conn = card.get_first_connected_connector()
 297 crtc = conn.get_current_crtc()
 298 modes = conn.get_modes()
 299 i = 0
 300 for m in modes:
 301         mode_buttons.append(urwid.Button(mode_to_str(m), on_press=mode_press, user_data=m))
 302         i += 1
 303
 304 modes_pile = urwid.LineBox(urwid.Pile(mode_buttons), title = "Video modes")
 305
 306 main_pile = urwid.Pile([modes_pile, pclk_columns, urwid.Columns([ h_columns, v_columns ]), info_box, flags_pile, apply_button])
 307
 308 main_columns = urwid.Filler(main_pile, valign='top')
 309
 310 loop = urwid.MainLoop(main_columns, unhandled_input=exit_on_q, handle_mouse=False)
 311
 312 # select the first mode
 313 mode_press(None, modes[0])
 314
 315 loop.run()
 316
 317 fb = None