OrangePi3588Media/tests/test_person_shoe_roi.cpp

26 lines
733 B
C++

#include <gtest/gtest.h>
#include "utils/person_shoe_roi.h"
namespace rk3588 {
namespace {
TEST(PersonShoeRoiTest, ComputesCenteredHorizontalOffsetFromWidthScale) {
EXPECT_FLOAT_EQ(ComputeCenteredXOffset(1.0f), 0.0f);
EXPECT_FLOAT_EQ(ComputeCenteredXOffset(1.6f), -0.3f);
EXPECT_FLOAT_EQ(ComputeCenteredXOffset(1.48f), -0.24f);
}
TEST(PersonShoeRoiTest, BuildsCenteredFootRegionHorizontally) {
const Rect person{100.0f, 200.0f, 200.0f, 400.0f};
const Rect roi = BuildPersonFootRegion(person, 1920, 1080, 0.8f, 1.6f, 0.3f);
EXPECT_FLOAT_EQ(roi.x, 40.0f);
EXPECT_FLOAT_EQ(roi.y, 520.0f);
EXPECT_FLOAT_EQ(roi.w, 320.0f);
EXPECT_FLOAT_EQ(roi.h, 120.0f);
}
} // namespace
} // namespace rk3588