aboutsummaryrefslogtreecommitdiff
path: root/tools/src
diff options
context:
space:
mode:
Diffstat (limited to 'tools/src')
-rw-r--r--tools/src/bin/create_item.rs32
-rw-r--r--tools/src/bin/gen_meta.rs133
-rw-r--r--tools/src/bin/import.rs187
3 files changed, 219 insertions, 133 deletions
diff --git a/tools/src/bin/create_item.rs b/tools/src/bin/create_item.rs
new file mode 100644
index 0000000..7628f95
--- /dev/null
+++ b/tools/src/bin/create_item.rs
@@ -0,0 +1,32 @@
+use std::{fs::File, io::Write, path::PathBuf};
+
+use clap::Parser;
+use jellycommon::ItemInfo;
+
+#[derive(Parser)]
+struct Args {
+ #[clap(short = 'I', long)]
+ item: PathBuf,
+ #[clap(short = 'd', long)]
+ dry: bool,
+ #[clap(short, long)]
+ title: String,
+}
+
+fn main() -> anyhow::Result<()> {
+ let args = Args::parse();
+
+ let iteminfo = ItemInfo {
+ title: args.title,
+ tracks: Default::default(),
+ };
+
+ let k = serde_json::to_string_pretty(&iteminfo)?;
+ if args.dry {
+ println!("{k}")
+ } else {
+ let mut f = File::create(args.item)?;
+ f.write_all(k.as_bytes())?;
+ }
+ Ok(())
+}
diff --git a/tools/src/bin/gen_meta.rs b/tools/src/bin/gen_meta.rs
deleted file mode 100644
index 4abbb8b..0000000
--- a/tools/src/bin/gen_meta.rs
+++ /dev/null
@@ -1,133 +0,0 @@
-use anyhow::{anyhow, bail};
-use clap::Parser;
-use jellycommon::{ItemInfo, Source, SourceTrack, SourceTrackKind};
-use jellyebml::{
- matroska::MatroskaTag,
- read::EbmlReader,
- unflatten::{Unflat, Unflatten},
- Master,
-};
-use log::error;
-use std::{collections::BTreeMap, fs::File, io::Write, path::PathBuf};
-
-#[derive(Parser)]
-struct Args {
- #[clap(short = 'I', long)]
- identifier: String,
- #[clap(short = 'O', long)]
- write_json: bool,
- #[clap(short, long)]
- title: String,
- #[clap(short = 'i', long)]
- input: PathBuf,
-}
-
-fn main() -> anyhow::Result<()> {
- env_logger::init_from_env("LOG");
- let args = Args::parse();
-
- let mut tracks = BTreeMap::new();
- let input = File::open(args.input.clone()).unwrap();
- let mut input = EbmlReader::new(input);
-
- // TODO dont traverse the entire file, if the tracks are listed at the end
- while let Some(item) = input.next() {
- let item = item?;
- match item {
- MatroskaTag::Tracks(_) => {
- let mut iter = Unflatten::new(&mut input);
- while let Some(Ok(Unflat { children, item })) = iter.next() {
- if !matches!(item, MatroskaTag::TrackEntry(_)) {
- panic!("no")
- }
- let mut children = children.unwrap();
- let (
- mut index,
- mut language,
- mut codec,
- mut kind,
- mut sample_rate,
- mut channels,
- mut width,
- mut height,
- mut name,
- ) = (None, None, None, None, None, None, None, None, None);
- while let Some(Ok(Unflat { children, item })) = children.next() {
- match item {
- MatroskaTag::CodecID(b) => codec = Some(b),
- MatroskaTag::Language(v) => language = Some(v),
- MatroskaTag::TrackNumber(v) => index = Some(v),
- MatroskaTag::TrackType(v) => kind = Some(v),
- MatroskaTag::Name(v) => name = Some(v),
- MatroskaTag::Audio(_) => {
- let mut children = children.unwrap();
- while let Some(Ok(Unflat { item, .. })) = children.next() {
- match item {
- MatroskaTag::Channels(v) => channels = Some(v as usize),
- MatroskaTag::SamplingFrequency(v) => sample_rate = Some(v),
- _ => (),
- }
- }
- }
- MatroskaTag::Video(_) => {
- let mut children = children.unwrap();
- while let Some(Ok(Unflat { item, .. })) = children.next() {
- match item {
- MatroskaTag::PixelWidth(v) => width = Some(v),
- MatroskaTag::PixelHeight(v) => height = Some(v),
- _ => (),
- }
- }
- }
- _ => (),
- }
- }
- tracks.insert(
- index.unwrap(),
- SourceTrack {
- name: name.unwrap_or_else(|| "unnamed".to_string()),
- codec: codec.unwrap(),
- language: language.unwrap_or_else(|| "none".to_string()),
- kind: match kind.ok_or(anyhow!("track type required"))? {
- 1 => SourceTrackKind::Video {
- fps: 0.0, // TODO
- width: width.unwrap(),
- height: height.unwrap(),
- },
- 2 => SourceTrackKind::Audio {
- bit_depth: 0, // TODO
- channels: channels.unwrap(),
- sample_rate: sample_rate.unwrap(),
- },
- 17 => SourceTrackKind::Subtitles,
- _ => bail!("invalid track type"),
- },
- },
- );
- }
- error!("break!");
- drop(iter);
- error!("break done!");
- break;
- }
- _ => (),
- }
- }
-
- let k = serde_json::to_string_pretty(&ItemInfo {
- title: args.title,
- source: Source {
- path: args.input.clone(),
- tracks,
- },
- })?;
-
- if args.write_json {
- let mut f = File::create(format!("{}.json", args.identifier))?;
- f.write_all(k.as_bytes())?;
- } else {
- println!("{k}")
- }
-
- Ok(())
-}
diff --git a/tools/src/bin/import.rs b/tools/src/bin/import.rs
new file mode 100644
index 0000000..dfa5267
--- /dev/null
+++ b/tools/src/bin/import.rs
@@ -0,0 +1,187 @@
+use anyhow::{anyhow, bail, Result};
+use clap::Parser;
+use jellycommon::{ItemInfo, SourceTrack, SourceTrackKind};
+use jellymatroska::{
+ matroska::MatroskaTag,
+ read::EbmlReader,
+ unflatten::{Unflat, Unflatten},
+ Master,
+};
+use log::{debug, error, info, trace, warn};
+use std::{fs::File, io::Write, path::PathBuf};
+
+#[derive(Parser)]
+struct Args {
+ #[clap(short = 'I', long)]
+ item: PathBuf,
+ #[clap(short = 'd', long)]
+ dry: bool,
+ #[clap(short = 'i', long)]
+ input: PathBuf,
+}
+
+fn main() -> anyhow::Result<()> {
+ env_logger::init_from_env("LOG");
+ let args = Args::parse();
+
+ let mut iteminfo: ItemInfo = serde_json::from_reader(File::open(&args.item)?)?;
+ // let iteminfo_orig = iteminfo.clone();
+
+ let input = File::open(args.input.clone()).unwrap();
+ let mut input = EbmlReader::new(input);
+
+ read(&mut input, &mut iteminfo)?;
+
+ let k = serde_json::to_string_pretty(&iteminfo)?;
+ if args.dry {
+ println!("{k}")
+ } else {
+ let mut f = File::create(args.item)?;
+ f.write_all(k.as_bytes())?;
+ }
+
+ Ok(())
+}
+
+pub fn read(input: &mut EbmlReader, iteminfo: &mut ItemInfo) -> Result<()> {
+ // TODO dont traverse the entire file, if the tracks are listed at the end
+ let mut sbc = 0;
+ while let Some(item) = input.next() {
+ let item = item?;
+ match item {
+ MatroskaTag::Cluster(_) => {
+ info!("start of cluster found");
+ let mut iter = Unflatten::new_with_end(input, item);
+ while let Some(Ok(Unflat { children, item })) = iter.next() {
+ match item {
+ MatroskaTag::BlockGroup(_) => {
+ debug!("group");
+ let mut iter = children.unwrap();
+ while let Some(Ok(Unflat { children, item })) = iter.next() {
+ match item {
+ MatroskaTag::Block(_) => (),
+ _ => trace!("{item:?}"),
+ }
+ }
+ }
+ MatroskaTag::SimpleBlock(_) => {
+ // debug!("simple");
+ }
+ _ => debug!("(rc) tag ignored: {item:?}"),
+ }
+ }
+ sbc += 1;
+ }
+ MatroskaTag::Tags(_) => {
+ Unflatten::new_with_end(input, item);
+ }
+ MatroskaTag::Cues(_) => {
+ let mut iter = Unflatten::new_with_end(input, item);
+ while let Some(Ok(Unflat { children, item })) = iter.next() {
+ match item {
+ MatroskaTag::CuePoint(_) => {
+ let mut children = children.unwrap();
+ while let Some(Ok(Unflat { children, item })) = children.next() {
+ // error!("{item:?}")
+ }
+ }
+ _ => (),
+ }
+ }
+ }
+ MatroskaTag::Chapters(_) => {
+ Unflatten::new_with_end(input, item);
+ }
+ MatroskaTag::Tracks(_) => {
+ let mut iter = Unflatten::new_with_end(input, item);
+ while let Some(Ok(Unflat { children, item })) = iter.next() {
+ match item {
+ MatroskaTag::TrackEntry(_) => {
+ let mut children = children.unwrap();
+ let (
+ mut index,
+ mut language,
+ mut codec,
+ mut kind,
+ mut sample_rate,
+ mut channels,
+ mut width,
+ mut height,
+ mut name,
+ mut fps,
+ mut bit_depth,
+ ) = (
+ None, None, None, None, None, None, None, None, None, None, None,
+ );
+ while let Some(Ok(Unflat { children, item })) = children.next() {
+ match item {
+ MatroskaTag::CodecID(b) => codec = Some(b),
+ MatroskaTag::Language(v) => language = Some(v),
+ MatroskaTag::TrackNumber(v) => index = Some(v),
+ MatroskaTag::TrackType(v) => kind = Some(v),
+ MatroskaTag::Name(v) => name = Some(v),
+ MatroskaTag::Audio(_) => {
+ let mut children = children.unwrap();
+ while let Some(Ok(Unflat { item, .. })) = children.next() {
+ match item {
+ MatroskaTag::Channels(v) => {
+ channels = Some(v as usize)
+ }
+ MatroskaTag::SamplingFrequency(v) => {
+ sample_rate = Some(v)
+ }
+ MatroskaTag::BitDepth(v) => bit_depth = Some(v),
+ _ => (),
+ }
+ }
+ }
+ MatroskaTag::Video(_) => {
+ let mut children = children.unwrap();
+ while let Some(Ok(Unflat { item, .. })) = children.next() {
+ match item {
+ MatroskaTag::PixelWidth(v) => width = Some(v),
+ MatroskaTag::PixelHeight(v) => height = Some(v),
+ MatroskaTag::FrameRate(v) => fps = Some(v),
+ _ => (),
+ }
+ }
+ }
+ _ => (),
+ }
+ }
+ let index = index.unwrap();
+ let kind = match kind.ok_or(anyhow!("track type required"))? {
+ 1 => SourceTrackKind::Video {
+ fps: fps.unwrap_or(f64::NAN), // TODO
+ width: width.unwrap(),
+ height: height.unwrap(),
+ },
+ 2 => SourceTrackKind::Audio {
+ bit_depth: bit_depth.unwrap_or(0) as usize, // TODO
+ channels: channels.unwrap(),
+ sample_rate: sample_rate.unwrap(),
+ },
+ 17 => SourceTrackKind::Subtitles,
+ _ => bail!("invalid track type"),
+ };
+ iteminfo.tracks.insert(
+ index,
+ SourceTrack {
+ name: name.unwrap_or_else(|| "unnamed".to_string()),
+ codec: codec.unwrap(),
+ language: language.unwrap_or_else(|| "none".to_string()),
+ kind,
+ },
+ );
+ }
+ _ => debug!("(rt) tag ignored: {item:?}"),
+ }
+ }
+ }
+ MatroskaTag::Segment(Master::End) => break,
+ _ => debug!("(r) tag ignored: {item:?}"),
+ }
+ }
+ info!("clusters {sbc}");
+ Ok(())
+}
generated by cgit v1.2.3-70-g09d2 (git 2.51.0) at 2025-10-08 14:23:40 +0000