tail效果

.env.example

@@ -7,5 +7,7 @@ TEMP_DIR=tmp/
 ENCODER_ARGS="-c:v h264_qsv -global_quality 28 -look_ahead 1"
 # NVENC
 #ENCODER_ARGS="-c:v h264_nvenc -cq:v 24 -preset:v p7 -tune:v hq -profile:v high"
+# HEVC
+#VIDEO_ARGS="-profile:v main
 UPLOAD_METHOD="rclone"
 RCLONE_REPLACE_MAP="https://oss.zhentuai.com|alioss://frametour-assets,https://frametour-assets.oss-cn-shanghai.aliyuncs.com|alioss://frametour-assets"

biz/ffmpeg.py

@@ -27,10 +27,6 @@ def parse_ffmpeg_task(task_info, template_info):
         
         # 统计only_if占位符的使用次数
         only_if_usage_count = {}
-        for part in template_info.get("video_parts", []):
-            only_if = part.get('only_if', '')
-            if only_if:
-                only_if_usage_count[only_if] = only_if_usage_count.get(only_if, 0) + 1
         with tracer.start_as_current_span("parse_ffmpeg_task.download_all") as sub_span:
             with ThreadPoolExecutor(max_workers=8) as executor:
                 param_list: list[dict]
@@ -48,7 +44,8 @@ def parse_ffmpeg_task(task_info, template_info):
                 continue
             only_if = part.get('only_if', '')
             if only_if:
-                required_count = only_if_usage_count.get(only_if, 1)
+                only_if_usage_count[only_if] = only_if_usage_count.get(only_if, 0) + 1
+                required_count = only_if_usage_count.get(only_if)
                 if not check_placeholder_exist_with_count(only_if, task_params_orig, required_count):
                     logger.info("because only_if exist, placeholder: %s insufficient (need %d), skip part: %s", only_if, required_count, part)
                     continue

constant/__init__.py

@@ -1,8 +1,9 @@
 SUPPORT_FEATURE = (
     'simple_render_algo',
     'gpu_accelerate',
-    'gpu_accelerate',
+    'hevc_encode',
     'rapid_download',
     'rclone_upload',
+    'custom_re_encode',
 )
-SOFTWARE_VERSION = '0.0.2'
+SOFTWARE_VERSION = '0.0.4'

entity/ffmpeg.py

@@ -6,11 +6,22 @@ from typing import Any
 
 DEFAULT_ARGS = ("-shortest",)
 ENCODER_ARGS = ("-c:v", "h264", ) if not os.getenv("ENCODER_ARGS", False) else os.getenv("ENCODER_ARGS", "").split(" ")
-VIDEO_ARGS = ("-profile:v", "high", "-level:v", "4", )
+VIDEO_ARGS = ("-profile:v", "high", "-level:v", "4", ) if not os.getenv("VIDEO_ARGS", False) else os.getenv("VIDEO_ARGS", "").split(" ")
 AUDIO_ARGS = ("-c:a", "aac", "-b:a", "128k", "-ar", "48000", "-ac", "2", )
 MUTE_AUDIO_INPUT = ("-f", "lavfi", "-i", "anullsrc=cl=stereo:r=48000", )
 
 
+def get_mp4toannexb_filter():
+    """
+    Determine which mp4toannexb filter to use based on ENCODER_ARGS.
+    Returns 'hevc_mp4toannexb' if ENCODER_ARGS contains 'hevc', otherwise 'h264_mp4toannexb'.
+    """
+    encoder_args_str = os.getenv("ENCODER_ARGS", "").lower()
+    if "hevc" in encoder_args_str:
+        return "hevc_mp4toannexb"
+    return "h264_mp4toannexb"
+
+
 class FfmpegTask(object):
 
     effects: list[str]
@@ -164,7 +175,7 @@ class FfmpegTask(object):
             output_args = [*VIDEO_ARGS, *AUDIO_ARGS, *ENCODER_ARGS, *DEFAULT_ARGS]
             if self.annexb:
                 output_args.append("-bsf:v")
-                output_args.append("h264_mp4toannexb")
+                output_args.append(get_mp4toannexb_filter())
                 output_args.append("-reset_timestamps")
                 output_args.append("1")
             video_output_str = "[0:v]"
@@ -238,9 +249,10 @@ class FfmpegTask(object):
                     filter_args.append(f"{_start_out_str}select=lt(n\\,{int(start * self.frame_rate)}){_start_out_str}")
                     filter_args.append(f"{_end_out_str}select=gt(n\\,{int(start * self.frame_rate)}){_end_out_str}")
                     filter_args.append(f"{_mid_out_str}select=eq(n\\,{int(start * self.frame_rate)}){_mid_out_str}")
-                    filter_args.append(f"{_mid_out_str}tpad=start_mode=clone:start_duration={duration:.4f}{_mid_out_str}")
+                    filter_args.append(
+                        f"{_mid_out_str}tpad=start_mode=clone:start_duration={duration:.4f}{_mid_out_str}")
                     if rotate_deg != 0:
