sources/includes/nvll_obb.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 *
 * 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.
 */
 
#ifndef __NVLL_OBB_H__
#define __NVLL_OBB_H__
 
#include "nvll_osd_api.h"
#include "nvbufsurface.h"
#include <cuda_runtime.h>
 
#ifdef __cplusplus
extern "C" {
#endif
 
typedef struct _NvOBBDumpCtx NvOBBDumpCtx;
 
NvOBBDumpCtx* nvobb_dump_create_context(int gpu_id, cudaStream_t cuda_stream);
 
void nvobb_dump_destroy_context(NvOBBDumpCtx *ctx);
 
NvOBBDumpCtx* nvobb_dump_get_singleton(int gpu_id, cudaStream_t cuda_stream);
 
void nvobb_dump_cleanup_singleton(void);
 
bool nvobb_dump_frame(
    NvOBBDumpCtx *ctx,
    NvBufSurface *input_surf,
    void *rect_params_array,
    int num_rects,
    unsigned int frame_num);
 
cudaError_t nvobb_rotate_gpu(
    const uint8_t *aabb_pixels,
    int aabb_width,
    int aabb_height,
    int aabb_pitch,
    int obb_width,
    int obb_height,
    float obb_cx,
    float obb_cy,
    float angle_rad,
    cudaStream_t stream,
    uint8_t **d_output);
 
uint8_t *nvobb_rotate_cpu(
    const uint8_t *aabb_pixels,
    int aabb_width,
    int aabb_height,
    int aabb_pitch,
    int obb_width,
    int obb_height,
    float obb_cx,
    float obb_cy,
    float angle_rad);
 
#ifdef __cplusplus
}
#endif
 
#endif /* __NVLL_OBB_H__ */