split into files
This commit is contained in:
klemek
@@ -0,0 +1,254 @@
|
||||
import tkinter as tk
|
||||
from tkinter import ttk, filedialog
|
||||
from typing import Optional
|
||||
|
||||
from .explorer import Explorer
|
||||
from .image_view import ImageView
|
||||
from .file import File
|
||||
from .image import Image
|
||||
from .bitmap import Bitmap
|
||||
|
||||
|
||||
class App(ttk.Frame):
|
||||
def __init__(self, parent) -> None:
|
||||
super().__init__(parent)
|
||||
|
||||
parent.option_add("*tearOff", tk.FALSE)
|
||||
parent.resizable(False, False)
|
||||
|
||||
self.parent = parent
|
||||
self.current_file = None
|
||||
|
||||
self.explorer = Explorer(self, self.update)
|
||||
self.explorer.grid(column=0, row=0, sticky="nsw")
|
||||
|
||||
self.image_view = ImageView(self)
|
||||
self.image_view.grid(column=1, row=0, sticky="nsew")
|
||||
|
||||
self.init_menus()
|
||||
|
||||
self.grid_rowconfigure(0, weight=1)
|
||||
self.grid_columnconfigure(0, weight=1)
|
||||
|
||||
self.open_file(None)
|
||||
|
||||
self.pack(fill="both", expand=True)
|
||||
|
||||
@property
|
||||
def current_image(self) -> Optional[Image]:
|
||||
if self.current_file is None:
|
||||
return None
|
||||
else:
|
||||
return self.explorer.current_image
|
||||
|
||||
def update(self, force: bool = False) -> None:
|
||||
self.update_title()
|
||||
self.update_menus()
|
||||
self.image_view.update(self.current_image)
|
||||
self.explorer.update(self.current_file, force)
|
||||
|
||||
def update_title(self) -> None:
|
||||
title = "Watchy Image Editor"
|
||||
if self.current_file is not None:
|
||||
title += "- "
|
||||
if self.current_file.path is None:
|
||||
title += "New file"
|
||||
else:
|
||||
title += self.current_file.filename
|
||||
if self.current_file.modified:
|
||||
title += "*"
|
||||
self.parent.title(title)
|
||||
|
||||
def init_menus(self) -> None:
|
||||
self.menubar = tk.Menu(self.parent)
|
||||
self.parent["menu"] = self.menubar
|
||||
|
||||
self.menu_file = tk.Menu(self.menubar)
|
||||
self.menubar.add_cascade(menu=self.menu_file, label="File")
|
||||
|
||||
# TODO better handling of menu items
|
||||
|
||||
self.menu_file_need_file = []
|
||||
|
||||
self.menu_file.add_command(label="New File", command=lambda: self.open_file(""))
|
||||
i = 0
|
||||
self.menu_file.add_command(
|
||||
label="Open File...",
|
||||
command=lambda: self.open_file(
|
||||
filedialog.askopenfilename(
|
||||
filetypes=File.FILE_TYPES,
|
||||
defaultextension=File.FILE_TYPES,
|
||||
)
|
||||
),
|
||||
)
|
||||
i += 1
|
||||
self.menu_file.add_command(
|
||||
label="Save File",
|
||||
command=lambda: self.save_file(self.current_file.path),
|
||||
)
|
||||
i += 1
|
||||
self.menu_file_need_file += [i]
|
||||
self.menu_file.add_command(
|
||||
label="Save File As...",
|
||||
command=lambda: self.save_file(
|
||||
filedialog.asksaveasfilename(
|
||||
filetypes=File.FILE_TYPES, defaultextension=File.FILE_TYPES
|
||||
)
|
||||
),
|
||||
)
|
||||
i += 1
|
||||
self.menu_file_need_file += [i]
|
||||
self.menu_file.add_command(
|
||||
label="Close File",
|
||||
command=lambda: self.open_file(None),
|
||||
)
|
||||
i += 1
|
||||
self.menu_file_need_file += [i]
|
||||
self.menu_file.add_separator()
|
||||
i += 1
|
||||
self.menu_file.add_command(
|
||||
label="New image...",
|
||||
command=self.add_image,
|
||||
state="disabled",
|
||||
)
|
||||
i += 1
|
||||
self.menu_file_need_file += [i]
|
||||
|
||||
self.menu_image = tk.Menu(self.menubar)
|
||||
self.menubar.add_cascade(menu=self.menu_image, label="Image")
|
||||
|
||||
self.menu_image.add_command(
|
||||
label="Edit name...",
|
||||
command=self.edit_image_name,
|
||||
state="disabled",
|
||||
)
|
||||
self.menu_image.add_command(
|
||||
label="Edit size...",
|
||||
command=self.edit_image_size,
|
||||
state="disabled",
|
||||
)
|
||||
self.menu_image.add_command(
|
||||
label="Move up",
|
||||
command=self.move_image_up,
|
||||
state="disabled",
|
||||
)
|
||||
self.menu_image.add_command(
|
||||
label="Move down",
|
||||
command=self.move_image_down,
|
||||
state="disabled",
|
||||
)
|
||||
self.menu_image.add_command(
|
||||
label="Delete",
|
||||
command=self.delete_image,
|
||||
state="disabled",
|
||||
)
|
||||
|
||||
self.menu_bmp = tk.Menu(self.menubar)
|
||||
self.menubar.add_cascade(menu=self.menu_bmp, label="Bitmap")
|
||||
|
||||
self.menu_bmp_need_image = []
|
||||
|
||||
self.menu_bmp.add_command(
|
||||
label="Bulk .bmp import...",
|
||||
command=self.import_all_bmp,
|
||||
)
|
||||
i = 0
|
||||
self.menu_bmp.add_command(
|
||||
label="Export all to .bmp...",
|
||||
command=self.export_all_bmp,
|
||||
)
|
||||
i += 1
|
||||
self.menu_bmp.add_separator()
|
||||
i += 1
|
||||
self.menu_bmp.add_command(
|
||||
label="Import .bmp into image...",
|
||||
command=self.import_bmp,
|
||||
state="disabled",
|
||||
)
|
||||
i += 1
|
||||
self.menu_bmp_need_image += [i]
|
||||
self.menu_bmp.add_command(
|
||||
label="Export image to .bmp...",
|
||||
command=self.export_bmp,
|
||||
state="disabled",
|
||||
)
|
||||
i += 1
|
||||
self.menu_bmp_need_image += [i]
|
||||
|
||||
def update_menus(self) -> None:
|
||||
for index in self.menu_file_need_file:
|
||||
self.menu_file.entryconfigure(
|
||||
index,
|
||||
state=("normal" if self.current_file is not None else "disabled"),
|
||||
)
|
||||
self.menubar.entryconfigure(
|
||||
"Image", state=("normal" if self.current_image is not None else "disabled")
|
||||
)
|
||||
self.menubar.entryconfigure(
|
||||
"Bitmap", state=("normal" if self.current_file is not None else "disabled")
|
||||
)
|
||||
for index in self.menu_bmp_need_image:
|
||||
self.menu_bmp.entryconfigure(
|
||||
index,
|
||||
state=("normal" if self.current_image is not None else "disabled"),
|
||||
)
|
||||
|
||||
def open_file(self, path: Optional[str]) -> None:
|
||||
if path is None:
|
||||
self.current_file = None
|
||||
else:
|
||||
self.current_file = File(path if path != "" else None)
|
||||
self.update(force=True)
|
||||
|
||||
def save_file(self, path: Optional[str] = None) -> None:
|
||||
if path == "":
|
||||
path = filedialog.asksaveasfilename()
|
||||
self.current_file.export(path)
|
||||
self.open_file(path)
|
||||
|
||||
def add_image(self) -> None:
|
||||
pass # TODO add image action
|
||||
|
||||
def edit_image_name(self) -> None:
|
||||
pass # TODO edit image name action
|
||||
|
||||
def edit_image_size(self) -> None:
|
||||
pass # TODO edit image size action
|
||||
|
||||
def move_image_up(self) -> None:
|
||||
pass # TODO move image actions
|
||||
|
||||
def move_image_down(self) -> None:
|
||||
pass # TODO move image actions
|
||||
|
||||
def delete_image(self) -> None:
|
||||
pass # TODO delete image action
|
||||
|
||||
def import_all_bmp(self) -> None:
|
||||
pass # TODO import all bmp action
|
||||
|
||||
def export_all_bmp(self) -> None:
|
||||
pass # TODO export all bmp action
|
||||
|
||||
def import_bmp(self) -> None:
|
||||
if self.current_image is None:
|
||||
return
|
||||
path = filedialog.askopenfilename(
|
||||
filetypes=Bitmap.FILE_TYPES,
|
||||
defaultextension=Bitmap.FILE_TYPES,
|
||||
)
|
||||
if path is not None:
|
||||
# TODO error handling
|
||||
self.current_image.import_bmp(path)
|
||||
self.update()
|
||||
|
||||
def export_bmp(self) -> None:
|
||||
if self.current_image is None:
|
||||
return
|
||||
path = filedialog.asksaveasfilename(
|
||||
filetypes=Bitmap.FILE_TYPES,
|
||||
defaultextension=Bitmap.FILE_TYPES,
|
||||
initialfile=f"{self.current_image.name}.bmp",
|
||||
)
|
||||
if path is not None:
|
||||
self.current_image.export_bmp(path)
|
||||
@@ -0,0 +1,119 @@
|
||||
from typing import Tuple
|
||||
|
||||
|
||||
class BitmapError(Exception):
|
||||
pass
|
||||
|
||||
|
||||
class Bitmap:
|
||||
HEADER_SIZE = 54
|
||||
FILE_TYPES = [("Bitmap Image", "*.bmp"), ("All Files", "*.*")]
|
||||
|
||||
@classmethod
|
||||
def write_bmp(cls, path: str, width: int, color_depth: int, data: bytes) -> None:
|
||||
with open(path, mode="wb") as f:
|
||||
f.write(cls.__get_bmp_data(width, color_depth, data))
|
||||
|
||||
@classmethod
|
||||
def __get_bmp_data(cls, width: int, color_depth: int, data: bytes) -> bytes:
|
||||
height = len(data) // (width * 3)
|
||||
return cls.__get_header(
|
||||
width, height, color_depth, len(data)
|
||||
) + cls.__format_data(width, height, color_depth, data)
|
||||
|
||||
@classmethod
|
||||
def __get_header(
|
||||
cls, width: int, height: int, color_depth: int, data_len: int
|
||||
) -> bytes:
|
||||
header = bytes()
|
||||
# BMP header
|
||||
header += "BM".encode() # (0, 2) BM
|
||||
header += (cls.HEADER_SIZE + data_len).to_bytes(
|
||||
4, byteorder="little"
|
||||
) # (2, 4) file size
|
||||
header += bytes([0]) * 4 # (6, 4) application reserved
|
||||
header += (cls.HEADER_SIZE).to_bytes(
|
||||
4, byteorder="little"
|
||||
) # (10, 4) data offset
|
||||
# DIB header
|
||||
header += (40).to_bytes(4, byteorder="little") # (14, 4) DIB header size
|
||||
header += width.to_bytes(4, byteorder="little") # (18, 4) width
|
||||
header += height.to_bytes(4, byteorder="little") # (22, 4) height
|
||||
header += (1).to_bytes(2, byteorder="little") # (26, 2) color panes
|
||||
header += (color_depth * 8).to_bytes(
|
||||
2, byteorder="little"
|
||||
) # (28, 2) bits per pixel
|
||||
header += bytes([0]) * 4 # (30, 4) BI_RGB, no compression
|
||||
header += (data_len).to_bytes(
|
||||
4, byteorder="little"
|
||||
) # (34, 4) size of raw bitmap data
|
||||
header += (2835).to_bytes(
|
||||
4, byteorder="little"
|
||||
) # (38, 4) horizontal print resolution
|
||||
header += (2835).to_bytes(
|
||||
4, byteorder="little"
|
||||
) # (42, 4) vertical print resolution
|
||||
header += bytes([0]) * 4 # (46, 4) color in palette
|
||||
header += bytes([0]) * 4 # (50, 4) 0 important colors
|
||||
return header
|
||||
|
||||
@classmethod
|
||||
def __format_data(
|
||||
cls, width: int, height: int, color_depth: int, data: bytes
|
||||
) -> bytes:
|
||||
size = width * height * color_depth
|
||||
if len(data) < size:
|
||||
data += bytes([0]) * (size - len(data))
|
||||
elif len(data) > size:
|
||||
data = data[:size]
|
||||
line_padding = (width * color_depth) % 4
|
||||
output_data = bytes()
|
||||
for y in range(height):
|
||||
start = (height - y - 1) * color_depth * width
|
||||
output_data += data[start : start + width * color_depth]
|
||||
if line_padding > 0:
|
||||
output_data += bytes([0]) * (4 - line_padding)
|
||||
return output_data
|
||||
|
||||
@classmethod
|
||||
def read_bmp(cls, path: str) -> Tuple[int, int, int, bytes]:
|
||||
with open(path, mode="rb") as f:
|
||||
bmp_data = f.read()
|
||||
width, height, color_depth, data_start, data_size = cls.__read_header(bmp_data)
|
||||
content_data = bmp_data[data_start:]
|
||||
if data_size > 0:
|
||||
content_data = content_data[:data_size]
|
||||
output_data = cls.__read_formated_data(width, height, color_depth, content_data)
|
||||
return width, height, color_depth, output_data
|
||||
|
||||
@classmethod
|
||||
def __read_header(cls, bmp_data: bytes) -> Tuple[int, int, int, int, int]:
|
||||
if bmp_data[0:2].decode() != "BM":
|
||||
raise BitmapError("Not a Bitmap Image")
|
||||
if int.from_bytes(bmp_data[30:34], byteorder="little") != 0:
|
||||
raise BitmapError("Cannot read Bitmap: need no compression")
|
||||
if int.from_bytes(bmp_data[26:28], byteorder="little") != 1:
|
||||
raise BitmapError("Cannot read Bitmap: need 1 color panes")
|
||||
width = int.from_bytes(bmp_data[18:22], byteorder="little")
|
||||
height = int.from_bytes(bmp_data[22:26], byteorder="little")
|
||||
color_depth = int.from_bytes(bmp_data[28:30], byteorder="little") // 8
|
||||
if color_depth < 1:
|
||||
raise BitmapError("Cannot read Bitmap: bits per pixels is < 8")
|
||||
data_start = int.from_bytes(bmp_data[10:14], byteorder="little")
|
||||
data_size = int.from_bytes(bmp_data[34:38], byteorder="little")
|
||||
return width, height, color_depth, data_start, data_size
|
||||
|
||||
@classmethod
|
||||
def __read_formated_data(
|
||||
cls, width: int, height: int, color_depth: int, bmp_data: bytes
|
||||
) -> bytes:
|
||||
line_padding = (width * color_depth) % 4
|
||||
if line_padding > 0:
|
||||
real_width = width * color_depth + (4 - line_padding)
|
||||
else:
|
||||
real_width = width * color_depth
|
||||
output_data = bytes()
|
||||
for y in range(height):
|
||||
start = (height - y - 1) * real_width
|
||||
output_data += bmp_data[start : start + width * color_depth]
|
||||
return output_data
|
||||
@@ -0,0 +1,63 @@
|
||||
from tkinter import ttk
|
||||
from typing import Optional
|
||||
|
||||
from .file import File
|
||||
from .image import Image
|
||||
|
||||
|
||||
class Explorer(ttk.Frame):
|
||||
def __init__(self, parent, update_callback) -> None:
|
||||
super().__init__(parent)
|
||||
|
||||
self.current_file = None
|
||||
self.update_callback = update_callback
|
||||
|
||||
self.explorer = ttk.Treeview(self, columns=("size"))
|
||||
self.explorer.heading("#0", text="name")
|
||||
self.explorer.heading("size", text="size")
|
||||
self.explorer.column("#0", width=100, anchor="w")
|
||||
self.explorer.column("size", width=100, anchor="w")
|
||||
self.explorer.grid(row=0, column=0, sticky="nsw")
|
||||
self.explorer.bind("<<TreeviewSelect>>", self.explorer_item_click)
|
||||
|
||||
yscrollbar = ttk.Scrollbar(self, orient="vertical", command=self.explorer.yview)
|
||||
yscrollbar.grid(row=0, column=1, sticky="nsw")
|
||||
self.explorer.configure(yscrollcommand=yscrollbar.set)
|
||||
|
||||
self.grid_rowconfigure(0, weight=1)
|
||||
self.grid_columnconfigure(0, weight=1)
|
||||
|
||||
@property
|
||||
def current_image(self) -> Optional[Image]:
|
||||
if self.current_file is None or self.explorer.focus() == "":
|
||||
return None
|
||||
else:
|
||||
return self.current_file.images[int(self.explorer.focus())]
|
||||
|
||||
def update(self, file: File, force: bool):
|
||||
if force or file != self.current_file:
|
||||
ids = self.explorer.get_children()
|
||||
if len(ids) > 0:
|
||||
self.explorer.delete(*ids)
|
||||
|
||||
self.current_file = file
|
||||
|
||||
if self.current_file is not None:
|
||||
for i, image in enumerate(self.current_file.images):
|
||||
if self.explorer.exists(str(i)):
|
||||
self.explorer.item(
|
||||
str(i),
|
||||
text=f"{image.name}{'*' if image.modified else ''}",
|
||||
values=[f"{image.width}x{image.height}"],
|
||||
)
|
||||
else:
|
||||
self.explorer.insert(
|
||||
"",
|
||||
"end",
|
||||
iid=str(i),
|
||||
text=f"{image.name}{'*' if image.modified else ''}",
|
||||
values=[f"{image.width}x{image.height}"],
|
||||
)
|
||||
|
||||
def explorer_item_click(self, event) -> None:
|
||||
self.update_callback()
|
||||
@@ -0,0 +1,64 @@
|
||||
from typing import List
|
||||
import re
|
||||
import os.path
|
||||
|
||||
from .image import Image
|
||||
|
||||
|
||||
class File:
|
||||
FILE_TYPES = [("Header File", "*.h"), ("All Files", "*.*")]
|
||||
|
||||
def __init__(self, path: str) -> None:
|
||||
self.path = path
|
||||
if path is None:
|
||||
self.images = []
|
||||
else:
|
||||
self.images = self.__read_file()
|
||||
|
||||
@property
|
||||
def filename(self) -> str:
|
||||
if self.path is None:
|
||||
return None
|
||||
return os.path.basename(self.path)
|
||||
|
||||
@property
|
||||
def modified(self) -> bool:
|
||||
return any(image.modified for image in self.images)
|
||||
|
||||
def __read_file(self) -> List[Image]:
|
||||
images = []
|
||||
|
||||
current_image = None
|
||||
|
||||
with open(self.path) as f:
|
||||
for line in f:
|
||||
header = re.match(
|
||||
r"const unsigned char (\w+) \[\] PROGMEM \= \{",
|
||||
line,
|
||||
)
|
||||
if header:
|
||||
groups = header.groups()
|
||||
if current_image is None:
|
||||
current_image = Image(groups[0], 0, 0)
|
||||
current_image.name = groups[0]
|
||||
elif current_image is not None and current_image.name is not None:
|
||||
data = re.match(r"((0x\w+,? ?)+)", line.strip())
|
||||
if data:
|
||||
current_image.add_data(
|
||||
data.groups()[0].strip().strip(",").split(", ")
|
||||
)
|
||||
else:
|
||||
images += [current_image]
|
||||
current_image.finalize()
|
||||
current_image = None
|
||||
comment_header = re.match(r"// '(\w+)', (\d+)x(\d+)px", line)
|
||||
if comment_header:
|
||||
groups = comment_header.groups()
|
||||
current_image = Image(groups[0], int(groups[1]), int(groups[2]))
|
||||
|
||||
return images
|
||||
|
||||
def export(self, path: str) -> None:
|
||||
with open(path, mode="w") as f:
|
||||
for image in self.images:
|
||||
f.write(image.export_cpp())
|
||||
@@ -0,0 +1,89 @@
|
||||
from typing import List
|
||||
from math import sqrt
|
||||
|
||||
from .bitmap import Bitmap
|
||||
|
||||
|
||||
class Image:
|
||||
def __init__(self, name: str, width: int, height: int, empty: bool = False) -> None:
|
||||
self.name = name
|
||||
self.width = width
|
||||
self.height = height
|
||||
self.modified = False
|
||||
if empty:
|
||||
self.data = [0] * (width * height) // 8
|
||||
self.modified = True
|
||||
else:
|
||||
self.data = []
|
||||
|
||||
def finalize(self) -> None:
|
||||
if self.width == 0:
|
||||
pixels = len(self.data) * 8
|
||||
width = int(sqrt(pixels))
|
||||
while width > 1 and pixels % width != 0:
|
||||
width -= 1
|
||||
self.width = width
|
||||
self.height = pixels // width
|
||||
|
||||
def add_data(self, raw_data: List[str]) -> None:
|
||||
for v in raw_data:
|
||||
self.data += [int(v, 16)]
|
||||
|
||||
def __get_position(self, x: int, y: int) -> int:
|
||||
real_width = (len(self.data) * 8) // self.height
|
||||
return y * real_width + x
|
||||
|
||||
def get_pixel(self, x: int, y: int) -> bool:
|
||||
position = self.__get_position(x, y)
|
||||
chunk_id = position // 8
|
||||
return self.data[chunk_id] & (1 << (7 - position % 8)) > 0
|
||||
|
||||
def set_pixel(self, x: int, y: int, v: bool) -> None:
|
||||
position = self.__get_position(x, y)
|
||||
chunk_id = position // 8
|
||||
# TODO fix, invalid chunk id when drawing
|
||||
if v != self.get_pixel(x, y):
|
||||
if v:
|
||||
self.data[chunk_id] |= 1 << (7 - position % 8)
|
||||
else:
|
||||
self.data[chunk_id] &= ~(1 << (7 - position % 8))
|
||||
self.modified = True
|
||||
|
||||
def __get_color_bytes(self) -> bytes:
|
||||
output = bytes()
|
||||
for y in range(self.height):
|
||||
for x in range(self.width):
|
||||
if self.get_pixel(x, y):
|
||||
output += bytes([0, 0, 0])
|
||||
else:
|
||||
output += bytes([255, 255, 255])
|
||||
return output
|
||||
|
||||
def export_bmp(self, path: str) -> None:
|
||||
Bitmap.write_bmp(path, self.width, 3, self.__get_color_bytes())
|
||||
|
||||
def __set_color_bytes(self, color_depth: int, data: bytes) -> None:
|
||||
for y in range(self.height):
|
||||
for x in range(self.width):
|
||||
position = (y * self.width + x) * color_depth
|
||||
colors = data[position : position + color_depth]
|
||||
mean_color = sum(c for c in colors) / color_depth
|
||||
if mean_color < 128:
|
||||
self.set_pixel(x, y, True)
|
||||
|
||||
def import_bmp(self, path: str) -> None:
|
||||
self.width, self.height, color_depth, bmp_data = Bitmap.read_bmp(path)
|
||||
self.data = [0] * ((self.width * self.height) // 8)
|
||||
self.__set_color_bytes(color_depth, bmp_data)
|
||||
|
||||
def export_cpp(self) -> str:
|
||||
# 16 per line
|
||||
output = [
|
||||
f"// '{self.name}', {self.width}x{self.height}px",
|
||||
f"const unsigned char {self.name} [] PROGMEM = {{",
|
||||
]
|
||||
while len(self.data) > 16:
|
||||
output += ["\t" + ", ".join(f"0x{v:02x}" for v in self.data[0:16]) + ","]
|
||||
self.data = self.data[16:]
|
||||
output += ["\t" + ", ".join(f"0x{v:02x}" for v in self.data), "};", ""]
|
||||
return "\n".join(output)
|
||||
@@ -0,0 +1,97 @@
|
||||
import tkinter as tk
|
||||
from tkinter import ttk
|
||||
|
||||
from .image import Image
|
||||
|
||||
|
||||
class ImageView(ttk.Frame):
|
||||
INITIAL_DRAW_SCALE = 3
|
||||
|
||||
def __init__(self, parent) -> None:
|
||||
super().__init__(parent, height=650, width=650)
|
||||
|
||||
self.draw_scale = self.INITIAL_DRAW_SCALE
|
||||
|
||||
self.current_image = None
|
||||
|
||||
self.canvas = tk.Canvas(self, width=0, height=0, background="white")
|
||||
self.canvas.place(in_=self, anchor="c", relx=0.5, rely=0.5)
|
||||
self.canvas.bind("<Button-1>", self.click_canvas_b1)
|
||||
self.canvas.bind("<B1-Motion>", self.click_canvas_b1)
|
||||
self.canvas.bind("<ButtonRelease-1>", self.update)
|
||||
self.canvas.bind("<Button-3>", self.click_canvas_b3)
|
||||
self.canvas.bind("<B3-Motion>", self.click_canvas_b3)
|
||||
self.canvas.bind("<ButtonRelease-3>", self.update)
|
||||
|
||||
self.canvas.bind("<MouseWheel>", self.zoom_canvas)
|
||||
self.canvas.bind("<Button-4>", self.zoom_canvas_up)
|
||||
self.canvas.bind("<Button-5>", self.zoom_canvas_down)
|
||||
|
||||
self.bind("<MouseWheel>", self.zoom_canvas)
|
||||
self.bind("<Button-4>", self.zoom_canvas_up)
|
||||
self.bind("<Button-5>", self.zoom_canvas_down)
|
||||
|
||||
def update(self, image: Image) -> None:
|
||||
if self.current_image != image:
|
||||
self.draw_scale = self.INITIAL_DRAW_SCALE
|
||||
if image is None:
|
||||
self.canvas.configure(
|
||||
width=0,
|
||||
height=0,
|
||||
background="white",
|
||||
)
|
||||
else:
|
||||
self.canvas.configure(
|
||||
width=(image.width * self.draw_scale),
|
||||
height=(image.height * self.draw_scale),
|
||||
background="white",
|
||||
)
|
||||
# TODO fix / handle, "bad screen distance "??????""
|
||||
self.canvas.delete("all")
|
||||
for x in range(image.width):
|
||||
for y in range(image.height):
|
||||
if image.get_pixel(x, y):
|
||||
self.canvas.create_rectangle(
|
||||
x * self.draw_scale + 1,
|
||||
y * self.draw_scale + 1,
|
||||
(x + 1) * self.draw_scale + 1,
|
||||
(y + 1) * self.draw_scale + 1,
|
||||
fill="black",
|
||||
outline="",
|
||||
)
|
||||
self.current_image = image
|
||||
|
||||
def click_canvas_b1(self, event):
|
||||
self.click_canvas(True, event)
|
||||
|
||||
def click_canvas_b3(self, event):
|
||||
self.click_canvas(False, event)
|
||||
|
||||
def click_canvas(self, value, event):
|
||||
if self.current_image is None:
|
||||
return
|
||||
x = int(event.x / self.draw_scale)
|
||||
y = int(event.y / self.draw_scale)
|
||||
self.current_image.set_pixel(x, y, value)
|
||||
self.canvas.create_rectangle(
|
||||
x * self.draw_scale + 1,
|
||||
y * self.draw_scale + 1,
|
||||
(x + 1) * self.draw_scale + 1,
|
||||
(y + 1) * self.draw_scale + 1,
|
||||
fill=("black" if value else "white"),
|
||||
outline="",
|
||||
)
|
||||
|
||||
def zoom_canvas(self, event):
|
||||
if event.delta > 0:
|
||||
self.zoom_canvas_up()
|
||||
else:
|
||||
self.zoom_canvas_down()
|
||||
|
||||
def zoom_canvas_up(self, event=None):
|
||||
self.draw_scale *= 2
|
||||
self.update(self.current_image)
|
||||
|
||||
def zoom_canvas_down(self, event=None):
|
||||
self.draw_scale /= 2
|
||||
self.update(self.current_image)
|
||||
+3
-666
@@ -1,672 +1,9 @@
|
||||
from tkinter import *
|
||||
from tkinter import ttk
|
||||
from tkinter import filedialog
|
||||
from typing import List, Optional, Tuple
|
||||
import re
|
||||
import os.path
|
||||
from math import sqrt
|
||||
|
||||
|
||||
class BitmapError(Exception):
|
||||
pass
|
||||
|
||||
|
||||
class Bitmap:
|
||||
HEADER_SIZE = 54
|
||||
FILE_TYPES = [("Bitmap Image", "*.bmp"), ("All Files", "*.*")]
|
||||
|
||||
@classmethod
|
||||
def write_bmp(cls, path: str, width: int, color_depth: int, data: bytes) -> None:
|
||||
with open(path, mode="wb") as f:
|
||||
f.write(cls.__get_bmp_data(width, color_depth, data))
|
||||
|
||||
@classmethod
|
||||
def __get_bmp_data(cls, width: int, color_depth: int, data: bytes) -> bytes:
|
||||
height = len(data) // (width * 3)
|
||||
return cls.__get_header(
|
||||
width, height, color_depth, len(data)
|
||||
) + cls.__format_data(width, height, color_depth, data)
|
||||
|
||||
@classmethod
|
||||
def __get_header(
|
||||
cls, width: int, height: int, color_depth: int, data_len: int
|
||||
) -> bytes:
|
||||
header = bytes()
|
||||
# BMP header
|
||||
header += "BM".encode() # (0, 2) BM
|
||||
header += (cls.HEADER_SIZE + data_len).to_bytes(
|
||||
4, byteorder="little"
|
||||
) # (2, 4) file size
|
||||
header += bytes([0]) * 4 # (6, 4) application reserved
|
||||
header += (cls.HEADER_SIZE).to_bytes(
|
||||
4, byteorder="little"
|
||||
) # (10, 4) data offset
|
||||
# DIB header
|
||||
header += (40).to_bytes(4, byteorder="little") # (14, 4) DIB header size
|
||||
header += width.to_bytes(4, byteorder="little") # (18, 4) width
|
||||
header += height.to_bytes(4, byteorder="little") # (22, 4) height
|
||||
header += (1).to_bytes(2, byteorder="little") # (26, 2) color panes
|
||||
header += (color_depth * 8).to_bytes(
|
||||
2, byteorder="little"
|
||||
) # (28, 2) bits per pixel
|
||||
header += bytes([0]) * 4 # (30, 4) BI_RGB, no compression
|
||||
header += (data_len).to_bytes(
|
||||
4, byteorder="little"
|
||||
) # (34, 4) size of raw bitmap data
|
||||
header += (2835).to_bytes(
|
||||
4, byteorder="little"
|
||||
) # (38, 4) horizontal print resolution
|
||||
header += (2835).to_bytes(
|
||||
4, byteorder="little"
|
||||
) # (42, 4) vertical print resolution
|
||||
header += bytes([0]) * 4 # (46, 4) color in palette
|
||||
header += bytes([0]) * 4 # (50, 4) 0 important colors
|
||||
return header
|
||||
|
||||
@classmethod
|
||||
def __format_data(
|
||||
cls, width: int, height: int, color_depth: int, data: bytes
|
||||
) -> bytes:
|
||||
size = width * height * color_depth
|
||||
if len(data) < size:
|
||||
data += bytes([0]) * (size - len(data))
|
||||
elif len(data) > size:
|
||||
data = data[:size]
|
||||
line_padding = (width * color_depth) % 4
|
||||
output_data = bytes()
|
||||
for y in range(height):
|
||||
start = (height - y - 1) * color_depth * width
|
||||
output_data += data[start : start + width * color_depth]
|
||||
if line_padding > 0:
|
||||
output_data += bytes([0]) * (4 - line_padding)
|
||||
return output_data
|
||||
|
||||
@classmethod
|
||||
def read_bmp(cls, path: str) -> Tuple[int, int, int, bytes]:
|
||||
with open(path, mode="rb") as f:
|
||||
bmp_data = f.read()
|
||||
width, height, color_depth, data_start, data_size = cls.__read_header(bmp_data)
|
||||
content_data = bmp_data[data_start:]
|
||||
if data_size > 0:
|
||||
content_data = content_data[:data_size]
|
||||
output_data = cls.__read_formated_data(width, height, color_depth, content_data)
|
||||
return width, height, color_depth, output_data
|
||||
|
||||
@classmethod
|
||||
def __read_header(cls, bmp_data: bytes) -> Tuple[int, int, int, int, int]:
|
||||
if bmp_data[0:2].decode() != "BM":
|
||||
raise BitmapError("Not a Bitmap Image")
|
||||
if int.from_bytes(bmp_data[30:34], byteorder="little") != 0:
|
||||
raise BitmapError("Cannot read Bitmap: need no compression")
|
||||
if int.from_bytes(bmp_data[26:28], byteorder="little") != 1:
|
||||
raise BitmapError("Cannot read Bitmap: need 1 color panes")
|
||||
width = int.from_bytes(bmp_data[18:22], byteorder="little")
|
||||
height = int.from_bytes(bmp_data[22:26], byteorder="little")
|
||||
color_depth = int.from_bytes(bmp_data[28:30], byteorder="little") // 8
|
||||
if color_depth < 1:
|
||||
raise BitmapError("Cannot read Bitmap: bits per pixels is < 8")
|
||||
data_start = int.from_bytes(bmp_data[10:14], byteorder="little")
|
||||
data_size = int.from_bytes(bmp_data[34:38], byteorder="little")
|
||||
return width, height, color_depth, data_start, data_size
|
||||
|
||||
@classmethod
|
||||
def __read_formated_data(
|
||||
cls, width: int, height: int, color_depth: int, bmp_data: bytes
|
||||
) -> bytes:
|
||||
line_padding = (width * color_depth) % 4
|
||||
if line_padding > 0:
|
||||
real_width = width * color_depth + (4 - line_padding)
|
||||
else:
|
||||
real_width = width * color_depth
|
||||
output_data = bytes()
|
||||
for y in range(height):
|
||||
start = (height - y - 1) * real_width
|
||||
output_data += bmp_data[start : start + width * color_depth]
|
||||
return output_data
|
||||
|
||||
|
||||
class Image:
|
||||
def __init__(self, name: str, width: int, height: int, empty: bool = False) -> None:
|
||||
self.name = name
|
||||
self.width = width
|
||||
self.height = height
|
||||
self.modified = False
|
||||
if empty:
|
||||
self.data = [0] * (width * height) // 8
|
||||
self.modified = True
|
||||
else:
|
||||
self.data = []
|
||||
|
||||
def finalize(self) -> None:
|
||||
if self.width == 0:
|
||||
pixels = len(self.data) * 8
|
||||
width = int(sqrt(pixels))
|
||||
while width > 1 and pixels % width != 0:
|
||||
width -= 1
|
||||
self.width = width
|
||||
self.height = pixels // width
|
||||
|
||||
def add_data(self, raw_data: List[str]) -> None:
|
||||
for v in raw_data:
|
||||
self.data += [int(v, 16)]
|
||||
|
||||
def __get_position(self, x: int, y: int) -> int:
|
||||
real_width = (len(self.data) * 8) // self.height
|
||||
return y * real_width + x
|
||||
|
||||
def get_pixel(self, x: int, y: int) -> bool:
|
||||
position = self.__get_position(x, y)
|
||||
chunk_id = position // 8
|
||||
return self.data[chunk_id] & (1 << (7 - position % 8)) > 0
|
||||
|
||||
def set_pixel(self, x: int, y: int, v: bool) -> None:
|
||||
position = self.__get_position(x, y)
|
||||
chunk_id = position // 8
|
||||
# TODO fix, invalid chunk id when drawing
|
||||
if v != self.get_pixel(x, y):
|
||||
if v:
|
||||
self.data[chunk_id] |= 1 << (7 - position % 8)
|
||||
else:
|
||||
self.data[chunk_id] &= ~(1 << (7 - position % 8))
|
||||
self.modified = True
|
||||
|
||||
def __get_color_bytes(self) -> bytes:
|
||||
output = bytes()
|
||||
for y in range(self.height):
|
||||
for x in range(self.width):
|
||||
if self.get_pixel(x, y):
|
||||
output += bytes([0, 0, 0])
|
||||
else:
|
||||
output += bytes([255, 255, 255])
|
||||
return output
|
||||
|
||||
def export_bmp(self, path: str) -> None:
|
||||
Bitmap.write_bmp(path, self.width, 3, self.__get_color_bytes())
|
||||
|
||||
def __set_color_bytes(self, color_depth: int, data: bytes) -> None:
|
||||
for y in range(self.height):
|
||||
for x in range(self.width):
|
||||
position = (y * self.width + x) * color_depth
|
||||
colors = data[position : position + color_depth]
|
||||
mean_color = sum(c for c in colors) / color_depth
|
||||
if mean_color < 128:
|
||||
self.set_pixel(x, y, True)
|
||||
|
||||
def import_bmp(self, path: str) -> None:
|
||||
self.width, self.height, color_depth, bmp_data = Bitmap.read_bmp(path)
|
||||
self.data = [0] * ((self.width * self.height) // 8)
|
||||
self.__set_color_bytes(color_depth, bmp_data)
|
||||
|
||||
def export_cpp(self) -> str:
|
||||
# 16 per line
|
||||
output = [
|
||||
f"// '{self.name}', {self.width}x{self.height}px",
|
||||
f"const unsigned char {self.name} [] PROGMEM = {{",
|
||||
]
|
||||
while len(self.data) > 16:
|
||||
output += ["\t" + ", ".join(f"0x{v:02x}" for v in self.data[0:16]) + ","]
|
||||
self.data = self.data[16:]
|
||||
output += ["\t" + ", ".join(f"0x{v:02x}" for v in self.data), "};", ""]
|
||||
return "\n".join(output)
|
||||
|
||||
|
||||
class File:
|
||||
FILE_TYPES = [("Header File", "*.h"), ("All Files", "*.*")]
|
||||
|
||||
def __init__(self, path: str) -> None:
|
||||
self.path = path
|
||||
if path is None:
|
||||
self.images = []
|
||||
else:
|
||||
self.images = self.__read_file()
|
||||
|
||||
@property
|
||||
def filename(self) -> str:
|
||||
if self.path is None:
|
||||
return None
|
||||
return os.path.basename(self.path)
|
||||
|
||||
@property
|
||||
def modified(self) -> bool:
|
||||
return any(image.modified for image in self.images)
|
||||
|
||||
def __read_file(self) -> List[Image]:
|
||||
images = []
|
||||
|
||||
current_image = None
|
||||
|
||||
with open(self.path) as f:
|
||||
for line in f:
|
||||
header = re.match(
|
||||
r"const unsigned char (\w+) \[\] PROGMEM \= \{",
|
||||
line,
|
||||
)
|
||||
if header:
|
||||
groups = header.groups()
|
||||
if current_image is None:
|
||||
current_image = Image(groups[0], 0, 0)
|
||||
current_image.name = groups[0]
|
||||
elif current_image is not None and current_image.name is not None:
|
||||
data = re.match(r"((0x\w+,? ?)+)", line.strip())
|
||||
if data:
|
||||
current_image.add_data(
|
||||
data.groups()[0].strip().strip(",").split(", ")
|
||||
)
|
||||
else:
|
||||
images += [current_image]
|
||||
current_image.finalize()
|
||||
current_image = None
|
||||
comment_header = re.match(r"// '(\w+)', (\d+)x(\d+)px", line)
|
||||
if comment_header:
|
||||
groups = comment_header.groups()
|
||||
current_image = Image(groups[0], int(groups[1]), int(groups[2]))
|
||||
|
||||
return images
|
||||
|
||||
def export(self, path: str) -> None:
|
||||
with open(path, mode="w") as f:
|
||||
for image in self.images:
|
||||
f.write(image.export_cpp())
|
||||
|
||||
|
||||
class ImageView(ttk.Frame):
|
||||
INITIAL_DRAW_SCALE = 3
|
||||
|
||||
def __init__(self, parent) -> None:
|
||||
super().__init__(parent, height=650, width=650)
|
||||
|
||||
self.draw_scale = self.INITIAL_DRAW_SCALE
|
||||
|
||||
self.current_image = None
|
||||
|
||||
self.canvas = Canvas(self, width=0, height=0, background="white")
|
||||
self.canvas.place(in_=self, anchor="c", relx=0.5, rely=0.5)
|
||||
self.canvas.bind("<Button-1>", self.click_canvas_b1)
|
||||
self.canvas.bind("<B1-Motion>", self.click_canvas_b1)
|
||||
self.canvas.bind("<ButtonRelease-1>", self.update)
|
||||
self.canvas.bind("<Button-3>", self.click_canvas_b3)
|
||||
self.canvas.bind("<B3-Motion>", self.click_canvas_b3)
|
||||
self.canvas.bind("<ButtonRelease-3>", self.update)
|
||||
|
||||
self.canvas.bind("<MouseWheel>", self.zoom_canvas)
|
||||
self.canvas.bind("<Button-4>", self.zoom_canvas_up)
|
||||
self.canvas.bind("<Button-5>", self.zoom_canvas_down)
|
||||
|
||||
self.bind("<MouseWheel>", self.zoom_canvas)
|
||||
self.bind("<Button-4>", self.zoom_canvas_up)
|
||||
self.bind("<Button-5>", self.zoom_canvas_down)
|
||||
|
||||
def update(self, image: Image) -> None:
|
||||
if self.current_image != image:
|
||||
self.draw_scale = self.INITIAL_DRAW_SCALE
|
||||
if image is None:
|
||||
self.canvas.configure(
|
||||
width=0,
|
||||
height=0,
|
||||
background="white",
|
||||
)
|
||||
else:
|
||||
self.canvas.configure(
|
||||
width=(image.width * self.draw_scale),
|
||||
height=(image.height * self.draw_scale),
|
||||
background="white",
|
||||
)
|
||||
# TODO fix / handle, "bad screen distance "??????""
|
||||
self.canvas.delete("all")
|
||||
for x in range(image.width):
|
||||
for y in range(image.height):
|
||||
if image.get_pixel(x, y):
|
||||
self.canvas.create_rectangle(
|
||||
x * self.draw_scale + 1,
|
||||
y * self.draw_scale + 1,
|
||||
(x + 1) * self.draw_scale + 1,
|
||||
(y + 1) * self.draw_scale + 1,
|
||||
fill="black",
|
||||
outline="",
|
||||
)
|
||||
self.current_image = image
|
||||
|
||||
def click_canvas_b1(self, event):
|
||||
self.click_canvas(True, event)
|
||||
|
||||
def click_canvas_b3(self, event):
|
||||
self.click_canvas(False, event)
|
||||
|
||||
def click_canvas(self, value, event):
|
||||
if self.current_image is None:
|
||||
return
|
||||
x = int(event.x / self.draw_scale)
|
||||
y = int(event.y / self.draw_scale)
|
||||
self.current_image.set_pixel(x, y, value)
|
||||
self.canvas.create_rectangle(
|
||||
x * self.draw_scale + 1,
|
||||
y * self.draw_scale + 1,
|
||||
(x + 1) * self.draw_scale + 1,
|
||||
(y + 1) * self.draw_scale + 1,
|
||||
fill=("black" if value else "white"),
|
||||
outline="",
|
||||
)
|
||||
|
||||
def zoom_canvas(self, event):
|
||||
if event.delta > 0:
|
||||
self.zoom_canvas_up()
|
||||
else:
|
||||
self.zoom_canvas_down()
|
||||
|
||||
def zoom_canvas_up(self, event=None):
|
||||
self.draw_scale *= 2
|
||||
self.update(self.current_image)
|
||||
|
||||
def zoom_canvas_down(self, event=None):
|
||||
self.draw_scale /= 2
|
||||
self.update(self.current_image)
|
||||
|
||||
|
||||
class Explorer(ttk.Frame):
|
||||
def __init__(self, parent, update_callback) -> None:
|
||||
super().__init__(parent)
|
||||
|
||||
self.current_file = None
|
||||
self.update_callback = update_callback
|
||||
|
||||
self.explorer = ttk.Treeview(self, columns=("size"))
|
||||
self.explorer.heading("#0", text="name")
|
||||
self.explorer.heading("size", text="size")
|
||||
self.explorer.column("#0", width=100, anchor="w")
|
||||
self.explorer.column("size", width=100, anchor="w")
|
||||
self.explorer.grid(row=0, column=0, sticky=(N, S, W))
|
||||
self.explorer.bind("<<TreeviewSelect>>", self.explorer_item_click)
|
||||
|
||||
yscrollbar = ttk.Scrollbar(self, orient="vertical", command=self.explorer.yview)
|
||||
yscrollbar.grid(row=0, column=1, sticky=(N, S, W))
|
||||
self.explorer.configure(yscrollcommand=yscrollbar.set)
|
||||
|
||||
self.grid_rowconfigure(0, weight=1)
|
||||
self.grid_columnconfigure(0, weight=1)
|
||||
|
||||
@property
|
||||
def current_image(self) -> Optional[Image]:
|
||||
if self.current_file is None or self.explorer.focus() == "":
|
||||
return None
|
||||
else:
|
||||
return self.current_file.images[int(self.explorer.focus())]
|
||||
|
||||
def update(self, file: File, force: bool):
|
||||
if force or file != self.current_file:
|
||||
ids = self.explorer.get_children()
|
||||
if len(ids) > 0:
|
||||
self.explorer.delete(*ids)
|
||||
|
||||
self.current_file = file
|
||||
|
||||
if self.current_file is not None:
|
||||
for i, image in enumerate(self.current_file.images):
|
||||
if self.explorer.exists(str(i)):
|
||||
self.explorer.item(
|
||||
str(i),
|
||||
text=f"{image.name}{'*' if image.modified else ''}",
|
||||
values=[f"{image.width}x{image.height}"],
|
||||
)
|
||||
else:
|
||||
self.explorer.insert(
|
||||
"",
|
||||
"end",
|
||||
iid=str(i),
|
||||
text=f"{image.name}{'*' if image.modified else ''}",
|
||||
values=[f"{image.width}x{image.height}"],
|
||||
)
|
||||
|
||||
def explorer_item_click(self, event) -> None:
|
||||
self.update_callback()
|
||||
|
||||
|
||||
class App(ttk.Frame):
|
||||
def __init__(self, parent) -> None:
|
||||
super().__init__(parent)
|
||||
|
||||
parent.option_add("*tearOff", FALSE)
|
||||
parent.resizable(False, False)
|
||||
|
||||
self.parent = parent
|
||||
self.current_file = None
|
||||
|
||||
self.explorer = Explorer(self, self.update)
|
||||
self.explorer.grid(column=0, row=0, sticky=(N, S, W))
|
||||
|
||||
self.image_view = ImageView(self)
|
||||
self.image_view.grid(column=1, row=0, sticky=(N, S, E, W))
|
||||
|
||||
self.init_menus()
|
||||
|
||||
self.grid_rowconfigure(0, weight=1)
|
||||
self.grid_columnconfigure(0, weight=1)
|
||||
|
||||
self.open_file(None)
|
||||
|
||||
self.pack(fill="both", expand=True)
|
||||
|
||||
@property
|
||||
def current_image(self) -> Optional[Image]:
|
||||
if self.current_file is None:
|
||||
return None
|
||||
else:
|
||||
return self.explorer.current_image
|
||||
|
||||
def update(self, force: bool = False) -> None:
|
||||
self.update_title()
|
||||
self.update_menus()
|
||||
self.image_view.update(self.current_image)
|
||||
self.explorer.update(self.current_file, force)
|
||||
|
||||
def update_title(self) -> None:
|
||||
title = "Watchy Image Editor"
|
||||
if self.current_file is not None:
|
||||
title += "- "
|
||||
if self.current_file.path is None:
|
||||
title += "New file"
|
||||
else:
|
||||
title += self.current_file.filename
|
||||
if self.current_file.modified:
|
||||
title += "*"
|
||||
self.parent.title(title)
|
||||
|
||||
def init_menus(self) -> None:
|
||||
self.menubar = Menu(self.parent)
|
||||
self.parent["menu"] = self.menubar
|
||||
|
||||
self.menu_file = Menu(self.menubar)
|
||||
self.menubar.add_cascade(menu=self.menu_file, label="File")
|
||||
|
||||
# TODO better handling of menu items
|
||||
|
||||
self.menu_file_need_file = []
|
||||
|
||||
self.menu_file.add_command(label="New File", command=lambda: self.open_file(""))
|
||||
i = 0
|
||||
self.menu_file.add_command(
|
||||
label="Open File...",
|
||||
command=lambda: self.open_file(
|
||||
filedialog.askopenfilename(
|
||||
filetypes=File.FILE_TYPES,
|
||||
defaultextension=File.FILE_TYPES,
|
||||
)
|
||||
),
|
||||
)
|
||||
i += 1
|
||||
self.menu_file.add_command(
|
||||
label="Save File",
|
||||
command=lambda: self.save_file(self.current_file.path),
|
||||
)
|
||||
i += 1
|
||||
self.menu_file_need_file += [i]
|
||||
self.menu_file.add_command(
|
||||
label="Save File As...",
|
||||
command=lambda: self.save_file(
|
||||
filedialog.asksaveasfilename(
|
||||
filetypes=File.FILE_TYPES, defaultextension=File.FILE_TYPES
|
||||
)
|
||||
),
|
||||
)
|
||||
i += 1
|
||||
self.menu_file_need_file += [i]
|
||||
self.menu_file.add_command(
|
||||
label="Close File",
|
||||
command=lambda: self.open_file(None),
|
||||
)
|
||||
i += 1
|
||||
self.menu_file_need_file += [i]
|
||||
self.menu_file.add_separator()
|
||||
i += 1
|
||||
self.menu_file.add_command(
|
||||
label="New image...",
|
||||
command=self.add_image,
|
||||
state="disabled",
|
||||
)
|
||||
i += 1
|
||||
self.menu_file_need_file += [i]
|
||||
|
||||
self.menu_image = Menu(self.menubar)
|
||||
self.menubar.add_cascade(menu=self.menu_image, label="Image")
|
||||
|
||||
self.menu_image.add_command(
|
||||
label="Edit name...",
|
||||
command=self.edit_image_name,
|
||||
state="disabled",
|
||||
)
|
||||
self.menu_image.add_command(
|
||||
label="Edit size...",
|
||||
command=self.edit_image_size,
|
||||
state="disabled",
|
||||
)
|
||||
self.menu_image.add_command(
|
||||
label="Move up",
|
||||
command=self.move_image_up,
|
||||
state="disabled",
|
||||
)
|
||||
self.menu_image.add_command(
|
||||
label="Move down",
|
||||
command=self.move_image_down,
|
||||
state="disabled",
|
||||
)
|
||||
self.menu_image.add_command(
|
||||
label="Delete",
|
||||
command=self.delete_image,
|
||||
state="disabled",
|
||||
)
|
||||
|
||||
self.menu_bmp = Menu(self.menubar)
|
||||
self.menubar.add_cascade(menu=self.menu_bmp, label="Bitmap")
|
||||
|
||||
self.menu_bmp_need_image = []
|
||||
|
||||
self.menu_bmp.add_command(
|
||||
label="Bulk .bmp import...",
|
||||
command=self.import_all_bmp,
|
||||
)
|
||||
i = 0
|
||||
self.menu_bmp.add_command(
|
||||
label="Export all to .bmp...",
|
||||
command=self.export_all_bmp,
|
||||
)
|
||||
i += 1
|
||||
self.menu_bmp.add_separator()
|
||||
i += 1
|
||||
self.menu_bmp.add_command(
|
||||
label="Import .bmp into image...",
|
||||
command=self.import_bmp,
|
||||
state="disabled",
|
||||
)
|
||||
i += 1
|
||||
self.menu_bmp_need_image += [i]
|
||||
self.menu_bmp.add_command(
|
||||
label="Export image to .bmp...",
|
||||
command=self.export_bmp,
|
||||
state="disabled",
|
||||
)
|
||||
i += 1
|
||||
self.menu_bmp_need_image += [i]
|
||||
|
||||
def update_menus(self) -> None:
|
||||
for index in self.menu_file_need_file:
|
||||
self.menu_file.entryconfigure(
|
||||
index,
|
||||
state=("normal" if self.current_file is not None else "disabled"),
|
||||
)
|
||||
self.menubar.entryconfigure(
|
||||
"Image", state=("normal" if self.current_image is not None else "disabled")
|
||||
)
|
||||
self.menubar.entryconfigure(
|
||||
"Bitmap", state=("normal" if self.current_file is not None else "disabled")
|
||||
)
|
||||
for index in self.menu_bmp_need_image:
|
||||
self.menu_bmp.entryconfigure(
|
||||
index,
|
||||
state=("normal" if self.current_image is not None else "disabled"),
|
||||
)
|
||||
|
||||
def open_file(self, path: Optional[str]) -> None:
|
||||
if path is None:
|
||||
self.current_file = None
|
||||
else:
|
||||
self.current_file = File(path if path != "" else None)
|
||||
self.update(force=True)
|
||||
|
||||
def save_file(self, path: Optional[str] = None) -> None:
|
||||
if path == "":
|
||||
path = filedialog.asksaveasfilename()
|
||||
self.current_file.export(path)
|
||||
self.open_file(path)
|
||||
|
||||
def add_image(self) -> None:
|
||||
pass # TODO add image action
|
||||
|
||||
def edit_image_name(self) -> None:
|
||||
pass # TODO edit image name action
|
||||
|
||||
def edit_image_size(self) -> None:
|
||||
pass # TODO edit image size action
|
||||
|
||||
def move_image_up(self) -> None:
|
||||
pass # TODO move image actions
|
||||
|
||||
def move_image_down(self) -> None:
|
||||
pass # TODO move image actions
|
||||
|
||||
def delete_image(self) -> None:
|
||||
pass # TODO delete image action
|
||||
|
||||
def import_all_bmp(self) -> None:
|
||||
pass # TODO import all bmp action
|
||||
|
||||
def export_all_bmp(self) -> None:
|
||||
pass # TODO export all bmp action
|
||||
|
||||
def import_bmp(self) -> None:
|
||||
if self.current_image is None:
|
||||
return
|
||||
path = filedialog.askopenfilename(
|
||||
filetypes=Bitmap.FILE_TYPES,
|
||||
defaultextension=Bitmap.FILE_TYPES,
|
||||
)
|
||||
if path is not None:
|
||||
# TODO error handling
|
||||
self.current_image.import_bmp(path)
|
||||
self.update()
|
||||
|
||||
def export_bmp(self) -> None:
|
||||
if self.current_image is None:
|
||||
return
|
||||
path = filedialog.asksaveasfilename(
|
||||
filetypes=Bitmap.FILE_TYPES,
|
||||
defaultextension=Bitmap.FILE_TYPES,
|
||||
initialfile=f"{self.current_image.name}.bmp",
|
||||
)
|
||||
if path is not None:
|
||||
self.current_image.export_bmp(path)
|
||||
import tkinter as tk
|
||||
|
||||
from .app import App
|
||||
|
||||
if __name__ == "__main__":
|
||||
app = App(Tk())
|
||||
app = App(tk.Tk())
|
||||
|
||||
# TODO remove debug
|
||||
app.open_file("../watchfaces/tetris-2.0/tetris.h")
|
||||
|
||||
Reference in New Issue
Block a user