178 lines
6.6 KiB
C++
178 lines
6.6 KiB
C++
// Copyright 2020 Google LLC. All rights reserved.
|
|
//
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file or at
|
|
// https://developers.google.com/open-source/licenses/bsd
|
|
|
|
#include <packager/media/formats/dvb/subtitle_composer.h>
|
|
|
|
#include <gmock/gmock.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
namespace shaka {
|
|
namespace media {
|
|
|
|
namespace {
|
|
|
|
constexpr const int kNoBgColor = -1;
|
|
|
|
void CreateDefaultImage(SubtitleComposer* composer, uint16_t object_id) {
|
|
auto* image = composer->GetObjectImage(object_id);
|
|
EXPECT_TRUE(image->AddPixel(BitDepth::k8Bit, 1, true));
|
|
image->NewRow(true);
|
|
}
|
|
|
|
} // namespace
|
|
|
|
TEST(SubtitleComposerTest, PositionsSamples) {
|
|
const uint8_t kRegionId = 3, kColorSpaceId = 9;
|
|
const uint16_t kObjectId1 = 5, kObjectId2 = 11;
|
|
SubtitleComposer composer;
|
|
composer.SetDisplaySize(100, 100);
|
|
ASSERT_TRUE(composer.SetRegionPosition(kRegionId, 4, 5));
|
|
ASSERT_TRUE(composer.SetRegionInfo(kRegionId, kColorSpaceId, 96, 95));
|
|
ASSERT_TRUE(composer.SetObjectInfo(kObjectId1, kRegionId, 8, 9, kNoBgColor));
|
|
ASSERT_TRUE(
|
|
composer.SetObjectInfo(kObjectId2, kRegionId, 12, 14, kNoBgColor));
|
|
CreateDefaultImage(&composer, kObjectId1);
|
|
CreateDefaultImage(&composer, kObjectId2);
|
|
|
|
std::vector<std::shared_ptr<TextSample>> samples;
|
|
ASSERT_TRUE(composer.GetSamples(0, 1, &samples));
|
|
ASSERT_EQ(samples.size(), 2u);
|
|
for (size_t i = 0; i < samples.size(); i++) {
|
|
ASSERT_TRUE(samples[i]->settings().line);
|
|
ASSERT_EQ(samples[i]->settings().line->type, TextUnitType::kPercent);
|
|
ASSERT_TRUE(samples[i]->settings().position);
|
|
ASSERT_EQ(samples[i]->settings().position->type, TextUnitType::kPercent);
|
|
ASSERT_TRUE(samples[i]->settings().width);
|
|
ASSERT_EQ(samples[i]->settings().width->type, TextUnitType::kPercent);
|
|
ASSERT_EQ(samples[i]->settings().width->value, 1);
|
|
ASSERT_TRUE(samples[i]->settings().height);
|
|
ASSERT_EQ(samples[i]->settings().height->type, TextUnitType::kPercent);
|
|
ASSERT_EQ(samples[i]->settings().height->value, 1);
|
|
ASSERT_FALSE(samples[i]->body().image.empty());
|
|
}
|
|
// Allow in either order.
|
|
if (samples[0]->settings().position->value == 12) {
|
|
EXPECT_EQ(samples[0]->settings().line->value, 14);
|
|
EXPECT_EQ(samples[1]->settings().position->value, 16);
|
|
EXPECT_EQ(samples[1]->settings().line->value, 19);
|
|
} else {
|
|
EXPECT_EQ(samples[0]->settings().position->value, 16);
|
|
EXPECT_EQ(samples[0]->settings().line->value, 19);
|
|
EXPECT_EQ(samples[1]->settings().position->value, 12);
|
|
EXPECT_EQ(samples[1]->settings().line->value, 14);
|
|
}
|
|
}
|
|
|
|
TEST(SubtitleComposerTest, EnsuresRegionsFit) {
|
|
SubtitleComposer composer;
|
|
composer.SetDisplaySize(0xfff0, 0xfff0);
|
|
ASSERT_FALSE(composer.SetRegionPosition(1, 0xffff, 0xffff));
|
|
ASSERT_TRUE(composer.SetRegionPosition(1, 0xff00, 0xff00));
|
|
ASSERT_FALSE(composer.SetRegionInfo(1, 0, 0xf1, 0));
|
|
ASSERT_FALSE(composer.SetRegionInfo(1, 0, 0, 0xf1));
|
|
ASSERT_FALSE(composer.SetRegionInfo(1, 0, 0xff1, 0));
|
|
ASSERT_FALSE(composer.SetRegionInfo(1, 0, 0, 0xff1));
|
|
ASSERT_TRUE(composer.SetRegionInfo(1, 0, 25, 25));
|
|
|
|
ASSERT_FALSE(composer.SetObjectInfo(1, 1, 0xf1, 0, kNoBgColor));
|
|
ASSERT_FALSE(composer.SetObjectInfo(1, 1, 0, 0xf1, kNoBgColor));
|
|
ASSERT_FALSE(composer.SetObjectInfo(1, 1, 0xff1, 0, kNoBgColor));
|
|
ASSERT_FALSE(composer.SetObjectInfo(1, 1, 0, 0xff1, kNoBgColor));
|
|
ASSERT_TRUE(composer.SetObjectInfo(1, 1, 20, 20, kNoBgColor));
|
|
}
|
|
|
|
TEST(SubtitleComposerTest, MustInitRegionFirst) {
|
|
SubtitleComposer composer;
|
|
EXPECT_FALSE(composer.SetObjectInfo(0, 0, 0, 0, kNoBgColor));
|
|
}
|
|
|
|
TEST(SubtitleComposerTest, ReturnsConsistentColorSpace) {
|
|
const uint8_t kColorSpaceId = 2;
|
|
const uint16_t kObjectId = 5;
|
|
const uint16_t kRegionId = 1;
|
|
|
|
// Initially created in GetColorSpace.
|
|
{
|
|
SubtitleComposer composer;
|
|
auto* color_space = composer.GetColorSpace(kColorSpaceId);
|
|
ASSERT_TRUE(composer.SetRegionInfo(kRegionId, kColorSpaceId, 1, 1));
|
|
ASSERT_TRUE(composer.SetObjectInfo(kObjectId, kRegionId, 0, 0, kNoBgColor));
|
|
ASSERT_EQ(composer.GetColorSpace(kColorSpaceId), color_space);
|
|
ASSERT_EQ(composer.GetColorSpaceForObject(kObjectId), color_space);
|
|
}
|
|
|
|
// Initially created in SetRegionInfo.
|
|
{
|
|
SubtitleComposer composer;
|
|
ASSERT_TRUE(composer.SetRegionInfo(kRegionId, kColorSpaceId, 1, 1));
|
|
ASSERT_TRUE(composer.SetObjectInfo(kObjectId, kRegionId, 0, 0, kNoBgColor));
|
|
auto* color_space = composer.GetColorSpace(kColorSpaceId);
|
|
ASSERT_EQ(composer.GetColorSpace(kColorSpaceId), color_space);
|
|
ASSERT_EQ(composer.GetColorSpaceForObject(kObjectId), color_space);
|
|
}
|
|
}
|
|
|
|
TEST(SubtitleComposerTest, ClearObjects) {
|
|
const uint8_t kColorSpaceId = 2;
|
|
const uint16_t kObjectId1 = 5;
|
|
const uint16_t kObjectId2 = 6;
|
|
const uint16_t kRegionId = 1;
|
|
|
|
SubtitleComposer composer;
|
|
ASSERT_TRUE(composer.SetRegionInfo(kRegionId, kColorSpaceId, 10, 10));
|
|
ASSERT_TRUE(composer.SetObjectInfo(kObjectId1, kRegionId, 0, 0, kNoBgColor));
|
|
CreateDefaultImage(&composer, kObjectId1);
|
|
|
|
std::vector<std::shared_ptr<TextSample>> samples;
|
|
ASSERT_TRUE(composer.GetSamples(0, 1, &samples));
|
|
EXPECT_EQ(samples.size(), 1u);
|
|
|
|
composer.ClearObjects();
|
|
samples.clear();
|
|
|
|
// Should clear regions too.
|
|
ASSERT_FALSE(composer.SetObjectInfo(kObjectId2, kRegionId, 3, 3, kNoBgColor));
|
|
ASSERT_TRUE(composer.SetRegionInfo(kRegionId, kColorSpaceId, 10, 10));
|
|
ASSERT_TRUE(composer.SetObjectInfo(kObjectId2, kRegionId, 3, 3, kNoBgColor));
|
|
CreateDefaultImage(&composer, kObjectId2);
|
|
|
|
ASSERT_TRUE(composer.GetSamples(0, 1, &samples));
|
|
EXPECT_EQ(samples.size(), 1u);
|
|
}
|
|
|
|
TEST(SubtitleComposerTest, IgnoresEmptyImages) {
|
|
const uint8_t kColorSpaceId = 1;
|
|
const uint16_t kRegionId = 1;
|
|
const uint16_t kObjectId1 = 2;
|
|
const uint16_t kObjectId2 = 3;
|
|
const uint16_t kObjectId3 = 4;
|
|
const uint8_t kColorId = 10;
|
|
|
|
SubtitleComposer composer;
|
|
ASSERT_TRUE(composer.SetRegionInfo(kRegionId, kColorSpaceId, 10, 10));
|
|
ASSERT_TRUE(composer.SetObjectInfo(kObjectId1, kRegionId, 0, 0, kNoBgColor));
|
|
ASSERT_TRUE(composer.SetObjectInfo(kObjectId2, kRegionId, 5, 0, kNoBgColor));
|
|
ASSERT_TRUE(composer.SetObjectInfo(kObjectId3, kRegionId, 0, 5, kNoBgColor));
|
|
// Leave kObjectId1 with nothing.
|
|
CreateDefaultImage(&composer, kObjectId2);
|
|
{
|
|
// Add a transparent color.
|
|
auto* color_space = composer.GetColorSpace(kColorSpaceId);
|
|
color_space->SetColor(BitDepth::k8Bit, kColorId, RgbaColor{0, 0, 0, 0});
|
|
|
|
auto* image = composer.GetObjectImage(kObjectId3);
|
|
EXPECT_TRUE(image->AddPixel(BitDepth::k8Bit, kColorId, true));
|
|
image->NewRow(true);
|
|
}
|
|
|
|
std::vector<std::shared_ptr<TextSample>> samples;
|
|
ASSERT_TRUE(composer.GetSamples(0, 1, &samples));
|
|
EXPECT_EQ(samples.size(), 1u);
|
|
}
|
|
|
|
} // namespace media
|
|
} // namespace shaka
|