From be29c99d5a9b70ed691a7bf6185a75a540687b7c Mon Sep 17 00:00:00 2001 From: Vadim Levin Date: Fri, 21 Jul 2023 17:50:48 +0300 Subject: [PATCH] feat: add cuda_GpuMat to big types This patch enables passing GpuMat as an in/out argument in several functions. --- modules/python/src2/gen2.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/modules/python/src2/gen2.py b/modules/python/src2/gen2.py index fdcae20187..95bd0495cf 100755 --- a/modules/python/src2/gen2.py +++ b/modules/python/src2/gen2.py @@ -511,14 +511,17 @@ class ArgInfo(object): return self.enclosing_arg.name + '.' + self.name def isbig(self): - return self.tp in ["Mat", "vector_Mat", "cuda::GpuMat", "GpuMat", "vector_GpuMat", "UMat", "vector_UMat"] # or self.tp.startswith("vector") + return self.tp in ["Mat", "vector_Mat", + "cuda::GpuMat", "cuda_GpuMat", "GpuMat", + "vector_GpuMat", "vector_cuda_GpuMat", + "UMat", "vector_UMat"] # or self.tp.startswith("vector") def crepr(self): return "ArgInfo(\"%s\", %d)" % (self.name, self.outputarg) def find_argument_class_info(argument_type, function_namespace, - function_class_name, known_classes): + function_class_name, known_classes): # type: (str, str, str, dict[str, ClassInfo]) -> ClassInfo | None """Tries to find corresponding class info for the provided argument type