OpenSystemsDevelopment / qpdf

Commit 41b5c46497997d45ba2e7521b66a262ea7a1afd4

Authored by Jay Berkenbilt
1 parent b3c91f64

refactor: split write_outfile and do_split_pages

Inline Side-by-side
Showing 1 changed file with 86 additions and 84 deletions
qpdf/qpdf.cc
  View file @41b5c46
@@ -5024,107 +5024,105 @@ static void set_writer_options(QPDF& pdf, Options& o, QPDFWriter& w) @@ -5024,107 +5024,105 @@ static void set_writer_options(QPDF& pdf, Options& o, QPDFWriter& w)
5024 } 5024 }
5025 } 5025 }
5026 5026
5027 -static void write_outfile(QPDF& pdf, Options& o) 5027 +static void do_split_pages(QPDF& pdf, Options& o)
5028 { 5028 {
5029 - if (o.split_pages) 5029 + // Generate output file pattern
  5030 + std::string before;
  5031 + std::string after;
  5032 + size_t len = strlen(o.outfilename);
  5033 + char* num_spot = strstr(const_cast<char*>(o.outfilename), "%d");
  5034 + if (num_spot != 0)
5030 { 5035 {
5031 - // Generate output file pattern  
5032 - std::string before;  
5033 - std::string after;  
5034 - size_t len = strlen(o.outfilename);  
5035 - char* num_spot = strstr(const_cast<char*>(o.outfilename), "%d");  
5036 - if (num_spot != 0)  
5037 - {  
5038 - QTC::TC("qpdf", "qpdf split-pages %d");  
5039 - before = std::string(o.outfilename,  
5040 - QIntC::to_size(num_spot - o.outfilename));  
5041 - after = num_spot + 2;  
5042 - }  
5043 - else if ((len >= 4) &&  
5044 - (QUtil::str_compare_nocase(  
5045 - o.outfilename + len - 4, ".pdf") == 0))  
5046 - {  
5047 - QTC::TC("qpdf", "qpdf split-pages .pdf");  
5048 - before = std::string(o.outfilename, len - 4) + "-";  
5049 - after = o.outfilename + len - 4;  
5050 - }  
5051 - else 5036 + QTC::TC("qpdf", "qpdf split-pages %d");
  5037 + before = std::string(o.outfilename,
  5038 + QIntC::to_size(num_spot - o.outfilename));
  5039 + after = num_spot + 2;
  5040 + }
  5041 + else if ((len >= 4) &&
  5042 + (QUtil::str_compare_nocase(
  5043 + o.outfilename + len - 4, ".pdf") == 0))
  5044 + {
  5045 + QTC::TC("qpdf", "qpdf split-pages .pdf");
  5046 + before = std::string(o.outfilename, len - 4) + "-";
  5047 + after = o.outfilename + len - 4;
  5048 + }
  5049 + else
  5050 + {
  5051 + QTC::TC("qpdf", "qpdf split-pages other");
  5052 + before = std::string(o.outfilename) + "-";
  5053 + }
  5054 +
  5055 + if (! o.preserve_unreferenced_page_resources)
  5056 + {
  5057 + QPDFPageDocumentHelper dh(pdf);
  5058 + dh.removeUnreferencedResources();
  5059 + }
  5060 + QPDFPageLabelDocumentHelper pldh(pdf);
  5061 + std::vector<QPDFObjectHandle> const& pages = pdf.getAllPages();
  5062 + size_t pageno_len = QUtil::uint_to_string(pages.size()).length();
  5063 + size_t num_pages = pages.size();
  5064 + for (size_t i = 0; i < num_pages; i += QIntC::to_size(o.split_pages))
  5065 + {
  5066 + size_t first = i + 1;
  5067 + size_t last = i + QIntC::to_size(o.split_pages);
  5068 + if (last > num_pages)
5052 { 5069 {
5053 - QTC::TC("qpdf", "qpdf split-pages other");  
5054 - before = std::string(o.outfilename) + "-"; 5070 + last = num_pages;
5055 } 5071 }
5056 -  
5057 - if (! o.preserve_unreferenced_page_resources) 5072 + QPDF outpdf;
  5073 + outpdf.emptyPDF();
  5074 + for (size_t pageno = first; pageno <= last; ++pageno)
5058 { 5075 {
5059 - QPDFPageDocumentHelper dh(pdf);  
5060 - dh.removeUnreferencedResources(); 5076 + QPDFObjectHandle page = pages.at(pageno - 1);
  5077 + outpdf.addPage(page, false);
5061 } 5078 }
5062 - QPDFPageLabelDocumentHelper pldh(pdf);  
5063 - std::vector<QPDFObjectHandle> const& pages = pdf.getAllPages();  
5064 - size_t pageno_len = QUtil::uint_to_string(pages.size()).length();  
5065 - size_t num_pages = pages.size();  
5066 - for (size_t i = 0; i < num_pages; i += QIntC::to_size(o.split_pages)) 5079 + if (pldh.hasPageLabels())
5067 { 5080 {
5068 - size_t first = i + 1;  
5069 - size_t last = i + QIntC::to_size(o.split_pages);  
5070 - if (last > num_pages)  
5071 - {  
5072 - last = num_pages;  
5073 - }  
5074 - QPDF outpdf;  
5075 - outpdf.emptyPDF();  
5076 - for (size_t pageno = first; pageno <= last; ++pageno)  
5077 - {  
5078 - QPDFObjectHandle page = pages.at(pageno - 1);  
5079 - outpdf.addPage(page, false);  
5080 - }  
5081 - if (pldh.hasPageLabels())  
5082 - {  
5083 - std::vector<QPDFObjectHandle> labels;  
5084 - pldh.getLabelsForPageRange(  
5085 - QIntC::to_longlong(first - 1),  
5086 - QIntC::to_longlong(last - 1),  
5087 - 0, labels);  
5088 - QPDFObjectHandle page_labels =  
5089 - QPDFObjectHandle::newDictionary();  
5090 - page_labels.replaceKey(  
5091 - "/Nums", QPDFObjectHandle::newArray(labels));  
5092 - outpdf.getRoot().replaceKey("/PageLabels", page_labels);  
5093 - }  
5094 - std::string page_range =  
5095 - QUtil::uint_to_string(first, QIntC::to_int(pageno_len));  
5096 - if (o.split_pages > 1)  
5097 - {  
5098 - page_range += "-" +  
5099 - QUtil::uint_to_string(last, QIntC::to_int(pageno_len));  
5100 - }  
5101 - std::string outfile = before + page_range + after;  
5102 - QPDFWriter w(outpdf, outfile.c_str());  
5103 - set_writer_options(outpdf, o, w);  
5104 - w.write();  
5105 - if (o.verbose)  
5106 - {  
5107 - std::cout << whoami << ": wrote file " << outfile << std::endl;  
5108 - } 5081 + std::vector<QPDFObjectHandle> labels;
  5082 + pldh.getLabelsForPageRange(
  5083 + QIntC::to_longlong(first - 1),
  5084 + QIntC::to_longlong(last - 1),
  5085 + 0, labels);
  5086 + QPDFObjectHandle page_labels =
  5087 + QPDFObjectHandle::newDictionary();
  5088 + page_labels.replaceKey(
  5089 + "/Nums", QPDFObjectHandle::newArray(labels));
  5090 + outpdf.getRoot().replaceKey("/PageLabels", page_labels);
5109 } 5091 }
5110 - }  
5111 - else  
5112 - {  
5113 - if (strcmp(o.outfilename, "-") == 0) 5092 + std::string page_range =
  5093 + QUtil::uint_to_string(first, QIntC::to_int(pageno_len));
  5094 + if (o.split_pages > 1)
5114 { 5095 {
5115 - o.outfilename = 0; 5096 + page_range += "-" +
  5097 + QUtil::uint_to_string(last, QIntC::to_int(pageno_len));
5116 } 5098 }
5117 - QPDFWriter w(pdf, o.outfilename);  
5118 - set_writer_options(pdf, o, w); 5099 + std::string outfile = before + page_range + after;
  5100 + QPDFWriter w(outpdf, outfile.c_str());
  5101 + set_writer_options(outpdf, o, w);
5119 w.write(); 5102 w.write();
5120 if (o.verbose) 5103 if (o.verbose)
5121 { 5104 {
5122 - std::cout << whoami << ": wrote file "  
5123 - << o.outfilename << std::endl; 5105 + std::cout << whoami << ": wrote file " << outfile << std::endl;
5124 } 5106 }
5125 } 5107 }
5126 } 5108 }
5127 5109
  5110 +static void write_outfile(QPDF& pdf, Options& o)
  5111 +{
  5112 + if (strcmp(o.outfilename, "-") == 0)
  5113 + {
  5114 + o.outfilename = 0;
  5115 + }
  5116 + QPDFWriter w(pdf, o.outfilename);
  5117 + set_writer_options(pdf, o, w);
  5118 + w.write();
  5119 + if (o.verbose)
  5120 + {
  5121 + std::cout << whoami << ": wrote file "
  5122 + << o.outfilename << std::endl;
  5123 + }
  5124 +}
  5125 +
5128 int realmain(int argc, char* argv[]) 5126 int realmain(int argc, char* argv[])
5129 { 5127 {
5130 whoami = QUtil::getWhoami(argv[0]); 5128 whoami = QUtil::getWhoami(argv[0]);
@@ -5162,6 +5160,10 @@ int realmain(int argc, char* argv[]) @@ -5162,6 +5160,10 @@ int realmain(int argc, char* argv[])
5162 { 5160 {
5163 do_inspection(pdf, o); 5161 do_inspection(pdf, o);
5164 } 5162 }
  5163 + else if (o.split_pages)
  5164 + {
  5165 + do_split_pages(pdf, o);
  5166 + }
5165 else 5167 else
5166 { 5168 {
5167 write_outfile(pdf, o); 5169 write_outfile(pdf, o);