From 5f0be4b7a558faeed6b260fb73224ae59f747dd5 Mon Sep 17 00:00:00 2001
From: gcw_4z3jiPvB <huweinan1@h-partners.com>
Date: Thu, 13 Aug 2026 17:06:09 +0800
Subject: [PATCH] =?UTF-8?q?update:=20=E6=9B=B4=E6=96=B0=E6=96=87=E4=BB=B6?=
 =?UTF-8?q?=20extension=5Fmanager.h?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: gcw_4z3jiPvB <huweinan1@h-partners.com>
---
 .../algorithm/extension_manager/include/extension_manager.h | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/framework/algorithm/extension_manager/include/extension_manager.h b/framework/algorithm/extension_manager/include/extension_manager.h
index dee51c3..435c781 100644
--- a/framework/algorithm/extension_manager/include/extension_manager.h
+++ b/framework/algorithm/extension_manager/include/extension_manager.h
@@ -43,6 +43,7 @@
 #include "colorspace_converter_display_extension.h"
 #include "metadata_generator_extension.h"
 #include "detail_enhancer_extension.h"
+#include "auto_effect_aisr_extension.h"
 #include "contrast_enhancer_extension.h"
 #include "video_refreshrate_prediction_extension.h"
 #include "utils.h"
@@ -84,6 +85,7 @@ using AihdrEnhancerCapabilityMap =
         std::tuple<ColorSpaceDescription, GraphicPixelFormat>,
         std::vector<std::tuple<Rank, int32_t, size_t>>>;
 using ContrastEnhancerCapabilityMap = std::map<ContrastEnhancerType, size_t>;
+using AutoEffectAisrCapabilityMap = std::map<uint32_t, size_t>;
 
 using ColorSpaceConverterDisplaySet = std::set<std::shared_ptr<ColorSpaceConverterDisplayBase>>;
 using ColorSpaceConverterDisplayExtensionSet = std::set<std::shared_ptr<ColorSpaceConverterDisplayExtension>>;
@@ -104,6 +106,7 @@ public:
     std::shared_ptr<DetailEnhancerBase> CreateDetailEnhancer(uint32_t level) const;
     std::shared_ptr<VideoRefreshRatePredictionBase> CreateVideoRefreshRatePredictor() const;
     std::shared_ptr<ContrastEnhancerBase> CreateContrastEnhancer(ContrastEnhancerType type) const;
+    std::shared_ptr<DetailEnhancerBase> CreateAutoEffectAisr() const;
 
     using InstanceVariableType = std::variant<std::shared_ptr<ColorSpaceConverter>,
                                                 std::shared_ptr<MetadataGenerator>,
@@ -133,6 +136,7 @@ private:
     std::shared_ptr<AihdrEnhancerExtension> FindAihdrEnhancerExtension(const FrameInfo &inputInfo) const;
     std::shared_ptr<DetailEnhancerExtension> FindDetailEnhancerExtension(uint32_t level) const;
     std::shared_ptr<ContrastEnhancerExtension> FindContrastEnhancerExtension(ContrastEnhancerType type) const;
+    std::shared_ptr<ExtensionBase> FindAutoEffectAisrExtension() const;
     ExtensionList LoadExtensions() const;
     VPEAlgoErrCode LoadStaticExtensions(ExtensionList& extensionList) const;
     ExtensionList LoadStaticImageExtensions(
@@ -152,6 +156,8 @@ private:
         AihdrEnhancerCapabilityMap& aihdrEnhancerCapabilityMap) const;
     VPEAlgoErrCode BuildDetailEnhancerCaps(const std::shared_ptr<ExtensionBase>& ext, size_t idx,
         DetailEnhancerCapabilityMap& detailEnhancerCapabilityMap) const;
+    VPEAlgoErrCode BuildAutoEffectAisrCaps(const std::shared_ptr<ExtensionBase>& ext, size_t idx,
+        AutoEffectAisrCapabilityMap& autoEffectAisrCapabilityMap) const;
     VPEAlgoErrCode BuildContrastEnhancerCaps(const std::shared_ptr<ExtensionBase>& ext, size_t idx,
     ContrastEnhancerCapabilityMap& contrastEnhancerCapabilityMap) const;
     VPEAlgoErrCode ExtractColorSpaceConverterCap(const ColorSpaceConverterCapability& cap, size_t idx,