gitadmin
/
tuoheng_alg_move_20240715
Obserwuj 1
Polub 0
Forkuj 0
Kod Zgłoszenia 0 Oczekujące zmiany 0 Wydania 11 Wiki Aktywność
Nie możesz wybrać więcej, niż 25 tematów Tematy muszą się zaczynać od litery lub cyfry, mogą zawierać myślniki ('-') i mogą mieć do 35 znaków.
48 Commity
8 Gałęzie
Drzewo: a9ac6b2953
Gałęzie Tagi
${ item.name }
Utwórz gałąź ${ searchTerm }
z 'a9ac6b2953'
${ noResults }
tuoheng_alg_move_20240715/util/Cv2Utils.py

495 lines
23KB
Czysty Wina Historia 

  1. # -*- coding: utf-8 -*-

  2. import time

  3. 

  4. import cv2

  5. import subprocess as sp

  6. 

  7. import ffmpeg

  8. import numpy as np

  9. from loguru import logger

  10. from exception.CustomerException import ServiceException

  11. from enums.ExceptionEnum import ExceptionType

  12. 

  13. 

  14. class Cv2Util():

  15. 

  16.     def __init__(self, pullUrl, pushUrl=None, orFilePath=None, aiFilePath=None, requestId=None):

  17.         self.pullUrl = pullUrl

  18.         self.pushUrl = pushUrl

  19.         self.orFilePath = orFilePath

  20.         self.aiFilePath = aiFilePath

  21.         self.cap = None

  22.         self.p = None

  23.         self.or_video_file = None

  24.         self.ai_video_file = None

  25.         self.fps = None

  26.         self.width = None

  27.         self.height = None

  28.         self.wah = None

  29.         self.wh = None

  30.         self.h = None

  31.         self.hn = None

  32.         self.w = None

  33.         self.all_frames = None

  34.         self.bit_rate = None

  35.         self.pull_p = None

  36.         self.requestId = requestId

  37.         self.p_push_retry_num = 0

  38.         self.resize_status = False

  39.         self.current_frame = 0

  40. 

  41.     def getFrameConfig(self, fps, width, height):

  42.         if self.fps is None:

  43.             self.fps = fps

  44.             self.width = width

  45.             self.height = height

  46.             self.wh = int(width * height * 3 // 8)

  47.             self.wah = '%sx%s' % (int(self.width / 2), int(self.height / 2))

  48.             self.h = int(self.height * 3 // 4)

  49.             self.w = int(self.width // 2)

  50.             self.hn = int(self.height // 2)

  51.             self.wn = int(self.width // 2)

  52.             w_f = self.wh != width * height * 3 / 8

  53.             h_f = self.h != self.height * 3 / 4

  54.             wd_f = self.w != self.width / 2

  55.             if w_f or h_f or wd_f:

  56.                 self.resize_status = True

  57.                 self.wh = int(width * height * 3 // 2)

  58.                 self.wah = '%sx%s' % (int(self.width), int(self.height))

  59.                 self.h = int(self.height * 3 // 2)

  60.                 self.w = int(self.width)

  61. 

  62.     '''

  63.         获取视频信息

  64.     '''

  65. 

  66.     def get_video_info(self):

  67.         try:

  68.             if self.pullUrl is None:

  69.                 logger.error("拉流地址不能为空, requestId:{}", self.requestId)

  70.                 raise ServiceException(ExceptionType.PULL_STREAM_URL_EXCEPTION.value[0],

  71.                                        ExceptionType.PULL_STREAM_URL_EXCEPTION.value[1])

  72.             probe = ffmpeg.probe(self.pullUrl)

  73.             if probe is None or probe.get("streams") is None:

  74.                 return

  75.             # 视频大小

  76.             # format = probe['format']

  77.             # size = int(format['size'])/1024/1024

  78.             video_stream = next((stream for stream in probe['streams'] if stream.get('codec_type') == 'video'), None)

  79.             if video_stream is None:

  80.                 logger.error("根据拉流地址未获取到视频流, requestId:{}", self.requestId)

  81.                 return

  82.             width = video_stream.get('width')

  83.             height = video_stream.get('height')

  84.             nb_frames = video_stream.get('nb_frames')

  85.             fps = video_stream.get('r_frame_rate')

  86.             # duration = video_stream.get('duration')

  87.             bit_rate = video_stream.get('bit_rate')

  88.             if width:

  89.                 self.width = int(width)

  90.             if height:

  91.                 self.height = int(height)

  92.             if width is not None and height is not None:

  93.                 self.wh = int(width * height * 3 // 8)

  94.                 self.wah = '%sx%s' % (int(self.width / 2), int(self.height / 2))

  95.                 self.h = int(self.height * 3 // 4)

  96.                 self.w = int(self.width / 2)

  97.                 self.hn = int(self.height / 2)

  98.                 self.wn = int(self.width // 2)

  99.                 w_f = self.wh != width * height * 3 / 8

  100.                 h_f = self.h != self.height * 3 / 4

  101.                 wd_f = self.w != self.width / 2

  102.                 if w_f or h_f or wd_f:

  103.                     self.resize_status = True

  104.                     self.wh = int(width * height * 3 // 2)

  105.                     self.wah = '%sx%s' % (int(self.width), int(self.height))

  106.                     self.h = int(self.height * 3 // 2)

  107.                     self.w = int(self.width)

  108.             if nb_frames:

  109.                 self.all_frames = int(nb_frames)

  110.             if fps:

  111.                 up, down = str(fps).split('/')

  112.                 self.fps = int(eval(up) / eval(down))

  113.             # if duration:

  114.             #    self.duration = float(video_stream['duration'])

  115.             if bit_rate:

  116.                 self.bit_rate = int(bit_rate) / 1000

  117.             logger.info("视频信息, width:{}|height:{}|fps:{}|all_frames:{}|bit_rate:{}, requestId:{}", self.width,

  118.                         self.height, self.fps, self.all_frames, self.bit_rate, self.requestId)

  119.         except ServiceException as s:

  120.             logger.error("获取视频信息异常: {}, requestId:{}", s.msg, self.requestId)

  121.             raise s

  122.         except ffmpeg._run.Error as er:

  123.             logger.error("获取视频信息异常: {}, requestId:{}", er.stderr.decode(encoding='utf-8'), self.requestId)

  124.         except Exception as e:

  125.             logger.exception("获取视频信息异常：{}, requestId:{}", e, self.requestId)

  126. 

  127.     '''

  128.         拉取视频

  129.     '''

  130. 

  131.     def build_pull_p(self):

  132.         try:

  133.             if self.wah is None:

  134.                 return

  135.             if self.pull_p:

  136.                 logger.info("重试, 关闭拉流管道, requestId:{}", self.requestId)

  137.                 self.pull_p.stdout.close()

  138.                 self.pull_p.terminate()

  139.                 self.pull_p.wait()

  140.             # command = ['ffmpeg',

  141.             #            # '-b:v', '3000k',

  142.             #            '-i', self.pullUrl,

  143.             #            '-f', 'rawvideo',

  144.             #            '-vcodec', 'rawvideo',

  145.             #            '-pix_fmt', 'bgr24',

  146.             #            # '-s', "{}x{}".format(int(width), int(height)),

  147.             #            '-an',

  148.             #            '-']

  149.             # input_config = {'c:v': 'h264_cuvid', 'resize': self.wah}

  150.             # process = (

  151.             #     ffmpeg

  152.             #         .input(self.pullUrl, **input_config)

  153.             #         .output('pipe:', format='rawvideo', r=str(self.fps)) # pix_fmt='bgr24'

  154.             #         .overwrite_output()

  155.             #         .global_args('-an')

  156.             #         .run_async(pipe_stdout=True)

  157.             # )

  158.             command = ['ffmpeg',

  159.                        '-re',

  160.                        '-y',

  161.                        '-c:v', 'h264_cuvid',

  162.                        '-resize', self.wah,

  163.                        '-i', self.pullUrl,

  164.                        '-f', 'rawvideo',

  165.                        '-an',

  166.                        '-']

  167.             self.pull_p = sp.Popen(command, stdout=sp.PIPE)

  168.             # self.pull_p = sp.Popen(command, stdout=sp.PIPE, stderr=sp.PIPE)

  169.             # self.pull_p = process

  170.         except ServiceException as s:

  171.             logger.exception("构建拉流管道异常: {}, requestId:{}", s, self.requestId)

  172.             raise s

  173.         except Exception as e:

  174.             logger.exception("构建拉流管道异常：{}, requestId:{}", e, self.requestId)

  175. 

  176.     def checkconfig(self):

  177.         if self.fps is None or self.width is None or self.height is None:

  178.             return True

  179.         return False

  180. 

  181.     def read(self):

  182.         result = None

  183.         try:

  184.             # if self.pull_p is None:

  185.             #     logger.error("拉流管道为空, requestId:{}", self.requestId)

  186.             #     raise ServiceException(ExceptionType.PULL_PIPELINE_INIT_EXCEPTION.value[0],

  187.             #                            ExceptionType.PULL_PIPELINE_INIT_EXCEPTION.value[1])

  188.             in_bytes = self.pull_p.stdout.read(self.wh)

  189.             self.current_frame += 1

  190.             if in_bytes is not None and len(in_bytes) > 0:

  191.                 # result = (np.frombuffer(in_bytes, np.uint8).reshape([int(self.height), int(self.width), 3]))

  192.                 try:

  193.                     img = (np.frombuffer(in_bytes, np.uint8)).reshape((self.h, self.w))

  194.                 except Exception as ei:

  195.                     logger.exception("视频格式异常：{}, requestId:{}", ei, self.requestId)

  196.                     raise ServiceException(ExceptionType.VIDEO_RESOLUTION_EXCEPTION.value[0],

  197.                                            ExceptionType.VIDEO_RESOLUTION_EXCEPTION.value[1])

  198.                 result = cv2.cvtColor(img, cv2.COLOR_YUV2BGR_NV12)

  199.                 if self.resize_status:

  200.                     result = cv2.resize(result, (int(self.width / 2), int(self.height / 2)),

  201.                                         interpolation=cv2.INTER_LINEAR)

  202.         except ServiceException as s:

  203.             raise s

  204.         except Exception as e:

  205.             logger.exception("读流异常：{}, requestId:{}", e, self.requestId)

  206.         return result

  207. 

  208.     def close(self):

  209.         if self.pull_p:

  210.             if self.pull_p.stdout:

  211.                 self.pull_p.stdout.close()

  212.             self.pull_p.terminate()

  213.             self.pull_p.wait()

  214.             logger.info("关闭拉流管道完成, requestId:{}", self.requestId)

  215.         if self.p:

  216.             if self.p.stdin:

  217.                 self.p.stdin.close()

  218.             self.p.terminate()

  219.             self.p.wait()

  220.             # self.p.communicate()

  221.             # self.p.kill()

  222.             logger.info("关闭管道完成, requestId:{}", self.requestId)

  223.         if self.or_video_file:

  224.             self.or_video_file.release()

  225.             logger.info("关闭原视频写入流完成, requestId:{}", self.requestId)

  226.         if self.ai_video_file:

  227.             self.ai_video_file.release()

  228.             logger.info("关闭AI视频写入流完成, requestId:{}", self.requestId)

  229. 

  230.     # 构建 cv2

  231.     # def build_cv2(self):

  232.     #     try:

  233.     #         if self.cap is not None:

  234.     #             logger.info("重试, 关闭cap, requestId:{}", self.requestId)

  235.     #             self.cap.release()

  236.     #         if self.p is not None:

  237.     #             logger.info("重试, 关闭管道, requestId:{}", self.requestId)

  238.     #             self.p.stdin.close()

  239.     #             self.p.terminate()

  240.     #             self.p.wait()

  241.     #         if self.pullUrl is None:

  242.     #             logger.error("拉流地址不能为空, requestId:{}", self.requestId)

  243.     #             raise ServiceException(ExceptionType.PULL_STREAM_URL_EXCEPTION.value[0],

  244.     #                                    ExceptionType.PULL_STREAM_URL_EXCEPTION.value[1])

  245.     #         if self.pushUrl is None:

  246.     #             logger.error("推流地址不能为空, requestId:{}", self.requestId)

  247.     #             raise ServiceException(ExceptionType.PUSH_STREAM_URL_EXCEPTION.value[0],

  248.     #                                    ExceptionType.PUSH_STREAM_URL_EXCEPTION.value[1])

  249.     #         self.cap = cv2.VideoCapture(self.pullUrl)

  250.     #         if self.fps is None or self.fps == 0:

  251.     #             self.fps = int(self.cap.get(cv2.CAP_PROP_FPS))

  252.     #         if self.width is None or self.width == 0:

  253.     #             self.width = int(self.cap.get(cv2.CAP_PROP_FRAME_WIDTH))

  254.     #         if self.height is None or self.height == 0:

  255.     #             self.height = int(self.cap.get(cv2.CAP_PROP_FRAME_HEIGHT))

  256.     #         command = ['/usr/bin/ffmpeg',

  257.     #                    # '-y',  # 不经过确认，输出时直接覆盖同名文件。

  258.     #                    '-f', 'rawvideo',

  259.     #                    '-vcodec', 'rawvideo',

  260.     #                    '-pix_fmt', 'bgr24',  # 显示可用的像素格式

  261.     #                    # '-s', "{}x{}".format(self.width * 2, self.height),

  262.     #                    '-s', "{}x{}".format(int(self.width), int(self.height/2)),

  263.     #                    # '-r', str(15),

  264.     #                    '-i', '-',  # 指定输入文件

  265.     #                    '-g', '25',

  266.     #                    '-b:v', '3000k',

  267.     #                    '-tune', 'zerolatency',  # 加速编码速度

  268.     #                    '-c:v', 'libx264',  # 指定视频编码器

  269.     #                    '-sc_threshold', '0',

  270.     #                    '-pix_fmt', 'yuv420p',

  271.     #                    '-an',

  272.     #                    '-preset', 'ultrafast',  # 指定输出的视频质量，会影响文件的生成速度，有以下几个可用的值 ultrafast,

  273.     #                    # superfast, veryfast, faster, fast, medium, slow, slower, veryslow。

  274.     #                    '-f', 'flv',

  275.     #                    self.pushUrl]

  276.     #         # 管道配置

  277.     #         logger.info("fps:{}|height:{}|width:{}|requestId:{}", self.fps, self.height, self.width, self.requestId)

  278.     #         self.p = sp.Popen(command, stdin=sp.PIPE)

  279.     #     except ServiceException as s:

  280.     #         logger.exception("构建cv2异常: {}, requestId:{}", s, self.requestId)

  281.     #         raise s

  282.     #     except Exception as e:

  283.     #         logger.exception("初始化cv2异常：{}, requestId:{}", e, self.requestId)

  284. 

  285.     # 构建 cv2

  286.     def build_p(self):

  287.         try:

  288.             if self.p:

  289.                 logger.info("重试, 关闭管道, requestId:{}", self.requestId)

  290.                 if self.p.stdin:

  291.                     self.p.stdin.close()

  292.                 self.p.terminate()

  293.                 self.p.wait()

  294.                 # self.p.communicate()

  295.                 # self.p.kill()

  296.             if self.pushUrl is None:

  297.                 logger.error("推流地址不能为空, requestId:{}", self.requestId)

  298.                 raise ServiceException(ExceptionType.PUSH_STREAM_URL_EXCEPTION.value[0],

  299.                                        ExceptionType.PUSH_STREAM_URL_EXCEPTION.value[1])

  300.             command = ['ffmpeg',

  301.                        # '-loglevel', 'debug',

  302.                        '-y',

  303.                        '-f', 'rawvideo',

  304.                        '-vcodec', 'rawvideo',

  305.                        '-pix_fmt', 'bgr24',

  306.                        '-thread_queue_size', '1024',

  307.                        # '-s', "{}x{}".format(self.width * 2, self.height),

  308.                        '-s', "{}x{}".format(int(self.width), int(self.hn)),

  309.                        '-r', str(self.fps),

  310.                        '-i', '-',  # 指定输入文件

  311.                        '-g', str(self.fps),

  312.                        # '-maxrate', '15000k',

  313.                        # '-profile:v', 'high',

  314.                        # '-b:v', '4000k',

  315.                        # '-crf', '18',

  316.                        '-rc:v', 'vbr',

  317.                        '-cq:v', '30',

  318.                        '-qmin', '30',

  319.                        '-qmax', '30',

  320.                        '-c:v', 'h264_nvenc',  #

  321.                        # '-bufsize', '4000k',

  322.                        # '-c:v', 'libx264',  # 指定视频编码器

  323.                        # '-tune', 'zerolatency',  # 加速编码速度

  324.                        # '-sc_threshold', '0',

  325.                        '-pix_fmt', 'yuv420p',

  326.                        "-an",

  327.                        # '-flvflags', 'no_duration_filesize',

  328.                        '-preset', 'fast',  # 指定输出的视频质量，会影响文件的生成速度，有以下几个可用的值 ultrafast,

  329.                        # superfast, veryfast, faster, fast, medium, slow, slower, veryslow。

  330.                        '-f', 'flv',

  331.                        self.pushUrl]

  332.             # command = 'ffmpeg -loglevel debug -y -f rawvideo -vcodec rawvideo -pix_fmt bgr24' +\

  333.             #            ' -s ' +  "{}x{}".format(int(self.width), int(self.height/2))\

  334.             #            + ' -i - ' + '-g ' + str(self.fps)+\

  335.             #            ' -b:v 6000k -tune zerolatency -c:v libx264 -pix_fmt yuv420p -preset ultrafast'+\

  336.             #            ' -f flv ' + self.pushUrl

  337. 

  338.             # kwargs = {'format': 'rawvideo',

  339.             #           # 'vcodec': 'rawvideo',

  340.             #           'pix_fmt': 'bgr24',

  341.             #           's': '{}x{}'.format(int(self.width), int(self.height/2))}

  342.             # out = {

  343.             #     'r': str(self.fps),

  344.             #     'g': str(self.fps),

  345.             #     'b:v': '5500k', # 恒定码率

  346.             #     # 'maxrate': '15000k',

  347.             #     # 'crf': '18',

  348.             #     'bufsize': '5500k',

  349.             #     'tune': 'zerolatency', # 加速编码速度

  350.             #     'c:v': 'libx264',  # 指定视频编码器

  351.             #     'sc_threshold': '0',

  352.             #     'pix_fmt': 'yuv420p',

  353.             #     # 'flvflags': 'no_duration_filesize',

  354.             #     'preset': 'medium',  # 指定输出的视频质量，会影响文件的生成速度，有以下几个可用的值 ultrafast,

  355.             #     # superfast, veryfast, faster, fast, medium, slow, slower, veryslow。

  356.             #     'format': 'flv'}

  357.             # 管道配置

  358.             # process2 = (

  359.             #     ffmpeg

  360.             #         .input('pipe:', **kwargs)

  361.             #         .output(self.pushUrl, **out)

  362.             #         .global_args('-y', '-an')

  363.             #         .overwrite_output()

  364.             #         .run_async(pipe_stdin=True)

  365.             # )

  366.             logger.info("fps:{}|height:{}|width:{}|requestId:{}", self.fps, self.height, self.width, self.requestId)

  367.             self.p = sp.Popen(command, stdin=sp.PIPE, shell=False)

  368.             # self.p = process2

  369.         except ServiceException as s:

  370.             logger.exception("构建p管道异常: {}, requestId:{}", s, self.requestId)

  371.             raise s

  372.         except Exception as e:

  373.             logger.exception("初始化p管道异常：{}, requestId:{}", e, self.requestId)

  374. 

  375.     async def push_stream_write(self, frame):

  376.         self.p.stdin.write(frame.tostring())

  377. 

  378.     async def push_stream(self, frame):

  379.         if self.p is None:

  380.             self.build_p()

  381.         try:

  382.             await self.push_stream_write(frame)

  383.             return True

  384.         except Exception as ex:

  385.             logger.exception("推流进管道异常：{}, requestId: {}", ex, self.requestId)

  386.             current_retry_num = 0

  387.             while True:

  388.                 try:

  389.                     time.sleep(1)

  390.                     self.p_push_retry_num += 1

  391.                     current_retry_num += 1

  392.                     if current_retry_num > 3 or self.p_push_retry_num > 600:

  393.                         return False

  394.                     self.build_p()

  395.                     await self.push_stream_write(frame)

  396.                     logger.info("构建p管道重试成功, 当前重试次数: {}, requestId: {}", current_retry_num,

  397.                                 self.requestId)

  398.                     return True

  399.                 except Exception as e:

  400.                     logger.exception("构建p管道异常：{}, 开始重试, 当前重试次数：{}, requestId: {}", e,

  401.                                      current_retry_num, self.requestId)

  402.         return False

  403. 

  404.     async def video_frame_write(self, or_frame, ai_frame):

  405.         if or_frame is not None:

  406.             self.or_video_file.write(or_frame)

  407.         if ai_frame is not None:

  408.             self.ai_video_file.write(ai_frame)

  409. 

  410.     async def video_write(self, or_frame, ai_frame):

  411.         try:

  412.             self.build_write()

  413.             if or_frame is not None and len(or_frame) > 0:

  414.                 await self.video_frame_write(or_frame, None)

  415.             if ai_frame is not None and len(ai_frame) > 0:

  416.                 await self.video_frame_write(None, ai_frame)

  417.             return True

  418.         except Exception as ex:

  419.             ai_retry_num = 0

  420.             while True:

  421.                 try:

  422.                     ai_retry_num += 1

  423.                     if ai_retry_num > 3:

  424.                         logger.exception("重新写入离线分析后视频到本地,重试失败：{}, requestId: {}", e, self.requestId)

  425.                         return False

  426.                     if or_frame is not None and len(or_frame) > 0:

  427.                         await self.or_video_file.write(or_frame)

  428.                     if ai_frame is not None and len(ai_frame) > 0:

  429.                         await self.ai_video_file.write(ai_frame)

  430.                     logger.info("重新写入离线分析后视频到本地, 当前重试次数: {}, requestId: {}", ai_retry_num,

  431.                                 self.requestId)

  432.                     return True

  433.                 except Exception as e:

  434.                     logger.exception("重新写入离线分析后视频到本地：{}, 开始重试, 当前重试次数：{}, requestId: {}", e,

  435.                                      ai_retry_num, self.requestId)

  436. 

  437.     def build_write(self):

  438.         try:

  439.             if self.fps is None or self.width is None or self.height is None:

  440.                 raise ServiceException(ExceptionType.VIDEO_CONFIG_EXCEPTION.value[0],

  441.                                        ExceptionType.VIDEO_CONFIG_EXCEPTION.value[1])

  442.             if self.orFilePath is not None and self.or_video_file is None:

  443.                 self.or_video_file = cv2.VideoWriter(self.orFilePath, cv2.VideoWriter_fourcc(*'mp4v'), self.fps,

  444.                                                      (int(self.wn), int(self.hn)))

  445.                 if self.or_video_file is None:

  446.                     raise ServiceException(ExceptionType.OR_WRITE_OBJECT_EXCEPTION.value[0],

  447.                                            ExceptionType.OR_WRITE_OBJECT_EXCEPTION.value[1])

  448.             if self.aiFilePath is not None and self.ai_video_file is None:

  449.                 self.ai_video_file = cv2.VideoWriter(self.aiFilePath, cv2.VideoWriter_fourcc(*'mp4v'), self.fps,

  450.                                                      (int(self.width), int(self.hn)))

  451.                 if self.ai_video_file is None:

  452.                     raise ServiceException(ExceptionType.AI_WRITE_OBJECT_EXCEPTION.value[0],

  453.                                            ExceptionType.AI_WRITE_OBJECT_EXCEPTION.value[1])

  454.         except ServiceException as s:

  455.             logger.exception("构建文件写对象异常: {}, requestId:{}", s, self.requestId)

  456.             raise s

  457.         except Exception as e:

  458.             logger.exception("构建文件写对象异常: {}, requestId:{}", e, self.requestId)

  459.             raise e

  460. 

  461.     def video_merge(self, frame1, frame2):

  462.         # frameLeft = cv2.resize(frame1, (int(self.width / 2), int(self.height / 2)), interpolation=cv2.INTER_LINEAR)

  463.         # frameRight = cv2.resize(frame2, (int(self.width / 2), int(self.height / 2)), interpolation=cv2.INTER_LINEAR)

  464.         # frame_merge = np.hstack((frameLeft, frameRight))

  465.         frame_merge = np.hstack((frame1, frame2))

  466.         return frame_merge

  467. 

  468.     def getP(self):

  469.         if self.p is None:

  470.             logger.error("获取管道为空, requestId:{}", self.requestId)

  471.             raise ServiceException(ExceptionType.PULL_PIPELINE_INIT_EXCEPTION.value[0],

  472.                                    ExceptionType.PULL_PIPELINE_INIT_EXCEPTION.value[1])

  473.         return self.p

  474. 

  475.     def getCap(self):

  476.         if self.cap is None:

  477.             logger.error("获取cv2为空, requestId:{}", self.requestId)

  478.             raise ServiceException(ExceptionType.CV2_IS_NULL_EXCEPTION.value[0],

  479.                                    ExceptionType.CV2_IS_NULL_EXCEPTION.value[1])

  480.         return self.cap

  481. 

  482.     def getOrVideoFile(self):

  483.         if self.or_video_file is None:

  484.             logger.error("获取原视频写入对象为空, requestId:{}", self.requestId)

  485.             raise ServiceException(ExceptionType.OR_WRITE_OBJECT_EXCEPTION.value[0],

  486.                                    ExceptionType.OR_WRITE_OBJECT_EXCEPTION.value[1])

  487.         return self.or_video_file

  488. 

  489.     def getAiVideoFile(self):

  490.         if self.ai_video_file is None:

  491.             logger.error("获取AI视频写入对象为空, requestId:{}", self.requestId)

  492.             raise ServiceException(ExceptionType.AI_WRITE_OBJECT_EXCEPTION.value[0],

  493.                                    ExceptionType.AI_WRITE_OBJECT_EXCEPTION.value[1])

  494.         return self.ai_video_file