/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/wrappers/interpolation_wrapper.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/wrappers/interpolation_wrapper.hpp Source File#

5 min read time

Applies to Linux

rocCV: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/core/wrappers/interpolation_wrapper.hpp Source File

Go to the documentation of this file.

 /*
  * Copyright (c) 2025 Advanced Micro Devices, Inc. All rights reserved.
  * Permission is hereby granted, free of charge, to any person obtaining a copy
  * of this software and associated documentation files (the "Software"), to deal
  * in the Software without restriction, including without limitation the rights
  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  * copies of the Software, and to permit persons to whom the Software is
  * furnished to do so, subject to the following conditions:
  *
  * The above copyright notice and this permission notice shall be included in
  * all copies or substantial portions of the Software.
  *
  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  * THE SOFTWARE.
  */
  
 #pragma once
  
 #include "core/detail/casting.hpp"
 #include "core/detail/math/vectorized_type_math.hpp"
 #include "core/wrappers/border_wrapper.hpp"
 #include "operator_types.h"
  
 namespace roccv {
  
 template <typename T, eBorderType B, eInterpolationType I>
 class InterpolationWrapper {
    public:
     InterpolationWrapper(const Tensor& tensor, T border_value) : m_desc(tensor, border_value) {}
  
     InterpolationWrapper(BorderWrapper<T, B> borderWrapper) : m_desc(borderWrapper) {}
  
     inline __device__ __host__ const T at(int64_t n, float h, float w, int64_t c) const {
         if constexpr (I == eInterpolationType::INTERP_TYPE_NEAREST) {
             // Nearest neighbor interpolation implementation
             return m_desc.at(n, lroundf(h), lroundf(w), c);
         } else if constexpr (I == eInterpolationType::INTERP_TYPE_LINEAR) {
             // Bilinear interpolation implementation
             // v1 -- v2
             // -     -
             // v3 -- v4
  
             using WorkType = detail::MakeType<float, detail::NumElements<T>>;
  
             int64_t x0 = static_cast<int64_t>(floorf(w));
             int64_t x1 = x0 + 1;
             int64_t y0 = static_cast<int64_t>(floorf(h));
             int64_t y1 = y0 + 1;
  
             auto v1 = detail::RangeCast<WorkType>(m_desc.at(n, y0, x0, c));
             auto v2 = detail::RangeCast<WorkType>(m_desc.at(n, y0, x1, c));
             auto v3 = detail::RangeCast<WorkType>(m_desc.at(n, y1, x0, c));
             auto v4 = detail::RangeCast<WorkType>(m_desc.at(n, y1, x1, c));
  
             auto q1 = v1 * (x1 - w) + v2 * (w - x0);
             auto q2 = v3 * (x1 - w) + v4 * (w - x0);
             auto q = q1 * (y1 - h) + q2 * (h - y0);
  
             return detail::RangeCast<T>(q);
         }
  
         // TODO: Support other interpolation methods.
  
         else {
             static_assert(false, "Provided interpolation type is not supported.");
         }
     }
  
     __device__ __host__ inline int64_t height() const { return m_desc.height(); }
     __device__ __host__ inline int64_t width() const { return m_desc.width(); }
     __device__ __host__ inline int64_t batches() const { return m_desc.batches(); }
     __device__ __host__ inline int64_t channels() const { return m_desc.channels(); }
  
    private:
     BorderWrapper<T, B> m_desc;
 };
 }  // namespace roccv