main.py

#!/usr/bin/#!/usr/bin/env python3
"""
Author: Ana Berthel
Date Last Modified: Aug 19, 2022

This program is an interactive implementation of the abelian sandpile model,
a type of cellular automaton. Users can click on the canvas area to
deposit 'grains of sand' onto the field, which will spill over into adjacent
areas following a specified pattern. The patterns can be saved to PNG files.
"""
import tkinter as tk
import numpy as np
from PIL import Image, ImageTk, ImagePalette
from tkinter.filedialog import asksaveasfilename
import settings
import sandboxes



### MAIN WINDOW KEY BINDS ###

def open_settings(event):
    """ Open settings window """
    global sw
    sw = settings.Window(main_canvas, sandbox)

def open_save(event):
    """ Open save dialog box """

    # only PNG images are supported at the moment
    supported_filetypes = [('PNG Image', '*.png')]

    filename = asksaveasfilename(filetypes=supported_filetypes, defaultextension=supported_filetypes)
    sandbox.image.save(filename, "PNG")

def place_sand(event):
    """ Handles button press event on main canvas """
    # TODO: can be streamlined - im_to_coords called within place_sand
    sandbox.place_sand(sandbox.im_to_coords(event.x, event.y))

def return_focus(event):
    """ Called whenever settings window is closed so that zoom and color palette automatically change """
    main_canvas.configure(height=sandbox.zoom*sandbox.height, width=sandbox.zoom*sandbox.width)
    sandbox.update_canvas()

### MAIN WINDOW SETUP ###

main_window = tk.Tk()

header_frame = tk.Frame()
drawing_frame = tk.Frame()

# buttons at top of window - settings, save, etc.
title_label = tk.Label(master=header_frame, text="Abelian Sandpile Model")
settings_button = tk.Button(master=header_frame, text="Settings")
settings_button.bind("<Button-1>", open_settings)
save_button = tk.Button(master=header_frame, text="Save Image")
save_button.bind("<Button-1>", open_save)

title_label.pack(side=tk.TOP)
settings_button.pack(side=tk.LEFT)
save_button.pack(side=tk.RIGHT)

# canvas holds the sandpile image and is interactive by clicking with left mouse button
main_canvas = tk.Canvas(master=drawing_frame, bg="grey", height=500, width=500)
main_canvas.bind("<Button-1>", place_sand)
main_canvas.pack()

header_frame.pack()
drawing_frame.pack()

# TODO: interchange triangular and hexagonal sandboxes
sandbox = sandboxes.Sandbox(100, 100, main_canvas, main_window)

# draw image from sandbox array
sandbox.to_image()
image_container = main_canvas.create_image(0, 0, image=sandbox.photoimage, anchor="nw")
sandbox.image_container = image_container
sandbox.update_canvas()

sw = ""
main_window.bind("<FocusIn>", return_focus)
main_window.mainloop()