Basic text editing implemented in pygame

.gitignore

@@ -1,2 +1 @@
-/rust-version/
-/main
+/cosmic-text/

fide.py

@@ -0,0 +1,173 @@
+import pygame as pg
+from pygame import Rect
+
+from piece_table import PieceTable
+
+import os
+import json
+import sys
+import time
+import copy
+from dataclasses import dataclass
+
+class Cursor:
+    def __init__(self, row, col):
+        self.row: int = 0
+        self.col: int = 0
+        self.want_col: int = 0
+
+    def pos(self):
+        return (self.row, self.col)
+
+class Row:
+    def __init__(self, text):
+        self.text = PieceTable(text)
+        self.length = len(text)
+
+class Buffer:
+    def __init__(self, raw_text):
+        pass
+
+oswinx, oswiny = 1940,50
+os.environ['SDL_VIDEO_WINDOW_POS'] = "%d,%d" % (oswinx,oswiny)
+clock = pg.time.Clock()
+
+pg.init()
+pg.font.init()
+text_renderer = pg.font.SysFont('JetBrains Mono', 14)
+screen = pg.display.set_mode((900, 900), pg.RESIZABLE)
+pg.key.set_repeat(240, 10)
+
+rows = []
+cursor = Cursor(0, 0)
+line_height = 20
+char_width = 8
+
+with open('../gux/test.c', 'r', encoding='utf-8') as f:
+    text = f.read()
+    for line in text.split('\n'):
+        rows.append(Row(line))
+    # print(current_buffer[120:125])
+    f.close()
+
+def draw_cursor():
+    rect = Rect(cursor.col * char_width, cursor.row * line_height, 3, line_height)
+    pg.draw.rect(screen, "black", rect, width=2)
+
+if __name__ == '__main__':
+    running = True
+    # print(current_buffer.display())
+    frame_count = 0
+    cursor_flash = 0
+    cursor_on = True
+    while running:
+        dt = clock.tick(120) / 1000.0
+        screen.fill('white')
+        frame_count += 1
+        for event in pg.event.get():
+            if event.type == pg.QUIT:
+                running = False
+            if event.type == pg.KEYDOWN:
+                # print(frame_count, event.unicode, pg.key.name(event.key), event.mod, end="")
+                # print("")
+                # print(event.mod)
+                cursor_on = True
+                cursor_flash = 0
+                row = rows[cursor.row]
+                if event.key == pg.K_RETURN:
+                    if cursor.col < row.length:
+                        remainder = row.length - cursor.col
+                        line = row.text.string_at(cursor.col, remainder)
+                        row.text.delete(cursor.col, remainder)
+                        row.length -= remainder
+                    else:
+                        line = ""
+                    rows.insert(cursor.row+1, Row(str(line)))
+                    cursor.row += 1
+                    # Eventually when we need to be indentation aware
+                    cursor.col = 0
+                    cursor.want_col = 0
+                elif event.key == pg.K_BACKSPACE:
+                    if cursor.pos() != (0,0):
+                        if cursor.col == 0:
+                            row = rows.pop(cursor.row)
+                            cursor.row -= 1
+                            above_row = rows[cursor.row]
+                            above_row.text.insert(row.text.get_text(), above_row.length)
+                            cursor.col = above_row.length
+                            cursor.want_col = cursor.col
+                            above_row.length += row.length
+                        else:
+                            row.text.delete(cursor.col - 1, 1)
+                            row.length -= 1
+                            cursor.col = max(0, cursor.col - 1)
+                            cursor.want_col = cursor.col
+                elif event.key == pg.K_DELETE:
+                    if cursor.col >= rows[cursor.row].length:
+                        if cursor.row < len(rows) - 1:
+                            below_row = rows.pop(cursor.row+1)
+                            row = rows[cursor.row]
+                            row.text.insert(below_row.text.get_text(), row.length)
+                            cursor.want_col = cursor.col
+                            row.length += below_row.length
+                    else:
+                        row.text.delete(cursor.col, 1)
+                        row.length -= 1
+                        cursor.want_col = cursor.col
+                elif event.key == pg.K_LEFT:
+                    if cursor.col == 0:
+                        if cursor.row > 0:
+                            cursor.col = rows[cursor.row-1].length
+                            cursor.want_col = cursor.col
+                            cursor.row -= 1
+                    else:
+                        cursor.col = max(0, cursor.col - 1)
+                        cursor.want_col = cursor.col
+                elif event.key == pg.K_RIGHT:
+                    if cursor.col == rows[cursor.row].length:
+                        if cursor.row < len(rows):
+                            cursor.col = 0
+                            cursor.want_col = cursor.col
+                            cursor.row += 1
+                    else:
+                        cursor.col = min(rows[cursor.row].length, cursor.col + 1)
+                        cursor.want_col = cursor.col
+                elif event.key == pg.K_UP:
+                    cursor.row = max(0, cursor.row - 1)
+                    cursor.col = min(cursor.want_col, rows[cursor.row].length)
+                elif event.key == pg.K_DOWN:
+                    cursor.row = min(len(rows) - 1, cursor.row + 1)
+                    cursor.col = min(cursor.want_col, rows[cursor.row].length)
+                elif event.key == pg.K_SPACE:
+                    row.text.insert(u' ', cursor.col)
+                    row.length += 1
+                    cursor.col += 1
+                    cursor.want_col = cursor.col
+                elif event.key >= 33 and event.key <= 126 and event.mod < 2:
+                    row.text.insert(event.unicode, cursor.col)
+                    row.length += 1
+                    cursor.col += 1
+                    cursor.want_col = cursor.col
+                elif event.mod == 64: # Right Ctrl
+                    if event.key == pg.K_e:
+                        cursor.col = rows[cursor.row].length
+                        cursor.want_col = cursor.col
+                    if event.key == pg.K_a:
+                        cursor.col = 0
+                        cursor.want_col = cursor.col
+
+        for i,row in enumerate(rows):
+            render = text_renderer.render(row.text.get_text(), True, 'black')
+            screen.blit(render, (0, i * line_height))
+
+        if cursor_on and cursor_flash < 0.5:
+            draw_cursor()
+        elif cursor_on:
+            cursor_on = False
+            cursor_flash = 0
+        elif not cursor_on and cursor_flash > 0.5:
+            cursor_on = True
+            cursor_flash = 0
+        cursor_flash += dt
+
+        pg.display.flip()

requirements.txt

@@ -0,0 +1,2 @@
+pygame-ce==2.3.2
+pyropes==1.4