script.py

import os
from datetime import datetime
import csv

# Directorio raíz donde se encuentran las carpetas con las secuencias de png
directorio_raiz = r"I:\_Unreal"

# Velocidad de cuadro (FPS)
fps = 30

# Función para calcular el tiempo en segundos para un frame dado
def calcular_tiempo(frame, fps):
    return frame / fps

# Función para aplicar el formato a los segundos
def formato_segundos(segundos):
    horas = int(segundos // 3600)
    segundos_restantes = int(segundos % 3600)
    minutos = int(segundos_restantes // 60)
    segundos = int(segundos_restantes % 60)
    return f"{horas:02d}:{minutos:02d}:{segundos:02d}"


# Nombre del archivo CSV
nombre_csv = f"{datetime.now().strftime("%H:%M:%S")}.csv".replace(":", "-")


# Abre el archivo CSV en modo de escritura
with open(f"{directorio_raiz}\{nombre_csv}", mode='a', newline='') as file:  
    # Escribe la cabecera del CSV
    csv.writer(file).writerow([
        'File Name', 'Scene', 'Take'
    ])


# Recorre todas las carpetas en el directorio raíz
for carpeta in os.listdir(directorio_raiz):
    carpeta_path = directorio_raiz + '/' + carpeta + '/'
    
    # Verifica si la carpeta contiene archivos png
    if os.path.isdir(carpeta_path):
        archivos_png = [archivo for archivo in os.listdir(carpeta_path) if archivo.endswith(".png")]
        
        # Ordena los archivos png alfabéticamente
        archivos_png.sort()
        
        # Verifica si hay al menos un archivo png en la carpeta
        if archivos_png:
            primer_archivo = archivos_png[0]
            nombre_archivo = os.path.splitext(primer_archivo)[0]
            nombre_escena, numero_frame = nombre_archivo.split(".")
            print("\nnombre escena: ", nombre_escena)
            print("\nnumero frame: ", numero_frame, "\n")
            numero_frame = int(numero_frame)
            tiempo_segundos = calcular_tiempo(numero_frame, fps)
            hhmmss = formato_segundos(tiempo_segundos)
            
            # Imprime el tiempo de inicio para el primer frame de la carpeta
            print(f"Carpeta: {carpeta}")
            print(f"  Tiempo de inicio (segundos): {tiempo_segundos}")
            print(hhmmss)


            # Convertir a .mov con ffmpeg

            import subprocess
            import os

            # Directorio de las imágenes png
            input_directory = carpeta_path

            # Nombre del archivo de salida
            output_file = directorio_raiz + "/" + f"{nombre_escena}.mov"

            # Timecode en formato HH:MM:SS:FF (horas, minutos, segundos, fotogramas)
            timecode = hhmmss  # Por ejemplo, comenzar en 0 horas, 0 minutos, 0 segundos y 0 fotogramas

            # Obtener la lista de archivos en el directorio de entrada
            image_files = sorted([f for f in os.listdir(input_directory) if f.endswith(".png")])

            partes_nombre = nombre_escena.split("_")
            scene = partes_nombre[1]
            shot = partes_nombre[2]

            # Construir una lista de argumentos para FFmpeg
            ffmpeg_args = [
                "ffmpeg",
                "-framerate", str(fps),                                 # Tasa de fotogramas de salida (ajusta según sea necesario)
                "-start_number", str(numero_frame),                     # Número de inicio para los nombres de archivo
                "-i", input_directory + f"{nombre_escena}.%07d.png",    # Patrón de nombres de archivo
                "-c:v", "libx264",                                      # Códec de video (H.264)
                "-pix_fmt", "yuv420p",                                  # Formato de píxeles
                "-crf", "18",                                           # Calidad de compresión (ajusta según sea necesario)
                "-timecode", timecode + ":00",                          # Agrega ":00" al final del timecode
                output_file                                             # Nombre del archivo de salida
            ]

            print("Argumentos FFmpeg:", str(ffmpeg_args))

            # Ejecutar el comando FFmpeg para crear el video
            subprocess.run(ffmpeg_args)

            print(f"\nVideo creado con timecode y guardado como '{output_file}'.")

            # Print comando ffmpeg
            print(" ".join(ffmpeg_args))

            with open(f"{directorio_raiz}\{nombre_csv}", mode='a', newline='') as file:
                # Escribe la línea CSV con los datos relevantes
                csv.writer(file).writerow([f"{nombre_escena}.mov", scene, shot])