/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/kernels/host/composite_host.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/kernels/host/composite_host.hpp Source File#

3 min read time

Applies to Linux

rocCV: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/kernels/host/composite_host.hpp Source File
composite_host.hpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2025 Advanced Micro Devices, Inc. All rights reserved.
3  * Permission is hereby granted, free of charge, to any person obtaining a copy
4  * of this software and associated documentation files (the "Software"), to deal
5  * in the Software without restriction, including without limitation the rights
6  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
7  * copies of the Software, and to permit persons to whom the Software is
8  * furnished to do so, subject to the following conditions:
9  *
10  * The above copyright notice and this permission notice shall be included in
11  * all copies or substantial portions of the Software.
12  *
13  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
17  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
18  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
19  * THE SOFTWARE.
20  */
21 
22 #pragma once
23 
24 #include "core/detail/casting.hpp"
25 
26 namespace Kernels {
27 namespace Host {
28 template <typename SrcWrapper, typename MaskWrapper, typename DstWrapper>
29 inline void composite(SrcWrapper foreground, SrcWrapper background, MaskWrapper mask, DstWrapper output) {
30  using namespace roccv::detail; // For RangeCast, NumElements, etc.
31  using src_type = typename SrcWrapper::ValueType;
32  using dst_type = typename DstWrapper::ValueType;
33  using work_type = MakeType<float, NumElements<src_type>>;
34 
35  for (int batch = 0; batch < output.batches(); batch++) {
36 #pragma omp parallel for
37  for (int y = 0; y < output.height(); y++) {
38  for (int x = 0; x < output.width(); x++) {
39  // Range cast all input values to float to avoid overflowing values and keep them in the same range.
40  auto maskFactor = RangeCast<float1>(mask.at(batch, y, x, 0));
41  auto fgVal = RangeCast<work_type>(foreground.at(batch, y, x, 0));
42  auto bgVal = RangeCast<work_type>(background.at(batch, y, x, 0));
43 
44  work_type result = fgVal * maskFactor.x + (1.0f - maskFactor.x) * bgVal;
45 
46  // If number of channels in output is 4, ensure that the last channel (alpha in this case) is always
47  // fully on.
48  if constexpr (NumElements<dst_type> == 4) {
49  output.at(batch, y, x, 0) =
50  RangeCast<dst_type>((MakeType<float, 4>){result.x, result.y, result.z, 1.0f});
51  } else {
52  output.at(batch, y, x, 0) = RangeCast<dst_type>(result);
53  }
54  }
55  }
56  }
57 }
58 } // namespace Host
59 } // namespace Kernels
Kernels::Host::composite
void composite(SrcWrapper foreground, SrcWrapper background, MaskWrapper mask, DstWrapper output)
Definition: composite_host.hpp:29
Kernels
Definition: non_max_suppression_helpers.hpp:26
roccv::detail
Definition: strided_data_wrap.hpp:33
roccv::detail::MakeType
MakeType_t< T, C >::type MakeType
Definition: type_traits.hpp:97