13

2023/sketch_2023_10_13/folder_browser_selection.txt

@@ -0,0 +1,4 @@
+/home/villares/GitHub/sketch-a-day/2022/sketch_2022_01_01a
+/home/villares/GitHub/sketch-a-day/2022/sketch_2022_01_02a
+/home/villares/GitHub/sketch-a-day/2022/sketch_2022_01_03a
+/home/villares/GitHub/sketch-a-day/2022/sketch_2022_01_04a

2023/sketch_2023_10_13/folder_diff.py

@@ -0,0 +1,161 @@
+#!/home/villares/thonny-python-env/bin/python3
+
+"""
+Experiment on diffing sketches
+
+TODO: Improve image files & code files accepted
+
+"""
+
+import PIL.Image
+import py5
+import difflib
+from pathlib import Path
+
+SAVED_SELECTION = Path(__file__).parent / 'folder_browser_selection.txt'
+
+image_paths = []
+img = None
+current = 0
+code = previous_code = ''
+font_size, line_step = 12, 12
+run = False
+diff = True
+scroll_offset = 0
+
+def setup():
+    py5.size(1800, 1000)
+    setup_folder_diff()
+    image_paths[:] = py5.load_strings(SAVED_SELECTION)
+    walk_images(0)
+
+def setup_folder_diff():    
+    global fr, fb
+    fr = py5.create_font("Source Code Pro", font_size)
+    fb = py5.create_font("Source Code Pro Bold", font_size)
+
+def draw():
+    py5.text_align(py5.LEFT, py5.TOP)
+    py5.background(240)
+    if image_paths and img:
+        py5.image(img, 50, 50)
+        draw_code(img.width + 100, 80)
+        py5.no_stroke()
+        py5.fill(240)
+        py5.rect(0, 0, py5.width, 50)
+        py5.fill(0)
+        py5.text(image_paths[current], 50, 30)
+        if diff:
+            py5.fill(0, 100, 0)
+            py5.text(f'previous: {image_paths[previous]}', 600, 30)
+
+    if run:
+        py5.save(f'a-{current}-{current_path.stem}.png')
+        walk_images(1)
+        if current == 0:
+            py5.exit_sketch()
+
+def walk_images(i):
+    global current, previous, img, code, previous_code
+    if len(image_paths) == 0:
+        return
+    previous, current = current, (current + i) % len(image_paths)
+    current_path = Path(image_paths[current])
+    img, new_code = get_img_and_code(current_path)
+    previous_code, code = code, new_code
+
+def get_img_and_code(path, resize=None):
+    if resize is None:
+        resize = (py5.height - 100, py5.height - 100)
+    img, code = None, ''
+    try:
+        if path.is_file():
+            img, code = load_image_and_data(path, resize)
+        elif path.is_dir():
+            # TODO: search any valid image format
+            image_path = path / (path.name + '.png')
+            if not image_path.is_file():
+                image_path = path / (path.name + '.gif')
+            img, code = load_image_and_data(image_path, resize)
+            if len(code) == 0:
+                # TODO: search any code/text file
+                code_path = path / (path.name + '.py')
+                if not code_path.is_file():
+                    code_path = path / (path.name + '.pyde')                
+                code = code_path.read_text()
+    except Exception as e:
+        print(str(e))
+    return img, code
+
+def load_image_and_data(image_file, resize=None):
+    img = PIL.Image.open(image_file)
+    if resize:
+        img.thumbnail(resize)
+    code = img.info.get('code', '').replace('    \n', '\n').replace('\n\n\n', '')
+    return py5.convert_image(img), code
+
+def list_files(folder):
+    global image_paths
+    try:
+        file_list = sorted(Path(folder).iter_dir())  # get list of files in folder
+    except:
+        file_list = []
+    image_paths = [
+            f for f in file_list
+            if f.isfile() and f.suffix.lower()
+            in ('.png', '.jpg', 'jpeg', '.tga', '.webp',
+               '.tiff', '.tif', '.bmp', '.gif')]
+
+def draw_code(x, y):
+    with py5.push():
+        py5.translate(0, line_step * scroll_offset)
+        py5.text_font(fb)
+        differ = difflib.Differ()
+        colors = {' ': 0, '-': 0xFFFF0000, '+': 0xFF0000FF, '?': 0xFF00CC00}
+        if previous_code and diff:
+            code_lines = differ.compare(previous_code.splitlines(), code.splitlines())
+        else:
+            code_lines = code.splitlines()
+        for li in code_lines:
+            py5.fill(colors.get(li[:1], 0))
+            py5.text(li, x, y)
+            py5.translate(0, line_step )
+
+
+def key_pressed():
+    global run, current, previous, previous_code, diff
+    global scroll_offset
+    print(py5.key)
+    if py5.key == 'o':
+        py5.select_folder('Select a folder', list_files)
+    elif py5.key == 'u':
+        image_paths[:] = py5.load_strings(SAVED_SELECTION)
+        previous_code = ''
+        current = 0
+        walk_images(0)
+    elif py5.key == 'd':
+        diff = not diff
+    elif py5.key == 'p':
+        current_path = Path(image_paths[current])
+        py5.save(f'{current}-{current_path.stem}.png')
+    elif py5.key == 'r':
+        previous_code = ''
+        current = 0
+        run = True
+    elif py5.key == py5.ESC:
+        py5.exit_sketch()
+    elif py5.key_code == py5.DOWN:
+        walk_images(1)
+    elif py5.key_code == py5.UP:
+        walk_images(-1)
+    elif py5.key_code == py5.LEFT:
+        scroll_offset = 0
+
+
+def mouse_wheel(e):
+    global scroll_offset
+    d = e.get_count()
+    scroll_offset += d
+
+if __name__ == '__main__':
+    py5.run_sketch()