370 lines
No EOL
14 KiB
Python
370 lines
No EOL
14 KiB
Python
"""
|
||
Диалог для создания таймлапс-анимации
|
||
"""
|
||
|
||
from PySide6.QtWidgets import (
|
||
QDialog, QVBoxLayout, QHBoxLayout, QGroupBox,
|
||
QComboBox, QDateEdit, QPushButton, QLabel,
|
||
QProgressBar, QFileDialog, QMessageBox, QCheckBox,
|
||
QTextEdit
|
||
)
|
||
from PySide6.QtCore import Qt, QDateTime, QSettings, QThread, Signal
|
||
from pathlib import Path
|
||
from datetime import datetime, timedelta
|
||
import tempfile
|
||
import shutil
|
||
|
||
|
||
class TimelapseWorker(QThread):
|
||
progress = Signal(int, int, str)
|
||
log = Signal(str)
|
||
finished = Signal(bool, str)
|
||
|
||
def __init__(self, source_id, start_date, end_date, output_path, fps=10):
|
||
super().__init__()
|
||
self.source_id = source_id
|
||
self.start_date = start_date
|
||
self.end_date = end_date
|
||
self.output_path = output_path
|
||
self.fps = fps
|
||
self._is_cancelled = False
|
||
|
||
def cancel(self):
|
||
self._is_cancelled = True
|
||
|
||
def run(self):
|
||
from models.api_model import HelioviewerAPI
|
||
from utils.video_creator import VideoCreator
|
||
|
||
try:
|
||
# Создаем временную папку
|
||
temp_dir = Path(tempfile.mkdtemp(prefix="helioviewer_timelapse_"))
|
||
self.log.emit(f"Создана временная папка: {temp_dir}")
|
||
|
||
# Генерируем даты - ИСПРАВЛЕНО
|
||
dates = []
|
||
|
||
# Проверяем тип и преобразуем в datetime если нужно
|
||
from datetime import datetime as dt
|
||
if isinstance(self.start_date, dt):
|
||
current = self.start_date.replace(hour=12, minute=0, second=0)
|
||
else:
|
||
# Если это date, конвертируем в datetime
|
||
current = dt.combine(self.start_date, dt.min.time()).replace(hour=12)
|
||
|
||
# Проверяем конец
|
||
if isinstance(self.end_date, dt):
|
||
end = self.end_date
|
||
else:
|
||
end = dt.combine(self.end_date, dt.max.time())
|
||
|
||
while current <= end:
|
||
dates.append(current)
|
||
current += timedelta(days=1)
|
||
|
||
total = len(dates)
|
||
self.log.emit(f"Всего файлов в очереди: {total}")
|
||
|
||
downloaded = []
|
||
|
||
for i, date in enumerate(dates):
|
||
if self._is_cancelled:
|
||
self.log.emit("Отменено пользователем")
|
||
shutil.rmtree(temp_dir, ignore_errors=True)
|
||
self.finished.emit(False, "Отменено")
|
||
return
|
||
|
||
current_num = i + 1
|
||
percent = int(current_num / total * 100)
|
||
self.progress.emit(current_num, total, f"Скачивание {current_num}/{total} ({percent}%)")
|
||
self.log.emit(f"Скачивание {current_num}/{total}: {date.strftime('%Y-%m-%d')}")
|
||
|
||
filepath = HelioviewerAPI.download_image(self.source_id, date, temp_dir)
|
||
if filepath:
|
||
downloaded.append(filepath)
|
||
self.log.emit(f" ✓ Успешно")
|
||
else:
|
||
self.log.emit(f" ✗ Ошибка")
|
||
|
||
if self._is_cancelled:
|
||
self.log.emit("Отменено пользователем")
|
||
shutil.rmtree(temp_dir, ignore_errors=True)
|
||
self.finished.emit(False, "Отменено")
|
||
return
|
||
|
||
if not downloaded:
|
||
shutil.rmtree(temp_dir, ignore_errors=True)
|
||
self.finished.emit(False, "Не удалось скачать ни одного изображения")
|
||
return
|
||
|
||
self.log.emit(f"Создание видео из {len(downloaded)} кадров...")
|
||
self.progress.emit(total, total, "Создание видео...")
|
||
|
||
video_path = VideoCreator.create_timelapse(downloaded, self.output_path, self.fps)
|
||
|
||
# Очищаем временную папку
|
||
shutil.rmtree(temp_dir, ignore_errors=True)
|
||
self.log.emit(f"Временные файлы удалены")
|
||
|
||
self.finished.emit(True, str(video_path))
|
||
|
||
except Exception as e:
|
||
self.log.emit(f"Ошибка: {str(e)}")
|
||
self.finished.emit(False, str(e))
|
||
|
||
|
||
class TimelapseDialog(QDialog):
|
||
"""Диалог для создания таймлапса"""
|
||
|
||
def __init__(self, controller, parent=None):
|
||
super().__init__(parent)
|
||
self.controller = controller
|
||
self.settings = QSettings("SolarViewer", "Helioviewer")
|
||
self.worker = None
|
||
self.init_ui()
|
||
self.load_settings()
|
||
|
||
def init_ui(self):
|
||
"""Инициализация UI"""
|
||
self.setWindowTitle("Создание таймлапс-анимации")
|
||
self.setModal(True)
|
||
self.setMinimumWidth(500)
|
||
|
||
layout = QVBoxLayout(self)
|
||
|
||
# Группа выбора спектра
|
||
spectrum_group = QGroupBox("🌞 Спектральный канал")
|
||
spectrum_layout = QVBoxLayout(spectrum_group)
|
||
|
||
self.spectrum_combo = QComboBox()
|
||
self.populate_spectrums()
|
||
spectrum_layout.addWidget(self.spectrum_combo)
|
||
|
||
layout.addWidget(spectrum_group)
|
||
|
||
# Группа выбора дат
|
||
date_group = QGroupBox("📅 Период")
|
||
date_layout = QVBoxLayout(date_group)
|
||
|
||
# Начальная дата
|
||
start_layout = QHBoxLayout()
|
||
start_layout.addWidget(QLabel("Начало:"))
|
||
self.start_date = QDateEdit()
|
||
self.start_date.setDateTime(QDateTime.currentDateTime().addDays(-7))
|
||
self.start_date.setCalendarPopup(True)
|
||
start_layout.addWidget(self.start_date)
|
||
date_layout.addLayout(start_layout)
|
||
|
||
# Конечная дата
|
||
end_layout = QHBoxLayout()
|
||
end_layout.addWidget(QLabel("Конец:"))
|
||
self.end_date = QDateEdit()
|
||
self.end_date.setDateTime(QDateTime.currentDateTime())
|
||
self.end_date.setCalendarPopup(True)
|
||
end_layout.addWidget(self.end_date)
|
||
date_layout.addLayout(end_layout)
|
||
|
||
layout.addWidget(date_group)
|
||
|
||
# Параметры видео
|
||
video_group = QGroupBox("Параметры видео")
|
||
video_layout = QHBoxLayout(video_group)
|
||
|
||
video_layout.addWidget(QLabel("FPS:"))
|
||
self.fps_spin = QComboBox()
|
||
self.fps_spin.addItems(['5', '10', '15', '24', '30'])
|
||
self.fps_spin.setCurrentText('10')
|
||
video_layout.addWidget(self.fps_spin)
|
||
|
||
video_layout.addStretch()
|
||
|
||
layout.addWidget(video_group)
|
||
|
||
# Выбор папки сохранения
|
||
folder_group = QGroupBox("Папка сохранения")
|
||
folder_layout = QVBoxLayout(folder_group)
|
||
|
||
folder_select_layout = QHBoxLayout()
|
||
self.folder_path = QLabel("Не выбрана")
|
||
self.folder_path.setStyleSheet("color: gray;")
|
||
folder_select_layout.addWidget(self.folder_path, 1)
|
||
|
||
browse_button = QPushButton("Обзор...")
|
||
browse_button.clicked.connect(self.browse_folder)
|
||
folder_select_layout.addWidget(browse_button)
|
||
|
||
folder_layout.addLayout(folder_select_layout)
|
||
|
||
# Чекбокс "Запомнить путь"
|
||
self.remember_path_checkbox = QCheckBox("Запомнить этот путь")
|
||
self.remember_path_checkbox.setChecked(True)
|
||
folder_layout.addWidget(self.remember_path_checkbox)
|
||
|
||
layout.addWidget(folder_group)
|
||
|
||
# Прогресс бар
|
||
self.progress_bar = QProgressBar()
|
||
self.progress_bar.setVisible(False)
|
||
layout.addWidget(self.progress_bar)
|
||
|
||
# Лог сообщений
|
||
self.log_text = QTextEdit()
|
||
self.log_text.setMaximumHeight(150)
|
||
self.log_text.setReadOnly(True)
|
||
self.log_text.setVisible(False)
|
||
layout.addWidget(self.log_text)
|
||
|
||
# Кнопки
|
||
button_layout = QHBoxLayout()
|
||
|
||
self.create_button = QPushButton("Создать")
|
||
self.create_button.clicked.connect(self.start_timelapse)
|
||
button_layout.addWidget(self.create_button)
|
||
|
||
self.cancel_button = QPushButton("Отмена")
|
||
self.cancel_button.clicked.connect(self.cancel_timelapse)
|
||
self.cancel_button.setEnabled(False)
|
||
button_layout.addWidget(self.cancel_button)
|
||
|
||
close_button = QPushButton("Закрыть")
|
||
close_button.clicked.connect(self.reject)
|
||
button_layout.addWidget(close_button)
|
||
|
||
layout.addLayout(button_layout)
|
||
|
||
self.selected_folder = None
|
||
|
||
def populate_spectrums(self):
|
||
"""Заполняет список доступных спектров"""
|
||
from models.api_model import HelioviewerAPI
|
||
|
||
sources = HelioviewerAPI.get_available_sources()
|
||
for source_id, info in sources.items():
|
||
display_text = f"{info['observatory']} - {info['name']} ({info['wavelength']})"
|
||
self.spectrum_combo.addItem(display_text, source_id)
|
||
|
||
def browse_folder(self):
|
||
"""Выбор папки для сохранения"""
|
||
folder = QFileDialog.getExistingDirectory(self, "Выберите папку для сохранения")
|
||
if folder:
|
||
self.selected_folder = Path(folder)
|
||
self.folder_path.setText(str(self.selected_folder))
|
||
self.folder_path.setStyleSheet("color: green;")
|
||
|
||
if self.remember_path_checkbox.isChecked():
|
||
self.save_settings()
|
||
|
||
def save_settings(self):
|
||
"""Сохраняет настройки"""
|
||
if self.selected_folder:
|
||
self.settings.setValue("timelapse/last_folder", str(self.selected_folder))
|
||
self.settings.setValue("timelapse/remember_path", self.remember_path_checkbox.isChecked())
|
||
|
||
def load_settings(self):
|
||
"""Загружает настройки"""
|
||
remember = self.settings.value("timelapse/remember_path", True, type=bool)
|
||
self.remember_path_checkbox.setChecked(remember)
|
||
|
||
if remember:
|
||
last_folder = self.settings.value("timelapse/last_folder", "")
|
||
if last_folder and Path(last_folder).exists():
|
||
self.selected_folder = Path(last_folder)
|
||
self.folder_path.setText(str(self.selected_folder))
|
||
self.folder_path.setStyleSheet("color: green;")
|
||
|
||
def start_timelapse(self):
|
||
"""Запускает создание таймлапса"""
|
||
if not self.selected_folder:
|
||
QMessageBox.warning(self, "Внимание", "Выберите папку для сохранения")
|
||
return
|
||
|
||
source_id = self.spectrum_combo.currentData()
|
||
start_date = self.start_date.date().toPython()
|
||
end_date = self.end_date.date().toPython()
|
||
fps = int(self.fps_spin.currentText())
|
||
|
||
from datetime import datetime as dt
|
||
|
||
# Конвертируем date в datetime с правильным временем
|
||
start_datetime = dt(start_date.year, start_date.month, start_date.day, 12, 0, 0)
|
||
end_datetime = dt(end_date.year, end_date.month, end_date.day, 23, 59, 59)
|
||
|
||
days = (end_date - start_date).days + 1
|
||
filename = f"timelapse_{source_id}_{start_date.strftime('%Y%m%d')}_to_{end_date.strftime('%Y%m%d')}.mp4"
|
||
output_path = self.selected_folder / filename
|
||
|
||
self.save_settings()
|
||
|
||
# Обновляем UI
|
||
self.create_button.setEnabled(False)
|
||
self.cancel_button.setEnabled(True)
|
||
self.progress_bar.setVisible(True)
|
||
self.log_text.setVisible(True)
|
||
self.progress_bar.setValue(0)
|
||
self.log_text.clear()
|
||
|
||
self.add_log(f"🚀 Запуск таймлапса: {days} файлов")
|
||
self.add_log(f"📁 Папка: {self.selected_folder}")
|
||
|
||
# Создаем и запускаем поток
|
||
self.worker = TimelapseWorker(source_id, start_datetime, end_datetime, output_path, fps)
|
||
self.worker.progress.connect(self.update_progress)
|
||
self.worker.log.connect(self.add_log)
|
||
self.worker.finished.connect(self.on_finished)
|
||
self.worker.start()
|
||
|
||
def cancel_timelapse(self):
|
||
"""Отмена создания таймлапса"""
|
||
if self.worker and self.worker.isRunning():
|
||
reply = QMessageBox.question(
|
||
self, "Отмена",
|
||
"Вы уверены? Все уже скачанные файлы будут удалены.",
|
||
QMessageBox.Yes | QMessageBox.No
|
||
)
|
||
if reply == QMessageBox.Yes:
|
||
self.worker.cancel()
|
||
self.add_log("⏹️ Отмена процесса...")
|
||
self.cancel_button.setEnabled(False)
|
||
|
||
def update_progress(self, current, total, message):
|
||
"""Обновление прогресса"""
|
||
if total > 0:
|
||
percent = int((current / total) * 100)
|
||
self.progress_bar.setValue(percent)
|
||
|
||
def add_log(self, message):
|
||
"""Добавление сообщения в лог"""
|
||
timestamp = datetime.now().strftime("%H:%M:%S")
|
||
self.log_text.append(f"[{timestamp}] {message}")
|
||
self.log_text.verticalScrollBar().setValue(
|
||
self.log_text.verticalScrollBar().maximum()
|
||
)
|
||
|
||
def on_finished(self, success, message):
|
||
"""Обработка завершения"""
|
||
self.create_button.setEnabled(True)
|
||
self.cancel_button.setEnabled(False)
|
||
|
||
if success:
|
||
self.add_log(f"✅ ГОТОВО: {message}")
|
||
QMessageBox.information(self, "Готово", f"Таймлапс успешно создан!\n{message}")
|
||
self.accept()
|
||
else:
|
||
self.add_log(f"❌ Ошибка: {message}")
|
||
QMessageBox.critical(self, "Ошибка", message)
|
||
|
||
def closeEvent(self, event):
|
||
"""При закрытии окна"""
|
||
if self.worker and self.worker.isRunning():
|
||
reply = QMessageBox.question(
|
||
self, "Процесс выполняется",
|
||
"Создание таймлапса еще не завершено.\n\nЗакрыть окно?",
|
||
QMessageBox.Yes | QMessageBox.No
|
||
)
|
||
if reply == QMessageBox.Yes:
|
||
self.worker.cancel()
|
||
event.accept()
|
||
else:
|
||
event.ignore()
|
||
else:
|
||
event.accept() |