drivers_framework/tools/hdi-gen/codegen/code_generator.cpp

/*
 * Copyright (c) 2021-2022 Huawei Device Co., Ltd.
 *
 * HDF is dual licensed: you can use it either under the terms of
 * the GPL, or the BSD license, at your option.
 * See the LICENSE file in the root of this repository for complete details.
 */

#include "codegen/code_generator.h"
#include "codegen/c_client_proxy_code_emitter.h"
#include "codegen/c_custom_types_code_emitter.h"
#include "codegen/c_interface_code_emitter.h"
#include "codegen/c_service_driver_code_emitter.h"
#include "codegen/c_service_impl_code_emitter.h"
#include "codegen/c_service_stub_code_emitter.h"
#include "codegen/cpp_client_proxy_code_emitter.h"
#include "codegen/cpp_custom_types_code_emitter.h"
#include "codegen/cpp_interface_code_emitter.h"
#include "codegen/cpp_service_driver_code_emitter.h"
#include "codegen/cpp_service_impl_code_emitter.h"
#include "codegen/cpp_service_stub_code_emitter.h"
#include "codegen/java_client_interface_code_emitter.h"
#include "codegen/java_client_proxy_code_emitter.h"
#include "util/options.h"

namespace OHOS {
namespace HDI {
CodeEmitMap CodeGenerator::cCodeEmitters_ = {
    {"types",     new CCustomTypesCodeEmitter()  },
    {"interface", new CInterfaceCodeEmitter()    },
    {"proxy",     new CClientProxyCodeEmitter()  },
    {"driver",    new CServiceDriverCodeEmitter()},
    {"stub",      new CServiceStubCodeEmitter()  },
    {"impl",      new CServiceImplCodeEmitter()  },
};

CodeEmitMap CodeGenerator::cppCodeEmitters_ = {
    {"types",     new CppCustomTypesCodeEmitter()  },
    {"interface", new CppInterfaceCodeEmitter()    },
    {"proxy",     new CppClientProxyCodeEmitter()  },
    {"driver",    new CppServiceDriverCodeEmitter()},
    {"stub",      new CppServiceStubCodeEmitter()  },
    {"impl",      new CppServiceImplCodeEmitter()  },
};

CodeEmitMap CodeGenerator::javaCodeEmitters_ = {
    {"clientIface", new JavaClientInterfaceCodeEmitter()},
    {"proxy",       new JavaClientProxyCodeEmitter()    },
};

bool CodeGenerator::Generate()
{
    const Options &options = Options::GetInstance();
    String dir = options.GetGenerationDirectory();
    String language = options.GetTargetLanguage();
    bool isModeKernel = options.DoGenerateKernelCode();
    String codePart = options.GetCodePart();

    for (auto &astPair : allAst_) {
        AutoPtr<AST> ast = astPair.second;
        if (language.Equals("c")) {
            GenerateCCode(ast, dir, codePart, isModeKernel);
        } else if (language.Equals("cpp")) {
            GenerateCppCode(ast, dir, codePart);
        } else if (language.Equals("java")) {
            GenerateJavaCode(ast, dir, codePart);
        }
    }

    return true;
}

void CodeGenerator::GenerateCCode(const AutoPtr<AST> &ast, const String &outDir, const String &codePart, bool isKernel)
{
    switch (ast->GetASTFileType()) {
        case ASTFileType::AST_TYPES: {
            cCodeEmitters_["types"]->OutPut(ast, outDir, isKernel);
            break;
        }
        case ASTFileType::AST_IFACE: {
            if (codePart.Equals("client")) {
                cCodeEmitters_["interface"]->OutPut(ast, outDir, isKernel);
                cCodeEmitters_["proxy"]->OutPut(ast, outDir, isKernel);
                break;
            } else if (codePart.Equals("server")) {
                cCodeEmitters_["interface"]->OutPut(ast, outDir, isKernel);
                cCodeEmitters_["driver"]->OutPut(ast, outDir, isKernel);
                cCodeEmitters_["stub"]->OutPut(ast, outDir, isKernel);
                cCodeEmitters_["impl"]->OutPut(ast, outDir, isKernel);
                break;
            } else {
                cCodeEmitters_["interface"]->OutPut(ast, outDir, isKernel);
                cCodeEmitters_["proxy"]->OutPut(ast, outDir, isKernel);
                cCodeEmitters_["driver"]->OutPut(ast, outDir, isKernel);
                cCodeEmitters_["stub"]->OutPut(ast, outDir, isKernel);
                cCodeEmitters_["impl"]->OutPut(ast, outDir, isKernel);
            }
            break;
        }
        case ASTFileType::AST_ICALLBACK:
            // khdf doesn't support callback
            cCodeEmitters_["interface"]->OutPut(ast, outDir);
            cCodeEmitters_["proxy"]->OutPut(ast, outDir);
            cCodeEmitters_["driver"]->OutPut(ast, outDir);
            cCodeEmitters_["stub"]->OutPut(ast, outDir);
            cCodeEmitters_["impl"]->OutPut(ast, outDir);
            break;
        default:
            break;
    }
}

void CodeGenerator::GenerateCppCode(const AutoPtr<AST> &ast, const String &outDir, const String &codePart)
{
    switch (ast->GetASTFileType()) {
        case ASTFileType::AST_TYPES:
            cppCodeEmitters_["types"]->OutPut(ast, outDir);
            break;
        case ASTFileType::AST_IFACE: {
            if (codePart.Equals("client")) {
                cppCodeEmitters_["interface"]->OutPut(ast, outDir);
                cppCodeEmitters_["proxy"]->OutPut(ast, outDir);
            } else if (codePart.Equals("server")) {
                cppCodeEmitters_["interface"]->OutPut(ast, outDir);
                cppCodeEmitters_["driver"]->OutPut(ast, outDir);
                cppCodeEmitters_["stub"]->OutPut(ast, outDir);
                cppCodeEmitters_["impl"]->OutPut(ast, outDir);
            } else {
                cppCodeEmitters_["interface"]->OutPut(ast, outDir);
                cppCodeEmitters_["proxy"]->OutPut(ast, outDir);
                cppCodeEmitters_["driver"]->OutPut(ast, outDir);
                cppCodeEmitters_["stub"]->OutPut(ast, outDir);
                cppCodeEmitters_["impl"]->OutPut(ast, outDir);
            }
            break;
        }
        case ASTFileType::AST_ICALLBACK:
            cppCodeEmitters_["interface"]->OutPut(ast, outDir);
            cppCodeEmitters_["proxy"]->OutPut(ast, outDir);
            cppCodeEmitters_["driver"]->OutPut(ast, outDir);
            cppCodeEmitters_["stub"]->OutPut(ast, outDir);
            cppCodeEmitters_["impl"]->OutPut(ast, outDir);
            break;
        default:
            break;
    }
}

void CodeGenerator::GenerateJavaCode(const AutoPtr<AST> &ast, const String &outDir, const String &codePart)
{
    switch (ast->GetASTFileType()) {
        case ASTFileType::AST_IFACE:
            javaCodeEmitters_["clientIface"]->OutPut(ast, outDir);
            javaCodeEmitters_["proxy"]->OutPut(ast, outDir);
            break;
        case ASTFileType::AST_ICALLBACK:
            javaCodeEmitters_["clientIface"]->OutPut(ast, outDir);
            javaCodeEmitters_["proxy"]->OutPut(ast, outDir);
            break;
        default:
            break;
    }
}
} // namespace HDI
} // namespace OHOS