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

3 min read time

Applies to Linux

• include
• kernels
• host

rotate_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 namespace Kernels::Host {

25 template <typename SrcWrapper, typename DstWrapper, typename MatWrapper>

26 void rotate(SrcWrapper src, DstWrapper dst, MatWrapper affineMat) {

27 #pragma omp parallel for

28  for (int b = 0; b < dst.batches(); b++) {

29  for (int y = 0; y < dst.height(); y++) {

30  for (int x = 0; x < dst.width(); x++) {

31  const auto xShift = x - affineMat[2];

32  const auto yShift = y - affineMat[5];

33 

34  const float srcX = static_cast<float>(xShift * affineMat[0] + yShift * -affineMat[1]);

35  const float srcY = static_cast<float>(xShift * -affineMat[3] + yShift * affineMat[4]);

36 

37  dst.at(b, y, x, 0) = src.at(b, srcY, srcX, 0);

38  }

39  }

40  }

41 }

42 } // namespace Kernels::Host

Kernels::Host

Definition: bilateral_filter_host.hpp:34

Kernels::Host::rotate

void rotate(SrcWrapper src, DstWrapper dst, MatWrapper affineMat)

Definition: rotate_host.hpp:26