summaryrefslogtreecommitdiff
path: root/py/kmsmodeview.py
diff options
context:
space:
mode:
Diffstat (limited to 'py/kmsmodeview.py')
-rwxr-xr-xpy/kmsmodeview.py317
1 files changed, 317 insertions, 0 deletions
diff --git a/py/kmsmodeview.py b/py/kmsmodeview.py
new file mode 100755
index 0000000..355db02
--- /dev/null
+++ b/py/kmsmodeview.py
@@ -0,0 +1,317 @@
+#!/usr/bin/python3
+
+import urwid
+import pykms
+
+def exit_on_q(key):
+ if key in ('q', 'Q'):
+ raise urwid.ExitMainLoop()
+ elif key == 'a':
+ apply_mode()
+
+alarm_handle = None
+
+def recalc_info(l, d):
+ global alarm_handle
+
+ alarm_handle = None
+
+ for w in recalc_list:
+ w.recalc()
+
+def div_or_zero(n, d):
+ if d == 0:
+ return 0
+ else:
+ return n / d
+
+class MyIntEdit(urwid.IntEdit):
+ _metaclass_ = urwid.signals.MetaSignals
+ signals = ['value_change']
+
+ def __init__(self, caption, calc=None):
+ self._myval = 0
+ self._disable_change = False
+ self._calc = calc
+ self._updlist = None
+
+ super().__init__(caption, 0)
+
+ def set_edit_text(self, text):
+ global alarm_handle
+
+ super().set_edit_text(text)
+ newtext = super().get_edit_text()
+ new_val = int(newtext) if newtext != "" else 0
+ if new_val != self._myval:
+ self._myval = new_val
+ if not self._disable_change:
+ urwid.emit_signal(self, 'value_change', self, self._myval)
+
+ if alarm_handle == None:
+ alarm_handle = loop.set_alarm_in(0, recalc_info)
+
+ if self._updlist != None:
+ for w in self._updlist:
+ w.recalc()
+
+ def recalc(self):
+ self._disable_change = True
+ self.set_val(self._calc())
+ self._disable_change = False
+
+ def set_val(self, val):
+ self.set_edit_text(str(int(val)))
+
+ def get_val(self):
+ return self._myval
+
+ def set_updlist(self, list):
+ self._updlist = list
+
+ def keypress(self, size, key):
+ if key == '+':
+ self.set_edit_text(str(self.value() + 1))
+ elif key == '-':
+ self.set_edit_text(str(self.value() - 1))
+ else:
+ return super().keypress(size, key)
+
+class MyIntText(urwid.Text):
+ def __init__(self, fmt, calc=None):
+ super().__init__("")
+ self._fmt = fmt
+ self._calc = calc
+
+ def recalc(self):
+ val = self._calc()
+ super().set_text(self._fmt.format(val))
+
+def khz_to_ps(khz):
+ if khz == 0:
+ return 0
+ else:
+ return 1.0 / khz * 1000 * 1000 * 1000
+
+def khz_to_us(khz):
+ if khz == 0:
+ return 0
+ else:
+ return 1.0 / khz * 1000
+
+pclk_khz_widget = MyIntEdit(u"pclk (kHz) ")
+pclk_ps_widget = MyIntText(fmt="pclk {:.2f} ps", calc = lambda: khz_to_ps(pclk_khz_widget.get_val()))
+
+pclk_widgets = [pclk_khz_widget, pclk_ps_widget]
+
+pclk_columns = urwid.LineBox(urwid.Columns(pclk_widgets), title = "Pixel clock")
+
+# Horizontal widgets
+
+hdisp_widget = MyIntEdit(u"hdisp ", calc = lambda: hdisp2_widget.get_val())
+hfp_widget = MyIntEdit(u"hfp ", calc = lambda: hss_widget.get_val() - hdisp_widget.get_val())
+hsw_widget = MyIntEdit(u"hsw ", calc = lambda: hse_widget.get_val() - hss_widget.get_val())
+hbp_widget = MyIntEdit(u"hbp ", calc = lambda: htot_widget.get_val() - hse_widget.get_val())
+
+hdisp2_widget = MyIntEdit(u"hdisp ", calc = lambda: hdisp_widget.get_val())
+hss_widget = MyIntEdit(u"hss ",
+ calc = lambda: hdisp_widget.get_val() + hfp_widget.get_val())
+hse_widget = MyIntEdit(u"hse ",
+ calc = lambda: hdisp_widget.get_val() + hfp_widget.get_val() + hsw_widget.get_val())
+htot_widget = MyIntEdit(u"htot ",
+ calc = lambda: hdisp_widget.get_val() + hfp_widget.get_val() + hsw_widget.get_val() + hbp_widget.get_val())
+
+hwidgets1 = [hdisp_widget, hfp_widget, hsw_widget, hbp_widget]
+hwidgets2 = [hdisp2_widget, hss_widget, hse_widget, htot_widget]
+
+horiz_pile1 = urwid.Pile(hwidgets1)
+horiz_pile2 = urwid.Pile(hwidgets2)
+
+h_columns = urwid.LineBox(urwid.Columns([(15, horiz_pile1), (15, horiz_pile2)]), title = "Horizontal")
+
+# Vertical columns
+
+vdisp_widget = MyIntEdit(u"vdisp ", calc = lambda: vdisp2_widget.get_val())
+vfp_widget = MyIntEdit(u"vfp ", calc = lambda: vss_widget.get_val() - vdisp_widget.get_val())
+vsw_widget = MyIntEdit(u"vsw ", calc = lambda: vse_widget.get_val() - vss_widget.get_val())
+vbp_widget = MyIntEdit(u"vbp ", calc = lambda: vtot_widget.get_val() - vse_widget.get_val())
+
+vdisp2_widget = MyIntEdit(u"vdisp ", calc = lambda: vdisp_widget.get_val())
+vss_widget = MyIntEdit(u"vss ",
+ calc = lambda: vdisp_widget.get_val() + vfp_widget.get_val())
+vse_widget = MyIntEdit(u"vse ",
+ calc = lambda: vdisp_widget.get_val() + vfp_widget.get_val() + vsw_widget.get_val())
+vtot_widget = MyIntEdit(u"vtot ",
+ calc = lambda: vdisp_widget.get_val() + vfp_widget.get_val() + vsw_widget.get_val() + vbp_widget.get_val())
+
+vwidgets1 = [vdisp_widget, vfp_widget, vsw_widget, vbp_widget]
+vwidgets2 = [vdisp2_widget, vss_widget, vse_widget, vtot_widget]
+
+vert_pile1 = urwid.Pile(vwidgets1)
+vert_pile2 = urwid.Pile(vwidgets2)
+
+v_columns = urwid.LineBox(urwid.Columns([(15, vert_pile1), (15, vert_pile2)]), title = "Vertical")
+
+# Info widgets
+
+line_us_widget = MyIntText(fmt="line {:.2f} us",
+ calc = lambda: khz_to_us(pclk_khz_widget.get_val()) * htot_widget.get_val())
+line_khz_widget = MyIntText(fmt="line {:.2f} kHz",
+ calc = lambda: div_or_zero(pclk_khz_widget.get_val(), htot_widget.get_val()))
+
+frame_tot_widget = MyIntText(fmt="tot {} pix",
+ calc = lambda: htot_widget.get_val() * vtot_widget.get_val())
+frame_us_widget = MyIntText(fmt="frame {:.2f} ms",
+ calc = lambda: khz_to_us(pclk_khz_widget.get_val()) * htot_widget.get_val() * vtot_widget.get_val() / 1000)
+frame_khz_widget = MyIntText(fmt="frame {:.2f} Hz",
+ calc = lambda: div_or_zero(pclk_khz_widget.get_val() * 1000, htot_widget.get_val() * vtot_widget.get_val()))
+
+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")
+
+# Set update lists
+
+recalc_list = [ pclk_ps_widget, line_us_widget, line_khz_widget, frame_tot_widget, frame_us_widget, frame_khz_widget ]
+
+hdisp_widget.set_updlist([hdisp2_widget, hss_widget, hse_widget, htot_widget])
+hfp_widget.set_updlist([hss_widget, hse_widget, htot_widget])
+hsw_widget.set_updlist([hse_widget, htot_widget])
+hbp_widget.set_updlist([htot_widget])
+hdisp2_widget.set_updlist([hdisp_widget, hfp_widget])
+hss_widget.set_updlist([hfp_widget, hsw_widget])
+hse_widget.set_updlist([hsw_widget, hbp_widget])
+htot_widget.set_updlist([hbp_widget])
+
+vdisp_widget.set_updlist([vdisp2_widget, vss_widget, vse_widget, vtot_widget])
+vfp_widget.set_updlist([vss_widget, vse_widget, vtot_widget])
+vsw_widget.set_updlist([vse_widget, vtot_widget])
+vbp_widget.set_updlist([vtot_widget])
+vdisp2_widget.set_updlist([vdisp_widget, vfp_widget])
+vss_widget.set_updlist([vfp_widget, vsw_widget])
+vse_widget.set_updlist([vsw_widget, vbp_widget])
+vtot_widget.set_updlist([vbp_widget])
+
+# Flags
+
+fb = None
+
+DRM_MODE_FLAG_PHSYNC = (1<<0)
+DRM_MODE_FLAG_NHSYNC = (1<<1)
+DRM_MODE_FLAG_PVSYNC = (1<<2)
+DRM_MODE_FLAG_NVSYNC = (1<<3)
+DRM_MODE_FLAG_INTERLACE = (1<<4)
+DRM_MODE_FLAG_DBLCLK = (1<<12)
+
+def mode_is_ilace(mode):
+ return (mode.flags & DRM_MODE_FLAG_INTERLACE) != 0
+
+def apply_mode():
+ global fb
+
+ mode = pykms.Videomode()
+ mode.clock = pclk_khz_widget.get_val()
+
+ mode.hdisplay = hdisp2_widget.get_val()
+ mode.hsync_start = hss_widget.get_val()
+ mode.hsync_end = hse_widget.get_val()
+ mode.htotal = htot_widget.get_val()
+
+ mode.vdisplay = vdisp2_widget.get_val()
+ mode.vsync_start = vss_widget.get_val()
+ mode.vsync_end = vse_widget.get_val()
+ mode.vtotal = vtot_widget.get_val()
+
+ if ilace_box.state:
+ mode.flags |= DRM_MODE_FLAG_INTERLACE
+
+ if dblclk_box.state:
+ mode.flags |= DRM_MODE_FLAG_DBLCLK
+
+ if hsync_pol.state == True:
+ mode.flags |= DRM_MODE_FLAG_PHSYNC
+ elif hsync_pol.state == False:
+ mode.flags |= DRM_MODE_FLAG_NHSYNC
+
+ if vsync_pol.state == True:
+ mode.flags |= DRM_MODE_FLAG_PVSYNC
+ elif vsync_pol.state == False:
+ mode.flags |= DRM_MODE_FLAG_NVSYNC
+
+ fb = pykms.DumbFramebuffer(card, mode.hdisplay, mode.vdisplay, "XR24");
+ pykms.draw_test_pattern(fb);
+
+ crtc.set_mode(conn, fb, mode)
+
+def read_mode(mode):
+ pclk_khz_widget.set_val(mode.clock)
+ hdisp2_widget.set_val(mode.hdisplay)
+ hss_widget.set_val(mode.hsync_start)
+ hse_widget.set_val(mode.hsync_end)
+ htot_widget.set_val(mode.htotal)
+
+ vdisp2_widget.set_val(mode.vdisplay)
+ vss_widget.set_val(mode.vsync_start)
+ vse_widget.set_val(mode.vsync_end)
+ vtot_widget.set_val(mode.vtotal)
+
+ ilace_box.set_state(mode_is_ilace(mode))
+ dblclk_box.set_state((mode.flags & DRM_MODE_FLAG_DBLCLK) != 0)
+
+ sync = 'mixed'
+ if (mode.flags & DRM_MODE_FLAG_PHSYNC) != 0:
+ sync = True
+ elif (mode.flags & DRM_MODE_FLAG_NHSYNC) != 0:
+ sync = False
+ hsync_pol.set_state(sync)
+
+ sync = 'mixed'
+ if (mode.flags & DRM_MODE_FLAG_PVSYNC) != 0:
+ sync = True
+ elif (mode.flags & DRM_MODE_FLAG_NVSYNC) != 0:
+ sync = False
+ vsync_pol.set_state(sync)
+
+def apply_press(w):
+ apply_mode()
+
+ilace_box = urwid.CheckBox('interlace')
+hsync_pol = urwid.CheckBox('hsync positive', has_mixed=True)
+vsync_pol = urwid.CheckBox('vsync positive', has_mixed=True)
+dblclk_box = urwid.CheckBox('double clock')
+
+flags_pile = urwid.LineBox(urwid.Pile([ilace_box, hsync_pol, vsync_pol, dblclk_box]), title = "Flags")
+
+apply_button = urwid.LineBox(urwid.Padding(urwid.Button('apply', on_press=apply_press)))
+
+# Main
+
+def mode_press(w, mode):
+ read_mode(mode)
+
+def mode_to_str(mode):
+ return "{}@{}{}".format(mode.name, mode.vrefresh, "i" if mode_is_ilace(mode) else "")
+
+mode_buttons = []
+
+card = pykms.Card()
+conn = card.get_first_connected_connector()
+crtc = conn.get_current_crtc()
+modes = conn.get_modes()
+i = 0
+for m in modes:
+ mode_buttons.append(urwid.Button(mode_to_str(m), on_press=mode_press, user_data=m))
+ i += 1
+
+modes_pile = urwid.LineBox(urwid.Pile(mode_buttons), title = "Video modes")
+
+main_pile = urwid.Pile([modes_pile, pclk_columns, urwid.Columns([ h_columns, v_columns ]), info_box, flags_pile, apply_button])
+
+main_columns = urwid.Filler(main_pile, valign='top')
+
+loop = urwid.MainLoop(main_columns, unhandled_input=exit_on_q, handle_mouse=False)
+
+# select the first mode
+mode_press(None, modes[0])
+
+loop.run()
+
+fb = None