File size: 4,630 Bytes
79ab348 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 |
import tkinter as tk
from PIL import Image, ImageTk
import json
class CoordinateVisualizer:
def __init__(self, master, image_path):
self.master = master
self.master.title("Video Area Coordinate Visualizer")
# Load the background image
self.original_image = Image.open(image_path)
self.image_ratio = self.original_image.width / self.original_image.height
# Get screen dimensions
screen_width = self.master.winfo_screenwidth() * 0.8 # Use 80% of screen width
screen_height = self.master.winfo_screenheight() * 0.8 # Use 80% of screen height
# Calculate scaling factor
width_ratio = screen_width / self.original_image.width
height_ratio = screen_height / self.original_image.height
self.scale_factor = min(width_ratio, height_ratio)
# Resize image
new_width = int(self.original_image.width * self.scale_factor)
new_height = int(self.original_image.height * self.scale_factor)
self.bg_image = self.original_image.resize((new_width, new_height), Image.LANCZOS)
self.bg_photo = ImageTk.PhotoImage(self.bg_image)
# Create a canvas and display the image
self.canvas = tk.Canvas(master, width=new_width, height=new_height)
self.canvas.pack(side=tk.LEFT)
self.canvas.create_image(0, 0, anchor=tk.NW, image=self.bg_photo)
# Initialize coordinates
self.start_x = self.start_y = self.end_x = self.end_y = 0
self.rect_id = None
# Bind mouse events
self.canvas.bind("<ButtonPress-1>", self.on_press)
self.canvas.bind("<B1-Motion>", self.on_drag)
self.canvas.bind("<ButtonRelease-1>", self.on_release)
# Create coordinate display labels
self.coord_frame = tk.Frame(master)
self.coord_frame.pack(side=tk.RIGHT, padx=10)
self.start_label = tk.Label(self.coord_frame, text="Start (x, y): (0, 0)")
self.start_label.pack()
self.end_label = tk.Label(self.coord_frame, text="End (x, y): (0, 0)")
self.end_label.pack()
self.size_label = tk.Label(self.coord_frame, text="Size (w, h): (0, 0)")
self.size_label.pack()
# Create a button to save coordinates
self.save_button = tk.Button(self.coord_frame, text="Save Coordinates", command=self.save_coordinates)
self.save_button.pack(pady=10)
def scaled_to_original(self, x, y):
return int(x / self.scale_factor), int(y / self.scale_factor)
def on_press(self, event):
self.start_x = self.canvas.canvasx(event.x)
self.start_y = self.canvas.canvasy(event.y)
if self.rect_id:
self.canvas.delete(self.rect_id)
self.rect_id = self.canvas.create_rectangle(self.start_x, self.start_y, self.start_x, self.start_y, outline='red')
def on_drag(self, event):
cur_x = self.canvas.canvasx(event.x)
cur_y = self.canvas.canvasy(event.y)
self.canvas.coords(self.rect_id, self.start_x, self.start_y, cur_x, cur_y)
self.update_labels(cur_x, cur_y)
def on_release(self, event):
self.end_x = self.canvas.canvasx(event.x)
self.end_y = self.canvas.canvasy(event.y)
self.update_labels(self.end_x, self.end_y)
def update_labels(self, cur_x, cur_y):
orig_start_x, orig_start_y = self.scaled_to_original(self.start_x, self.start_y)
orig_cur_x, orig_cur_y = self.scaled_to_original(cur_x, cur_y)
self.start_label.config(text=f"Start (x, y): ({orig_start_x}, {orig_start_y})")
self.end_label.config(text=f"End (x, y): ({orig_cur_x}, {orig_cur_y})")
width = abs(orig_cur_x - orig_start_x)
height = abs(orig_cur_y - orig_start_y)
self.size_label.config(text=f"Size (w, h): ({width}, {height})")
def save_coordinates(self):
start_x, start_y = self.scaled_to_original(min(self.start_x, self.end_x), min(self.start_y, self.end_y))
end_x, end_y = self.scaled_to_original(max(self.start_x, self.end_x), max(self.start_y, self.end_y))
coordinates = {
"start_x": start_x,
"start_y": start_y,
"end_x": end_x,
"end_y": end_y
}
with open("video_area_coordinates.json", "w") as f:
json.dump(coordinates, f)
print("Coordinates saved to video_area_coordinates.json")
if __name__ == "__main__":
root = tk.Tk()
app = CoordinateVisualizer(root, "background.png")
root.mainloop() |