mirror of
https://github.com/storytold/pdfcraft.git
synced 2026-11-06 18:52:36 +00:00
FredApps
52d68d7844
main's 3D-embed change (#5808) added crates/annot/src/threed.rs with its own pub MAX_MODEL_BYTES (same 256 MiB value), so both were re-exported from the crate root and pdfcraft-annot no longer compiled after merging main. model3d now uses threed's constant.
78 lines
3.3 KiB
Rust
78 lines
3.3 KiB
Rust
//! Reading the model a 3D annotation shows (ISO 32000-2 §13.6.3 and §13.6.4): its `/3DD` is a
|
|
//! 3D stream, or a 3D reference dictionary naming the stream another annotation shares.
|
|
|
|
use super::lookup::{annots, page_ref};
|
|
use super::threed::MAX_MODEL_BYTES;
|
|
use super::{AnnotError, Dict, Document, Object};
|
|
|
|
/// The format of a 3D stream, from its `/Subtype`.
|
|
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
|
pub enum Model3dKind {
|
|
/// Product Representation Compact (ISO 14739-1).
|
|
Prc,
|
|
/// Universal 3D (ECMA-363).
|
|
U3d,
|
|
}
|
|
|
|
/// A 3D annotation's model: its format and the decoded stream.
|
|
#[derive(Debug, Clone, PartialEq)]
|
|
pub struct Model3d {
|
|
pub kind: Model3dKind,
|
|
pub data: Vec<u8>,
|
|
}
|
|
|
|
/// The `/Annots` positions of the 3D annotations on 0-based `page`, in order.
|
|
///
|
|
/// # Errors
|
|
/// [`AnnotError::NoPageTree`] or [`AnnotError::NoSuchPage`].
|
|
pub fn model_3d_annotations(doc: &Document, page: usize) -> Result<Vec<usize>, AnnotError> {
|
|
let list = annots(doc, page_ref(doc, page)?);
|
|
Ok(list.iter().enumerate().filter(|(_, a)| doc.dict(a).is_some_and(|d| d.name(b"Subtype") == Some(b"3D"))).map(|(i, _)| i).collect())
|
|
}
|
|
|
|
/// The model of the 3D annotation at `/Annots` position `index` on 0-based `page`, decoded.
|
|
///
|
|
/// # Errors
|
|
/// [`AnnotError::NoSuchAnnotation`] when there is no annotation there; [`AnnotError::Invalid`]
|
|
/// when it is not a 3D annotation, has no readable 3D stream, names a format other than PRC or
|
|
/// U3D, or decodes past [`MAX_MODEL_BYTES`].
|
|
pub fn model_3d(doc: &Document, page: usize, index: usize) -> Result<Model3d, AnnotError> {
|
|
let list = annots(doc, page_ref(doc, page)?);
|
|
let entry = list.get(index).ok_or(AnnotError::NoSuchAnnotation { page, index })?;
|
|
let annot = doc.dict(entry).ok_or(AnnotError::NoSuchAnnotation { page, index })?;
|
|
let at = || format!("annotation {} on page {}", index.saturating_add(1), page.saturating_add(1));
|
|
if annot.name(b"Subtype") != Some(b"3D") {
|
|
return Err(AnnotError::Invalid(format!("{} is not a 3D annotation", at())));
|
|
}
|
|
let stream = stream_of(doc, &annot).ok_or_else(|| AnnotError::Invalid(format!("{} has no 3D stream", at())))?;
|
|
let Object::Stream(s) = stream.as_ref() else {
|
|
return Err(AnnotError::Invalid(format!("{} has no 3D stream", at())));
|
|
};
|
|
let kind = match s.dict.name(b"Subtype") {
|
|
Some(b"PRC") => Model3dKind::Prc,
|
|
Some(b"U3D") => Model3dKind::U3d,
|
|
other => {
|
|
let name = other.map_or_else(|| "none".to_owned(), |n| String::from_utf8_lossy(n).into_owned());
|
|
return Err(AnnotError::Invalid(format!("{}'s 3D stream is of an unknown format ({name})", at())));
|
|
}
|
|
};
|
|
let data = doc
|
|
.with_direct_parms(s)
|
|
.decoded_within(MAX_MODEL_BYTES)
|
|
.map_err(|e| AnnotError::Invalid(format!("{}'s 3D stream could not be decoded: {e}", at())))?;
|
|
Ok(Model3d { kind, data })
|
|
}
|
|
|
|
/// What an annotation's `/3DD` names: the 3D stream itself, or the `/3D` entry of a 3D reference
|
|
/// dictionary. A reference is followed once; the caller refuses anything that is not a stream.
|
|
fn stream_of(doc: &Document, annot: &Dict) -> Option<std::sync::Arc<Object>> {
|
|
let obj = doc.resolve(annot.get(b"3DD")?);
|
|
match obj.as_ref() {
|
|
Object::Dict(d) if d.name(b"Type") == Some(b"3DRef") => Some(doc.resolve(d.get(b"3D")?)),
|
|
_ => Some(obj),
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests;
|