80 lines
2.3 KiB
C++
80 lines
2.3 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
|
|
|
|
#ifndef PACKAGER_MEDIA_DVB_SUBTITLE_COMPOSER_H_
|
|
#define PACKAGER_MEDIA_DVB_SUBTITLE_COMPOSER_H_
|
|
|
|
#include <memory>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
#include "packager/base/macros.h"
|
|
#include "packager/media/base/text_sample.h"
|
|
#include "packager/media/formats/dvb/dvb_image.h"
|
|
|
|
namespace shaka {
|
|
namespace media {
|
|
|
|
/// Holds pixel/caption data for a single DVB-sub page. This composes
|
|
/// multiple objects and creates TextSample objects from it.
|
|
class SubtitleComposer {
|
|
public:
|
|
SubtitleComposer();
|
|
~SubtitleComposer();
|
|
|
|
DISALLOW_COPY_AND_ASSIGN(SubtitleComposer);
|
|
|
|
void SetDisplaySize(uint16_t width, uint16_t height);
|
|
bool SetRegionPosition(uint8_t region_id, uint16_t x, uint16_t y);
|
|
bool SetRegionInfo(uint8_t region_id,
|
|
uint8_t color_space_id,
|
|
uint16_t width,
|
|
uint16_t height);
|
|
bool SetObjectInfo(uint16_t object_id,
|
|
uint8_t region_id,
|
|
uint16_t x,
|
|
uint16_t y,
|
|
int default_color_code);
|
|
|
|
DvbImageColorSpace* GetColorSpace(uint8_t color_space_id);
|
|
DvbImageColorSpace* GetColorSpaceForObject(uint16_t object_id);
|
|
DvbImageBuilder* GetObjectImage(uint16_t object_id);
|
|
|
|
bool GetSamples(int64_t start,
|
|
int64_t end,
|
|
std::vector<std::shared_ptr<TextSample>>* samples) const;
|
|
void ClearObjects();
|
|
|
|
private:
|
|
struct RegionInfo {
|
|
DvbImageColorSpace* color_space = nullptr;
|
|
uint16_t x = 0;
|
|
uint16_t y = 0;
|
|
uint16_t width = 0;
|
|
uint16_t height = 0;
|
|
};
|
|
|
|
struct ObjectInfo {
|
|
RegionInfo* region = nullptr;
|
|
int default_color_code = -1;
|
|
uint16_t x = 0;
|
|
uint16_t y = 0;
|
|
};
|
|
|
|
// Maps of IDs to their respective object.
|
|
std::unordered_map<uint8_t, RegionInfo> regions_;
|
|
std::unordered_map<uint8_t, DvbImageColorSpace> color_spaces_;
|
|
std::unordered_map<uint16_t, ObjectInfo> objects_;
|
|
std::unordered_map<uint16_t, DvbImageBuilder> images_; // Uses object_id.
|
|
uint16_t display_width_;
|
|
uint16_t display_height_;
|
|
};
|
|
|
|
} // namespace media
|
|
} // namespace shaka
|
|
|
|
#endif // PACKAGER_MEDIA_DVB_SUBTITLE_COMPOSER_H_
|