Fix line-endings; Increase portability; Add speedtyper.py

python/speedtyper.py

@@ -0,0 +1,97 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+import sys
+import curses
+
+
+class Word():
+    def __init__(self, target, x, y):
+        self.target = target
+        self.input = ""
+        self.x = x
+        self.y = y
+
+
+def get_words(text, max_x, max_y):
+    curr_x = 0
+    curr_y = 0
+
+    words = []
+    for word in text.split():
+        if curr_x + len(word) > max_x:
+            curr_y += 1
+            curr_x = 0
+
+        if curr_y > max_y:
+            print("text too long, scrolling not implemented.")
+            exit(1)
+
+        words.append(Word(word, curr_x, curr_y))
+        curr_x += len(word) + 1
+
+    return words
+
+
+def main_curses(stdscr, text):
+    curses.use_default_colors()
+    curses.init_pair(1, curses.COLOR_GREEN, -1)
+    curses.init_pair(2, curses.COLOR_WHITE, curses.COLOR_RED)
+
+    key = old_max_y = old_max_x = curr_word = 0
+
+    while True:
+        # rewrap words on screen size change
+        max_y, max_x = stdscr.getmaxyx()
+        if max_y != old_max_y or max_x != old_max_x:
+            words = get_words(text, max_x, max_y)
+            old_max_x = max_x
+            old_max_y = max_y
+
+        if key in (curses.KEY_BACKSPACE, 'KEY_BACKSPACE', '\b', '\x7f', 127): # fml
+            words[curr_word].input = words[curr_word].input[:-1]
+        elif key >= 32 and key <= 126:
+            words[curr_word].input += chr(key)
+
+        # increment current word if its completed
+        if words[curr_word].input == words[curr_word].target:
+            curr_word += 1
+
+            # text complete
+            if curr_word == len(words):
+                break
+
+        stdscr.clear()
+
+        for word in words:
+            if word.input == word.target:
+                stdscr.addstr(word.y, word.x, word.target, curses.A_DIM)
+            elif word == words[curr_word]:
+                stdscr.addstr(word.y, word.x, word.target, curses.A_UNDERLINE)
+            else:
+                stdscr.addstr(word.y, word.x, word.target)
+
+        for i, c in enumerate(words[curr_word].input):
+            if i >= len(words[curr_word].target) or words[curr_word].input[i] != words[curr_word].target[i]:
+                stdscr.addstr(words[curr_word].y, words[curr_word].x + i, c, curses.color_pair(2) | curses.A_BOLD)
+            else:
+                stdscr.addstr(words[curr_word].y, words[curr_word].x + i, c, curses.color_pair(1))
+
+        stdscr.move(words[curr_word].y, words[curr_word].x + len(words[curr_word].input))
+
+        key = stdscr.getch()
+
+
+def main(argv):
+    from argparse import ArgumentParser
+    parser = ArgumentParser()
+    parser.add_argument("input")
+    args = parser.parse_args(argv[1:])
+
+    text = open(args.input).read()
+
+    curses.wrapper(main_curses, text)
+
+
+if __name__ == "__main__":
+    main(sys.argv)