def lines(self):
L = partial(Line, resolution=self.resolution)
lines = [
L(l) for l in self._script_data["dialog"] if l["comment"] == False
]
if self.progressbar:
return helpers.progressbar(lines)
return lines
def dialogs(self):
D = partial(Dialog, resolution=self.resolution)
dialogs = [
D(d) for d in self._script_data["dialog"] if d["comment"] == False
]
if self.progressbar:
return helpers.progressbar(dialogs)
return dialogs
def _dialog(self):
dialog_fmt = []
for dialog in helpers.progressbar(
self._assdict["dialog"], prefix='Writing'):
if dialog["comment"]:
key = "Comment"
else:
key = "Dialogue"
if dialog["text"]:
dialog_fmt.append(
"{:s}: {:d},{:s},{:s},{:s},{:s},{:s},{:s},{:s},{:s},"
"{:s}".format(
key,
dialog["layer"],
dialog["start"],
dialog["end"],
dialog["style"]["name"],
dialog["actor"],
"0000", "0000", "0000", # marginl, marginr, marginv
dialog["effect"],
dialog["text"]))
return "\n".join(dialog_fmt)
def _dialog(self):
dialog_fmt = []
for dialog in helpers.progressbar(self._assdict["dialog"],
prefix='Writing'):
if dialog["comment"]:
key = "Comment"
else:
key = "Dialogue"
if dialog["text"]:
dialog_fmt.append(
"{:s}: {:d},{:s},{:s},{:s},{:s},{:s},{:s},{:s},{:s},"
"{:s}".format(
key,
dialog["layer"],
dialog["start"],
dialog["end"],
dialog["style"]["name"],
dialog["actor"],
"0000",
"0000",
"0000", # marginl, marginr, marginv
dialog["effect"],
dialog["text"]))
return "\n".join(dialog_fmt)
def read(self, filename=None):
"""Read an parse the ass script
Parameters:
:param filename: filename of the script to read
"""
temp_dialog = []
style = {}
# Audio & Video Data
video, video_zoom, video_position, video_aspect_ratio = (
None, None, None, None)
playresx, playresy = None, None
audio = None
# Metadta
title, original_script, translation, timing = None, None, None, None
assfile = self._open(filename)
SKIP_CHARS = COMMENT_CHARS[:]
SKIP_CHARS.append("[")
for line in helpers.progressbar(
assfile.readlines(), prefix='Reading', sufix='Lines'):
line = line.strip()
if line.startswith(tuple(SKIP_CHARS)) or not line:
continue
else:
key, value = line.split(":", 1)
key = key.lower().replace(" ", "_")
value = value.strip()
# Dialogue, Comment, Style
if key == "dialogue" or key == "comment":
value = value.split(",", 9)
text = value[9].strip()
if not text: # Skip on Empty dialog
continue
dialog_item = {
"layer": int(value[0]),
"start": value[1],
"end": value[2],
"style": value[3],
"actor": value[4],
# marginl, marginr, marginl = "0000"
"effect": value[8],
"text": text,
"comment": key == "comment", # commented (True/False)
}
temp_dialog.append(dialog_item)
elif key == "style":
value = value.split(",", 22)
style_item = {
"name": value[0],
"font": {
"name": value[1],
"size": int(value[2])},
"color": {
"primary": value[3],
"secondary": value[4],
"bord": value[5],
"shadow": value[6]},
"bold": bool(int(value[7])),
"italic": bool(int(value[8])),
# Underline, StrikeOut = False (-1)
"scale": [float(value[11]), float(value[12])],
"spacing": int(value[13]),
# Angle = 0
# BorderStyle = 1 (Border + Shadow)
"bord": float(value[16]),
"shadow": float(value[17]),
"alignment": int(value[18]),
"margin": {
"l": int(value[19]),
"r": int(value[20]),
"v": int(value[21])},
# Encoding = 0,
}
style[value[0]] = style_item
elif key == "playresx":
playresx = int(value)
elif key == "playresy":
playresy = int(value)
elif key == "video_file":
video = value
elif key == "audio_uri" or key == "audio_file":
audio = value
elif key == "video_zoom_percent":
video_zoom = float(value)
elif key == "video_zoom":
video_zoom = float(value.replace(r"%", "")) / 100
elif (key == "video_aspect_ratio" or key == "video_ar_value" or
key == "aegisub_video_aspect_ratio"):
video_aspect_ratio = value.replace("c", "")
elif key == "title":
title = value
elif key == "original_script":
original_script = value
elif key == "translation":
translation = value
elif key == "timing":
timing = value
elif key == "video_position":
video_position = int(value)
else:
continue
assfile.close()
# Add style to dialogs
# FIX: Need for default style
dialog = []
if temp_dialog:
for d in temp_dialog:
d["style"] = style[d["style"]]
dialog.append(d)
else:
dialog = None
try:
num, den = video_aspect_ratio.split(':')
video_aspect_ratio = float(num / den)
except ValueError:
video_aspect_ratio = float(video_aspect_ratio)
# Default values: None
ass = {
"dialog": dialog,
"style": style,
"resolution": [playresx, playresy],
"video": {
"path": video,
"zoom": video_zoom,
"position": video_position,
"ar": video_aspect_ratio,
},
"audio": audio,
"metadata": {
"filename": filename,
"title": title,
"original_script": original_script,
"translation": translation,
"timing": timing,
}
}
return ass
def read(self, filename=None):
"""Read an parse the ass script
Parameters:
:param filename: filename of the script to read
"""
temp_dialog = []
style = {}
# Audio & Video Data
video, video_zoom, video_position, video_aspect_ratio = (
None, None, None, None)
playresx, playresy = None, None
audio = None
# Metadta
title, original_script, translation, timing = None, None, None, None
assfile = self._open(filename)
SKIP_CHARS = COMMENT_CHARS[:]
SKIP_CHARS.append("[")
for line in helpers.progressbar(
assfile.readlines(), prefix='Reading', sufix='Lines'):
line = line.strip()
if line.startswith(tuple(SKIP_CHARS)) or not line:
continue
else:
key, value = line.split(":", 1)
key = key.lower().replace("+", "p").replace(" ", "_")
value = value.strip()
# Dialogue, Comment, Style
if key == "dialogue" or key == "comment":
value = value.split(",", 9)
text = value[9]
if not text: # Skip on Empty dialog
continue
dialog_item = {
"layer": int(value[0]),
"start": value[1],
"end": value[2],
"style": value[3],
# marginl, marginr, marginl = "0000"
"actor": value[4],
"effect": value[8],
"text": text,
"comment": key == "comment", # commented (True/False)
}
temp_dialog.append(dialog_item)
elif key == "style":
value = value.split(",", 22)
style_item = {
"name": value[0],
"font": {
"name": value[1],
"size": int(value[2])},
"color": {
"primary": value[3],
"secondary": value[4],
"bord": value[5],
"shadow": value[6]},
"bold": bool(int(value[7])),
"italic": bool(int(value[8])),
# Underline, StrikeOut = False (-1)
"scale": [float(value[11]), float(value[12])],
"spacing": int(value[13]),
# Angle = 0
# BorderStyle = 1 (Border + Shadow)
"bord": float(value[16]),
"shadow": float(value[17]),
"alignment": int(value[18]),
"margin": {
"l": int(value[19]),
"r": int(value[20]),
"v": int(value[21])},
# Encoding = 0,
}
style[value[0]] = style_item
elif key == "playresx":
playresx = int(value)
elif key == "playresy":
playresy = int(value)
elif key == "video_file":
video = value
elif key == "audio_uri" or key == "audio_file":
audio = value
elif key == "video_zoom" or key == "video_zoom_percent":
video_zoom = value
elif (key == "video_aspect_ratio" or key == "video_ar_value" or
key == "aegisub_video_aspect_ratio"):
video_aspect_ratio = value.replace("c", "")
elif key == "title":
title = value
elif key == "original_script":
original_script = value
elif key == "translation":
translation = value
elif key == "timing":
timing = value
elif key == "video_position":
video_position = int(value)
else:
continue
assfile.close()
# Add style to dialogs
dialog = []
if temp_dialog:
for d in temp_dialog:
d["style"] = style[d["style"]]
dialog.append(d)
else:
dialog = None
try:
video_zoom = VIDEO_ZOOM_INV[VIDEO_ZOOM[video_zoom]]
except KeyError:
video_zoom = float(video_zoom)
try:
num, den = video_aspect_ratio.split(':')
video_aspect_ratio = float(num / den)
except ValueError:
video_aspect_ratio = float(video_aspect_ratio)
# Default values: None
ass = {
"dialog": dialog,
"style": style,
"resolution": [playresx, playresy],
"video": {
"path": video,
"zoom": video_zoom,
"position": video_position,
"ar": video_aspect_ratio,
},
"audio": audio,
"metadata": {
"filename": filename,
"title": title,
"original_script": original_script,
"translation": translation,
"timing": timing,
}
}
return ass
def dialogs(self):
D = partial(Dialog, resolution=self.resolution)
dialogs = list(map(D, self._script_data["dialog"]))
if self.progressbar:
return helpers.progressbar(dialogs)
return dialogs
