sources/libs/nvll_osd/cuda/drawBbox.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 DRAW_BBOX_H
#define DRAW_BBOX_H
 
#include <cstdint>
//#include <stdint.h>
#include <cuda_runtime.h>
 
 
void drawBoundingBox_cuda_unit_alpha(uint8_t *pBGRA, const int nWidth, const int nHeight, const int nBgraPitch, const int x_min, const int y_min, const int x_max, const int y_max, cudaStream_t stream,
    unsigned int r, unsigned int g, unsigned int b, unsigned int border_width);
 
void drawBoundingBox_cuda(uint8_t *pBGRA, const int nWidth, const int nHeight, const int nBgraPitch, const int x_min, const int y_min, const int x_max, const int y_max, cudaStream_t stream,
    unsigned int r, unsigned int g, unsigned int b, float a, unsigned int border_width);
 
void bboxAlphaFill_cuda(uint8_t *pBGRA, const int nWidth, const int nHeight, const int nBgraPitch, const int x_min, const int y_min, const int x_max, const int y_max, unsigned int r, unsigned int g, unsigned int b, float a, cudaStream_t stream);
 
void bgra_to_gray(uint8_t *dpBgra, const int nBgraPitch, uint8_t *dpGray, const int nGrayPitch, const int nWidth, const int nHeight, cudaStream_t stream);
 
void nv12_to_gray_batch(const uint8_t *pNv12, int nNv12Pitch, uint8_t *pGray, int nGrayPitch, int nWidth, int nHeight, int nBatchSize, cudaStream_t stream);
 
#endif // DRAW_BBOX_H