如何使用magick::image_read_pdf获取pdf文件的总页数？

Question

假设在一个文件夹main_path下，我们有多个不同页数的 pdf 文件，我使用下面的 function 循环所有文件并截图每个页面：

library(magick)
library(glue)

main_path <- './'

file_names <- list.files(path = main_path, pattern ='.pdf') 
file_paths <- file.path(main_path, file_names)
file_names_no_ext <- tools::file_path_sans_ext(file_names)

max_page <- 10
pdf2plot <- function(file_path, file_names_no_ext){
  pages <- magick::image_read_pdf(file_path)
  print(pages)
  num <- seq(1, max_page, 1)
  # num <- seq(1, nrow(data.frame(pages)), 1)
  for (i in num){
    pages[i] %>% image_write(., path = paste0(glue(main_path, '/plot/', {file_names_no_ext},
                                                   sprintf('_%02d.', i)), format = "png"))
  }
}

mapply(pdf2plot, file_paths, file_names_no_ext)

我遇到的问题是，如果文件夹中有一个文件的总页数小于max_page ，它将Error in magick_image_subset(x, i): subscript out of bounds 。 例如，我有一个有 2 页的文件，但我设置max_page=10 ，我会得到这个错误。

pages内容：

  format width height colorspace matte filesize density
  <chr>  <int>  <int> <chr>      <lgl>    <int> <chr>  
1 PNG     2250   3000 sRGB       TRUE         0 300x300
2 PNG     2250   3000 sRGB       TRUE         0 300x300
3 PNG     2250   3000 sRGB       TRUE         0 300x300
4 PNG     2250   3000 sRGB       TRUE         0 300x300
5 PNG     2250   3000 sRGB       TRUE         0 300x300
6 PNG     2250   3000 sRGB       TRUE         0 300x300
7 PNG     2250   3000 sRGB       TRUE         0 300x300
8 PNG     2250   3000 sRGB       TRUE         0 300x300
9 PNG     2250   3000 sRGB       TRUE         0 300x300
Error in magick_image_subset(x, i) : subscript out of bounds
Called from: magick_image_subset(x, i)

我认为可能有两种方法可以解决这个问题，但我还不知道如何去做：1. 使用try-catch ， 2. 通过使用magick::image_read_pdf获取总页数替换max_page 。

提前感谢您的帮助。

Answer 1

如果您查看?image_read的文档，我们可以看到：

所有标准的基本向量方法，例如 [、[[、c()、as.list()、as.raster()、rev()、length() 和 print()，都可以用于处理魔法图像对象。 使用标准 img[i] 语法从图像中提取帧的子集。

因此，您可以简单地使用length(pages)来获取该文档的页数。 这是使用lapply()的 function 的简单版本。 我认为你可以简化你的路径很多，但不会进入那个。

library(magick)
library(glue)

pdf2plot <- function(file_path, file_names_no_ext){
  pages <- magick::image_read_pdf(file_path)
  lapply(
    1:length(pages),
    \(i) image_write(pages[i], path = paste0(glue(main_path, '/plot/', {file_names_no_ext},
                                                   sprintf('_%02d.', i)), format = "png"))
  )
}