devedeng-extended/src/devedeng/dvdauthor_converter.py
Sergio Costas 501e8aac48 general: removed the "or any later version" statement
After the events happened recently in the management of the FSF,
I decided that it is safer to not rely on the good faith of other
people, so I decided to allow distribution of Devede only under
the current GPLv3 license, and not allow other, still unknown,
future versions of the license.
2019-10-13 18:18:11 +02:00

562 lines
23 KiB
Python

#!/usr/bin/env python3
# Copyright 2014 (C) Raster Software Vigo (Sergio Costas)
#
# This file is part of DeVeDe-NG
#
# DeVeDe-NG is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License.
#
# DeVeDe-NG is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>
import os
import devedeng.configuration_data
import devedeng.executor
class dvdauthor_converter(devedeng.executor.executor):
def __init__(self):
devedeng.executor.executor.__init__(self)
self.config = devedeng.configuration_data.configuration.get_config()
def create_dvd_project(self, path, name, file_movies, menu_entries, start_with_menu, play_all_opt, menuWide):
movie_path = os.path.join(path, "dvd_tree")
try:
os.makedirs(movie_path)
except:
pass
xml_file = self.create_dvdauthor_xml(path, file_movies, menu_entries, start_with_menu, play_all_opt, menuWide)
self.command_var = []
self.command_var.append("dvdauthor")
self.command_var.append("-o")
self.command_var.append(movie_path)
self.command_var.append("-x")
self.command_var.append(xml_file)
self.use_pulse_mode = True
self.text = _("Creating DVD structure")
def process_stdout(self, data):
return
def process_stderr(self, data):
if (data is not None) and (data[0] != ""):
self.progress_bar[1].set_text(data[0])
return
def create_dvdauthor_xml(self, movie_folder, file_movies_in, menu_entries, start_with_menu, play_all_opt, menuWide):
file_movies = []
for element in file_movies_in:
if element.element_type != "file_movie":
continue
file_movies.append(element)
xmlpath = os.path.join(movie_folder, "xml_data")
xml_file_path = os.path.join(xmlpath, "dvdauthor.xml")
datapath = os.path.join(movie_folder, "dvd_tree")
try:
os.makedirs(xmlpath)
except:
pass
if (len(file_movies) == 1) and (menu_entries is None):
onlyone = True
else:
onlyone = False
if (menu_entries is None):
elements_per_menu = 1000
else:
elements_per_menu = len(menu_entries[0]["chapters"])
xml_file = open(xml_file_path, "w")
xml_file.write('<dvdauthor dest="' + datapath + '">\n')
if onlyone:
xml_file.write('\t<vmgm>\n')
xml_file.write("\t\t<menus>\n")
xml_file.write('\t\t\t<video format="')
if self.config.PAL:
xml_file.write("pal")
else:
xml_file.write("ntsc")
if menuWide:
xml_file.write('" aspect="16:9"> ')
else:
xml_file.write('" aspect="4:3"> ')
xml_file.write('</video>\n')
xml_file.write("\t\t</menus>\n")
xml_file.write('\t</vmgm>\n')
else:
xml_file.write('\t<vmgm>\n')
# MENU
# in the FPC we do a jump to the first menu in the first titleset if we wanted MENU
# or we jump to the second titleset if we didn't want MENU at
# startup
xml_file.write('\t\t<fpc>\n')
xml_file.write('\t\t\tg0=100;\n')
xml_file.write('\t\t\tg1=') # goto variable
if (menu_entries is not None) and (start_with_menu):
xml_file.write('0;\n') # show menu
else:
xml_file.write('100;\n') # auto play
xml_file.write('\t\t\tg2=1024;\n') # highlight?
xml_file.write('\t\t\tg3=') # play all variable
if play_all_opt and (menu_entries is not None) and (start_with_menu):
xml_file.write('1;\n') # auto play all
else:
xml_file.write('0;\n') # do not play all
xml_file.write('\t\t\tjump menu 1;\n')
xml_file.write('\t\t</fpc>\n')
# in the VMGM menu we create a code to jump to the title specified in G0
# but if the title is 100, we jump to the menus. There we show the menu number
# contained in G1
xml_file.write("\t\t<menus>\n")
xml_file.write('\t\t\t<video format="')
if self.config.PAL:
xml_file.write("pal")
else:
xml_file.write("ntsc")
if menuWide:
xml_file.write('" aspect="16:9"> ')
else:
xml_file.write('" aspect="4:3"> ')
xml_file.write('</video>\n')
xml_file.write('\t\t\t<pgc>\n')
xml_file.write('\t\t\t\t<pre>\n')
counter = 1
for element in file_movies:
xml_file.write('\t\t\t\t\tif (g0 eq ' + str(counter) + ') {\n')
xml_file.write('\t\t\t\t\t\tjump titleset ' +
str(1 + counter) + ' menu;\n')
xml_file.write('\t\t\t\t\t}\n')
counter += 1
xml_file.write('\t\t\t\t\tif (g0 eq 100) {\n')
xml_file.write('\t\t\t\t\t\tg2=1024;\n')
xml_file.write('\t\t\t\t\t\tjump titleset 1 menu;\n')
xml_file.write('\t\t\t\t\t}\n')
xml_file.write('\t\t\t\t</pre>\n')
# fake video (one black picture with one second of sound) to ensure
# 100% compatibility
xml_file.write('\t\t\t\t<vob file="')
if menuWide:
if self.config.PAL:
xml_file.write(self.expand_xml(str(os.path.join(self.config.other_path, "base_pal_wide.mpg"))))
else:
xml_file.write(self.expand_xml(str(os.path.join(self.config.other_path, "base_ntsc_wide.mpg"))))
else:
if self.config.PAL:
xml_file.write(self.expand_xml(str(os.path.join(self.config.other_path, "base_pal.mpg"))))
else:
xml_file.write(self.expand_xml(str(os.path.join(self.config.other_path, "base_ntsc.mpg"))))
xml_file.write('"></vob>\n')
xml_file.write('\t\t\t</pgc>\n')
xml_file.write('\t\t</menus>\n')
xml_file.write("\t</vmgm>\n")
xml_file.write("\n")
# the first titleset contains all the menus. G1 allows us to jump
# to the desired menu
xml_file.write('\t<titleset>\n')
xml_file.write('\t\t<menus>\n')
xml_file.write('\t\t\t<video format="')
if self.config.PAL:
xml_file.write("pal")
else:
xml_file.write("ntsc")
if menuWide:
xml_file.write('" aspect="16:9"> ')
else:
xml_file.write('" aspect="4:3"> ')
xml_file.write('</video>\n')
first_entry = True
menu_number = 0
counter = 1
title_list = []
for element in file_movies:
if element.show_in_menu:
title_list.append(counter)
counter += 1
if (menu_entries is not None):
nmenues = len(menu_entries)
title_counter = 0 # A counter for all titles
for menu_page in menu_entries:
button_counter = 0 # A counter for all buttons on the actual page
xml_file.write('\t\t\t<pgc>\n')
xml_file.write('\t\t\t\t<pre>\n')
# first we recover the currently selected button
xml_file.write('\t\t\t\t\ts8=g2;\n')
xml_file.write('\t\t\t\t\tg3=0;\n') # turnoff play all
if first_entry: # here we add some code to jump to each menu
for menu2 in range(nmenues - 1):
xml_file.write(
'\t\t\t\t\tif (g1 eq ' + str(menu2 + 1) + ') {\n')
xml_file.write(
'\t\t\t\t\t\tjump menu ' + str(menu2 + 2) + ';\n')
xml_file.write('\t\t\t\t\t}\n')
# this code is to fix a bug in some players
xml_file.write('\t\t\t\t\tif (g1 eq 100) {\n')
# menu '+str(self.nmenues+1)+';\n')
xml_file.write('\t\t\t\t\t\tjump title 1;\n')
xml_file.write('\t\t\t\t\t}\n')
first_entry = False
xml_file.write('\t\t\t\t</pre>\n')
xml_file.write('\t\t\t\t<vob file="')
xml_file.write(self.expand_xml(menu_page["filename"]))
xml_file.write('"></vob>\n')
for nbutton in menu_page["chapters"]:
#xml_file.write('\t\t\t\t<button name="'+nbutton+'"> g0='+str(title_list[button_counter])+'; jump vmgm menu; </button>\n')
xml_file.write(
'\t\t\t\t<button name="' + nbutton + '">\n')
# If PlayAll option is set, the first button on each page
# will be the PlayAll button
if play_all_opt and button_counter == 0:
xml_file.write('\t\t\t\t\tg3=1;\n')
xml_file.write('\t\t\t\t\tg0=1;\n')
xml_file.write('\t\t\t\t\tjump vmgm menu;\n')
else:
xml_file.write('\t\t\t\t\tg0={:};\n'.format(str(title_list[title_counter])))
xml_file.write('\t\t\t\t\tjump vmgm menu;\n')
title_counter += 1
button_counter += 1
xml_file.write('\t\t\t\t</button>\n')
if (menu_page["left"] is not None):
xml_file.write('\t\t\t\t<button name="' + menu_page["left"] + '"> g1=')
xml_file.write(str(menu_number - 1))
xml_file.write('; g2=1024; jump menu ')
xml_file.write(str(menu_number))
xml_file.write('; </button>\n')
if (menu_page["right"] is not None):
xml_file.write('\t\t\t\t<button name="' +
menu_page["right"] + '"> g1=')
xml_file.write(str(menu_number + 1))
xml_file.write('; g2=1024; jump menu ')
xml_file.write(str(menu_number + 2))
xml_file.write('; </button>\n')
xml_file.write('\t\t\t\t<post>\n')
xml_file.write('\t\t\t\t\tg2=s8;\n')
xml_file.write('\t\t\t\t\tg1=' + str(menu_number) + ';\n')
xml_file.write('\t\t\t\t\tjump menu ' +
str(menu_number + 1) + ';\n')
xml_file.write('\t\t\t\t</post>\n')
xml_file.write('\t\t\t</pgc>\n')
menu_number += 1
xml_file.write('\t\t</menus>\n')
else:
xml_file.write('\t\t\t<pgc>\n')
xml_file.write('\t\t\t\t<pre>\n')
# first we recover the currently selected button
xml_file.write('\t\t\t\t\ts8=g2;\n')
# this code is to fix a bug in some players
xml_file.write('\t\t\t\t\tif (g1 eq 100) {\n')
# menu '+str(self.nmenues+1)+';\n')
xml_file.write('\t\t\t\t\t\tjump title 1;\n')
xml_file.write('\t\t\t\t\t}\n')
xml_file.write('\t\t\t\t</pre>\n')
xml_file.write('\t\t\t\t<vob file="')
if menuWide:
if self.config.PAL:
xml_file.write(self.expand_xml(str(os.path.join(self.config.other_path, "base_pal_wide.mpg"))))
else:
xml_file.write(self.expand_xml(str(os.path.join(self.config.other_path, "base_ntsc_wide.mpg"))))
else:
if self.config.PAL:
xml_file.write(self.expand_xml(str(os.path.join(self.config.other_path, "base_pal.mpg"))))
else:
xml_file.write(self.expand_xml(str(os.path.join(self.config.other_path, "base_ntsc.mpg"))))
xml_file.write('"></vob>\n')
xml_file.write('\t\t\t\t<post>\n')
xml_file.write('\t\t\t\t\tg2=s8;\n')
xml_file.write('\t\t\t\t\tg1=0;\n')
xml_file.write('\t\t\t\t\tjump menu 1;\n')
xml_file.write('\t\t\t\t</post>\n')
xml_file.write('\t\t\t</pgc>\n')
xml_file.write('\t\t</menus>\n')
xml_file.write('\t\t<titles>\n')
xml_file.write('\t\t\t<video format="')
if self.config.PAL:
xml_file.write("pal")
else:
xml_file.write("ntsc")
if menuWide:
xml_file.write('" aspect="16:9"> ')
else:
xml_file.write('" aspect="4:3"> ')
xml_file.write('</video>\n')
xml_file.write('\t\t\t<pgc>\n')
xml_file.write('\t\t\t\t<vob file="')
if menuWide:
if self.config.PAL:
xml_file.write(self.expand_xml(str(os.path.join(self.config.other_path, "base_pal_wide.mpg"))))
else:
xml_file.write(self.expand_xml(str(os.path.join(self.config.other_path, "base_ntsc_wide.mpg"))))
else:
if self.config.PAL:
xml_file.write(self.expand_xml(str(os.path.join(self.config.other_path, "base_pal.mpg"))))
else:
xml_file.write(self.expand_xml(str(os.path.join(self.config.other_path, "base_ntsc.mpg"))))
xml_file.write('"></vob>\n')
xml_file.write('\t\t\t\t<post>\n')
xml_file.write('\t\t\t\t\tg0=1;\n')
xml_file.write('\t\t\t\t\tg1=0;\n')
xml_file.write('\t\t\t\t\tg2=1024;\n')
xml_file.write('\t\t\t\t\tcall vmgm menu entry title;\n')
xml_file.write('\t\t\t\t</post>\n')
xml_file.write('\t\t\t</pgc>\n')
xml_file.write('\t\t</titles>\n')
xml_file.write("\t</titleset>\n")
xml_file.write("\n")
# Now, create the titleset for each video
total_t = len(file_movies)
titleset = 1
titles = 0
counter = 0
for element in file_movies:
files = 0
action = element.actions
xml_file.write("\n")
if element.is_mpeg_ps:
# if it's already an MPEG-2 compliant file, we use the original
# values
if element.original_fps == 25:
pal_ntsc = "pal"
ispal = True
else:
pal_ntsc = "ntsc"
ispal = False
if element.original_aspect_ratio > 1.6:
faspect = '16:9'
fwide = True
else:
faspect = '4:3'
fwide = False
else:
# but if we are converting it, we use the desired values
if element.format_pal:
pal_ntsc = "pal"
ispal = True
else:
pal_ntsc = "ntsc"
ispal = False
if element.aspect_ratio_final > 1.6:
faspect = '16:9'
fwide = True
else:
faspect = '4:3'
fwide = False
xml_file.write("\t<titleset>\n")
if not onlyone:
xml_file.write("\t\t<menus>\n")
xml_file.write('\t\t\t<video format="' +
pal_ntsc + '" aspect="' + faspect + '"')
if fwide:
xml_file.write(' widescreen="nopanscan"')
xml_file.write('> </video>\n')
xml_file.write("\t\t\t<pgc>\n")
xml_file.write("\t\t\t\t<pre>\n")
xml_file.write('\t\t\t\t\tif (g0 eq 100) {\n')
xml_file.write('\t\t\t\t\t\tjump vmgm menu entry title;\n')
xml_file.write('\t\t\t\t\t}\n')
xml_file.write('\t\t\t\t\tg0=100;\n')
xml_file.write('\t\t\t\t\tg1=' +
str(int(titles / elements_per_menu)) + ';\n')
xml_file.write('\t\t\t\t\tjump title 1;\n')
xml_file.write('\t\t\t\t</pre>\n')
# fake video to ensure compatibility
xml_file.write('\t\t\t\t<vob file="')
if ispal:
xml_file.write(self.expand_xml(
str(os.path.join(self.config.other_path, "base_pal"))))
else:
xml_file.write(self.expand_xml(
str(os.path.join(self.config.other_path, "base_ntsc"))))
if fwide:
xml_file.write("_wide")
xml_file.write('.mpg"></vob>\n')
xml_file.write("\t\t\t</pgc>\n")
xml_file.write("\t\t</menus>\n")
xml_file.write("\t\t<titles>\n")
xml_file.write('\t\t\t<video format="' +
pal_ntsc + '" aspect="' + faspect + '"')
if fwide:
xml_file.write(' widescreen="nopanscan"')
xml_file.write('> </video>\n')
# subtitles part
# for element3 in element2["sub_list"]:
# xml_file.write('\t\t\t<subpicture lang="'+self.expand_xml(str(element3["sub_language"][:2].lower()))+'" />\n')
xml_file.write('\t\t\t<pgc>\n')
# if (element2["force_subs"]) and (len(element2["sub_list"])!=0):
# xml_file.write('\t\t\t\t<pre>\n')
# xml_file.write('\t\t\t\t\tsubtitle=64;\n')
# xml_file.write('\t\t\t\t</pre>\n')
xml_file.write('\t\t\t\t<vob file="' +
self.expand_xml(element.converted_filename) + '" ')
xml_file.write('chapters="0')
if (element.original_length > 5):
if (element.divide_in_chapters): # add chapters
manual_chapter_list = []
if (element.chapter_list_entry):
manual_chapter_list = element.chapter_list_entry.split(
",")
toadd = element.chapter_size
seconds = toadd * 60
while seconds < (element.original_length - 4):
# Check manual_chapter_list for manual entries before
# set next chapter
for idx, m_chapter in enumerate(manual_chapter_list):
m_chapter = m_chapter.split(":")
# Change from mm:ss to seconds
m_chapter = int(
m_chapter[0]) * 60 + int(m_chapter[1])
if (m_chapter < seconds):
seconds = m_chapter
del manual_chapter_list[idx]
thetime = self.return_time(seconds, False)
xml_file.write("," + thetime)
seconds += (toadd * 60)
xml_file.write(
',' + self.return_time((element.original_length - 2), False))
xml_file.write('" />\n')
if not onlyone:
xml_file.write('\t\t\t\t<post>\n')
files += 1
xml_file.write('\t\t\t\t\tg1=' +
str(int(titles / elements_per_menu)) + ';\n')
# play all
xml_file.write('\t\t\t\t\tif (g3 eq 1) {\n') # if play all:
if titles == total_t - 1: # return to menu if last title
xml_file.write('\t\t\t\t\t\tg0=100;\n')
else: # play next title
xml_file.write('\t\t\t\t\t\tg0=' + str(titles + 2) + ';\n')
xml_file.write('\t\t\t\t\t} else {\n')
# end of play opt
xml_file.write('\t\t\t\t\t\tg0=')
if action == "action_stop":
xml_file.write('100')
elif action == "action_play_previous":
if titles == 0:
prev_t = total_t - 1
else:
prev_t = titles - 1
xml_file.write(str(prev_t + 1))
elif action == "action_play_again":
xml_file.write(str(titles + 1))
elif action == "action_play_next":
if titles == total_t - 1:
next_t = 0
else:
next_t = titles + 1
xml_file.write(str(next_t + 1))
elif action == "action_play_last":
xml_file.write(str(total_t))
else:
xml_file.write('1') # first
xml_file.write(';\n')
xml_file.write('\t\t\t\t\t}\n')
# preform action
xml_file.write('\t\t\t\t\tcall vmgm menu entry title;\n')
xml_file.write('\t\t\t\t</post>\n')
xml_file.write("\t\t\t</pgc>\n")
xml_file.write("\t\t</titles>\n")
xml_file.write("\t</titleset>\n")
counter += 1
titles += 1
xml_file.write("</dvdauthor>")
xml_file.close()
return xml_file_path
def return_time(self, seconds, empty):
""" cuts a time in seconds into seconds, minutes and hours """
seconds2 = int(seconds)
hours = str(int(seconds2 / 3600))
if empty:
if len(hours) == 1:
hours = "0" + hours
else:
if hours == "0":
hours = ""
if hours != "":
hours += ":"
minutes = str(int((seconds2 / 60) % 60))
if empty or (hours != ""):
if len(minutes) == 1:
minutes = "0" + minutes
elif (minutes == "0") and (hours == ""):
minutes = ""
if minutes != "":
minutes += ":"
secs = str(int(seconds2 % 60))
if (len(secs) == 1) and (minutes != ""):
secs = "0" + secs
return hours + minutes + secs