# Description:
#      VSI OVX wrapper logic

package(
    default_visibility = ["//visibility:public"],
    features = ["-parse_headers"],
)

load("@bazel_tools//tools/build_defs/pkg:pkg.bzl", "pkg_tar")

filegroup(
    name = "all_files",
    srcs = glob(
        ["**/*"],
        exclude = [
            "**/METADATA",
            "**/OWNERS",
            "external/*"
        ],
    ),
)

filegroup(
    name = "kernel_hdrs",
    srcs = glob([
        "include/kernel/cl/*.h",
        "include/kernel/evis/*.h",
    ])
)

filegroup(
    name = "kernel_srcs",
    srcs = glob([
        "src/kernel/cl/*.c",
        "src/kernel/evis/*.c",
        "src/kernel/vx/*.c",
    ])
)

filegroup(
    name = "operation_srcs",
    srcs = glob([
        "src/ops/*.c",
    ])
)

filegroup(
    name = "operation_hdrs",
    srcs = glob([
        "include/ops/*.h",
    ])
)

filegroup(
    name = "custom_hdrs",
    srcs = glob([
        "include/custom/ops/*.h",
    ]) + [
        #custom
        "include/custom/custom_node_type.def",
        "include/custom/custom_ops.def",
        "include/custom/vsi_nn_custom_node_type.h",
    ]
)

filegroup(
    name = "custom_srcs",
    srcs = glob([
        "src/custom/ops/*.c",
        "src/custom/ops/kernel/evis/*.c",
        "src/custom/ops/kernel/cl/*.c",
        "src/custom/ops/kernel/cpu/*.c",
    ])
)