-                        filter_args.append(f"{_mid_out_str}rotate=PI*{rotate_deg}/360{_mid_out_str}")
+                        filter_args.append(f"{_mid_out_str}rotate=PI*{rotate_deg}/180{_mid_out_str}")
                     # filter_args.append(f"{video_output_str}trim=start=0:end={start+duration},tpad=stop_mode=clone:stop_duration={duration},setpts=PTS-STARTPTS{_start_out_str}")
                     # filter_args.append(f"tpad=start_mode=clone:start_duration={duration},setpts=PTS-STARTPTS{_start_out_str}")
                     # filter_args.append(f"{_end_out_str}trim=start={start}{_end_out_str}")
@@ -258,7 +270,57 @@ class FfmpegTask(object):
                         filter_args.append(f"{video_output_str}setpts={param}*PTS[v_eff{effect_index}]")
                         video_output_str = f"[v_eff{effect_index}]"
                 elif effect.startswith("zoom:"):
-                    ...
+                    param = effect.split(":", 2)[1]
+                    if param == '':
+                        continue
+                    _split = param.split(",")
+                    if len(_split) < 3:
+                        continue
+                    try:
+                        start_time = float(_split[0])
+                        zoom_factor = float(_split[1])
+                        duration = float(_split[2])
+                        if start_time < 0:
+                            start_time = 0
+                        if duration < 0:
+                            duration = 0
+                        if zoom_factor <= 0:
+                            zoom_factor = 1
+                    except (ValueError, IndexError):
+                        start_time = 0
+                        duration = 0
+                        zoom_factor = 1
+                    if zoom_factor == 1:
+                        continue
+                    effect_index += 1
+                    if duration == 0:
+                        filter_args.append(f"{video_output_str}trim=start={start_time},scale=iw*{zoom_factor}:ih*{zoom_factor},crop=iw/{zoom_factor}:ih/{zoom_factor}:(iw-iw/{zoom_factor})/2:(ih-ih/{zoom_factor})/2,setpts=PTS-STARTPTS[v_eff{effect_index}]")
+                    else:
+                        zoom_expr = f"if(between(t,{start_time},{start_time + duration}),{zoom_factor},1)"
+                        filter_args.append(f"{video_output_str}scale=iw*({zoom_expr}):ih*({zoom_expr}),crop=iw:ih:(iw-iw)/2:(ih-ih)/2[v_eff{effect_index}]")
+                    video_output_str = f"[v_eff{effect_index}]"
+                elif effect.startswith("skip:"):
+                    param = effect.split(":", 2)[1]
+                    if param == '':
+                        param = "0"
+                    skip_seconds = float(param)
+                    if skip_seconds > 0:
+                        effect_index += 1
+                        filter_args.append(f"{video_output_str}trim=start={skip_seconds},setpts=PTS-STARTPTS[v_eff{effect_index}]")
+                        video_output_str = f"[v_eff{effect_index}]"
+                elif effect.startswith("tail:"):
+                    param = effect.split(":", 2)[1]
+                    if param == '':
+                        param = "0"
+                    tail_seconds = float(param)
+                    if tail_seconds > 0:
+                        effect_index += 1
+                        # 首先获取视频总时长，然后计算开始时间
+                        # 使用reverse+trim+reverse的方法来精确获取最后N秒
+                        filter_args.append(f"{video_output_str}reverse[v_rev{effect_index}]")
+                        filter_args.append(f"[v_rev{effect_index}]trim=duration={tail_seconds},setpts=PTS-STARTPTS[v_trim{effect_index}]")
+                        filter_args.append(f"[v_trim{effect_index}]reverse[v_eff{effect_index}]")
+                        video_output_str = f"[v_eff{effect_index}]"
                 ...
             if self.resolution:
                 filter_args.append(f"{video_output_str}scale={self.resolution.replace('x', ':')}[v]")
@@ -282,6 +344,8 @@ class FfmpegTask(object):
             output_args.append(video_output_str)
             output_args.append("-r")
             output_args.append(f"{self.frame_rate}")
+            output_args.append("-fps_mode")
+            output_args.append("cfr")
             if self.mute:
                 input_index = input_args.count("-i")
                 input_args += MUTE_AUDIO_INPUT
@@ -359,7 +423,7 @@ class FfmpegTask(object):
             output_args += AUDIO_ARGS
             if self.annexb:
                 output_args.append("-bsf:v")
-                output_args.append("h264_mp4toannexb")
+                output_args.append(get_mp4toannexb_filter())
                 output_args.append("-bsf:a")
                 output_args.append("setts=pts=DTS")
             output_args.append("-f")

index.py

@@ -1,15 +1,25 @@
 from time import sleep
+import sys
 
 import config
 import biz.task
 from telemetry import init_opentelemetry
-from template import load_local_template
+from template import load_local_template, download_template, TEMPLATES
 from util import api
 
 import os
 import glob
 
 load_local_template()
