docsplit images cmd - added the ability to specify page number delimiter via command line option

BrandonNoad · BrandonNoad · commit 2894363b1b0e · 2014-05-28T15:40:45.000-04:00
diff --git a/lib/docsplit/command_line.rb b/lib/docsplit/command_line.rb
@@ -101,6 +101,9 @@ def parse_options
         opts.on('-r', '--rolling', 'generate images from each previous image') do |r|
           @options[:rolling] = true
         end
+        opts.on('--delimiter [DELIMITER]', 'set page number delimiter (eg: _, -, -Page-...)') do |d|
+          @options[:delimiter] = d.tr('^','')
+        end
         opts.on_tail('-v', '--version', 'display docsplit version') do
           puts "Docsplit version #{Docsplit::VERSION}"
           exit
diff --git a/lib/docsplit/image_extractor.rb b/lib/docsplit/image_extractor.rb
@@ -7,6 +7,7 @@ class ImageExtractor
     MEMORY_ARGS     = "-limit memory 256MiB -limit map 512MiB"
     DEFAULT_FORMAT  = :png
     DEFAULT_DENSITY = '150'
+    DEFAULT_PAGE_DELIMITER = "_"
 
     # Extract a list of PDFs as rasterized page images, according to the
     # configuration in options.
@@ -41,7 +42,7 @@ def convert(pdf, size, format, previous=nil)
         raise ExtractionFailed, result if $? != 0
       else
         page_list(pages).each do |page|
-          out_file  = ESCAPE[File.join(directory, "#{basename}_#{page}.#{format}")]
+          out_file  = ESCAPE[File.join(directory, "#{basename}#{@delimiter}#{page}.#{format}")]
           cmd = "MAGICK_TMPDIR=#{tempdir} OMP_NUM_THREADS=2 gm convert +adjoin -define pdf:use-cropbox=true #{common} #{escaped_pdf}[#{page - 1}] #{out_file} 2>&1".chomp
           result = `#{cmd}`.chomp
           raise ExtractionFailed, result if $? != 0
@@ -63,6 +64,7 @@ def extract_options(options)
       @sizes   = [options[:size]].flatten.compact
       @sizes   = [nil] if @sizes.empty?
       @rolling = !!options[:rolling]
+      @delimiter = options[:delimiter] || DEFAULT_PAGE_DELIMITER
     end
 
     # If there's only one size requested, generate the images directly into