cc_library(
    name = "ovxlibimpl",
    copts = [
        "-Werror", "-Wmisleading-indentation",
        "-fvisibility=hidden", '-DOVXLIB_API=__attribute__((visibility(\\"default\\")))',
    ],
    linkopts = ["-ldl", "-lm"],
    alwayslink=True,
    linkstatic = True,
    includes = [
        "include",
    ],
    hdrs = [
        "include/vsi_nn_pub.h",
        "include/vsi_nn_ops.h",
        "include/vsi_nn_log.h",
        "include/vsi_nn_context.h",
        "include/vsi_nn_node_attr_template.h",
        "include/vsi_nn_tensor.h",
        "include/vsi_nn_prv.h",
        "include/vsi_nn_types.h",
        "include/vsi_nn_node.h",
        "include/vsi_nn_node_type.h",
        "include/vsi_nn_client_op.h",
        "include/vsi_nn_graph.h",
        "include/vsi_nn_test.h",
        "include/vsi_nn_tensor_util.h",
        "include/vsi_nn_version.h",
        "include/vsi_nn_compatibility.h",
        "include/vsi_nn_assert.h",
        "include/vsi_nn_feature.h",
        "include/vsi_nn_rnn.h",
        "include/vsi_nn_rnn_helper.h",
        "include/vsi_nn_rnn_prv.h",
        "include/vsi_nn_internal_node.h",
        "include/vsi_nn_daemon.h",
        "include/vsi_nn_pre_post_process.h",
        "include/vsi_nn_graph_optimization.h",
        "include/utils/vsi_nn_link_list.h",
        "include/utils/vsi_nn_math.h",
        "include/utils/vsi_nn_util.h",
        "include/utils/vsi_nn_code_generator.h",
        "include/utils/vsi_nn_binary_tree.h",
        "include/utils/vsi_nn_map.h",
        "include/utils/vsi_nn_hashmap.h",
        "include/utils/vsi_nn_limits.h",
        "include/utils/vsi_nn_dtype_util.h",
        "include/utils/vsi_nn_dtype_util_prv.h",
        "include/utils/vsi_nn_vdata.h",
        "include/utils/vsi_nn_tensor_op.h",
        "include/utils/vsi_nn_shape_util.h",
        "include/utils/vsi_nn_constraint_check.h",
        "include/quantization/vsi_nn_asymmetric_affine.h",
        "include/quantization/vsi_nn_dynamic_fixed_point.h",
        "include/quantization/vsi_nn_perchannel_symmetric_affine.h",
        "include/interface/ops.def",
        "include/kernel/vsi_nn_kernel.h",
        "include/kernel/vsi_nn_gpu.h",
        "include/kernel/vsi_nn_gpu_config.h",
        "include/kernel/vsi_nn_kernel_eltwise.h",
        "include/kernel/vsi_nn_kernel_node.h",
        "include/kernel/vsi_nn_kernel_gpu_shape_optimize.h",
        "include/kernel/vsi_nn_kernel_lut.h",
        "include/vsi_nn_error.h",

        # libnnext
        "include/libnnext/vsi_nn_vxkernel.h",
        "include/libnnext/vx_lib_nnext.h",
        "include/libnnext/vsi_nn_libnnext_resource.h",

        #internal
        "include/internal/internal_ops.def",
        "include/vsi_nn_feature_config.h"
    ] + [":kernel_hdrs"]
      + [":operation_hdrs"]
      + [":custom_hdrs"]
      + [
        "include/vsi_nn_platform.h",
    ],
    srcs = [
        "src/vsi_nn_graph.c",
        "src/vsi_nn_ops.c",
        "src/vsi_nn_context.c",
        "src/vsi_nn_node.c",
        "src/vsi_nn_tensor.c",
        "src/vsi_nn_client_op.c",
        "src/vsi_nn_node_attr_template.c",
        "src/vsi_nn_version.c",
        "src/vsi_nn_rnn.c",
        "src/vsi_nn_rnn_helper.c",
        "src/vsi_nn_log.c",
        "src/vsi_nn_internal_node.c",
        "src/vsi_nn_daemon.c",
        "src/vsi_nn_graph_optimization.c",
        "src/vsi_nn_pre_post_process.c",
        "src/utils/vsi_nn_link_list.c",
        "src/utils/vsi_nn_util.c",
        "src/utils/vsi_nn_math.c",
        "src/utils/vsi_nn_code_generator.c",
        "src/utils/vsi_nn_binary_tree.c",
        "src/utils/vsi_nn_map.c",
        "src/utils/vsi_nn_hashmap.c",
        "src/utils/vsi_nn_limits.c",
        "src/utils/vsi_nn_dtype_util.c",
        "src/utils/vsi_nn_vdata.c",
        "src/utils/vsi_nn_tensor_op.c",
        "src/utils/vsi_nn_shape_util.c",
        "src/utils/vsi_nn_dtype.c",
        "src/utils/vsi_nn_constraint_check.c",
        "src/quantization/vsi_nn_asymmetric_affine.c",
        "src/quantization/vsi_nn_dynamic_fixed_point.c",
        "src/quantization/vsi_nn_perchannel_symmetric_affine.c",
        "src/kernel/vsi_nn_kernel.c",
        "src/kernel/vsi_nn_kernel_util.c",
        "src/kernel/vsi_nn_kernel_backend.c",
        "src/kernel/vsi_nn_kernel_eltwise.c",
        "src/kernel/vsi_nn_kernel_selector.c",
        "src/kernel/vsi_nn_kernel_node.c",
        "src/kernel/vsi_nn_kernel_param.c",
        "src/kernel/vsi_nn_kernel_lut.c",
        "src/kernel/vsi_nn_gpu.c",
        "src/kernel/vsi_nn_kernel_gpu_shape_optimize.c",
        "src/libnnext/vsi_nn_libnnext_resource.c",
        "src/libnnext/vsi_nn_vxkernel.c",
    ] + [":kernel_srcs"]
      + [":operation_srcs"]
      + [":custom_srcs"],
    deps = ["//prebuilt-sdk:VIV_SDK_LIB"]
)