use crate::bindgen::FPDF_PAGE;
use crate::bindings::PdfiumLibraryBindings;
use crate::document::PdfDocument;
use crate::error::{PdfiumError, PdfiumInternalError};
use crate::page_annotation::PdfPageAnnotation;
use crate::page_annotation_private::internal::PdfPageAnnotationPrivate;
use std::ops::Range;
use std::os::raw::c_int;
pub type PdfPageAnnotationIndex = usize;
pub struct PdfPageAnnotations<'a> {
page_handle: FPDF_PAGE,
document: &'a PdfDocument<'a>,
do_regenerate_page_content_after_each_change: bool,
}
impl<'a> PdfPageAnnotations<'a> {
#[inline]
pub(crate) fn from_pdfium(page_handle: FPDF_PAGE, document: &'a PdfDocument<'a>) -> Self {
Self {
page_handle,
document,
do_regenerate_page_content_after_each_change: false,
}
}
#[inline]
pub(crate) fn do_regenerate_page_content_after_each_change(
&mut self,
do_regenerate_page_content_after_each_change: bool,
) {
for mut annotation in self.iter() {
annotation
.objects_mut_impl()
.do_regenerate_page_content_after_each_change(
do_regenerate_page_content_after_each_change,
);
}
self.do_regenerate_page_content_after_each_change =
do_regenerate_page_content_after_each_change;
}
#[inline]
pub fn bindings(&self) -> &'a dyn PdfiumLibraryBindings {
self.document.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.page_handle,
self.document,
))
}
}
#[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()
}
}