use crate::bindgen::FPDF_PAGE;
use crate::bindings::PdfiumLibraryBindings;
use crate::error::{PdfiumError, PdfiumInternalError};
use crate::page_annotation::PdfPageAnnotation;
use std::ops::Range;
use std::os::raw::c_int;
pub type PdfPageAnnotationIndex = usize;
pub struct PdfPageAnnotations<'a> {
page_handle: FPDF_PAGE,
bindings: &'a dyn PdfiumLibraryBindings,
}
impl<'a> PdfPageAnnotations<'a> {
#[inline]
pub(crate) fn from_pdfium(
page_handle: FPDF_PAGE,
bindings: &'a dyn PdfiumLibraryBindings,
) -> Self {
Self {
page_handle,
bindings,
}
}
#[inline]
pub fn len(&self) -> PdfPageAnnotationIndex {
self.bindings.FPDFPage_GetAnnotCount(self.page_handle) as PdfPageAnnotationIndex
}
#[inline]
pub fn is_empty(&self) -> bool {
self.len() == 0
}
#[inline]
pub fn as_range(&self) -> Range<PdfPageAnnotationIndex> {
0..self.len()
}
pub fn get(&self, index: PdfPageAnnotationIndex) -> Result<PdfPageAnnotation, PdfiumError> {
if index >= self.len() {
return Err(PdfiumError::PageAnnotationIndexOutOfBounds);
}
let annotation_handle = self
.bindings
.FPDFPage_GetAnnot(self.page_handle, index as c_int);
if annotation_handle.is_null() {
if let Some(error) = self.bindings.get_pdfium_last_error() {
Err(PdfiumError::PdfiumLibraryInternalError(error))
} else {
Err(PdfiumError::PdfiumLibraryInternalError(
PdfiumInternalError::Unknown,
))
}
} else {
Ok(PdfPageAnnotation::from_pdfium(
annotation_handle,
self.bindings,
))
}
}
#[inline]
pub fn iter(&self) -> PdfPageAnnotationsIterator {
PdfPageAnnotationsIterator::new(self)
}
}
pub struct PdfPageAnnotationsIterator<'a> {
annotations: &'a PdfPageAnnotations<'a>,
next_index: PdfPageAnnotationIndex,
}
impl<'a> PdfPageAnnotationsIterator<'a> {
#[inline]
pub(crate) fn new(annotations: &'a PdfPageAnnotations<'a>) -> Self {
PdfPageAnnotationsIterator {
annotations,
next_index: 0,
}
}
}
impl<'a> Iterator for PdfPageAnnotationsIterator<'a> {
type Item = PdfPageAnnotation<'a>;
fn next(&mut self) -> Option<Self::Item> {
let next = self.annotations.get(self.next_index);
self.next_index += 1;
next.ok()
}
}