diff options
Diffstat (limited to 'src/fileexporterpdf.cpp')
| -rw-r--r-- | src/fileexporterpdf.cpp | 197 |
1 files changed, 197 insertions, 0 deletions
diff --git a/src/fileexporterpdf.cpp b/src/fileexporterpdf.cpp new file mode 100644 index 0000000..47b8dad --- /dev/null +++ b/src/fileexporterpdf.cpp @@ -0,0 +1,197 @@ +/*************************************************************************** +* Copyright (C) 2004-2009 by Thomas Fischer * +* fischer@unix-ag.uni-kl.de * +* * +* This program is free software; you can redistribute it and/or modify * +* it under the terms of the GNU General Public License as published by * +* the Free Software Foundation; either version 2 of the License, or * +* (at your option) any later version. * +* * +* This program is distributed in the hope that it will be useful, * +* but WITHOUT ANY WARRANTY; without even the implied warranty of * +* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * +* GNU General Public License for more details. * +* * +* You should have received a copy of the GNU General Public License * +* along with this program; if not, write to the * +* Free Software Foundation, Inc., * +* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. * +***************************************************************************/ +#include <qfile.h> +#include <qstringlist.h> +#include <qurl.h> + +#include <settings.h> +#include <element.h> +#include <entry.h> +#include <fileexporterbibtex.h> +#include "fileexporterpdf.h" + +namespace BibTeX +{ + + FileExporterPDF::FileExporterPDF( bool embedFiles ) + : FileExporterToolchain(), m_latexLanguage( "english" ), m_latexBibStyle( "plain" ), m_embedFiles( embedFiles ) + { + laTeXFilename = QString( workingDir ).append( "/bibtex-to-pdf.tex" ); + bibTeXFilename = QString( workingDir ).append( "/bibtex-to-pdf.bib" ); + outputFilename = QString( workingDir ).append( "/bibtex-to-pdf.pdf" ); + } + + FileExporterPDF::~FileExporterPDF() + { + // nothing + } + + bool FileExporterPDF::save( QIODevice* iodevice, const File* bibtexfile, QStringList *errorLog ) + { + m_mutex.lock(); + bool result = FALSE; + m_embeddedFileList.clear(); + if ( m_embedFiles ) + { + m_embeddedFileList.append( QString( "%1|%2" ).arg( "BibTeX source" ).arg( bibTeXFilename ) ); + fillEmbeddedFileList( bibtexfile ); + } + + QFile bibtexFile( bibTeXFilename ); + if ( bibtexFile.open( IO_WriteOnly ) ) + { + FileExporter * bibtexExporter = new FileExporterBibTeX(); + result = bibtexExporter->save( &bibtexFile, bibtexfile, errorLog ); + bibtexFile.close(); + delete bibtexExporter; + } + + if ( result ) + result = generatePDF( iodevice, errorLog ); + + m_mutex.unlock(); + return result; + } + + bool FileExporterPDF::save( QIODevice* iodevice, const Element* element, QStringList *errorLog ) + { + m_mutex.lock(); + bool result = FALSE; + m_embeddedFileList.clear(); + if ( m_embedFiles ) + fillEmbeddedFileList( element ); + + QFile bibtexFile( bibTeXFilename ); + if ( bibtexFile.open( IO_WriteOnly ) ) + { + FileExporter * bibtexExporter = new FileExporterBibTeX(); + result = bibtexExporter->save( &bibtexFile, element, errorLog ); + bibtexFile.close(); + delete bibtexExporter; + } + + if ( result ) + result = generatePDF( iodevice, errorLog ); + + m_mutex.unlock(); + return result; + } + + void FileExporterPDF::setLaTeXLanguage( const QString& language ) + { + m_latexLanguage = language; + } + + void FileExporterPDF::setLaTeXBibliographyStyle( const QString& bibStyle ) + { + m_latexBibStyle = bibStyle; + } + + void FileExporterPDF::setDocumentSearchPaths( const QStringList& searchPaths ) + { + m_searchPaths = searchPaths; + } + + bool FileExporterPDF::generatePDF( QIODevice* iodevice, QStringList *errorLog ) + { + QStringList cmdLines = QStringList::split( '|', "pdflatex -halt-on-error bibtex-to-pdf.tex|bibtex bibtex-to-pdf|pdflatex -halt-on-error bibtex-to-pdf.tex|pdflatex -halt-on-error bibtex-to-pdf.tex" ); + + if ( writeLatexFile( laTeXFilename ) && runProcesses( cmdLines, errorLog ) && writeFileToIODevice( outputFilename, iodevice ) ) + return TRUE; + else + return FALSE; + } + + bool FileExporterPDF::writeLatexFile( const QString &filename ) + { + QFile latexFile( filename ); + if ( latexFile.open( IO_WriteOnly ) ) + { + m_embedFiles &= kpsewhich( "embedfile.sty" ); + QTextStream ts( &latexFile ); + ts.setEncoding( QTextStream::UnicodeUTF8 ); + ts << "\\documentclass{article}\n"; + if ( kpsewhich( "t1enc.dfu" ) ) + ts << "\\usepackage[T1]{fontenc}\n"; + if ( kpsewhich( "babel.sty" ) ) + ts << "\\usepackage[" << m_latexLanguage << "]{babel}\n"; + if ( kpsewhich( "hyperref.sty" ) ) + ts << "\\usepackage[pdfproducer={KBibTeX: http://www.t-fischer.net/kbibtex/},pdftex]{hyperref}\n"; + else if ( kpsewhich( "url.sty" ) ) + ts << "\\usepackage{url}\n"; + if ( m_latexBibStyle.startsWith( "apacite" ) && kpsewhich( "apacite.sty" ) ) + ts << "\\usepackage[bibnewpage]{apacite}\n"; + if ( m_embedFiles ) + ts << "\\usepackage{embedfile}\n"; + ts << "\\bibliographystyle{" << m_latexBibStyle << "}\n"; + ts << "\\begin{document}\n"; + + if ( m_embedFiles ) + for ( QStringList::ConstIterator it = m_embeddedFileList.begin(); it != m_embeddedFileList.end(); ++it ) + { + QStringList param = QStringList::split( "|", *it ); + QFile file( param[1] ); + if ( file.exists() ) + ts << "\\embedfile[desc={" << param[0] << "}]{" << param[1] << "}\n"; + } + + ts << "\\nocite{*}\n"; + ts << "\\bibliography{bibtex-to-pdf}\n"; + ts << "\\end{document}\n"; + latexFile.close(); + return TRUE; + } + else + return FALSE; + } + + void FileExporterPDF::fillEmbeddedFileList( const File* bibtexfile ) + { + for ( BibTeX::File::ElementList::ConstIterator it = bibtexfile->constBegin(); it != bibtexfile->constEnd(); ++it ) + fillEmbeddedFileList( *it ); + } + + void FileExporterPDF::fillEmbeddedFileList( const Element* element ) + { + const Entry *entry = dynamic_cast<const Entry*>( element ); + if ( entry != NULL ) + { + QString id = entry->id(); + QStringList urls = entry->urls(); + for ( QStringList::Iterator it = urls.begin(); it != urls.end(); ++it ) + { + QUrl url = QUrl( *it ); + if ( url.isValid() && url.isLocalFile() && !( *it ).endsWith( "/" ) && QFile( url.path() ).exists() ) + m_embeddedFileList.append( QString( "%1|%2" ).arg( id ).arg( url.path() ) ); + else + for ( QStringList::Iterator path_it = m_searchPaths.begin(); path_it != m_searchPaths.end(); ++path_it ) + { + url = QUrl( QString( *path_it ).append( "/" ).append( *it ) ); + if ( url.isValid() && url.isLocalFile() && !( *it ).endsWith( "/" ) && QFile( url.path() ).exists() ) + { + m_embeddedFileList.append( QString( "%1|%2" ).arg( id ).arg( url.path() ) ); + break; + } + } + } + } + } + +} |