mirror of
https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs.git
synced 2024-11-25 21:11:00 +00:00
closedcaption: add cea608tocea708 element
Implement an element that can take an input 608 caption stream and generate a valid 708 caption stream by parsing the 608 data and generating the equivalent DTVCCPackets and Service blocks. Part-of: <https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs/-/merge_requests/1112>
This commit is contained in:
parent
c0dc6eb35c
commit
a8b46f1bf4
7 changed files with 1300 additions and 0 deletions
|
@ -4600,6 +4600,31 @@
|
||||||
},
|
},
|
||||||
"rank": "primary"
|
"rank": "primary"
|
||||||
},
|
},
|
||||||
|
"cea608tocea708": {
|
||||||
|
"author": "Matthew Waters <matthew@centricular.com>",
|
||||||
|
"description": "Converts CEA-608 Closed Captions to CEA-708 Closed Captions",
|
||||||
|
"hierarchy": [
|
||||||
|
"GstCea608ToCea708",
|
||||||
|
"GstElement",
|
||||||
|
"GstObject",
|
||||||
|
"GInitiallyUnowned",
|
||||||
|
"GObject"
|
||||||
|
],
|
||||||
|
"klass": "Converter",
|
||||||
|
"pad-templates": {
|
||||||
|
"sink": {
|
||||||
|
"caps": "closedcaption/x-cea-608:\n format: s334-1a\nclosedcaption/x-cea-608:\n format: raw\n field: { (int)0, (int)1 }\n",
|
||||||
|
"direction": "sink",
|
||||||
|
"presence": "always"
|
||||||
|
},
|
||||||
|
"src": {
|
||||||
|
"caps": "closedcaption/x-cea-708:\n format: cc_data\n",
|
||||||
|
"direction": "src",
|
||||||
|
"presence": "always"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"rank": "none"
|
||||||
|
},
|
||||||
"cea608tojson": {
|
"cea608tojson": {
|
||||||
"author": "Mathieu Duponchelle <mathieu@centricular.com>",
|
"author": "Mathieu Duponchelle <mathieu@centricular.com>",
|
||||||
"description": "Converts CEA-608 Closed Captions to JSON",
|
"description": "Converts CEA-608 Closed Captions to JSON",
|
||||||
|
|
|
@ -22,6 +22,7 @@ pangocairo = { git = "https://github.com/gtk-rs/gtk-rs-core" }
|
||||||
byteorder = "1"
|
byteorder = "1"
|
||||||
serde = { version = "1.0", features = ["derive"] }
|
serde = { version = "1.0", features = ["derive"] }
|
||||||
serde_json = { version = "1.0", features = ["raw_value"] }
|
serde_json = { version = "1.0", features = ["raw_value"] }
|
||||||
|
cea708-types = "0.1"
|
||||||
|
|
||||||
[dependencies.gst]
|
[dependencies.gst]
|
||||||
git = "https://gitlab.freedesktop.org/gstreamer/gstreamer-rs"
|
git = "https://gitlab.freedesktop.org/gstreamer/gstreamer-rs"
|
||||||
|
|
201
video/closedcaption/src/cea608tocea708/fmt.rs
Normal file
201
video/closedcaption/src/cea608tocea708/fmt.rs
Normal file
|
@ -0,0 +1,201 @@
|
||||||
|
// Copyright (C) 2023 Matthew Waters <matthew@centricular.com>
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public License, v2.0.
|
||||||
|
// If a copy of the MPL was not distributed with this file, You can obtain one at
|
||||||
|
// <https://mozilla.org/MPL/2.0/>.
|
||||||
|
//
|
||||||
|
// SPDX-License-Identifier: MPL-2.0
|
||||||
|
|
||||||
|
use cea708_types::{tables::*, Service};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub enum WriteError {
|
||||||
|
// returns the number of characters/bytes written
|
||||||
|
WouldOverflow(usize),
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) struct Cea708ServiceWriter {
|
||||||
|
service: Option<Service>,
|
||||||
|
service_no: u8,
|
||||||
|
active_window: WindowBits,
|
||||||
|
hidden_window: WindowBits,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Cea708ServiceWriter {
|
||||||
|
pub fn new(service_no: u8) -> Self {
|
||||||
|
Self {
|
||||||
|
service: None,
|
||||||
|
service_no,
|
||||||
|
active_window: WindowBits::ZERO,
|
||||||
|
hidden_window: WindowBits::ONE,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn ensure_service(&mut self) {
|
||||||
|
if self.service.is_none() {
|
||||||
|
self.service = Some(Service::new(self.service_no));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn take_service(&mut self) -> Option<Service> {
|
||||||
|
self.service.take()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn popon_preamble(&mut self) -> Result<usize, WriteError> {
|
||||||
|
gst::trace!(super::imp::CAT, "popon_preamble");
|
||||||
|
let window = match self.hidden_window {
|
||||||
|
// switch up the newly defined window
|
||||||
|
WindowBits::ZERO => 0,
|
||||||
|
WindowBits::ONE => 1,
|
||||||
|
_ => unreachable!(),
|
||||||
|
};
|
||||||
|
let args = DefineWindowArgs::new(
|
||||||
|
window,
|
||||||
|
0,
|
||||||
|
Anchor::BottomMiddle,
|
||||||
|
false,
|
||||||
|
70,
|
||||||
|
105,
|
||||||
|
14,
|
||||||
|
31,
|
||||||
|
true,
|
||||||
|
true,
|
||||||
|
false,
|
||||||
|
1,
|
||||||
|
1,
|
||||||
|
);
|
||||||
|
gst::trace!(super::imp::CAT, "active window {:?}", self.active_window);
|
||||||
|
let codes = [
|
||||||
|
Code::DeleteWindows(!self.active_window),
|
||||||
|
Code::DefineWindow(args),
|
||||||
|
];
|
||||||
|
self.push_codes(&codes)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn clear_current_window(&mut self) -> Result<usize, WriteError> {
|
||||||
|
gst::trace!(
|
||||||
|
super::imp::CAT,
|
||||||
|
"clear_current_window {:?}",
|
||||||
|
self.active_window
|
||||||
|
);
|
||||||
|
self.push_codes(&[Code::ClearWindows(self.active_window)])
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn clear_hidden_window(&mut self) -> Result<usize, WriteError> {
|
||||||
|
gst::trace!(super::imp::CAT, "clear_hidden_window");
|
||||||
|
self.push_codes(&[Code::ClearWindows(self.hidden_window)])
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn end_of_caption(&mut self) -> Result<usize, WriteError> {
|
||||||
|
gst::trace!(super::imp::CAT, "end_of_caption");
|
||||||
|
let ret =
|
||||||
|
self.push_codes(&[Code::ToggleWindows(self.active_window | self.hidden_window)])?;
|
||||||
|
std::mem::swap(&mut self.active_window, &mut self.hidden_window);
|
||||||
|
gst::trace!(super::imp::CAT, "active window {:?}", self.active_window);
|
||||||
|
Ok(ret)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn paint_on_preamble(&mut self) -> Result<usize, WriteError> {
|
||||||
|
gst::trace!(super::imp::CAT, "paint_on_preamble");
|
||||||
|
let window = match self.active_window {
|
||||||
|
WindowBits::ZERO => 0,
|
||||||
|
WindowBits::ONE => 1,
|
||||||
|
_ => unreachable!(),
|
||||||
|
};
|
||||||
|
self.push_codes(&[
|
||||||
|
// FIXME: assumes positioning in a 16:9 ratio
|
||||||
|
Code::DefineWindow(DefineWindowArgs::new(
|
||||||
|
window,
|
||||||
|
0,
|
||||||
|
Anchor::BottomMiddle,
|
||||||
|
false,
|
||||||
|
70,
|
||||||
|
105,
|
||||||
|
14,
|
||||||
|
31,
|
||||||
|
true,
|
||||||
|
true,
|
||||||
|
true,
|
||||||
|
1,
|
||||||
|
1,
|
||||||
|
)),
|
||||||
|
])
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn rollup_preamble(&mut self, rollup_count: u8, base_row: u8) -> Result<usize, WriteError> {
|
||||||
|
let base_row = std::cmp::max(rollup_count, base_row);
|
||||||
|
let anchor_vertical = (base_row as u32 * 100 / 15) as u8;
|
||||||
|
gst::trace!(
|
||||||
|
super::imp::CAT,
|
||||||
|
"rollup_preamble base {base_row} count {rollup_count}, anchor-v {anchor_vertical}"
|
||||||
|
);
|
||||||
|
let codes = [
|
||||||
|
Code::DeleteWindows(!WindowBits::ZERO),
|
||||||
|
Code::DefineWindow(DefineWindowArgs::new(
|
||||||
|
0,
|
||||||
|
0,
|
||||||
|
Anchor::BottomMiddle,
|
||||||
|
true,
|
||||||
|
anchor_vertical,
|
||||||
|
50,
|
||||||
|
rollup_count - 1,
|
||||||
|
31,
|
||||||
|
true,
|
||||||
|
true,
|
||||||
|
true,
|
||||||
|
1,
|
||||||
|
1,
|
||||||
|
)),
|
||||||
|
Code::SetPenLocation(SetPenLocationArgs::new(rollup_count - 1, 0)),
|
||||||
|
];
|
||||||
|
self.active_window = WindowBits::ZERO;
|
||||||
|
self.hidden_window = WindowBits::ONE;
|
||||||
|
self.push_codes(&codes)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn write_char(&mut self, c: char) -> Result<usize, WriteError> {
|
||||||
|
if let Some(code) = Code::from_char(c) {
|
||||||
|
self.push_codes(&[code])
|
||||||
|
} else {
|
||||||
|
Ok(0)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn push_codes(&mut self, codes: &[Code]) -> Result<usize, WriteError> {
|
||||||
|
self.ensure_service();
|
||||||
|
let service = self.service.as_mut().unwrap();
|
||||||
|
let start_len = service.len();
|
||||||
|
if service.free_space() < codes.iter().map(|c| c.byte_len()).sum::<usize>() {
|
||||||
|
return Err(WriteError::WouldOverflow(0));
|
||||||
|
}
|
||||||
|
for code in codes.iter() {
|
||||||
|
gst::trace!(
|
||||||
|
super::imp::CAT,
|
||||||
|
"pushing for service:{} code: {code:?}",
|
||||||
|
service.number()
|
||||||
|
);
|
||||||
|
service.push_code(code).unwrap();
|
||||||
|
}
|
||||||
|
Ok(service.len() - start_len)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn etx(&mut self) -> Result<usize, WriteError> {
|
||||||
|
self.push_codes(&[Code::ETX])
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn carriage_return(&mut self) -> Result<usize, WriteError> {
|
||||||
|
self.push_codes(&[Code::CR])
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_pen_attributes(&mut self, args: SetPenAttributesArgs) -> Result<usize, WriteError> {
|
||||||
|
self.push_codes(&[Code::SetPenAttributes(args)])
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_pen_location(&mut self, args: SetPenLocationArgs) -> Result<usize, WriteError> {
|
||||||
|
self.push_codes(&[Code::SetPenLocation(args)])
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn set_pen_color(&mut self, args: SetPenColorArgs) -> Result<usize, WriteError> {
|
||||||
|
self.push_codes(&[Code::SetPenColor(args)])
|
||||||
|
}
|
||||||
|
}
|
846
video/closedcaption/src/cea608tocea708/imp.rs
Normal file
846
video/closedcaption/src/cea608tocea708/imp.rs
Normal file
|
@ -0,0 +1,846 @@
|
||||||
|
// Copyright (C) 2023 Matthew Waters <matthew@centricular.com>
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public License, v2.0.
|
||||||
|
// If a copy of the MPL was not distributed with this file, You can obtain one at
|
||||||
|
// <https://mozilla.org/MPL/2.0/>.
|
||||||
|
//
|
||||||
|
// SPDX-License-Identifier: MPL-2.0
|
||||||
|
|
||||||
|
use cea708_types::{tables::*, DTVCCPacket};
|
||||||
|
use gst::glib;
|
||||||
|
use gst::prelude::*;
|
||||||
|
use gst::subclass::prelude::*;
|
||||||
|
|
||||||
|
use atomic_refcell::AtomicRefCell;
|
||||||
|
|
||||||
|
use crate::cea608tocea708::fmt::Cea708ServiceWriter;
|
||||||
|
use crate::cea608utils::*;
|
||||||
|
|
||||||
|
use once_cell::sync::Lazy;
|
||||||
|
|
||||||
|
#[derive(Debug, Copy, Clone)]
|
||||||
|
enum Cea608Format {
|
||||||
|
S334_1A,
|
||||||
|
RawField0,
|
||||||
|
RawField1,
|
||||||
|
}
|
||||||
|
|
||||||
|
struct Cea608State {
|
||||||
|
tracker: [Cea608StateTracker; 2],
|
||||||
|
format: Cea608Format,
|
||||||
|
service: [Cea608ServiceState; 4],
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for Cea608State {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self {
|
||||||
|
tracker: [Cea608StateTracker::default(), Cea608StateTracker::default()],
|
||||||
|
format: Cea608Format::RawField0,
|
||||||
|
service: [
|
||||||
|
Cea608ServiceState::default(),
|
||||||
|
Cea608ServiceState::default(),
|
||||||
|
Cea608ServiceState::default(),
|
||||||
|
Cea608ServiceState::default(),
|
||||||
|
],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct Cea608ServiceState {
|
||||||
|
mode: Option<Cea608Mode>,
|
||||||
|
base_row: u8,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for Cea608ServiceState {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self {
|
||||||
|
mode: None,
|
||||||
|
base_row: 15,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct Cea708ServiceState {
|
||||||
|
writer: Cea708ServiceWriter,
|
||||||
|
pen_location: SetPenLocationArgs,
|
||||||
|
pen_color: SetPenColorArgs,
|
||||||
|
pen_attributes: SetPenAttributesArgs,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn textstyle_foreground_color(style: TextStyle) -> Color {
|
||||||
|
match style {
|
||||||
|
TextStyle::Red => Color {
|
||||||
|
r: ColorValue::Full,
|
||||||
|
g: ColorValue::None,
|
||||||
|
b: ColorValue::None,
|
||||||
|
},
|
||||||
|
TextStyle::Green => Color {
|
||||||
|
r: ColorValue::None,
|
||||||
|
g: ColorValue::Full,
|
||||||
|
b: ColorValue::None,
|
||||||
|
},
|
||||||
|
TextStyle::Blue => Color {
|
||||||
|
r: ColorValue::None,
|
||||||
|
g: ColorValue::None,
|
||||||
|
b: ColorValue::Full,
|
||||||
|
},
|
||||||
|
TextStyle::Cyan => Color {
|
||||||
|
r: ColorValue::None,
|
||||||
|
g: ColorValue::Full,
|
||||||
|
b: ColorValue::Full,
|
||||||
|
},
|
||||||
|
TextStyle::Yellow => Color {
|
||||||
|
r: ColorValue::Full,
|
||||||
|
g: ColorValue::Full,
|
||||||
|
b: ColorValue::None,
|
||||||
|
},
|
||||||
|
TextStyle::Magenta => Color {
|
||||||
|
r: ColorValue::Full,
|
||||||
|
g: ColorValue::None,
|
||||||
|
b: ColorValue::Full,
|
||||||
|
},
|
||||||
|
TextStyle::White | TextStyle::ItalicWhite => Color {
|
||||||
|
r: ColorValue::Full,
|
||||||
|
g: ColorValue::Full,
|
||||||
|
b: ColorValue::Full,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn textstyle_to_pen_color(style: TextStyle) -> SetPenColorArgs {
|
||||||
|
let black = Color {
|
||||||
|
r: ColorValue::None,
|
||||||
|
g: ColorValue::None,
|
||||||
|
b: ColorValue::None,
|
||||||
|
};
|
||||||
|
SetPenColorArgs {
|
||||||
|
foreground_color: textstyle_foreground_color(style),
|
||||||
|
foreground_opacity: Opacity::Solid,
|
||||||
|
background_color: black,
|
||||||
|
background_opacity: Opacity::Solid,
|
||||||
|
edge_color: black,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn textstyle_is_italics(style: TextStyle) -> bool {
|
||||||
|
style == TextStyle::ItalicWhite
|
||||||
|
}
|
||||||
|
|
||||||
|
fn cea608_mode_visible_rows(mode: Cea608Mode) -> u8 {
|
||||||
|
match mode {
|
||||||
|
Cea608Mode::RollUp2 => 2,
|
||||||
|
Cea608Mode::RollUp3 => 3,
|
||||||
|
Cea608Mode::RollUp4 => 4,
|
||||||
|
_ => unreachable!(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Cea708ServiceState {
|
||||||
|
fn new(service_no: u8) -> Self {
|
||||||
|
Self {
|
||||||
|
writer: Cea708ServiceWriter::new(service_no),
|
||||||
|
pen_location: SetPenLocationArgs::new(0, 0),
|
||||||
|
pen_color: textstyle_to_pen_color(TextStyle::White),
|
||||||
|
pen_attributes: SetPenAttributesArgs::new(
|
||||||
|
PenSize::Standard,
|
||||||
|
FontStyle::Default,
|
||||||
|
TextTag::Dialog,
|
||||||
|
TextOffset::Normal,
|
||||||
|
false,
|
||||||
|
false,
|
||||||
|
EdgeType::None,
|
||||||
|
),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn new_mode(&mut self, cea608_mode: Cea608Mode, base_row: u8) {
|
||||||
|
let new_row = if cea608_mode.is_rollup() {
|
||||||
|
cea608_mode_visible_rows(cea608_mode) - 1
|
||||||
|
} else {
|
||||||
|
0
|
||||||
|
};
|
||||||
|
match cea608_mode {
|
||||||
|
Cea608Mode::PopOn => self.writer.popon_preamble().unwrap(),
|
||||||
|
Cea608Mode::PaintOn => self.writer.paint_on_preamble().unwrap(),
|
||||||
|
Cea608Mode::RollUp2 => self.writer.rollup_preamble(2, base_row).unwrap(),
|
||||||
|
Cea608Mode::RollUp3 => self.writer.rollup_preamble(3, base_row).unwrap(),
|
||||||
|
Cea608Mode::RollUp4 => self.writer.rollup_preamble(4, base_row).unwrap(),
|
||||||
|
};
|
||||||
|
// we have redefined then window so all the attributes have been reset
|
||||||
|
self.pen_location.row = new_row;
|
||||||
|
self.pen_location.column = 0;
|
||||||
|
self.pen_color = textstyle_to_pen_color(TextStyle::White);
|
||||||
|
self.pen_attributes.underline = false;
|
||||||
|
self.pen_attributes.italics = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
fn handle_text(&mut self, text: Cea608Text) {
|
||||||
|
if text.code_space == CodeSpace::WestEU {
|
||||||
|
self.writer.push_codes(&[Code::BS]).unwrap();
|
||||||
|
}
|
||||||
|
if let Some(c) = text.char1 {
|
||||||
|
if self.pen_location.column > 31 {
|
||||||
|
self.writer.push_codes(&[Code::BS]).unwrap();
|
||||||
|
}
|
||||||
|
self.writer.write_char(c).unwrap();
|
||||||
|
self.pen_location.column = std::cmp::min(self.pen_location.column + 1, 32);
|
||||||
|
}
|
||||||
|
if let Some(c) = text.char2 {
|
||||||
|
if self.pen_location.column > 31 {
|
||||||
|
self.writer.push_codes(&[Code::BS]).unwrap();
|
||||||
|
}
|
||||||
|
self.writer.write_char(c).unwrap();
|
||||||
|
self.pen_location.column = std::cmp::min(self.pen_location.column + 1, 32);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn handle_preamble(&mut self, preamble: Preamble) {
|
||||||
|
let mut need_pen_location = false;
|
||||||
|
// TODO: may need a better algorithm then compressing the top four rows
|
||||||
|
let new_row = std::cmp::max(0, preamble.row) as u8;
|
||||||
|
if self.pen_location.row != new_row {
|
||||||
|
need_pen_location = true;
|
||||||
|
self.pen_location.row = new_row;
|
||||||
|
}
|
||||||
|
|
||||||
|
if self.pen_location.column != preamble.col as u8 {
|
||||||
|
need_pen_location = true;
|
||||||
|
self.pen_location.column = preamble.col as u8;
|
||||||
|
}
|
||||||
|
|
||||||
|
if need_pen_location {
|
||||||
|
self.writer.set_pen_location(self.pen_location).unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut need_pen_attributes = false;
|
||||||
|
if self.pen_attributes.italics != textstyle_is_italics(preamble.style) {
|
||||||
|
need_pen_attributes = true;
|
||||||
|
self.pen_attributes.italics = textstyle_is_italics(preamble.style);
|
||||||
|
}
|
||||||
|
|
||||||
|
if self.pen_attributes.underline != (preamble.underline > 0) {
|
||||||
|
need_pen_attributes = true;
|
||||||
|
self.pen_attributes.underline = preamble.underline > 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
if need_pen_attributes {
|
||||||
|
self.writer.set_pen_attributes(self.pen_attributes).unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
if self.pen_color.foreground_color != textstyle_foreground_color(preamble.style) {
|
||||||
|
self.pen_color.foreground_color = textstyle_foreground_color(preamble.style);
|
||||||
|
self.writer.set_pen_color(self.pen_color).unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn handle_midrowchange(&mut self, midrowchange: MidRowChange) {
|
||||||
|
self.writer.write_char(' ').unwrap();
|
||||||
|
if self.pen_color.foreground_color != textstyle_foreground_color(midrowchange.style) {
|
||||||
|
self.pen_color.foreground_color = textstyle_foreground_color(midrowchange.style);
|
||||||
|
self.writer.set_pen_color(self.pen_color).unwrap();
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut need_pen_attributes = false;
|
||||||
|
if self.pen_attributes.italics != textstyle_is_italics(midrowchange.style) {
|
||||||
|
need_pen_attributes = true;
|
||||||
|
self.pen_attributes.italics = textstyle_is_italics(midrowchange.style);
|
||||||
|
}
|
||||||
|
|
||||||
|
if self.pen_attributes.underline != midrowchange.underline {
|
||||||
|
need_pen_attributes = true;
|
||||||
|
self.pen_attributes.underline = midrowchange.underline;
|
||||||
|
}
|
||||||
|
|
||||||
|
if need_pen_attributes {
|
||||||
|
self.writer.set_pen_attributes(self.pen_attributes).unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn carriage_return(&mut self) {
|
||||||
|
self.writer.carriage_return().unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct Cea708State {
|
||||||
|
packet_counter: u8,
|
||||||
|
service_state: [Cea708ServiceState; 4],
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Cea708State {
|
||||||
|
fn take_buffer(&mut self, s334_1a_data: &[u8]) -> Option<gst::Buffer> {
|
||||||
|
let mut packet = DTVCCPacket::new(self.packet_counter);
|
||||||
|
self.packet_counter += 1;
|
||||||
|
self.packet_counter &= 0x3;
|
||||||
|
|
||||||
|
for state in self.service_state.iter_mut() {
|
||||||
|
if let Some(service) = state.writer.take_service() {
|
||||||
|
if let Err(e) = packet.push_service(service.clone()) {
|
||||||
|
gst::warning!(
|
||||||
|
CAT,
|
||||||
|
"failed to add service:{} to outgoing packet: {:?}",
|
||||||
|
service.number(),
|
||||||
|
e
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut cc_data = Vec::with_capacity(64);
|
||||||
|
assert!(s334_1a_data.len() % 3 == 0);
|
||||||
|
for triple in s334_1a_data.chunks(3) {
|
||||||
|
if (triple[0] & 0x80) > 0 {
|
||||||
|
cc_data.push(0xfd);
|
||||||
|
} else {
|
||||||
|
cc_data.push(0xfc);
|
||||||
|
}
|
||||||
|
cc_data.extend(&triple[1..]);
|
||||||
|
}
|
||||||
|
let mut ccp_data = Vec::with_capacity(128);
|
||||||
|
packet.write_cc_data(&mut ccp_data).ok()?;
|
||||||
|
gst::trace!(
|
||||||
|
CAT,
|
||||||
|
"take_buffer produced cc_data_len:{} cc_data:{cc_data:?}",
|
||||||
|
cc_data.len()
|
||||||
|
);
|
||||||
|
// ignore the 2 byte cc_data header that is unused in GStreamer
|
||||||
|
if ccp_data.len() > 2 {
|
||||||
|
cc_data.extend(&ccp_data[2..]);
|
||||||
|
} else if cc_data.is_empty() {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
Some(gst::Buffer::from_slice(cc_data))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct State {
|
||||||
|
cea608: Cea608State,
|
||||||
|
cea708: Cea708State,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for State {
|
||||||
|
fn default() -> Self {
|
||||||
|
State {
|
||||||
|
cea608: Cea608State::default(),
|
||||||
|
cea708: Cea708State {
|
||||||
|
packet_counter: 0,
|
||||||
|
service_state: [
|
||||||
|
Cea708ServiceState::new(1),
|
||||||
|
Cea708ServiceState::new(2),
|
||||||
|
Cea708ServiceState::new(3),
|
||||||
|
Cea708ServiceState::new(4),
|
||||||
|
],
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
enum BufferOrEvent {
|
||||||
|
Buffer(gst::Buffer),
|
||||||
|
Event(gst::Event),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl State {
|
||||||
|
fn field_channel_to_index(&self, field: u8, channel: i32) -> usize {
|
||||||
|
match (field, channel) {
|
||||||
|
(0, 0 | 2) => 0,
|
||||||
|
(0, 1 | 3) => 2,
|
||||||
|
(1, 0 | 2) => 1,
|
||||||
|
(1, 1 | 3) => 3,
|
||||||
|
_ => unreachable!(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn service_state_from_608_field_channel(
|
||||||
|
&mut self,
|
||||||
|
field: u8,
|
||||||
|
channel: i32,
|
||||||
|
) -> &mut Cea708ServiceState {
|
||||||
|
&mut self.cea708.service_state[self.field_channel_to_index(field, channel)]
|
||||||
|
}
|
||||||
|
|
||||||
|
fn new_mode(&mut self, field: u8, channel: i32, cea608_mode: Cea608Mode) {
|
||||||
|
let idx = self.field_channel_to_index(field, channel);
|
||||||
|
if let Some(old_mode) = self.cea608.service[idx].mode {
|
||||||
|
if cea608_mode.is_rollup()
|
||||||
|
&& matches!(old_mode, Cea608Mode::PopOn | Cea608Mode::PaintOn)
|
||||||
|
{
|
||||||
|
// https://www.law.cornell.edu/cfr/text/47/79.101 (f)(1)(x)
|
||||||
|
gst::trace!(CAT, "change to rollup from pop/paint-on");
|
||||||
|
self.cea708.service_state[idx]
|
||||||
|
.writer
|
||||||
|
.clear_hidden_window()
|
||||||
|
.unwrap();
|
||||||
|
self.cea708.service_state[idx]
|
||||||
|
.writer
|
||||||
|
.clear_current_window()
|
||||||
|
.unwrap();
|
||||||
|
self.cea608.service[idx].base_row = 15;
|
||||||
|
}
|
||||||
|
if old_mode.is_rollup() && cea608_mode.is_rollup() {
|
||||||
|
// https://www.law.cornell.edu/cfr/text/47/79.101 (f)(1)(x)
|
||||||
|
let old_count = cea608_mode_visible_rows(old_mode);
|
||||||
|
let new_count = cea608_mode_visible_rows(cea608_mode);
|
||||||
|
gst::trace!(
|
||||||
|
CAT,
|
||||||
|
"change of rollup row count from {old_count} to {new_count}",
|
||||||
|
);
|
||||||
|
if old_count > new_count {
|
||||||
|
// push the captions ot the top of the window before we shrink the size of the
|
||||||
|
// window
|
||||||
|
for _ in new_count..old_count {
|
||||||
|
self.cea708.service_state[idx]
|
||||||
|
.writer
|
||||||
|
.push_codes(&[Code::CR])
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
self.cea608.service[idx].mode = Some(cea608_mode);
|
||||||
|
let base_row = if cea608_mode.is_rollup() {
|
||||||
|
self.cea608.service[idx].base_row
|
||||||
|
} else {
|
||||||
|
0
|
||||||
|
};
|
||||||
|
self.cea708.service_state[idx].new_mode(cea608_mode, base_row);
|
||||||
|
}
|
||||||
|
|
||||||
|
fn handle_cc_data(&mut self, imp: &Cea608ToCea708, field: u8, cc_data: u16) {
|
||||||
|
self.cea608.tracker[field as usize].push_cc_data(cc_data);
|
||||||
|
|
||||||
|
let mut channel = None;
|
||||||
|
if let Some(cea608) = self.cea608.tracker[field as usize].pop() {
|
||||||
|
gst::trace!(
|
||||||
|
CAT,
|
||||||
|
imp: imp,
|
||||||
|
"have field:{field} channel:{} {cea608:?}",
|
||||||
|
cea608.channel()
|
||||||
|
);
|
||||||
|
if !matches!(cea608, Cea608::Duplicate) {
|
||||||
|
channel = Some(cea608.channel());
|
||||||
|
}
|
||||||
|
match cea608 {
|
||||||
|
Cea608::Duplicate => (),
|
||||||
|
Cea608::NewMode(chan, new_mode) => {
|
||||||
|
self.new_mode(field, chan, new_mode);
|
||||||
|
}
|
||||||
|
Cea608::Text(text) => {
|
||||||
|
let state = self.service_state_from_608_field_channel(field, text.chan);
|
||||||
|
state.handle_text(text);
|
||||||
|
}
|
||||||
|
Cea608::EndOfCaption(chan) => {
|
||||||
|
let state = self.service_state_from_608_field_channel(field, chan);
|
||||||
|
state.writer.end_of_caption().unwrap();
|
||||||
|
state.writer.etx().unwrap();
|
||||||
|
}
|
||||||
|
Cea608::Preamble(mut preamble) => {
|
||||||
|
let idx = self.field_channel_to_index(field, preamble.chan);
|
||||||
|
let rollup_count = self.cea608.service[idx]
|
||||||
|
.mode
|
||||||
|
.map(|mode| {
|
||||||
|
if mode.is_rollup() {
|
||||||
|
cea608_mode_visible_rows(mode)
|
||||||
|
} else {
|
||||||
|
0
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.unwrap_or(0);
|
||||||
|
if rollup_count > 0 {
|
||||||
|
// https://www.law.cornell.edu/cfr/text/47/79.101 (f)(1)(ii)
|
||||||
|
let old_base_row = self.cea608.service[idx].base_row;
|
||||||
|
self.cea608.service[idx].base_row = preamble.row as u8;
|
||||||
|
let state = self.service_state_from_608_field_channel(field, preamble.chan);
|
||||||
|
if old_base_row != preamble.row as u8 {
|
||||||
|
state
|
||||||
|
.writer
|
||||||
|
.rollup_preamble(rollup_count, preamble.row as u8)
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
state.pen_location.row = rollup_count - 1;
|
||||||
|
preamble.row = rollup_count as i32 - 1;
|
||||||
|
}
|
||||||
|
let state = self.service_state_from_608_field_channel(field, preamble.chan);
|
||||||
|
state.handle_preamble(preamble);
|
||||||
|
}
|
||||||
|
Cea608::MidRowChange(midrowchange) => {
|
||||||
|
let state = self.service_state_from_608_field_channel(field, midrowchange.chan);
|
||||||
|
state.handle_midrowchange(midrowchange);
|
||||||
|
}
|
||||||
|
Cea608::Backspace(chan) => {
|
||||||
|
let state = self.service_state_from_608_field_channel(field, chan);
|
||||||
|
// TODO: handle removing a midrowchange
|
||||||
|
state.pen_location.column = std::cmp::max(state.pen_location.column - 1, 0);
|
||||||
|
state
|
||||||
|
.writer
|
||||||
|
.push_codes(&[cea708_types::tables::Code::BS])
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
Cea608::CarriageReturn(chan) => {
|
||||||
|
if let Some(mode) =
|
||||||
|
self.cea608.service[self.field_channel_to_index(field, chan)].mode
|
||||||
|
{
|
||||||
|
if mode.is_rollup() {
|
||||||
|
let state = self.service_state_from_608_field_channel(field, chan);
|
||||||
|
state.carriage_return();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Cea608::EraseDisplay(chan) => {
|
||||||
|
let state = self.service_state_from_608_field_channel(field, chan);
|
||||||
|
state.writer.clear_current_window().unwrap();
|
||||||
|
}
|
||||||
|
Cea608::EraseNonDisplay(chan) => {
|
||||||
|
let state = self.service_state_from_608_field_channel(field, chan);
|
||||||
|
state.writer.clear_hidden_window().unwrap();
|
||||||
|
}
|
||||||
|
Cea608::TabOffset(chan, count) => {
|
||||||
|
let state = self.service_state_from_608_field_channel(field, chan);
|
||||||
|
state.pen_location.column =
|
||||||
|
std::cmp::min(state.pen_location.column + count as u8, 32);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if let Some(channel) = channel {
|
||||||
|
let idx = self.field_channel_to_index(field, channel);
|
||||||
|
if let Some(
|
||||||
|
Cea608Mode::RollUp2
|
||||||
|
| Cea608Mode::RollUp3
|
||||||
|
| Cea608Mode::RollUp4
|
||||||
|
| Cea608Mode::PaintOn,
|
||||||
|
) = self.cea608.service[idx].mode
|
||||||
|
{
|
||||||
|
// FIXME: actually track state for when things have changed
|
||||||
|
// and we need to send ETX
|
||||||
|
self.cea708.service_state[idx]
|
||||||
|
.writer
|
||||||
|
.push_codes(&[cea708_types::tables::Code::ETX])
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn take_buffer(
|
||||||
|
&mut self,
|
||||||
|
s334_1a_data: &[u8],
|
||||||
|
pts: gst::ClockTime,
|
||||||
|
duration: Option<gst::ClockTime>,
|
||||||
|
) -> BufferOrEvent {
|
||||||
|
if let Some(mut buffer) = self.cea708.take_buffer(s334_1a_data) {
|
||||||
|
{
|
||||||
|
let buffer_ref = buffer.get_mut().unwrap();
|
||||||
|
buffer_ref.set_pts(Some(pts));
|
||||||
|
buffer_ref.set_duration(duration);
|
||||||
|
}
|
||||||
|
BufferOrEvent::Buffer(buffer)
|
||||||
|
} else {
|
||||||
|
BufferOrEvent::Event(gst::event::Gap::builder(pts).duration(duration).build())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct Cea608ToCea708 {
|
||||||
|
srcpad: gst::Pad,
|
||||||
|
sinkpad: gst::Pad,
|
||||||
|
|
||||||
|
state: AtomicRefCell<State>,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) static CAT: Lazy<gst::DebugCategory> = Lazy::new(|| {
|
||||||
|
gst::DebugCategory::new(
|
||||||
|
"cea608tocea708",
|
||||||
|
gst::DebugColorFlags::empty(),
|
||||||
|
Some("CEA-608 to CEA-708 Element"),
|
||||||
|
)
|
||||||
|
});
|
||||||
|
|
||||||
|
impl Cea608ToCea708 {
|
||||||
|
fn sink_chain(
|
||||||
|
&self,
|
||||||
|
pad: &gst::Pad,
|
||||||
|
buffer: gst::Buffer,
|
||||||
|
) -> Result<gst::FlowSuccess, gst::FlowError> {
|
||||||
|
gst::log!(CAT, obj: pad, "Handling buffer {:?}", buffer);
|
||||||
|
|
||||||
|
let mut state = self.state.borrow_mut();
|
||||||
|
|
||||||
|
let buffer_pts = buffer.pts().ok_or_else(|| {
|
||||||
|
gst::error!(CAT, obj: pad, "Require timestamped buffers");
|
||||||
|
gst::FlowError::Error
|
||||||
|
})?;
|
||||||
|
|
||||||
|
let data = buffer.map_readable().map_err(|_| {
|
||||||
|
gst::error!(CAT, obj: pad, "Can't map buffer readable");
|
||||||
|
|
||||||
|
gst::FlowError::Error
|
||||||
|
})?;
|
||||||
|
let mut data_len = data.len();
|
||||||
|
|
||||||
|
let s334_1a_data = match state.cea608.format {
|
||||||
|
Cea608Format::S334_1A => {
|
||||||
|
if data_len % 3 != 0 {
|
||||||
|
gst::warning!(
|
||||||
|
CAT,
|
||||||
|
obj: pad,
|
||||||
|
"Invalid closed caption packet size, truncating"
|
||||||
|
);
|
||||||
|
data_len -= data_len % 3;
|
||||||
|
}
|
||||||
|
if data_len < 3 {
|
||||||
|
gst::warning!(
|
||||||
|
CAT,
|
||||||
|
obj: pad,
|
||||||
|
"Invalid closed caption packet size, dropping"
|
||||||
|
);
|
||||||
|
return Ok(gst::FlowSuccess::Ok);
|
||||||
|
}
|
||||||
|
for triple in data.chunks_exact(3) {
|
||||||
|
let cc_data = (triple[1] as u16) << 8 | triple[2] as u16;
|
||||||
|
let field = (triple[0] & 0x80) >> 7;
|
||||||
|
state.handle_cc_data(self, field, cc_data);
|
||||||
|
}
|
||||||
|
data.to_vec()
|
||||||
|
}
|
||||||
|
Cea608Format::RawField0 | Cea608Format::RawField1 => {
|
||||||
|
if data_len % 2 != 0 {
|
||||||
|
gst::warning!(
|
||||||
|
CAT,
|
||||||
|
obj: pad,
|
||||||
|
"Invalid closed caption packet size, truncating"
|
||||||
|
);
|
||||||
|
data_len -= data_len % 3;
|
||||||
|
}
|
||||||
|
if data_len < 2 {
|
||||||
|
gst::warning!(
|
||||||
|
CAT,
|
||||||
|
obj: pad,
|
||||||
|
"Invalid closed caption packet size, dropping"
|
||||||
|
);
|
||||||
|
return Ok(gst::FlowSuccess::Ok);
|
||||||
|
}
|
||||||
|
let field = match state.cea608.format {
|
||||||
|
Cea608Format::RawField0 => 0,
|
||||||
|
Cea608Format::RawField1 => 1,
|
||||||
|
_ => unreachable!(),
|
||||||
|
};
|
||||||
|
let mut s334_1a_data = Vec::with_capacity(data.len() / 2 * 3);
|
||||||
|
for pair in data.chunks_exact(2) {
|
||||||
|
let cc_data = (pair[0] as u16) << 8 | pair[1] as u16;
|
||||||
|
state.handle_cc_data(self, field, cc_data);
|
||||||
|
if field == 0 {
|
||||||
|
s334_1a_data.push(0x00);
|
||||||
|
} else {
|
||||||
|
s334_1a_data.push(0x80);
|
||||||
|
}
|
||||||
|
s334_1a_data.push(pair[0]);
|
||||||
|
s334_1a_data.push(pair[1]);
|
||||||
|
}
|
||||||
|
s334_1a_data
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
let buffer_or_event = state.take_buffer(&s334_1a_data, buffer_pts, buffer.duration());
|
||||||
|
drop(state);
|
||||||
|
|
||||||
|
match buffer_or_event {
|
||||||
|
BufferOrEvent::Buffer(buffer) => self.srcpad.push(buffer),
|
||||||
|
BufferOrEvent::Event(event) => {
|
||||||
|
self.srcpad.push_event(event);
|
||||||
|
Ok(gst::FlowSuccess::Ok)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn sink_event(&self, pad: &gst::Pad, event: gst::Event) -> bool {
|
||||||
|
use gst::EventView;
|
||||||
|
|
||||||
|
gst::log!(CAT, obj: pad, "Handling event {:?}", event);
|
||||||
|
match event.view() {
|
||||||
|
EventView::Caps(event) => {
|
||||||
|
let mut state = self.state.borrow_mut();
|
||||||
|
let caps = event.caps();
|
||||||
|
let structure = caps.structure(0).expect("Caps has no structure");
|
||||||
|
let framerate = structure.get::<gst::Fraction>("framerate").ok();
|
||||||
|
state.cea608.format = match structure.get::<&str>("format") {
|
||||||
|
Ok("raw") => {
|
||||||
|
if structure.has_field("field") {
|
||||||
|
match structure.get("field") {
|
||||||
|
Ok(0) => Cea608Format::RawField0,
|
||||||
|
Ok(1) => Cea608Format::RawField1,
|
||||||
|
_ => {
|
||||||
|
gst::error!(
|
||||||
|
CAT,
|
||||||
|
imp: self,
|
||||||
|
"unknown \'field\' value in caps, {caps:?}"
|
||||||
|
);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Cea608Format::RawField0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Ok("s334-1a") => Cea608Format::S334_1A,
|
||||||
|
v => {
|
||||||
|
gst::error!(
|
||||||
|
CAT,
|
||||||
|
imp: self,
|
||||||
|
"unknown or missing \'format\' value {v:?} in caps, {caps:?}"
|
||||||
|
);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
drop(state);
|
||||||
|
|
||||||
|
let mut caps_builder =
|
||||||
|
gst::Caps::builder("closedcaption/x-cea-708").field("format", "cc_data");
|
||||||
|
if let Some(framerate) = framerate {
|
||||||
|
caps_builder = caps_builder.field("framerate", framerate);
|
||||||
|
}
|
||||||
|
let new_event = gst::event::Caps::new(&caps_builder.build());
|
||||||
|
|
||||||
|
return self.srcpad.push_event(new_event);
|
||||||
|
}
|
||||||
|
EventView::FlushStop(..) => {
|
||||||
|
let mut state = self.state.borrow_mut();
|
||||||
|
let cea608_format = state.cea608.format;
|
||||||
|
*state = State::default();
|
||||||
|
state.cea608.format = cea608_format;
|
||||||
|
}
|
||||||
|
_ => (),
|
||||||
|
}
|
||||||
|
|
||||||
|
gst::Pad::event_default(pad, Some(&*self.obj()), event)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[glib::object_subclass]
|
||||||
|
impl ObjectSubclass for Cea608ToCea708 {
|
||||||
|
const NAME: &'static str = "GstCea608ToCea708";
|
||||||
|
type Type = super::Cea608ToCea708;
|
||||||
|
type ParentType = gst::Element;
|
||||||
|
|
||||||
|
fn with_class(klass: &Self::Class) -> Self {
|
||||||
|
let templ = klass.pad_template("sink").unwrap();
|
||||||
|
let sinkpad = gst::Pad::builder_with_template(&templ, Some("sink"))
|
||||||
|
.chain_function(|pad, parent, buffer| {
|
||||||
|
Cea608ToCea708::catch_panic_pad_function(
|
||||||
|
parent,
|
||||||
|
|| Err(gst::FlowError::Error),
|
||||||
|
|this| this.sink_chain(pad, buffer),
|
||||||
|
)
|
||||||
|
})
|
||||||
|
.event_function(|pad, parent, event| {
|
||||||
|
Cea608ToCea708::catch_panic_pad_function(
|
||||||
|
parent,
|
||||||
|
|| false,
|
||||||
|
|this| this.sink_event(pad, event),
|
||||||
|
)
|
||||||
|
})
|
||||||
|
.flags(gst::PadFlags::FIXED_CAPS)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
let templ = klass.pad_template("src").unwrap();
|
||||||
|
let srcpad = gst::Pad::builder_with_template(&templ, Some("src"))
|
||||||
|
.flags(gst::PadFlags::FIXED_CAPS)
|
||||||
|
.build();
|
||||||
|
|
||||||
|
Self {
|
||||||
|
srcpad,
|
||||||
|
sinkpad,
|
||||||
|
state: AtomicRefCell::new(State::default()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ObjectImpl for Cea608ToCea708 {
|
||||||
|
fn constructed(&self) {
|
||||||
|
self.parent_constructed();
|
||||||
|
|
||||||
|
let obj = self.obj();
|
||||||
|
obj.add_pad(&self.sinkpad).unwrap();
|
||||||
|
obj.add_pad(&self.srcpad).unwrap();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl GstObjectImpl for Cea608ToCea708 {}
|
||||||
|
|
||||||
|
impl ElementImpl for Cea608ToCea708 {
|
||||||
|
fn metadata() -> Option<&'static gst::subclass::ElementMetadata> {
|
||||||
|
static ELEMENT_METADATA: Lazy<gst::subclass::ElementMetadata> = Lazy::new(|| {
|
||||||
|
gst::subclass::ElementMetadata::new(
|
||||||
|
"CEA-608 to CEA-708",
|
||||||
|
"Converter",
|
||||||
|
"Converts CEA-608 Closed Captions to CEA-708 Closed Captions",
|
||||||
|
"Matthew Waters <matthew@centricular.com>",
|
||||||
|
)
|
||||||
|
});
|
||||||
|
|
||||||
|
Some(&*ELEMENT_METADATA)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn pad_templates() -> &'static [gst::PadTemplate] {
|
||||||
|
static PAD_TEMPLATES: Lazy<Vec<gst::PadTemplate>> = Lazy::new(|| {
|
||||||
|
let caps = gst::Caps::builder("closedcaption/x-cea-708")
|
||||||
|
.field("format", "cc_data")
|
||||||
|
.build();
|
||||||
|
|
||||||
|
let src_pad_template = gst::PadTemplate::new(
|
||||||
|
"src",
|
||||||
|
gst::PadDirection::Src,
|
||||||
|
gst::PadPresence::Always,
|
||||||
|
&caps,
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let sink_pad_template = gst::PadTemplate::new(
|
||||||
|
"sink",
|
||||||
|
gst::PadDirection::Sink,
|
||||||
|
gst::PadPresence::Always,
|
||||||
|
&[
|
||||||
|
gst::Structure::builder("closedcaption/x-cea-608")
|
||||||
|
.field("format", "s334-1a")
|
||||||
|
.build(),
|
||||||
|
gst::Structure::builder("closedcaption/x-cea-608")
|
||||||
|
.field("format", "raw")
|
||||||
|
.field("field", gst::List::new([0, 1]))
|
||||||
|
.build(),
|
||||||
|
]
|
||||||
|
.into_iter()
|
||||||
|
.collect::<gst::Caps>(),
|
||||||
|
)
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
vec![src_pad_template, sink_pad_template]
|
||||||
|
});
|
||||||
|
|
||||||
|
PAD_TEMPLATES.as_ref()
|
||||||
|
}
|
||||||
|
|
||||||
|
#[allow(clippy::single_match)]
|
||||||
|
fn change_state(
|
||||||
|
&self,
|
||||||
|
transition: gst::StateChange,
|
||||||
|
) -> Result<gst::StateChangeSuccess, gst::StateChangeError> {
|
||||||
|
gst::trace!(CAT, imp: self, "Changing state {:?}", transition);
|
||||||
|
|
||||||
|
match transition {
|
||||||
|
gst::StateChange::ReadyToPaused => {
|
||||||
|
let mut state = self.state.borrow_mut();
|
||||||
|
*state = State::default();
|
||||||
|
}
|
||||||
|
_ => (),
|
||||||
|
}
|
||||||
|
|
||||||
|
let ret = self.parent_change_state(transition)?;
|
||||||
|
|
||||||
|
match transition {
|
||||||
|
gst::StateChange::PausedToReady => {
|
||||||
|
let mut state = self.state.borrow_mut();
|
||||||
|
*state = State::default();
|
||||||
|
}
|
||||||
|
_ => (),
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(ret)
|
||||||
|
}
|
||||||
|
}
|
26
video/closedcaption/src/cea608tocea708/mod.rs
Normal file
26
video/closedcaption/src/cea608tocea708/mod.rs
Normal file
|
@ -0,0 +1,26 @@
|
||||||
|
// Copyright (C) 2023 Matthew Waters <matthew@centricular.com>
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public License, v2.0.
|
||||||
|
// If a copy of the MPL was not distributed with this file, You can obtain one at
|
||||||
|
// <https://mozilla.org/MPL/2.0/>.
|
||||||
|
//
|
||||||
|
// SPDX-License-Identifier: MPL-2.0
|
||||||
|
|
||||||
|
use gst::glib;
|
||||||
|
use gst::prelude::*;
|
||||||
|
|
||||||
|
mod fmt;
|
||||||
|
mod imp;
|
||||||
|
|
||||||
|
glib::wrapper! {
|
||||||
|
pub struct Cea608ToCea708(ObjectSubclass<imp::Cea608ToCea708>) @extends gst::Element, gst::Object;
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn register(plugin: &gst::Plugin) -> Result<(), glib::BoolError> {
|
||||||
|
gst::Element::register(
|
||||||
|
Some(plugin),
|
||||||
|
"cea608tocea708",
|
||||||
|
gst::Rank::None,
|
||||||
|
Cea608ToCea708::static_type(),
|
||||||
|
)
|
||||||
|
}
|
|
@ -26,6 +26,7 @@ mod caption_frame;
|
||||||
mod ccdetect;
|
mod ccdetect;
|
||||||
mod ccutils;
|
mod ccutils;
|
||||||
mod cea608overlay;
|
mod cea608overlay;
|
||||||
|
mod cea608tocea708;
|
||||||
mod cea608tojson;
|
mod cea608tojson;
|
||||||
mod cea608tott;
|
mod cea608tott;
|
||||||
mod cea608utils;
|
mod cea608utils;
|
||||||
|
@ -56,6 +57,7 @@ fn plugin_init(plugin: &gst::Plugin) -> Result<(), glib::BoolError> {
|
||||||
cea608tojson::register(plugin)?;
|
cea608tojson::register(plugin)?;
|
||||||
jsontovtt::register(plugin)?;
|
jsontovtt::register(plugin)?;
|
||||||
transcriberbin::register(plugin)?;
|
transcriberbin::register(plugin)?;
|
||||||
|
cea608tocea708::register(plugin)?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
199
video/closedcaption/tests/cea608tocea708.rs
Normal file
199
video/closedcaption/tests/cea608tocea708.rs
Normal file
|
@ -0,0 +1,199 @@
|
||||||
|
// Copyright (C) 2023 Matthew Waters <matthew@centricular.com>
|
||||||
|
//
|
||||||
|
// This Source Code Form is subject to the terms of the Mozilla Public License, v2.0.
|
||||||
|
// If a copy of the MPL was not distributed with this file, You can obtain one at
|
||||||
|
// <https://mozilla.org/MPL/2.0/>.
|
||||||
|
//
|
||||||
|
// SPDX-License-Identifier: MPL-2.0
|
||||||
|
|
||||||
|
use gst::prelude::*;
|
||||||
|
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
|
use cea708_types::tables::*;
|
||||||
|
use cea708_types::*;
|
||||||
|
|
||||||
|
fn init() {
|
||||||
|
use std::sync::Once;
|
||||||
|
static INIT: Once = Once::new();
|
||||||
|
|
||||||
|
INIT.call_once(|| {
|
||||||
|
gst::init().unwrap();
|
||||||
|
gstrsclosedcaption::plugin_register_static().unwrap();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
struct TestState {
|
||||||
|
cc_data_parser: CCDataParser,
|
||||||
|
h: gst_check::Harness,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl TestState {
|
||||||
|
fn new() -> Self {
|
||||||
|
let mut h = gst_check::Harness::new_parse("cea608tocea708");
|
||||||
|
h.set_src_caps_str("closedcaption/x-cea-608,format=raw,field=0");
|
||||||
|
h.set_sink_caps_str("closedcaption/x-cea-708,format=cc_data");
|
||||||
|
|
||||||
|
Self {
|
||||||
|
cc_data_parser: CCDataParser::new(),
|
||||||
|
h,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn push_data(&mut self, input: &[u8], pts: gst::ClockTime, output: &[Code]) {
|
||||||
|
let mut buf = gst::Buffer::from_mut_slice(input.to_vec());
|
||||||
|
{
|
||||||
|
let buf = buf.get_mut().unwrap();
|
||||||
|
buf.set_pts(pts);
|
||||||
|
}
|
||||||
|
assert_eq!(self.h.push(buf), Ok(gst::FlowSuccess::Ok));
|
||||||
|
let out_buf = self.h.try_pull().unwrap();
|
||||||
|
let data = out_buf.map_readable().unwrap();
|
||||||
|
|
||||||
|
// construct the two byte header for cc_data that GStreamer doesn't write
|
||||||
|
let mut complete_cc_data = vec![];
|
||||||
|
complete_cc_data.extend([0x80 | 0x40 | ((data.len() / 3) & 0x1f) as u8, 0xff]);
|
||||||
|
complete_cc_data.extend(&*data);
|
||||||
|
println!("{}, {input:X?} {complete_cc_data:X?}", pts.display());
|
||||||
|
|
||||||
|
self.cc_data_parser.push(&complete_cc_data).unwrap();
|
||||||
|
let mut output_iter = output.iter();
|
||||||
|
while let Some(packet) = self.cc_data_parser.pop_packet() {
|
||||||
|
for service in packet.services() {
|
||||||
|
for (ci, code) in service.codes().iter().enumerate() {
|
||||||
|
println!(
|
||||||
|
"{}, P{}: S{}: C{ci}: {code:?}",
|
||||||
|
pts.display(),
|
||||||
|
packet.sequence_no(),
|
||||||
|
service.number()
|
||||||
|
);
|
||||||
|
assert_eq!(output_iter.next(), Some(code));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
assert_eq!(out_buf.pts(), Some(pts));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_single_char() {
|
||||||
|
init();
|
||||||
|
|
||||||
|
let test_data = [([0xC1, 0x80], vec![Code::LatinCapitalA])];
|
||||||
|
|
||||||
|
let mut state = TestState::new();
|
||||||
|
|
||||||
|
for (i, d) in test_data.iter().enumerate() {
|
||||||
|
let ts = gst::ClockTime::from_mseconds((i as u64) * 13);
|
||||||
|
state.push_data(&d.0, ts, &d.1);
|
||||||
|
}
|
||||||
|
|
||||||
|
let caps = state
|
||||||
|
.h
|
||||||
|
.sinkpad()
|
||||||
|
.expect("harness has no sinkpad")
|
||||||
|
.current_caps()
|
||||||
|
.expect("pad has no caps");
|
||||||
|
assert_eq!(
|
||||||
|
caps,
|
||||||
|
gst::Caps::builder("closedcaption/x-cea-708")
|
||||||
|
.field("format", "cc_data")
|
||||||
|
.build()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_rollup() {
|
||||||
|
init();
|
||||||
|
|
||||||
|
let test_data = [
|
||||||
|
([0x94, 0x2C], vec![Code::ClearWindows(WindowBits::ZERO)]), // EDM -> ClearWindows(1)
|
||||||
|
(
|
||||||
|
[0x94, 0x26],
|
||||||
|
vec![
|
||||||
|
Code::DeleteWindows(!WindowBits::ZERO),
|
||||||
|
Code::DefineWindow(DefineWindowArgs::new(
|
||||||
|
0,
|
||||||
|
0,
|
||||||
|
Anchor::BottomMiddle,
|
||||||
|
true,
|
||||||
|
100,
|
||||||
|
50,
|
||||||
|
2,
|
||||||
|
31,
|
||||||
|
true,
|
||||||
|
true,
|
||||||
|
true,
|
||||||
|
1,
|
||||||
|
1,
|
||||||
|
)),
|
||||||
|
Code::SetPenLocation(SetPenLocationArgs::new(2, 0)),
|
||||||
|
Code::ETX,
|
||||||
|
],
|
||||||
|
), // RU3 -> DeleteWindows(!0), DefineWindow(0...), SetPenLocation(bottom-row)
|
||||||
|
([0x94, 0xAD], vec![Code::CR, Code::ETX]), // CR -> CR
|
||||||
|
(
|
||||||
|
[0x94, 0x70],
|
||||||
|
vec![
|
||||||
|
Code::DeleteWindows(!WindowBits::ZERO),
|
||||||
|
Code::DefineWindow(DefineWindowArgs::new(
|
||||||
|
0,
|
||||||
|
0,
|
||||||
|
Anchor::BottomMiddle,
|
||||||
|
true,
|
||||||
|
93,
|
||||||
|
50,
|
||||||
|
2,
|
||||||
|
31,
|
||||||
|
true,
|
||||||
|
true,
|
||||||
|
true,
|
||||||
|
1,
|
||||||
|
1,
|
||||||
|
)),
|
||||||
|
Code::SetPenLocation(SetPenLocationArgs::new(2, 0)),
|
||||||
|
Code::ETX,
|
||||||
|
],
|
||||||
|
), // PAC to bottom left -> SetPenLocation(...)
|
||||||
|
(
|
||||||
|
[0xA8, 0x43],
|
||||||
|
vec![Code::LeftParenthesis, Code::LatinCapitalC, Code::ETX],
|
||||||
|
), // text: (C -> (C
|
||||||
|
(
|
||||||
|
[0x94, 0x26],
|
||||||
|
vec![
|
||||||
|
Code::DeleteWindows(!WindowBits::ZERO),
|
||||||
|
Code::DefineWindow(DefineWindowArgs::new(
|
||||||
|
0,
|
||||||
|
0,
|
||||||
|
Anchor::BottomMiddle,
|
||||||
|
true,
|
||||||
|
93,
|
||||||
|
50,
|
||||||
|
2,
|
||||||
|
31,
|
||||||
|
true,
|
||||||
|
true,
|
||||||
|
true,
|
||||||
|
1,
|
||||||
|
1,
|
||||||
|
)),
|
||||||
|
Code::SetPenLocation(SetPenLocationArgs::new(2, 0)),
|
||||||
|
Code::ETX,
|
||||||
|
],
|
||||||
|
), // RU3
|
||||||
|
([0x94, 0xAD], vec![Code::CR, Code::ETX]), // CR -> CR
|
||||||
|
([0x94, 0x70], vec![Code::ETX]), // PAC to bottom left -> SetPenLocation(...)
|
||||||
|
(
|
||||||
|
[0xF2, 0xEF],
|
||||||
|
vec![Code::LatinLowerR, Code::LatinLowerO, Code::ETX],
|
||||||
|
), // ro
|
||||||
|
];
|
||||||
|
|
||||||
|
let mut state = TestState::new();
|
||||||
|
|
||||||
|
for (i, d) in test_data.iter().enumerate() {
|
||||||
|
let ts = gst::ClockTime::from_mseconds((i as u64) * 13);
|
||||||
|
state.push_data(&d.0, ts, &d.1);
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue