356 lines
12 KiB
Python
356 lines
12 KiB
Python
import kivy
|
|
from kivy.uix.screenmanager import Screen
|
|
import os
|
|
import json
|
|
import math
|
|
from kivy.clock import Clock
|
|
from kivy.properties import StringProperty, NumericProperty, AliasProperty
|
|
from py_scripts.utils import (
|
|
process_preview_util, optimize_route_entries_util
|
|
)
|
|
from py_scripts.video_3d_generator import generate_3d_video_animation
|
|
from kivy.uix.popup import Popup
|
|
from kivy.uix.button import Button
|
|
from kivy.uix.label import Label
|
|
from kivy.uix.boxlayout import BoxLayout
|
|
from kivy.uix.progressbar import ProgressBar
|
|
from kivy.uix.textinput import TextInput
|
|
from config import RESOURCES_FOLDER
|
|
|
|
class CreateAnimationScreen(Screen):
|
|
project_name = StringProperty("")
|
|
preview_html_path = StringProperty("") # Path to the HTML file for preview
|
|
preview_image_path = StringProperty("") # Add this line
|
|
preview_image_version = NumericProperty(0) # Add this line
|
|
|
|
def get_preview_image_source(self):
|
|
project_folder = os.path.join(RESOURCES_FOLDER, "projects", self.project_name)
|
|
img_path = os.path.join(project_folder, "preview.png")
|
|
if os.path.exists(img_path):
|
|
return img_path
|
|
return "resources/images/track.png"
|
|
|
|
preview_image_source = AliasProperty(
|
|
get_preview_image_source, None, bind=['project_name', 'preview_image_version']
|
|
)
|
|
|
|
def on_pre_enter(self):
|
|
# Update the route entries label with the actual number of entries
|
|
project_folder = os.path.join(RESOURCES_FOLDER, "projects", self.project_name)
|
|
positions_path = os.path.join(project_folder, "positions.json")
|
|
count = 0
|
|
if os.path.exists(positions_path):
|
|
with open(positions_path, "r") as f:
|
|
try:
|
|
positions = json.load(f)
|
|
count = len(positions)
|
|
except Exception:
|
|
count = 0
|
|
self.ids.route_entries_label.text = f"Your route has {count} entries,"
|
|
|
|
|
|
|
|
def open_rename_popup(self):
|
|
from kivy.uix.popup import Popup
|
|
from kivy.uix.boxlayout import BoxLayout
|
|
from kivy.uix.button import Button
|
|
from kivy.uix.textinput import TextInput
|
|
from kivy.uix.label import Label
|
|
|
|
layout = BoxLayout(orientation='vertical', spacing=10, padding=10)
|
|
label = Label(text="Enter new project name:")
|
|
input_field = TextInput(text=self.project_name, multiline=False)
|
|
btn_save = Button(text="Save", background_color=(0.008, 0.525, 0.290, 1))
|
|
btn_cancel = Button(text="Cancel")
|
|
|
|
layout.add_widget(label)
|
|
layout.add_widget(input_field)
|
|
layout.add_widget(btn_save)
|
|
layout.add_widget(btn_cancel)
|
|
|
|
popup = Popup(
|
|
title="Rename Project",
|
|
content=layout,
|
|
size_hint=(0.92, None),
|
|
size=(0, 260),
|
|
auto_dismiss=False
|
|
)
|
|
|
|
def do_rename(instance):
|
|
new_name = input_field.text.strip()
|
|
if new_name and new_name != self.project_name:
|
|
if self.rename_project_folder(self.project_name, new_name):
|
|
self.project_name = new_name
|
|
popup.dismiss()
|
|
self.on_pre_enter() # Refresh label
|
|
else:
|
|
label.text = "Rename failed (name exists?)"
|
|
else:
|
|
label.text = "Please enter a new name."
|
|
|
|
btn_save.bind(on_press=do_rename)
|
|
btn_cancel.bind(on_press=lambda x: popup.dismiss())
|
|
popup.open()
|
|
|
|
def rename_project_folder(self, old_name, new_name):
|
|
import os
|
|
old_path = os.path.join(RESOURCES_FOLDER, "projects", old_name)
|
|
new_path = os.path.join(RESOURCES_FOLDER, "projects", new_name)
|
|
if os.path.exists(old_path) and not os.path.exists(new_path):
|
|
os.rename(old_path, new_path)
|
|
return True
|
|
return False
|
|
|
|
def optimize_route_entries(self):
|
|
# Create the popup and UI elements
|
|
layout = BoxLayout(orientation='vertical', spacing=10, padding=10)
|
|
label = Label(text="Processing route entries...")
|
|
progress = ProgressBar(max=100, value=0)
|
|
layout.add_widget(label)
|
|
layout.add_widget(progress)
|
|
popup = Popup(
|
|
title="Optimizing Route",
|
|
content=layout,
|
|
size_hint=(0.92, None),
|
|
size=(0, 260),
|
|
auto_dismiss=False
|
|
)
|
|
popup.open()
|
|
|
|
# Now call the utility function with these objects
|
|
optimize_route_entries_util(
|
|
self.project_name,
|
|
RESOURCES_FOLDER,
|
|
label,
|
|
progress,
|
|
popup,
|
|
Clock,
|
|
on_save=lambda: self.on_pre_enter()
|
|
)
|
|
|
|
|
|
def preview_route(self):
|
|
# Show processing popup
|
|
layout = BoxLayout(orientation='vertical', spacing=10, padding=10)
|
|
label = Label(text="Processing route preview...")
|
|
progress = ProgressBar(max=100, value=0)
|
|
layout.add_widget(label)
|
|
layout.add_widget(progress)
|
|
popup = Popup(
|
|
title="Previewing Route",
|
|
content=layout,
|
|
size_hint=(0.8, None),
|
|
size=(0, 180),
|
|
auto_dismiss=False
|
|
)
|
|
popup.open()
|
|
|
|
def set_preview_image_path(path):
|
|
self.preview_image_path = path
|
|
self.preview_image_version += 1 # Force AliasProperty to update
|
|
self.property('preview_image_source').dispatch(self)
|
|
self.ids.preview_image.reload()
|
|
# Schedule the processing function
|
|
Clock.schedule_once(
|
|
lambda dt: process_preview_util(
|
|
self.project_name,
|
|
RESOURCES_FOLDER,
|
|
label,
|
|
progress,
|
|
popup,
|
|
self.ids.preview_image,
|
|
set_preview_image_path,
|
|
Clock
|
|
),
|
|
0.5
|
|
)
|
|
|
|
def open_pauses_popup(self):
|
|
"""Navigate to the pause edit screen"""
|
|
pause_edit_screen = self.manager.get_screen("pause_edit")
|
|
pause_edit_screen.set_project_and_callback(self.project_name, self.on_pre_enter)
|
|
self.manager.current = "pause_edit"
|
|
|
|
def generate_3d_video(self):
|
|
"""Show video generation mode selection popup"""
|
|
self.show_video_generation_options()
|
|
|
|
def generate_3d_video_test_mode(self):
|
|
"""Generate a 3D video animation in 720p test mode for faster processing"""
|
|
# Show processing popup with test mode indication
|
|
layout = BoxLayout(orientation='vertical', spacing=10, padding=10)
|
|
label = Label(text="Preparing 720p test video generation...")
|
|
progress = ProgressBar(max=100, value=0)
|
|
layout.add_widget(label)
|
|
layout.add_widget(progress)
|
|
popup = Popup(
|
|
title="Generating 3D Video Animation (720p Test Mode)",
|
|
content=layout,
|
|
size_hint=(0.9, None),
|
|
size=(0, 200),
|
|
auto_dismiss=False
|
|
)
|
|
popup.open()
|
|
|
|
# Schedule the 3D video generation in test mode
|
|
Clock.schedule_once(
|
|
lambda dt: generate_3d_video_animation(
|
|
self.project_name,
|
|
RESOURCES_FOLDER,
|
|
label,
|
|
progress,
|
|
popup,
|
|
Clock,
|
|
test_mode=True # Enable test mode
|
|
),
|
|
0.5
|
|
)
|
|
|
|
def generate_3d_video_production_mode(self):
|
|
"""Generate a 3D video animation in 2K production mode for high quality"""
|
|
# Show processing popup with production mode indication
|
|
layout = BoxLayout(orientation='vertical', spacing=10, padding=10)
|
|
label = Label(text="Preparing 2K production video generation...")
|
|
progress = ProgressBar(max=100, value=0)
|
|
layout.add_widget(label)
|
|
layout.add_widget(progress)
|
|
popup = Popup(
|
|
title="Generating 3D Video Animation (2K Production Mode)",
|
|
content=layout,
|
|
size_hint=(0.9, None),
|
|
size=(0, 200),
|
|
auto_dismiss=False
|
|
)
|
|
popup.open()
|
|
|
|
# Schedule the 3D video generation in production mode
|
|
Clock.schedule_once(
|
|
lambda dt: generate_3d_video_animation(
|
|
self.project_name,
|
|
RESOURCES_FOLDER,
|
|
label,
|
|
progress,
|
|
popup,
|
|
Clock,
|
|
test_mode=False # Disable test mode for production
|
|
),
|
|
0.5
|
|
)
|
|
|
|
def show_video_generation_options(self):
|
|
"""Show popup with video generation mode options"""
|
|
from kivy.uix.popup import Popup
|
|
from kivy.uix.boxlayout import BoxLayout
|
|
from kivy.uix.button import Button
|
|
from kivy.uix.label import Label
|
|
|
|
layout = BoxLayout(orientation='vertical', spacing=15, padding=15)
|
|
|
|
# Title
|
|
title_label = Label(
|
|
text="Choose Video Generation Mode",
|
|
font_size=18,
|
|
size_hint_y=None,
|
|
height=40,
|
|
color=(1, 1, 1, 1)
|
|
)
|
|
layout.add_widget(title_label)
|
|
|
|
# Test mode description
|
|
test_layout = BoxLayout(orientation='vertical', spacing=5)
|
|
test_title = Label(
|
|
text="🏃♂️ 720p Test Mode (Fast)",
|
|
font_size=16,
|
|
size_hint_y=None,
|
|
height=30,
|
|
color=(0.2, 0.8, 0.2, 1)
|
|
)
|
|
test_desc = Label(
|
|
text="• Resolution: 1280x720\n• Frame rate: 30 FPS\n• ~3x faster generation\n• Perfect for quick previews",
|
|
font_size=12,
|
|
size_hint_y=None,
|
|
height=80,
|
|
color=(0.9, 0.9, 0.9, 1),
|
|
halign="left",
|
|
valign="middle"
|
|
)
|
|
test_desc.text_size = (None, None)
|
|
test_layout.add_widget(test_title)
|
|
test_layout.add_widget(test_desc)
|
|
layout.add_widget(test_layout)
|
|
|
|
# Test mode button
|
|
test_btn = Button(
|
|
text="Generate 720p Test Video",
|
|
background_color=(0.2, 0.8, 0.2, 1),
|
|
size_hint_y=None,
|
|
height=50,
|
|
font_size=14
|
|
)
|
|
layout.add_widget(test_btn)
|
|
|
|
# Production mode description
|
|
prod_layout = BoxLayout(orientation='vertical', spacing=5)
|
|
prod_title = Label(
|
|
text="🎯 2K Production Mode (High Quality)",
|
|
font_size=16,
|
|
size_hint_y=None,
|
|
height=30,
|
|
color=(0.8, 0.2, 0.2, 1)
|
|
)
|
|
prod_desc = Label(
|
|
text="• Resolution: 2560x1440\n• Frame rate: 60 FPS\n• Cinema-quality results\n• Ultra-detailed visuals",
|
|
font_size=12,
|
|
size_hint_y=None,
|
|
height=80,
|
|
color=(0.9, 0.9, 0.9, 1),
|
|
halign="left",
|
|
valign="middle"
|
|
)
|
|
prod_desc.text_size = (None, None)
|
|
prod_layout.add_widget(prod_title)
|
|
prod_layout.add_widget(prod_desc)
|
|
layout.add_widget(prod_layout)
|
|
|
|
# Production mode button
|
|
prod_btn = Button(
|
|
text="Generate 2K Production Video",
|
|
background_color=(0.8, 0.2, 0.2, 1),
|
|
size_hint_y=None,
|
|
height=50,
|
|
font_size=14
|
|
)
|
|
layout.add_widget(prod_btn)
|
|
|
|
# Cancel button
|
|
cancel_btn = Button(
|
|
text="Cancel",
|
|
background_color=(0.5, 0.5, 0.5, 1),
|
|
size_hint_y=None,
|
|
height=40,
|
|
font_size=12
|
|
)
|
|
layout.add_widget(cancel_btn)
|
|
|
|
popup = Popup(
|
|
title="Select Video Generation Mode",
|
|
content=layout,
|
|
size_hint=(0.9, 0.8),
|
|
auto_dismiss=False
|
|
)
|
|
|
|
def start_test_mode(instance):
|
|
popup.dismiss()
|
|
self.generate_3d_video_test_mode()
|
|
|
|
def start_production_mode(instance):
|
|
popup.dismiss()
|
|
self.generate_3d_video_production_mode()
|
|
|
|
test_btn.bind(on_press=start_test_mode)
|
|
prod_btn.bind(on_press=start_production_mode)
|
|
cancel_btn.bind(on_press=lambda x: popup.dismiss())
|
|
|
|
popup.open()
|
|
|