arkanalyzer 1.0.85 → 1.0.87

package/lib/frontend/cppFrontend/common/IRInference.js

@@ -0,0 +1,1087 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IRInference = void 0;
+/*
+ * Copyright (c) 2024-2025 Huawei Device Co., Ltd.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importStar(require("path"));
+const const_1 = require("../ast/const");
+const ArkMethod_1 = require("../../../core/model/ArkMethod");
+const Type_1 = require("../../../core/base/Type");
+const Local_1 = require("../../../core/base/Local");
+const TypeInference_1 = require("./TypeInference");
+const Expr_1 = require("../../../core/base/Expr");
+const logger_1 = __importStar(require("../../../utils/logger"));
+const ArkClass_1 = require("../../../core/model/ArkClass");
+const ModelUtils_1 = require("../../../core/common/ModelUtils");
+const ArkField_1 = require("../../../core/model/ArkField");
+const EtsConst_1 = require("../../../core/common/EtsConst");
+const ArkSignature_1 = require("../../../core/model/ArkSignature");
+const TSConst_1 = require("../../../core/common/TSConst");
+const Builtin_1 = require("../../../core/common/Builtin");
+const Stmt_1 = require("../../../core/base/Stmt");
+const Ref_1 = require("../../../core/base/Ref");
+const Constant_1 = require("../../../core/base/Constant");
+const Const_1 = require("../../../core/common/Const");
+const ValueUtil_1 = require("../../../core/common/ValueUtil");
+const TypeExpr_1 = require("../../../core/base/TypeExpr");
+const ArkBaseModel_1 = require("../../../core/model/ArkBaseModel");
+const FileUtils_1 = require("../../../utils/FileUtils");
+const Type_2 = require("../base/Type");
+const SdkUtils_1 = require("../../../core/common/SdkUtils");
+const ModelUtils_2 = require("./ModelUtils");
+const Builtin_2 = require("./Builtin");
+const logger = logger_1.default.getLogger(logger_1.LOG_MODULE_TYPE.ARKANALYZER, 'IRInference');
+class IRInference {
+    static inferExportInfos(file) {
+        file.getExportInfos().forEach(exportInfo => {
+            if (exportInfo.getArkExport() === undefined) {
+                let arkExport = (0, ModelUtils_1.findArkExport)(exportInfo);
+                exportInfo.setArkExport(arkExport);
+                if (arkExport) {
+                    exportInfo.setExportClauseType(arkExport.getExportType());
+                }
+            }
+        });
+        file.getNamespaces().forEach(namespace => {
+            namespace.getExportInfos().forEach(exportInfo => {
+                if (exportInfo.getArkExport() === undefined) {
+                    let arkExport = (0, ModelUtils_1.findArkExport)(exportInfo);
+                    exportInfo.setArkExport(arkExport);
+                    arkExport !== null ? exportInfo.setExportClauseType(arkExport.getExportType()) : true;
+                }
+            });
+        });
+    }
+    static inferImportInfos(file) {
+        file.getImportInfos().forEach(importInfo => {
+            importInfo.getLazyExportInfo();
+        });
+    }
+    static inferFile(file) {
+        this.inferImportInfos(file);
+        ModelUtils_1.ModelUtils.getAllClassesInFile(file).forEach(arkClass => {
+            TypeInference_1.TypeInference.inferGenericType(arkClass.getGenericsTypes(), arkClass);
+            arkClass.getAllHeritageClasses();
+            arkClass.getFields().forEach(arkField => TypeInference_1.TypeInference.inferTypeInArkField(arkField));
+            const methods = arkClass.getMethods().sort((a, b) => {
+                const name = a.getName().split(Const_1.NAME_DELIMITER).reverse().join();
+                const anotherName = b.getName().split(Const_1.NAME_DELIMITER).reverse().join();
+                if (name.startsWith(anotherName)) {
+                    return 1;
+                }
+                else if (anotherName.startsWith(name)) {
+                    return -1;
+                }
+                return 0;
+            });
+            methods.forEach(arkMethod => TypeInference_1.TypeInference.inferTypeInMethod(arkMethod));
+        });
+        this.inferExportInfos(file);
+    }
+    static inferStaticInvokeExpr(expr, arkMethod) {
+        const fileSignature = expr.getMethodSignature().getDeclaringClassSignature().getDeclaringFileSignature();
+        if (fileSignature !== ArkSignature_1.FileSignature.DEFAULT && fileSignature !== Builtin_1.Builtin.BUILT_IN_CLASSES_FILE_SIGNATURE) {
+            return expr;
+        }
+        const arkClass = arkMethod.getDeclaringArkClass();
+        const methodName = expr.getMethodSignature().getMethodSubSignature().getMethodName();
+        expr.getArgs().forEach(arg => TypeInference_1.TypeInference.inferValueType(arg, arkMethod));
+        if (methodName === TSConst_1.IMPORT) {
+            const arg = expr.getArg(0);
+            let type;
+            if (arg instanceof Constant_1.Constant) {
+                type = TypeInference_1.TypeInference.inferDynamicImportType(arg.getValue(), arkClass);
+            }
+            if (type) {
+                expr.getMethodSignature().getMethodSubSignature().setReturnType(type);
+            }
+            return expr;
+        }
+        else if (methodName === TSConst_1.SUPER_NAME) {
+            const superClass = arkClass.getSuperClass();
+            if (superClass !== null) {
+                const newMethodSignature = new ArkSignature_1.MethodSignature(superClass.getSignature(), expr.getMethodSignature().getMethodSubSignature());
+                expr.setMethodSignature(newMethodSignature);
+            }
+            return expr;
+        }
+        const className = expr.getMethodSignature().getDeclaringClassSignature().getClassName();
+        if (className && className !== Const_1.UNKNOWN_CLASS_NAME) {
+            const baseType = TypeInference_1.TypeInference.inferBaseType(className, arkClass);
+            if (baseType) {
+                let result = this.inferInvokeExpr(expr, baseType, methodName, arkClass.getDeclaringArkFile().getScene());
+                if (result) {
+                    this.inferArgs(result, arkMethod);
+                    return result;
+                }
+            }
+            return expr;
+        }
+        return this.inferStaticInvokeExprByMethodName(methodName, arkMethod, expr);
+    }
+    static inferStaticInvokeExprByMethodName(methodName, arkMethod, expr) {
+        var _a, _b, _c, _d;
+        const arkClass = arkMethod.getDeclaringArkClass();
+        const arkExport = (_d = (_c = (_b = (_a = ModelUtils_2.CxxModelUtils.getStaticMethodWithName(methodName, arkClass)) !== null && _a !== void 0 ? _a : arkMethod.getFunctionLocal(methodName)) !== null && _b !== void 0 ? _b : ModelUtils_1.ModelUtils.findDeclaredLocal(new Local_1.Local(methodName), arkMethod)) !== null && _c !== void 0 ? _c : ModelUtils_2.CxxModelUtils.getArkExportInImportInfoWithName(methodName, arkClass.getDeclaringArkFile(), arkClass)) !== null && _d !== void 0 ? _d : arkClass.getDeclaringArkFile().getScene().getSdkGlobal(methodName);
+        let { mtd: method, sig: signature } = this.processArkExportForMethodAndSignature(arkExport, arkClass);
+        if (method) {
+            signature = method.matchMethodSignature(expr.getArgs());
+            // Handle Standard Operator Overloading
+            if ([Builtin_2.BuiltinCxx.OPERATOR_ISTREAM, Builtin_2.BuiltinCxx.OPERATOR_OSTREAM].includes(methodName) &&
+                !ModelUtils_2.CxxModelUtils.isIOStreamObjectMatched(signature.getMethodSubSignature().getParameters(), expr.getArgs(), method.getDeclaringArkFile().getScene())) {
+                signature = undefined;
+            }
+            if (signature) {
+                TypeInference_1.TypeInference.inferSignatureReturnType(signature, method);
+            }
+        }
+        if (signature) {
+            if (arkExport instanceof Local_1.Local) {
+                expr = new Expr_1.ArkPtrInvokeExpr(signature, arkExport, expr.getArgs(), expr.getRealGenericTypes());
+            }
+            else {
+                expr.setMethodSignature(signature);
+            }
+            this.inferArgs(expr, arkMethod);
+        }
+        return expr;
+    }
+    static processArkExportForMethodAndSignature(arkExport, arkClass) {
+        var _a;
+        let method;
+        let signature;
+        if (arkExport instanceof ArkMethod_1.ArkMethod) {
+            method = arkExport;
+        }
+        else if (arkExport instanceof ArkClass_1.ArkClass) {
+            method = arkExport.getMethodWithName(TSConst_1.CONSTRUCTOR_NAME);
+        }
+        else if (arkExport instanceof Local_1.Local) {
+            const type = TypeInference_1.TypeInference.replaceAliasType(arkExport.getType());
+            if (type instanceof Type_1.ClassType) {
+                const cls = arkClass.getDeclaringArkFile().getScene().getClass(type.getClassSignature());
+                method = (_a = cls === null || cls === void 0 ? void 0 : cls.getMethodWithName(TSConst_1.CONSTRUCTOR_NAME)) !== null && _a !== void 0 ? _a : cls === null || cls === void 0 ? void 0 : cls.getMethodWithName(Const_1.CALL_SIGNATURE_NAME);
+            }
+            else if (type instanceof Type_1.FunctionType) {
+                signature = type.getMethodSignature();
+            }
+            else if (type instanceof Type_2.PointerType && type.getBaseType() instanceof Type_1.FunctionType) {
+                // The type of a function pointer in CXX is 'PointerType(FunctionType, 1)'
+                signature = type.getBaseType().getMethodSignature();
+            }
+        }
+        else if (arkExport instanceof Type_1.AliasType && arkExport.getOriginalType() instanceof Type_1.FunctionType) {
+            signature = arkExport.getOriginalType().getMethodSignature();
+        }
+        return { mtd: method, sig: signature };
+    }
+    static inferInstanceInvokeExpr(expr, arkMethod) {
+        var _a, _b, _c;
+        const arkClass = arkMethod.getDeclaringArkClass();
+        TypeInference_1.TypeInference.inferRealGenericTypes(expr.getRealGenericTypes(), arkClass);
+        this.inferBase(expr, arkMethod);
+        const baseType = TypeInference_1.TypeInference.replaceAliasType(expr.getBase().getType());
+        let methodName = expr.getMethodSignature().getMethodSubSignature().getMethodName();
+        if (methodName === TSConst_1.CONSTRUCTOR_NAME &&
+            expr.getMethodSignature().getDeclaringClassSignature().getDeclaringFileSignature().getFileName() !== Const_1.UNKNOWN_FILE_NAME) {
+            return expr;
+        }
+        if (methodName.startsWith(Const_1.NAME_PREFIX)) {
+            const declaringStmt = (_b = (_a = arkMethod.getBody()) === null || _a === void 0 ? void 0 : _a.getLocals().get(methodName)) === null || _b === void 0 ? void 0 : _b.getDeclaringStmt();
+            if (declaringStmt instanceof Stmt_1.ArkAssignStmt && declaringStmt.getRightOp() instanceof Ref_1.ArkInstanceFieldRef) {
+                const rightOp = declaringStmt.getRightOp();
+                methodName = rightOp.getBase().getName() + '.' + rightOp.getFieldName();
+            }
+        }
+        const scene = arkClass.getDeclaringArkFile().getScene();
+        if (methodName === 'forEach' && baseType instanceof Type_1.ArrayType) {
+            this.processForEach(expr.getArg(0), baseType, scene);
+            return expr;
+        }
+        expr.getArgs().forEach(arg => TypeInference_1.TypeInference.inferValueType(arg, arkMethod));
+        let result = (_c = this.inferInvokeExpr(expr, baseType, methodName, scene)) !== null && _c !== void 0 ? _c : this.processExtendFunc(expr, arkMethod, methodName);
+        if (result) {
+            this.inferArgs(result, arkMethod);
+            return result;
+        }
+        logger.warn('invoke ArkInstanceInvokeExpr MethodSignature type fail: ', expr.toString());
+        return expr;
+    }
+    /**
+     * process arkUI function with Annotation @Extend @Styles @AnimatableExtend
+     * @param expr
+     * @param arkMethod
+     * @param methodName
+     */
+    static processExtendFunc(expr, arkMethod, methodName) {
+        var _a, _b;
+        const type = TypeInference_1.TypeInference.inferBaseType(methodName, arkMethod.getDeclaringArkClass());
+        if (type instanceof Type_1.FunctionType) {
+            const methodSignature = type.getMethodSignature();
+            // because of last stmt is ArkReturnVoidStmt, the ArkInvokeStmt at -2 before ArkReturnVoidStmt.
+            const stmts = (_b = (_a = arkMethod.getDeclaringArkFile().getScene().getMethod(methodSignature)) === null || _a === void 0 ? void 0 : _a.getCfg()) === null || _b === void 0 ? void 0 : _b.getStmts();
+            if (stmts) {
+                const endStmt = stmts[stmts.length - 2];
+                if (endStmt instanceof Stmt_1.ArkInvokeStmt) {
+                    methodSignature.getMethodSubSignature().setReturnType(endStmt.getInvokeExpr().getType());
+                }
+            }
+            expr.setMethodSignature(methodSignature);
+            return expr;
+        }
+        return null;
+    }
+    static inferFieldRef(ref, arkMethod) {
+        this.inferBase(ref, arkMethod);
+        const baseType = TypeInference_1.TypeInference.replaceAliasType(ref.getBase().getType());
+        if (baseType instanceof Type_1.ArrayType && ref.getFieldName() !== 'length') {
+            return new Ref_1.ArkArrayRef(ref.getBase(), ValueUtil_1.ValueUtil.createConst(ref.getFieldName()));
+        }
+        let newFieldSignature = this.generateNewFieldSignature(ref, arkMethod.getDeclaringArkClass(), baseType);
+        if (newFieldSignature) {
+            if (newFieldSignature.isStatic()) {
+                return new Ref_1.ArkStaticFieldRef(newFieldSignature);
+            }
+            ref.setFieldSignature(newFieldSignature);
+        }
+        return ref;
+    }
+    static inferBase(instance, arkMethod) {
+        const base = instance.getBase();
+        if (base.getName() === TSConst_1.THIS_NAME) {
+            const name = instance instanceof Ref_1.ArkInstanceFieldRef ? instance.getFieldName() : instance.getMethodSignature().getMethodSubSignature().getMethodName();
+            if (name.includes('.')) {
+                return;
+            }
+            const declaringArkClass = arkMethod.getDeclaringArkClass();
+            if (declaringArkClass.isAnonymousClass()) {
+                let newBase = this.inferThisLocal(arkMethod);
+                if (newBase) {
+                    instance.setBase(newBase);
+                }
+            }
+            else if (base.getType() instanceof Type_1.UnknownType) {
+                base.setType(new Type_1.ClassType(declaringArkClass.getSignature(), declaringArkClass.getRealTypes()));
+            }
+        }
+        else {
+            this.inferLocal(instance.getBase(), arkMethod);
+        }
+    }
+    static inferThisLocal(arkMethod) {
+        var _a, _b, _c, _d;
+        const arkClass = arkMethod.getDeclaringArkClass();
+        if (!arkClass.isAnonymousClass()) {
+            return null;
+        }
+        const value = (_b = (_a = arkMethod.getBody()) === null || _a === void 0 ? void 0 : _a.getUsedGlobals()) === null || _b === void 0 ? void 0 : _b.get(TSConst_1.THIS_NAME);
+        if (value instanceof Local_1.Local) {
+            return value;
+        }
+        else {
+            const thisType = TypeInference_1.TypeInference.inferBaseType(arkClass.getSignature().getDeclaringClassName(), arkClass);
+            if (thisType instanceof Type_1.ClassType) {
+                const newBase = new Local_1.Local(TSConst_1.THIS_NAME, thisType);
+                let usedGlobals = (_c = arkMethod.getBody()) === null || _c === void 0 ? void 0 : _c.getUsedGlobals();
+                if (!usedGlobals) {
+                    usedGlobals = new Map();
+                    (_d = arkMethod.getBody()) === null || _d === void 0 ? void 0 : _d.setUsedGlobals(usedGlobals);
+                }
+                usedGlobals.set(TSConst_1.THIS_NAME, newBase);
+                return newBase;
+            }
+        }
+        return null;
+    }
+    static inferArgs(expr, arkMethod) {
+        const scene = arkMethod.getDeclaringArkFile().getScene();
+        const parameters = expr.getMethodSignature().getMethodSubSignature().getParameters();
+        let realTypes = [];
+        const len = expr.getArgs().length;
+        for (let index = 0; index < len; index++) {
+            const arg = expr.getArg(index);
+            if (index >= parameters.length) {
+                break;
+            }
+            const argType = arg.getType();
+            const paramType = parameters[index].getType();
+            this.inferArg(expr, argType, paramType, scene, realTypes);
+        }
+        if (realTypes.length > 0 && !expr.getRealGenericTypes()) {
+            expr.setRealGenericTypes(realTypes);
+        }
+    }
+    static inferArg(expr, argType, paramType, scene, realTypes) {
+        var _a;
+        if (paramType instanceof Type_1.UnionType) {
+            paramType.getTypes().forEach(t => this.inferArg(expr, argType, t, scene, realTypes));
+        }
+        else if (paramType instanceof Type_1.AliasType) {
+            this.inferArg(expr, argType, paramType.getOriginalType(), scene, realTypes);
+        }
+        else if (paramType instanceof Type_1.ArrayType && argType instanceof Type_1.ArrayType) {
+            this.inferArg(expr, argType.getBaseType(), paramType.getBaseType(), scene, realTypes);
+        }
+        else if (expr instanceof Expr_1.ArkInstanceInvokeExpr && expr.getBase().getType() instanceof Type_1.ArrayType) {
+            if (paramType instanceof Type_1.ArrayType && paramType.getBaseType() instanceof Type_1.GenericType) {
+                this.inferArg(expr, argType, expr.getBase().getType().getBaseType(), scene, realTypes);
+            }
+        }
+        if (paramType instanceof Type_1.ClassType && scene.getProjectSdkMap().has(paramType.getClassSignature().getDeclaringFileSignature().getProjectName())) {
+            this.inferArgTypeWithSdk(paramType, scene, argType);
+        }
+        else if (paramType instanceof Type_1.GenericType || paramType instanceof Type_1.AnyType) {
+            realTypes.push(argType);
+        }
+        else if (paramType instanceof Type_1.FunctionType && argType instanceof Type_1.FunctionType) {
+            const returnType = paramType.getMethodSignature().getType();
+            if (paramType.getMethodSignature().getParamLength() > 0 && returnType instanceof Type_1.GenericType) {
+                const paramMethod = scene.getMethod(expr.getMethodSignature());
+                const argMethod = scene.getMethod(argType.getMethodSignature());
+                if (argMethod && ((_a = paramMethod === null || paramMethod === void 0 ? void 0 : paramMethod.getGenericTypes()) === null || _a === void 0 ? void 0 : _a.find(t => t === returnType))) {
+                    TypeInference_1.TypeInference.inferTypeInMethod(argMethod);
+                }
+            }
+            const realTypes = expr.getRealGenericTypes();
+            TypeInference_1.TypeInference.inferFunctionType(argType, paramType.getMethodSignature().getMethodSubSignature(), realTypes);
+        }
+    }
+    static inferRightWithSdkType(leftType, rightType, ackClass) {
+        if (leftType instanceof Type_1.AliasType) {
+            this.inferRightWithSdkType(TypeInference_1.TypeInference.replaceAliasType(leftType), rightType, ackClass);
+        }
+        else if (leftType instanceof Type_1.UnionType) {
+            leftType.getTypes().forEach(t => this.inferRightWithSdkType(t, rightType, ackClass));
+        }
+        else if (leftType instanceof Type_1.ClassType) {
+            IRInference.inferArgTypeWithSdk(leftType, ackClass.getDeclaringArkFile().getScene(), rightType);
+        }
+        else if (rightType instanceof Type_1.ArrayType && leftType instanceof Type_1.ArrayType) {
+            const baseType = TypeInference_1.TypeInference.replaceAliasType(leftType.getBaseType());
+            if (baseType instanceof Type_1.ClassType) {
+                IRInference.inferArgTypeWithSdk(baseType, ackClass.getDeclaringArkFile().getScene(), rightType.getBaseType());
+            }
+        }
+    }
+    static inferArgTypeWithSdk(sdkType, scene, argType) {
+        var _a, _b;
+        const sdkProjectName = sdkType.getClassSignature().getDeclaringFileSignature().getProjectName();
+        const className = sdkType.getClassSignature().getClassName();
+        // When leftOp is local with Function annotation, the rightOp is a lambda function, which should be inferred as method later.
+        if (!scene.getProjectSdkMap().has(sdkProjectName) || (sdkProjectName === SdkUtils_1.SdkUtils.BUILT_IN_NAME && className === 'Function')) {
+            return;
+        }
+        if (argType instanceof Type_1.UnionType) {
+            argType.getTypes().forEach(t => this.inferArgTypeWithSdk(sdkType, scene, t));
+        }
+        else if (argType instanceof Type_1.ClassType && argType.getClassSignature().getClassName().startsWith(Const_1.ANONYMOUS_CLASS_PREFIX)) {
+            this.inferAnonymousClass(scene.getClass(argType.getClassSignature()), sdkType.getClassSignature());
+        }
+        else if (argType instanceof Type_1.FunctionType) {
+            const param = (_b = (_a = scene.getClass(sdkType.getClassSignature())) === null || _a === void 0 ? void 0 : _a.getMethodWithName(Const_1.CALL_SIGNATURE_NAME)) === null || _b === void 0 ? void 0 : _b.getSignature().getMethodSubSignature();
+            const realTypes = sdkType.getRealGenericTypes();
+            TypeInference_1.TypeInference.inferFunctionType(argType, param, realTypes);
+        }
+    }
+    static inferInvokeExpr(expr, baseType, methodName, scene) {
+        let typeWithoutPtrOrRef = baseType;
+        // If it is a Cxx pointer or reference type, it is necessary to obtain its baseType and determine whether type inference is required.
+        if (baseType instanceof Type_2.PointerType || baseType instanceof Type_2.ReferenceType) {
+            typeWithoutPtrOrRef = baseType.getBaseType();
+        }
+        if (typeWithoutPtrOrRef instanceof Type_1.AliasType) {
+            return this.inferInvokeExpr(expr, typeWithoutPtrOrRef.getOriginalType(), methodName, scene);
+        }
+        else if (typeWithoutPtrOrRef instanceof Type_1.UnionType) {
+            for (let type of typeWithoutPtrOrRef.flatType()) {
+                if (type instanceof Type_1.UndefinedType || type instanceof Type_1.NullType) {
+                    continue;
+                }
+                let result = this.inferInvokeExpr(expr, type, methodName, scene);
+                if (result) {
+                    return result;
+                }
+            }
+        }
+        if (typeWithoutPtrOrRef instanceof Type_1.ClassType) {
+            return this.inferInvokeExprWithDeclaredClass(expr, typeWithoutPtrOrRef, methodName, scene);
+        }
+        else if (typeWithoutPtrOrRef instanceof Type_1.AnnotationNamespaceType) {
+            const namespace = scene.getNamespace(typeWithoutPtrOrRef.getNamespaceSignature());
+            if (namespace) {
+                const foundMethod = ModelUtils_2.CxxModelUtils.findPropertyInNamespace(methodName, namespace);
+                if (foundMethod instanceof ArkMethod_1.ArkMethod) {
+                    let signature = foundMethod.matchMethodSignature(expr.getArgs());
+                    TypeInference_1.TypeInference.inferSignatureReturnType(signature, foundMethod);
+                    expr.setMethodSignature(signature);
+                    return expr instanceof Expr_1.ArkInstanceInvokeExpr ? new Expr_1.ArkStaticInvokeExpr(signature, expr.getArgs(), expr.getRealGenericTypes()) : expr;
+                }
+            }
+        }
+        else if (typeWithoutPtrOrRef instanceof Type_1.FunctionType) {
+            return IRInference.inferInvokeExprWithFunction(methodName, expr, typeWithoutPtrOrRef, scene);
+        }
+        else if (typeWithoutPtrOrRef instanceof Type_1.ArrayType) {
+            return IRInference.inferInvokeExprWithArray(methodName, expr, typeWithoutPtrOrRef, scene);
+        }
+        return null;
+    }
+    static inferInvokeExprWithArray(methodName, expr, baseType, scene) {
+        const arrayInterface = scene.getSdkGlobal(Builtin_1.Builtin.ARRAY);
+        if (arrayInterface instanceof ArkClass_1.ArkClass) {
+            return this.inferInvokeExpr(expr, new Type_1.ClassType(arrayInterface.getSignature(), [baseType.getBaseType()]), methodName, scene);
+        }
+        else if (methodName === Builtin_1.Builtin.ITERATOR_FUNCTION) {
+            expr.getMethodSignature().getMethodSubSignature().setReturnType(Builtin_1.Builtin.ITERATOR_CLASS_TYPE);
+            expr.setRealGenericTypes([baseType.getBaseType()]);
+            return expr;
+        }
+        return null;
+    }
+    static inferInvokeExprWithFunction(methodName, expr, baseType, scene) {
+        if (methodName === Const_1.CALL_SIGNATURE_NAME) {
+            expr.setMethodSignature(baseType.getMethodSignature());
+            return expr;
+        }
+        const funcInterface = scene.getSdkGlobal(TSConst_1.FUNCTION);
+        if (funcInterface instanceof ArkClass_1.ArkClass) {
+            const method = ModelUtils_1.ModelUtils.findPropertyInClass(methodName, funcInterface);
+            if (method instanceof ArkMethod_1.ArkMethod) {
+                expr.setRealGenericTypes([baseType]);
+                expr.setMethodSignature(method.getSignature());
+                return expr;
+            }
+        }
+        return null;
+    }
+    static inferInvokeExprWithDeclaredClass(expr, baseType, methodName, scene) {
+        var _a, _b;
+        let declaredClass = scene.getClass(baseType.getClassSignature());
+        if (!declaredClass) {
+            const globalClass = scene.getSdkGlobal(baseType.getClassSignature().getClassName());
+            if (globalClass instanceof ArkClass_1.ArkClass) {
+                declaredClass = globalClass;
+            }
+        }
+        const method = declaredClass ? ModelUtils_1.ModelUtils.findPropertyInClass(methodName, declaredClass) : null;
+        if (method instanceof ArkMethod_1.ArkMethod) {
+            const methodSignature = method.matchMethodSignature(expr.getArgs());
+            TypeInference_1.TypeInference.inferSignatureReturnType(methodSignature, method);
+            expr.setMethodSignature(this.replaceMethodSignature(expr.getMethodSignature(), methodSignature));
+            expr.setRealGenericTypes(IRInference.getRealTypes(expr, declaredClass, baseType, method));
+            if (method.isStatic() && expr instanceof Expr_1.ArkInstanceInvokeExpr) {
+                return new Expr_1.ArkStaticInvokeExpr(methodSignature, expr.getArgs(), expr.getRealGenericTypes());
+            }
+            return expr;
+        }
+        else if (method instanceof ArkField_1.ArkField) {
+            return (_a = this.changePtrInvokeExpr(method, scene, expr)) !== null && _a !== void 0 ? _a : expr;
+        }
+        else if (methodName === TSConst_1.CONSTRUCTOR_NAME) {
+            const constructor = (_b = declaredClass === null || declaredClass === void 0 ? void 0 : declaredClass.getMethodWithName('construct-signature')) !== null && _b !== void 0 ? _b : declaredClass === null || declaredClass === void 0 ? void 0 : declaredClass.getMethodWithName(Const_1.CALL_SIGNATURE_NAME);
+            if (constructor) {
+                const methodSignature = constructor.matchMethodSignature(expr.getArgs());
+                TypeInference_1.TypeInference.inferSignatureReturnType(methodSignature, constructor);
+                expr.setMethodSignature(this.replaceMethodSignature(expr.getMethodSignature(), methodSignature));
+                expr.setRealGenericTypes(IRInference.getRealTypes(expr, declaredClass, baseType, constructor));
+            }
+            else {
+                const subSignature = new ArkSignature_1.MethodSubSignature(methodName, [], new Type_1.ClassType(baseType.getClassSignature()));
+                expr.setMethodSignature(new ArkSignature_1.MethodSignature(baseType.getClassSignature(), subSignature));
+            }
+            return expr;
+        }
+        else if (methodName === Builtin_1.Builtin.ITERATOR_NEXT &&
+            baseType.getClassSignature().getDeclaringFileSignature().getProjectName() === Builtin_1.Builtin.DUMMY_PROJECT_NAME) {
+            expr.getMethodSignature().getMethodSubSignature().setReturnType(Builtin_1.Builtin.ITERATOR_RESULT_CLASS_TYPE);
+            expr.setRealGenericTypes(baseType.getRealGenericTypes());
+            return expr;
+        }
+        return null;
+    }
+    static changePtrInvokeExpr(method, scene, expr) {
+        var _a;
+        const type = method.getType();
+        let methodSignature;
+        if (type instanceof Type_1.FunctionType) {
+            methodSignature = type.getMethodSignature();
+        }
+        else if (type instanceof Type_1.ClassType && type.getClassSignature().getClassName().endsWith(EtsConst_1.CALL_BACK)) {
+            const callback = (_a = scene.getClass(type.getClassSignature())) === null || _a === void 0 ? void 0 : _a.getMethodWithName(Const_1.CALL_SIGNATURE_NAME);
+            if (callback) {
+                methodSignature = callback.getSignature();
+            }
+        }
+        if (methodSignature) {
+            const ptr = expr instanceof Expr_1.ArkInstanceInvokeExpr
+                ? new Ref_1.ArkInstanceFieldRef(expr.getBase(), method.getSignature())
+                : new Ref_1.ArkStaticFieldRef(method.getSignature());
+            return new Expr_1.ArkPtrInvokeExpr(methodSignature, ptr, expr.getArgs(), expr.getRealGenericTypes());
+        }
+        return null;
+    }
+    static getRealTypes(expr, declaredClass, baseType, method) {
+        var _a;
+        let realTypes;
+        const tmp = [];
+        if (method.getGenericTypes()) {
+            expr.getMethodSignature()
+                .getMethodSubSignature()
+                .getParameters()
+                .filter(p => !p.getName().startsWith(Const_1.LEXICAL_ENV_NAME_PREFIX))
+                .forEach((p, i) => {
+                if (TypeInference_1.TypeInference.checkType(p.getType(), t => t instanceof Type_1.GenericType)) {
+                    tmp.push(expr.getArg(i).getType());
+                }
+            });
+        }
+        if (tmp.length > 0) {
+            realTypes = tmp;
+        }
+        else if (declaredClass === null || declaredClass === void 0 ? void 0 : declaredClass.hasComponentDecorator()) {
+            realTypes = [new Type_1.ClassType(declaredClass === null || declaredClass === void 0 ? void 0 : declaredClass.getSignature())];
+        }
+        else {
+            realTypes = (_a = baseType.getRealGenericTypes()) !== null && _a !== void 0 ? _a : declaredClass === null || declaredClass === void 0 ? void 0 : declaredClass.getRealTypes();
+        }
+        return realTypes;
+    }
+    static replaceMethodSignature(init, declared) {
+        const className = init.getDeclaringClassSignature().getClassName();
+        let classSignature;
+        if (declared.getDeclaringClassSignature().getClassName().endsWith('Interface')) {
+            classSignature = new ArkSignature_1.AliasClassSignature(className, declared.getDeclaringClassSignature());
+        }
+        let newSubSignature;
+        if (classSignature || newSubSignature) {
+            return new ArkSignature_1.MethodSignature(classSignature !== null && classSignature !== void 0 ? classSignature : declared.getDeclaringClassSignature(), newSubSignature !== null && newSubSignature !== void 0 ? newSubSignature : declared.getMethodSubSignature());
+        }
+        return declared;
+    }
+    static processForEach(arg, baseType, scene) {
+        const argType = arg.getType();
+        if (argType instanceof Type_1.FunctionType) {
+            const argMethodSignature = argType.getMethodSignature();
+            const argMethod = scene.getMethod(argMethodSignature);
+            if (argMethod !== null && argMethod.getBody()) {
+                const body = argMethod.getBody();
+                const firstStmt = body.getCfg().getStartingStmt();
+                if (firstStmt instanceof Stmt_1.ArkAssignStmt && firstStmt.getRightOp() instanceof Ref_1.ArkParameterRef) {
+                    const parameterRef = firstStmt.getRightOp();
+                    parameterRef.setType(baseType.getBaseType());
+                    const argMethodParams = argMethod.getSignature().getMethodSubSignature().getParameters();
+                    const actualParam = argMethodParams[argMethodParams.length - 1];
+                    actualParam.setType(baseType.getBaseType());
+                }
+                TypeInference_1.TypeInference.inferTypeInMethod(argMethod);
+            }
+        }
+        else {
+            logger.warn(`arg of forEach must be callable`);
+        }
+    }
+    static inferLocal(base, arkMethod) {
+        var _a, _b, _c;
+        const arkClass = arkMethod.getDeclaringArkClass();
+        let baseType = base.getType();
+        if (baseType instanceof Type_1.UnclearReferenceType) {
+            baseType = TypeInference_1.TypeInference.inferUnclearRefName(baseType.getName(), arkClass);
+        }
+        if (TypeInference_1.TypeInference.isUnclearType(baseType)) {
+            const declaringStmt = base.getDeclaringStmt();
+            if (!declaringStmt || !declaringStmt.getOriginalText() || ((_a = declaringStmt.getOriginalText()) === null || _a === void 0 ? void 0 : _a.startsWith(base.getName()))) {
+                baseType = (_c = (_b = ModelUtils_1.ModelUtils.findDeclaredLocal(base, arkMethod)) === null || _b === void 0 ? void 0 : _b.getType()) !== null && _c !== void 0 ? _c : TypeInference_1.TypeInference.inferBaseType(base.getName(), arkClass);
+            }
+        }
+        if (baseType && !TypeInference_1.TypeInference.isUnclearType(baseType)) {
+            base.setType(baseType);
+        }
+        // If the type of value is functionType and the current file is a CXX file,
+        // it should be represented as a CXX function pointer type ==> PointerType(FunctionType, 1).
+        if (baseType instanceof Type_1.FunctionType) {
+            base.setType(new Type_2.PointerType(baseType, 1));
+        }
+    }
+    static generateNewFieldSignature(ref, arkClass, baseType) {
+        if (baseType instanceof Type_1.AliasType) {
+            return this.generateNewFieldSignature(ref, arkClass, baseType.getOriginalType());
+        }
+        const fieldName = ref.getFieldName().replace(/[\"|\']/g, '');
+        const propertyAndType = TypeInference_1.TypeInference.inferFieldType(baseType, fieldName, arkClass);
+        let propertyType = IRInference.repairType(propertyAndType === null || propertyAndType === void 0 ? void 0 : propertyAndType[1], fieldName, arkClass);
+        let staticFlag;
+        let signature;
+        let typeWithoutPtrOrRef = baseType;
+        // If it is a Cxx pointer or reference type, it is necessary to obtain its baseType and determine whether type inference is required.
+        if (baseType instanceof Type_2.PointerType || baseType instanceof Type_2.ReferenceType) {
+            typeWithoutPtrOrRef = baseType.getBaseType();
+        }
+        if (typeWithoutPtrOrRef instanceof Type_1.ClassType) {
+            const property = propertyAndType === null || propertyAndType === void 0 ? void 0 : propertyAndType[0];
+            if (property instanceof ArkField_1.ArkField && property.getCategory() !== ArkField_1.FieldCategory.ENUM_MEMBER && !(property.getType() instanceof Type_1.GenericType)) {
+                return property.getSignature();
+            }
+            staticFlag =
+                typeWithoutPtrOrRef.getClassSignature().getClassName() === Const_1.DEFAULT_ARK_CLASS_NAME ||
+                    ((property instanceof ArkField_1.ArkField || property instanceof ArkMethod_1.ArkMethod) && property.isStatic());
+            signature = property instanceof ArkMethod_1.ArkMethod ? property.getSignature().getDeclaringClassSignature() : typeWithoutPtrOrRef.getClassSignature();
+        }
+        else if (typeWithoutPtrOrRef instanceof Type_1.ArrayType) {
+            const property = propertyAndType === null || propertyAndType === void 0 ? void 0 : propertyAndType[0];
+            if (property instanceof ArkField_1.ArkField) {
+                return property.getSignature();
+            }
+            return null;
+        }
+        else if (typeWithoutPtrOrRef instanceof Type_1.AnnotationNamespaceType) {
+            staticFlag = true;
+            signature = typeWithoutPtrOrRef.getNamespaceSignature();
+        }
+        else {
+            return null;
+        }
+        return new ArkSignature_1.FieldSignature(fieldName, signature, propertyType !== null && propertyType !== void 0 ? propertyType : ref.getType(), staticFlag);
+    }
+    static repairType(propertyType, fieldName, arkClass) {
+        if (!propertyType || propertyType instanceof Type_1.UnknownType) {
+            const newType = TypeInference_1.TypeInference.inferBaseType(fieldName, arkClass);
+            if (newType) {
+                propertyType = newType;
+            }
+        }
+        else if (TypeInference_1.TypeInference.isUnclearType(propertyType)) {
+            const newType = TypeInference_1.TypeInference.inferUnclearedType(propertyType, arkClass);
+            if (newType) {
+                propertyType = newType;
+            }
+        }
+        return propertyType;
+    }
+    static inferAnonymousClass(anon, declaredSignature, set = new Set()) {
+        if (!anon) {
+            return;
+        }
+        const key = anon.getSignature().toString();
+        if (set.has(key)) {
+            return;
+        }
+        else {
+            set.add(key);
+        }
+        const scene = anon.getDeclaringArkFile().getScene();
+        const declaredClass = scene.getClass(declaredSignature);
+        if (!declaredClass) {
+            return;
+        }
+        for (const anonField of anon.getFields()) {
+            const property = ModelUtils_1.ModelUtils.findPropertyInClass(anonField.getName(), declaredClass);
+            if (property instanceof ArkField_1.ArkField) {
+                this.assignAnonField(property, anonField, scene, set);
+            }
+            else if (property instanceof ArkMethod_1.ArkMethod) {
+                const type = anonField.getType();
+                if (type instanceof Type_1.FunctionType) {
+                    this.assignAnonMethod(scene.getMethod(type.getMethodSignature()), property);
+                }
+                if (type instanceof Type_1.UnknownType) {
+                    anonField.setSignature(new ArkSignature_1.FieldSignature(anonField.getName(), property.getDeclaringArkClass().getSignature(), new Type_1.FunctionType(property.getSignature())));
+                }
+            }
+        }
+        for (const anonMethod of anon.getMethods()) {
+            this.assignAnonMethod(anonMethod, declaredClass.getMethodWithName(anonMethod.getName()));
+        }
+    }
+    static assignAnonMethod(anonMethod, declaredMethod) {
+        if (declaredMethod && anonMethod) {
+            anonMethod.setDeclareSignatures(declaredMethod.matchMethodSignature(anonMethod.getSubSignature().getParameters()));
+        }
+    }
+    static assignAnonField(property, anonField, scene, set) {
+        function deepInfer(anonType, declaredSignature) {
+            if (anonType instanceof Type_1.ClassType && anonType.getClassSignature().getClassName().startsWith(Const_1.ANONYMOUS_CLASS_PREFIX)) {
+                IRInference.inferAnonymousClass(scene.getClass(anonType.getClassSignature()), declaredSignature, set);
+            }
+        }
+        const type = property.getSignature().getType();
+        const fieldInitializer = anonField.getInitializer();
+        const lastStmt = fieldInitializer[fieldInitializer.length - 1];
+        if (lastStmt instanceof Stmt_1.ArkAssignStmt) {
+            const rightType = lastStmt.getRightOp().getType();
+            if (type instanceof Type_1.ClassType) {
+                deepInfer(rightType, type.getClassSignature());
+            }
+            else if (type instanceof Type_1.ArrayType && type.getBaseType() instanceof Type_1.ClassType && rightType instanceof Type_1.ArrayType) {
+                const baseType = rightType.getBaseType();
+                const classSignature = type.getBaseType().getClassSignature();
+                if (baseType instanceof Type_1.UnionType) {
+                    baseType.getTypes().forEach(t => deepInfer(t, classSignature));
+                }
+                else {
+                    deepInfer(rightType.getBaseType(), classSignature);
+                }
+            }
+            else if (type instanceof Type_1.FunctionType && rightType instanceof Type_1.FunctionType) {
+                TypeInference_1.TypeInference.inferFunctionType(rightType, type.getMethodSignature().getMethodSubSignature(), type.getRealGenericTypes());
+            }
+            const leftOp = lastStmt.getLeftOp();
+            if (leftOp instanceof Ref_1.AbstractFieldRef) {
+                leftOp.setFieldSignature(property.getSignature());
+            }
+        }
+        anonField.setSignature(property.getSignature());
+    }
+    static inferAliasTypeExpr(expr, arkMethod) {
+        const originalObject = expr.getOriginalObject();
+        let model;
+        if (originalObject instanceof Local_1.Local) {
+            model = ModelUtils_1.ModelUtils.findArkModelByRefName(originalObject.getName(), arkMethod.getDeclaringArkClass());
+        }
+        else if (originalObject instanceof TypeExpr_1.AbstractTypeExpr) {
+            originalObject.inferType(arkMethod);
+            model = originalObject;
+        }
+        else if (originalObject instanceof Type_1.Type) {
+            const type = TypeInference_1.TypeInference.inferUnclearedType(originalObject, arkMethod.getDeclaringArkClass());
+            // If original Object is ClassType, AliasType or UnclearReferenceType with real generic types,
+            // the type after infer should be revert back to the object itself.
+            if (type instanceof Type_1.ClassType) {
+                const scene = arkMethod.getDeclaringArkFile().getScene();
+                model = ModelUtils_1.ModelUtils.findArkModelBySignature(type.getClassSignature(), scene);
+            }
+            else if (type instanceof Type_1.AliasType) {
+                const scene = arkMethod.getDeclaringArkFile().getScene();
+                model = ModelUtils_1.ModelUtils.findArkModelBySignature(type.getSignature(), scene);
+            }
+            else if (type) {
+                model = type;
+            }
+            if (expr.getRealGenericTypes() !== undefined && originalObject instanceof Type_1.UnclearReferenceType) {
+                expr.setRealGenericTypes(originalObject.getGenericTypes());
+            }
+        }
+        if (Expr_1.AliasTypeExpr.isAliasTypeOriginalModel(model)) {
+            expr.setOriginalObject(model);
+        }
+        return expr;
+    }
+    static inferTypeQueryExpr(expr, arkMethod) {
+        var _a;
+        let gTypes = expr.getGenerateTypes();
+        if (gTypes) {
+            for (let i = 0; i < gTypes.length; i++) {
+                const newType = TypeInference_1.TypeInference.inferUnclearedType(gTypes[i], arkMethod.getDeclaringArkClass());
+                if (newType) {
+                    gTypes[i] = newType;
+                }
+            }
+        }
+        const opValue = expr.getOpValue();
+        let opValueType;
+        if (opValue instanceof ArkBaseModel_1.ArkBaseModel) {
+            opValueType = (_a = ModelUtils_1.ModelUtils.parseArkBaseModel2Type(opValue)) !== null && _a !== void 0 ? _a : Type_1.UnknownType.getInstance();
+        }
+        else {
+            opValueType = opValue.getType();
+        }
+        if (!TypeInference_1.TypeInference.isUnclearType(opValueType)) {
+            return;
+        }
+        if (opValue instanceof Local_1.Local) {
+            const newOpValueType = TypeInference_1.TypeInference.inferBaseType(opValue.getName(), arkMethod.getDeclaringArkClass());
+            const scene = arkMethod.getDeclaringArkFile().getScene();
+            if (newOpValueType instanceof Type_1.ClassType) {
+                const newOpValue = ModelUtils_1.ModelUtils.findArkModelBySignature(newOpValueType.getClassSignature(), scene);
+                if (newOpValue instanceof ArkBaseModel_1.ArkBaseModel) {
+                    expr.setOpValue(newOpValue);
+                }
+            }
+            else if (newOpValueType instanceof Type_1.FunctionType) {
+                const newOpValue = ModelUtils_1.ModelUtils.findArkModelBySignature(newOpValueType.getMethodSignature(), scene);
+                if (newOpValue instanceof ArkBaseModel_1.ArkBaseModel) {
+                    expr.setOpValue(newOpValue);
+                }
+            }
+            else {
+                this.inferLocal(opValue, arkMethod);
+            }
+        }
+        else if (opValue instanceof Ref_1.AbstractRef || opValue instanceof Expr_1.AbstractExpr) {
+            expr.setOpValue(opValue.inferType(arkMethod));
+        }
+    }
+    static inferKeyofTypeExpr(expr, arkMethod) {
+        const opType = expr.getOpType();
+        if (TypeInference_1.TypeInference.isUnclearType(opType)) {
+            if (opType instanceof TypeExpr_1.TypeQueryExpr) {
+                this.inferTypeQueryExpr(opType, arkMethod);
+            }
+            else {
+                const type = TypeInference_1.TypeInference.inferUnclearedType(opType, arkMethod.getDeclaringArkClass());
+                if (type) {
+                    expr.setOpType(type);
+                }
+            }
+        }
+    }
+    static inferParameterRef(ref, arkMethod) {
+        var _a, _b, _c;
+        const paramType = ref.getType();
+        let baseType;
+        // If it is a Cxx pointer or reference type, it is necessary to obtain its baseType and determine whether type inference is required.
+        if (paramType instanceof Type_2.PointerType || paramType instanceof Type_2.ReferenceType) {
+            baseType = paramType.getBaseType();
+            if (TypeInference_1.TypeInference.isUnclearType(baseType)) {
+                baseType = TypeInference_1.TypeInference.inferUnclearedType(baseType, arkMethod.getDeclaringArkClass());
+                if (baseType) {
+                    paramType.setBaseType(baseType);
+                    return ref;
+                }
+            }
+        }
+        if (paramType instanceof Type_1.UnknownType || paramType instanceof Type_1.UnclearReferenceType) {
+            const signature = (_b = (_a = arkMethod.getDeclareSignatures()) === null || _a === void 0 ? void 0 : _a[0]) !== null && _b !== void 0 ? _b : arkMethod.getSignature();
+            const type1 = (_c = signature.getMethodSubSignature().getParameters()[ref.getIndex()]) === null || _c === void 0 ? void 0 : _c.getType();
+            if (!TypeInference_1.TypeInference.isUnclearType(type1)) {
+                ref.setType(type1);
+                return ref;
+            }
+        }
+        else if (paramType instanceof Type_1.LexicalEnvType) {
+            paramType
+                .getClosures()
+                .filter(c => TypeInference_1.TypeInference.isUnclearType(c.getType()))
+                .forEach(e => this.inferLocal(e, arkMethod));
+            return ref;
+        }
+        let type = TypeInference_1.TypeInference.inferUnclearedType(paramType, arkMethod.getDeclaringArkClass());
+        if (type) {
+            ref.setType(type);
+        }
+        return ref;
+    }
+    /**
+     * map c++ function declaration to Implementation.
+     *
+     *This function traverses all C++header files to find the corresponding implementation file for each class and method,
+     *And establish the mapping relationship between methods and implementation files
+     */
+    static mapCxxDeclAndImpl(scene) {
+        const headerFileRefMap = this.getCxxHeaderFileRefMap(scene.getFiles(), scene.getIncludeDirs());
+        for (const [headerPath, refFiles] of headerFileRefMap) {
+            const headerArkFile = scene.getFile(new ArkSignature_1.FileSignature(scene.getProjectName(), path_1.default.relative(scene.getRealProjectDir(), headerPath)));
+            if (!headerArkFile) {
+                continue;
+            }
+            const sortedRefFiles = this.sortRefFiles(headerPath, refFiles);
+            for (const cls of ModelUtils_1.ModelUtils.getAllClassesInFile(headerArkFile)) {
+                const implClasses = this.mapClassDeclAndImpl(sortedRefFiles, cls);
+                for (const mtd of cls.getMethods(true)) {
+                    this.findMtdImpl(mtd, implClasses);
+                }
+            }
+        }
+    }
+    static mapClassDeclAndImpl(cppFiles, clsInHeader) {
+        var _a;
+        const implClasses = [];
+        const tgtClsName = clsInHeader.getName();
+        const tgtNamespaceName = (_a = clsInHeader.getDeclaringArkNamespace()) === null || _a === void 0 ? void 0 : _a.getName();
+        for (const cppFile of cppFiles) {
+            const refArkClasses = this.getClassWithNameAndNamespace(tgtClsName, cppFile, tgtNamespaceName);
+            if (refArkClasses.length !== 1) {
+                continue;
+            }
+            implClasses.push(refArkClasses[0]);
+            if (tgtClsName === Const_1.DEFAULT_ARK_CLASS_NAME) {
+                continue;
+            }
+            refArkClasses[0].setDeclareSignature(clsInHeader.getSignature());
+        }
+        return implClasses;
+    }
+    /**
+     *Implementation method of search method
+     *@ param mtd - Ark method object
+     *@ param headerPath - Header file path
+     *@ param sortedRefFiles - sorted reference file array
+     */
+    static findMtdImpl(mtd, implClasses) {
+        if (mtd.getImplementationSignature() || mtd.isDefaultArkMethod() ||
+            mtd.getName() === Const_1.INSTANCE_INIT_METHOD_NAME || mtd.getName() === Const_1.STATIC_INIT_METHOD_NAME) {
+            return;
+        }
+        this.mapHeaderToSource(mtd, implClasses);
+    }
+    /**
+     *Get the C++header file reference mapping table
+     *
+     *This function traverses all files and filters out C++source files (.cpp ,.c,.cxx),
+     *Then analyze the import information in these source files, and establish the reference relationship mapping from the header file to the source file.
+     *
+     *@ returns Map<string, string []>The mapping from the header file path to the source file path array that references the header file
+     */
+    static getCxxHeaderFileRefMap(files, includeDirs) {
+        const headerFileRefMap = new Map();
+        const cxxSuffixes = (0, const_1.getCxxImplementationFileExtensions)();
+        files.forEach(file => {
+            const filePath = (0, path_1.normalize)(file.getFilePath());
+            const extension = path_1.default.extname(filePath).toLowerCase();
+            if (!cxxSuffixes.some(suffix => extension === suffix)) {
+                return;
+            }
+            const importInfos = file.getImportInfos();
+            importInfos.forEach(im => {
+                this.processImportInfo(im, filePath, headerFileRefMap, includeDirs);
+            });
+        });
+        return headerFileRefMap;
+    }
+    /**
+     *Process import information and establish header file reference relationship mapping
+     *@ param im Import information object, including imported module information
+     *@ param filePath Path of the current file
+     *@ param headerFileRefMap header file reference relationship mapping table, used to record which files are referenced by each header file
+     */
+    static processImportInfo(im, filePath, headerFileRefMap, includeDirs) {
+        let imFrom = im.getFrom();
+        if (!imFrom) {
+            return;
+        }
+        if (!fs_1.default.existsSync(imFrom)) {
+            // If the path does not exist, try to use includeDirs to find the relative path
+            imFrom = (0, FileUtils_1.getFileAbsPath)(includeDirs, imFrom);
+            if (!imFrom) {
+                return;
+            }
+        }
+        if (!headerFileRefMap.has(imFrom)) {
+            headerFileRefMap.set(imFrom, []);
+        }
+        headerFileRefMap.get(imFrom).push(im.getDeclaringArkFile());
+    }
+    /**
+     *Sort the reference file array, and prioritize the reference files with the same name as the target file
+     *@ param headerFilePath Destination header file path
+     *@ param refFiles Array of reference file paths to be sorted
+     *@ returns The array of reference file paths sorted, with files with the same name first and other files second
+     */
+    static sortRefFiles(headerFilePath, refFiles) {
+        const targetFileName = path_1.default.parse(headerFilePath).name;
+        const prioritized = [];
+        const others = [];
+        for (const refFile of refFiles) {
+            if (path_1.default.parse(refFile.getFilePath()).name === targetFileName) {
+                prioritized.push(refFile);
+            }
+            else {
+                others.push(refFile);
+            }
+        }
+        return [...prioritized, ...others];
+    }
+    static mapHeaderToSource(mtdDecl, implClasses) {
+        const tgtMtdSubSig = mtdDecl.getSubSignature();
+        const matchKey = tgtMtdSubSig.toString().replace('[static]', '');
+        const matchedMtdImpl = this.getMatchedMtdImpl(mtdDecl, implClasses, matchKey);
+        if (matchedMtdImpl) {
+            this.processMethodDeclAndImpl(mtdDecl, matchedMtdImpl);
+            return;
+        }
+    }
+    static getClassWithNameAndNamespace(clsName, file, namespaceName) {
+        let tgtNamespacesInFile = [];
+        if (namespaceName) {
+            tgtNamespacesInFile = ModelUtils_1.ModelUtils.getAllNamespacesInFile(file).filter(ns => ns.getName() === namespaceName);
+        }
+        const searchedClasses = [];
+        if (tgtNamespacesInFile.length !== 0) {
+            for (const ns of tgtNamespacesInFile) {
+                const classInNamespaceWithName = ns.getClassWithName(clsName);
+                if (classInNamespaceWithName) {
+                    searchedClasses.push(classInNamespaceWithName);
+                }
+            }
+        }
+        else {
+            const classInFileWithName = file.getClassWithName(clsName);
+            if (classInFileWithName) {
+                searchedClasses.push(classInFileWithName);
+            }
+        }
+        return searchedClasses;
+    }
+    static getMatchedMtdImpl(mtdDecl, refArkClasses, matchKey) {
+        for (const refArkClass of refArkClasses) {
+            const nameMatchingMtds = refArkClass.getAllMethodsWithName(mtdDecl.getName());
+            if (nameMatchingMtds.length === 0) {
+                continue;
+            }
+            if (nameMatchingMtds.length === 1) {
+                return nameMatchingMtds[0];
+            }
+            for (const mtdImpl of nameMatchingMtds) {
+                const matchingMtdSubSig = mtdImpl.getSubSignature().toString().replace('[static]', '');
+                if (mtdImpl.getBody() && matchingMtdSubSig === matchKey) {
+                    return mtdImpl;
+                }
+            }
+        }
+        return null;
+    }
+    static processMethodDeclAndImpl(mtdDecl, mtdImpl) {
+        // 1. Set the signature of the function implementation corresponding to the current function declaration
+        const mtdImplSignature = mtdImpl.getImplementationSignature();
+        if (mtdImplSignature) {
+            mtdDecl.setImplementationSignature(mtdImplSignature);
+        }
+        // 2. Set the signature of the function Declaration corresponding to the current function Implementation
+        const declSignature = mtdDecl.getDeclareSignatures();
+        const declSignatureInMtdImpl = mtdImpl.getDeclareSignatures();
+        if (declSignature) {
+            declSignatureInMtdImpl ? declSignatureInMtdImpl.push(...declSignature) : mtdImpl.setDeclareSignatures(declSignature);
+        }
+        // 3.The function implementation may be missing the existing modifiers (such as static) in the function declaration. Here we add
+        mtdImpl.setModifiers(mtdDecl.getModifiers() | mtdImpl.getModifiers());
+    }
+}
+exports.IRInference = IRInference;