+
+# Check for redownload parameter
+if 'redownload' in sys.argv:
+    print("Redownloading all templates...")
+    for template_name in TEMPLATES.keys():
+        print(f"Redownloading template: {template_name}")
+        download_template(template_name)
+    print("All templates redownloaded successfully!")
+    sys.exit(0)
 import logging
 
 LOGGER = logging.getLogger(__name__)

template/__init__.py

@@ -79,6 +79,8 @@ def download_template(template_id):
     tracer = get_tracer(__name__)
     with tracer.start_as_current_span("download_template"):
         template_info = api.get_template_info(template_id)
+        if template_info is None:
+            return
         if not os.path.isdir(template_info['local_path']):
             os.makedirs(template_info['local_path'])
         # download template assets

util/api.py

@@ -87,6 +87,9 @@ def get_template_info(template_id):
         data = response.json()
         logger.debug("获取模板信息结果：【%s】", data)
         remote_template_info = data.get('data', {})
+        if not remote_template_info:
+            logger.warning("获取模板信息结果为空", data)
+            return None
         template = {
             'id': template_id,
             'updateTime': remote_template_info.get('updateTime', template_id),

util/ffmpeg.py

@@ -7,7 +7,7 @@ from typing import Optional, IO
 
 from opentelemetry.trace import Status, StatusCode
 
-from entity.ffmpeg import FfmpegTask, ENCODER_ARGS, VIDEO_ARGS, AUDIO_ARGS, MUTE_AUDIO_INPUT
+from entity.ffmpeg import FfmpegTask, ENCODER_ARGS, VIDEO_ARGS, AUDIO_ARGS, MUTE_AUDIO_INPUT, get_mp4toannexb_filter
 from telemetry import get_tracer
 
 logger = logging.getLogger(__name__)
@@ -20,12 +20,23 @@ def re_encode_and_annexb(file):
             return file
         logger.info("ReEncodeAndAnnexb: %s", file)
         has_audio = not not probe_video_audio(file)
-        ffmpeg_process = subprocess.run(["ffmpeg", "-y", "-hide_banner", "-vsync", "cfr", "-i", file,
+        # 优先使用RE_ENCODE_VIDEO_ARGS环境变量，其次使用默认的VIDEO_ARGS
+        if os.getenv("RE_ENCODE_VIDEO_ARGS", False):
+            _video_args = tuple(os.getenv("RE_ENCODE_VIDEO_ARGS", "").split(" "))
+        else:
+            _video_args = VIDEO_ARGS
+        # 优先使用RE_ENCODE_ENCODER_ARGS环境变量，其次使用默认的ENCODER_ARGS
+        if os.getenv("RE_ENCODE_ENCODER_ARGS", False):
+            _encoder_args = tuple(os.getenv("RE_ENCODE_ENCODER_ARGS", "").split(" "))
+        else:
+            _encoder_args = ENCODER_ARGS
+        ffmpeg_process = subprocess.run(["ffmpeg", "-y", "-hide_banner", "-i", file,
                                          *(set() if has_audio else MUTE_AUDIO_INPUT),
+                                         "-fps_mode", "cfr",
                                          "-map", "0:v", "-map", "0:a" if has_audio else "1:a",
-                                         *VIDEO_ARGS, "-bsf:v", "h264_mp4toannexb",
+                                         *_video_args, "-bsf:v", get_mp4toannexb_filter(),
                                          *AUDIO_ARGS, "-bsf:a", "setts=pts=DTS",
-                                         *ENCODER_ARGS, "-shortest", "-fflags", "+genpts",
+                                         *_encoder_args, "-shortest", "-fflags", "+genpts",
                                          "-f", "mpegts", file + ".ts"])
         logger.info(" ".join(ffmpeg_process.args))
         span.set_attribute("ffmpeg.args", json.dumps(ffmpeg_process.args))

util/oss.py

@@ -1,5 +1,6 @@
 import logging
 import os
+import sys
 
 import requests
 from opentelemetry.trace import Status, StatusCode
@@ -70,7 +71,7 @@ def upload_to_oss(url, file_path):
     return False
 
 
-def download_from_oss(url, file_path, skip_if_exist=False):
+def download_from_oss(url, file_path, skip_if_exist=None):
     """
     使用签名URL下载文件到OSS
     :param skip_if_exist: 如果存在就不下载了
@@ -82,6 +83,11 @@ def download_from_oss(url, file_path, skip_if_exist=False):
     with tracer.start_as_current_span("download_from_oss") as span:
         span.set_attribute("file.url", url)
         span.set_attribute("file.path", file_path)
+        
+        # 如果skip_if_exist为None，则从启动参数中读取
+        if skip_if_exist is None:
+            skip_if_exist = 'skip_if_exist' in sys.argv
+        
         if skip_if_exist and os.path.exists(file_path):
             span.set_attribute("file.exist", True)
             span.set_attribute("file.size", os.path.getsize(file_path))