mirror of
https://github.com/pdf2htmlEX/pdf2htmlEX.git
synced 2024-09-19 05:40:07 +00:00
281 lines
7.8 KiB
C++
281 lines
7.8 KiB
C++
/*
|
|
* general.cc
|
|
*
|
|
* Hanlding general stuffs
|
|
*
|
|
* by WangLu
|
|
* 2012.08.14
|
|
*/
|
|
|
|
#include <cstdio>
|
|
|
|
#include <splash/SplashBitmap.h>
|
|
|
|
#include "HTMLRenderer.h"
|
|
#include "BackgroundRenderer.h"
|
|
#include "namespace.h"
|
|
#include "ff.h"
|
|
#include "pdf2htmlEX-config.h"
|
|
|
|
using std::fixed;
|
|
using std::flush;
|
|
|
|
static void dummy(void *, enum ErrorCategory, int pos, char *)
|
|
{
|
|
}
|
|
|
|
HTMLRenderer::HTMLRenderer(const Param * param)
|
|
:line_opened(false)
|
|
,line_buf(this)
|
|
,image_count(0)
|
|
,param(param)
|
|
,dest_dir(param->dest_dir)
|
|
,tmp_dir(param->tmp_dir)
|
|
{
|
|
//disable error function of poppler
|
|
setErrorCallback(&dummy, nullptr);
|
|
|
|
ff_init();
|
|
cur_mapping = new int32_t [0x10000];
|
|
cur_mapping2 = new char* [0x100];
|
|
}
|
|
|
|
HTMLRenderer::~HTMLRenderer()
|
|
{
|
|
ff_fin();
|
|
clean_tmp_files();
|
|
delete [] cur_mapping;
|
|
delete [] cur_mapping2;
|
|
}
|
|
|
|
static GBool annot_cb(Annot *, void *) {
|
|
return false;
|
|
};
|
|
|
|
void HTMLRenderer::process(PDFDoc *doc)
|
|
{
|
|
xref = doc->getXRef();
|
|
|
|
cerr << "Preprocessing: ";
|
|
for(int i = param->first_page; i <= param->last_page ; ++i)
|
|
{
|
|
doc->displayPage(&font_preprocessor, i, param->h_dpi, param->v_dpi,
|
|
0, true, false, false,
|
|
nullptr, nullptr, nullptr, nullptr);
|
|
cerr << "." << flush;
|
|
}
|
|
cerr << endl;
|
|
|
|
cerr << "Working: ";
|
|
BackgroundRenderer * bg_renderer = nullptr;
|
|
if(param->process_nontext)
|
|
{
|
|
// Render non-text objects as image
|
|
// copied from poppler
|
|
SplashColor color;
|
|
color[0] = color[1] = color[2] = 255;
|
|
|
|
bg_renderer = new BackgroundRenderer(splashModeRGB8, 4, gFalse, color);
|
|
bg_renderer->startDoc(doc);
|
|
}
|
|
|
|
pre_process();
|
|
|
|
for(int i = param->first_page; i <= param->last_page ; ++i)
|
|
{
|
|
if(param->process_nontext)
|
|
{
|
|
doc->displayPage(bg_renderer, i, param->h_dpi, param->v_dpi,
|
|
0, true, false, false,
|
|
nullptr, nullptr, &annot_cb, nullptr);
|
|
|
|
{
|
|
auto fn = str_fmt("%s/p%llx.png", (param->single_html ? tmp_dir : dest_dir).c_str(), i);
|
|
if(param->single_html)
|
|
add_tmp_file((char*)fn);
|
|
|
|
bg_renderer->getBitmap()->writeImgFile(splashFormatPng,
|
|
(char*)fn,
|
|
param->h_dpi, param->v_dpi);
|
|
}
|
|
}
|
|
|
|
doc->displayPage(this, i, param->zoom * DEFAULT_DPI, param->zoom * DEFAULT_DPI,
|
|
0, true, false, false,
|
|
nullptr, nullptr, nullptr, nullptr);
|
|
|
|
cerr << "." << flush;
|
|
}
|
|
post_process();
|
|
|
|
if(bg_renderer)
|
|
delete bg_renderer;
|
|
|
|
cerr << endl;
|
|
}
|
|
|
|
void HTMLRenderer::pre_process()
|
|
{
|
|
// we may output utf8 characters, so use binary
|
|
if(param->single_html)
|
|
{
|
|
{
|
|
auto fn = str_fmt("%s/%s", tmp_dir.c_str(), CSS_FILENAME.c_str());
|
|
allcss_fout.open((char*)fn, ofstream::binary);
|
|
add_tmp_file((char*)fn);
|
|
}
|
|
|
|
{
|
|
// don't use output_file directly
|
|
// otherwise it'll be a disaster when tmp_dir == dest_dir
|
|
auto tmp_output_fn = str_fmt("%s/%s.part", tmp_dir.c_str(), param->output_filename.c_str());
|
|
add_tmp_file((char*)tmp_output_fn);
|
|
|
|
html_fout.open((char*)tmp_output_fn, ofstream::binary);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
html_fout.open(str_fmt("%s/%s", dest_dir.c_str(), param->output_filename.c_str()), ofstream::binary);
|
|
allcss_fout.open(str_fmt("%s/%s", dest_dir.c_str(), CSS_FILENAME.c_str()), ofstream::binary);
|
|
|
|
html_fout << ifstream(str_fmt("%s/%s", PDF2HTMLEX_DATA_PATH.c_str(), HEAD_HTML_FILENAME.c_str()), ifstream::binary).rdbuf();
|
|
html_fout << "<link rel=\"stylesheet\" type=\"text/css\" href=\"" << CSS_FILENAME << "\"/>" << endl;
|
|
html_fout << ifstream(str_fmt("%s/%s", PDF2HTMLEX_DATA_PATH.c_str(), NECK_HTML_FILENAME.c_str()), ifstream::binary).rdbuf();
|
|
}
|
|
|
|
fix_stream(html_fout);
|
|
fix_stream(allcss_fout);
|
|
|
|
allcss_fout << ifstream(str_fmt("%s/%s", PDF2HTMLEX_DATA_PATH.c_str(), CSS_FILENAME.c_str()), ifstream::binary).rdbuf();
|
|
}
|
|
|
|
void HTMLRenderer::post_process()
|
|
{
|
|
if(!param->single_html)
|
|
{
|
|
html_fout << ifstream(str_fmt("%s/%s", PDF2HTMLEX_DATA_PATH.c_str(), TAIL_HTML_FILENAME.c_str()), ifstream::binary).rdbuf();
|
|
}
|
|
|
|
html_fout.close();
|
|
allcss_fout.close();
|
|
|
|
if(param->single_html)
|
|
{
|
|
process_single_html();
|
|
}
|
|
}
|
|
|
|
void HTMLRenderer::startPage(int pageNum, GfxState *state)
|
|
{
|
|
this->pageNum = pageNum;
|
|
this->pageWidth = state->getPageWidth();
|
|
this->pageHeight = state->getPageHeight();
|
|
|
|
assert((!line_opened) && "Open line in startPage detected!");
|
|
|
|
html_fout << "<div class=\"b\" style=\"width:" << pageWidth << "px;height:" << pageHeight << "px;\">"
|
|
<< "<div id=\"p" << pageNum << "\" class=\"p\" style=\"width:" << pageWidth << "px;height:" << pageHeight << "px;";
|
|
|
|
html_fout << "background-image:url(";
|
|
|
|
{
|
|
if(param->single_html)
|
|
{
|
|
html_fout << "'data:image/png;base64," << base64stream(ifstream(str_fmt("%s/p%llx.png", tmp_dir.c_str(), pageNum) , ifstream::binary)) << "'";
|
|
}
|
|
else
|
|
{
|
|
html_fout << str_fmt("p%llx.png", pageNum);
|
|
}
|
|
}
|
|
|
|
html_fout << ");background-position:0 0;background-size:" << pageWidth << "px " << pageHeight << "px;background-repeat:no-repeat;\">";
|
|
|
|
draw_scale = 1.0;
|
|
|
|
cur_font_info = install_font(nullptr);
|
|
cur_font_size = draw_font_size = 0;
|
|
cur_fs_id = install_font_size(cur_font_size);
|
|
|
|
memcpy(cur_ctm, id_matrix, sizeof(cur_ctm));
|
|
memcpy(draw_ctm, id_matrix, sizeof(draw_ctm));
|
|
cur_tm_id = install_transform_matrix(draw_ctm);
|
|
|
|
cur_letter_space = cur_word_space = 0;
|
|
cur_ls_id = install_letter_space(cur_letter_space);
|
|
cur_ws_id = install_word_space(cur_word_space);
|
|
|
|
cur_color.r = cur_color.g = cur_color.b = 0;
|
|
cur_color_id = install_color(&cur_color);
|
|
|
|
cur_rise = 0;
|
|
cur_rise_id = install_rise(cur_rise);
|
|
|
|
cur_tx = cur_ty = 0;
|
|
draw_tx = draw_ty = 0;
|
|
|
|
reset_state_change();
|
|
all_changed = true;
|
|
}
|
|
|
|
void HTMLRenderer::endPage() {
|
|
close_line();
|
|
// close page
|
|
html_fout << "</div></div>" << endl;
|
|
}
|
|
|
|
void HTMLRenderer::process_single_html()
|
|
{
|
|
ofstream out (dest_dir + "/" + param->output_filename, ofstream::binary);
|
|
|
|
out << ifstream(PDF2HTMLEX_DATA_PATH + "/" + HEAD_HTML_FILENAME , ifstream::binary).rdbuf();
|
|
|
|
out << "<style type=\"text/css\">" << endl;
|
|
out << ifstream(tmp_dir + "/" + CSS_FILENAME, ifstream::binary).rdbuf();
|
|
out << "</style>" << endl;
|
|
|
|
out << ifstream(PDF2HTMLEX_DATA_PATH + "/" + NECK_HTML_FILENAME, ifstream::binary).rdbuf();
|
|
|
|
out << ifstream(tmp_dir + "/" + (param->output_filename + ".part"), ifstream::binary).rdbuf();
|
|
|
|
out << ifstream(PDF2HTMLEX_DATA_PATH + "/" + TAIL_HTML_FILENAME, ifstream::binary).rdbuf();
|
|
}
|
|
|
|
void HTMLRenderer::fix_stream (std::ostream & out)
|
|
{
|
|
out << fixed << hex;
|
|
}
|
|
|
|
void HTMLRenderer::add_tmp_file(const string & fn)
|
|
{
|
|
if(!param->clean_tmp)
|
|
return;
|
|
|
|
if(tmp_files.insert(fn).second && param->debug)
|
|
cerr << "Add new temporary file: " << fn << endl;
|
|
}
|
|
|
|
void HTMLRenderer::clean_tmp_files()
|
|
{
|
|
if(!param->clean_tmp)
|
|
return;
|
|
|
|
for(auto iter = tmp_files.begin(); iter != tmp_files.end(); ++iter)
|
|
{
|
|
const string & fn = *iter;
|
|
remove(fn.c_str());
|
|
if(param->debug)
|
|
cerr << "Remove temporary file: " << fn << endl;
|
|
}
|
|
|
|
remove(tmp_dir.c_str());
|
|
if(param->debug)
|
|
cerr << "Remove temporary directory: " << tmp_dir << endl;
|
|
}
|
|
|
|
const std::string HTMLRenderer::HEAD_HTML_FILENAME = "head.html";
|
|
const std::string HTMLRenderer::NECK_HTML_FILENAME = "neck.html";
|
|
const std::string HTMLRenderer::TAIL_HTML_FILENAME = "tail.html";
|
|
const std::string HTMLRenderer::CSS_FILENAME = "all.css";
|