Mercurial > clickerconvert
view src/converter.cpp @ 31:7bc75b1ff367
Removed tag 0.8
author | Andre Heinecke <andre.heinecke@intevation.de> |
---|---|
date | Mon, 11 Apr 2016 13:37:37 +0200 |
parents | 5acd601356ba |
children | ad54c98cb8d8 |
line wrap: on
line source
/* Copyright (C) 2016 by ETH Zürich * Software engineering by Intevation GmbH * * This file is Free Software under the GNU GPL (v>=2) * and comes with ABSOLUTELY NO WARRANTY! * See LICENSE.txt for details. */ #include "converter.h" #include <QDebug> #include <QRegularExpression> #include <QRegularExpressionMatch> #include <QTextDocument> #include <QPrinter> #include <QImage> #include <QPainter> #include "xlsxdocument.h" #include "xlsxconditionalformatting.h" #include "constants.h" QTXLSX_USE_NAMESPACE Converter::Converter(const QString &input, const QString &output, ConvertFormat fmt, const QString &title): QThread(Q_NULLPTR), mInput(input), mOutput(output), mFmt(fmt), mTitle(title) { mTitleFmt.setFontUnderline(Format::FontUnderlineSingle); mTitleFmt.setFontSize(18); mTitleFmt.setFontName("Calibri"); mTitleFmt.setFontBold(true); mTitleFmt.setVerticalAlignment(Format::AlignTop); mQuestionFmt.setFontSize(11); mQuestionFmt.setFontName("Calibri"); mQuestionFmt.setFontBold(true); mQuestionFmt.setTopBorderStyle(Format::BorderThin); mQuestionFmt.setBottomBorderStyle(Format::BorderThin); mQuestionFmt.setTextWarp(true); mAnswerChoiceFmt.setFontSize(11); mAnswerChoiceFmt.setFontName("Calibri"); mAnswerChoiceFmt.setHorizontalAlignment(Format::AlignLeft); mAnswerChoiceFmt.setTextWarp(true); mChoiceTextFmt = mAnswerChoiceFmt; mChoiceTextFmt.setVerticalAlignment(Format::AlignVCenter); mChoiceVotesFmt = mChoiceTextFmt; mChoiceVotesFmt.setFontSize(10); mFreeTextFmt = mQuestionFmt; mFreeTextFmt.setFontBold(false); mAnswerTextFmt = mQuestionFmt; mAnswerTextFmt.setVerticalAlignment(Format::AlignVCenter); mAnswerTextFmt.setHorizontalAlignment(Format::AlignLeft); mTitleStyle = QStringLiteral("<tr><td colspan='3' style='vertical-align: top;" "font-weight: bold; text-decoration:underline; font-size: 18pt;'>" "%1</td></tr><tr/>"); mQuestionStyle = QStringLiteral("<tr><td colspan='3' style='text-decoration: underline;" "font-size: 11pt;font-weight: bold;'><hr/>%1<hr/></td></tr>"); mAnswerChoiceStyle= QStringLiteral("<tr><td colspan='3' style='text-align: left; font-size: 11pt;'>Answer</td></tr>"); mChoiceTextStyle= QStringLiteral("<tr><td align='right' style='vertical-align: middle; font-size:11pt;'>%1</td>"); mChoiceVotesStyle = QStringLiteral("<td style='text-align: left;vertical-algin: center;" "font-size:10pt;'</td>%1</td></tr>"); mAnswerTextStyle = QStringLiteral("<tr><td colspan='3' style='font-weight: bold;vertical-algin: middle;" "font-size:11pt;'>Answer<hr/></td></tr>"); mFreeTextStyle = QStringLiteral("<tr><td colspan='3'; font-size:11pt;'>%1<hr/></td></tr>"); mEmptyRow = QStringLiteral("<tr style='height: %1px'/>").arg(CHOICE_ROW_HEIGHT); } void Converter::run() { QFile infile; if (mInput.isEmpty()) { if (!infile.open(stdin, QIODevice::ReadOnly)) { mErrors << tr("Failed to open standard input and no input file provided."); return; } } else { infile.setFileName(mInput); if (!infile.open(QIODevice::ReadOnly)) { mErrors << tr("Failed to open %1 for reading.").arg(mInput); return; } } QTextStream instream(&infile); QFile outfile; if (mOutput.isEmpty()) { if (!outfile.open(stdout, QIODevice::WriteOnly)) { mErrors << tr("Failed to open standard output and no output file provided."); return; } } else { outfile.setFileName(mOutput); if (!outfile.open(QIODevice::WriteOnly)) { mErrors << tr("Failed to open %1 for writing.").arg(mOutput); return; } } convertToXSLX(instream, outfile); } static void makeBar(QTextStream &html, double percent, int width, QTextDocument &doc, bool forPDF) { static int barCnt; if (!forPDF) { html << QStringLiteral("<td style='background:linear-gradient(to right," BAR_COLOR ", " BAR_COLOR " %1%, #ffffff %1%)'></td>").arg(percent); return; } QImage image(QSize(width, 25), QImage::Format_RGB32); QPainter painter(&image); QRect rect = image.rect(); qDebug() << "Image of " << width; rect.setRight(rect.right() / (100 / percent)); painter.fillRect(rect, QColor(BAR_COLOR)); qDebug() << "Filled " << rect << " with color"; rect.setLeft(rect.right()); rect.setRight(width); painter.fillRect(rect, Qt::white); qDebug() << "Filled " << rect << " with white"; doc.addResource(QTextDocument::ImageResource, QUrl(QStringLiteral("internal://bar%1.png").arg(barCnt)), QVariant(image)); html << QStringLiteral("<td><img src=\"internal://bar%1.png\"/></td>").arg(barCnt++); image.save(QStringLiteral("/tmp/foo.png")); return; } void Converter::convertToXSLX(QTextStream& instream, QFile &output) { Document xlsx; QTextDocument doc; QString htmlString; QTextStream html (&htmlString); ConditionalFormatting bars; bars.addDataBarRule(QColor(0xFF, 0x99, 0x33), ConditionalFormatting::VOT_Num, "0", ConditionalFormatting::VOT_Num, "100", false); const double colWidth[] = COLUMN_WIDTHS; double sum = 0; for (int i = 1; i <= COLUMN_CNT; i++) { xlsx.setColumnWidth(i, colWidth[i-1]); sum += colWidth[i-1]; } double xlsx2htmlFactor = HTML_WIDTH / sum; int col1Width = colWidth[0] * xlsx2htmlFactor; int col2Width = colWidth[1] * xlsx2htmlFactor; int col3Width = colWidth[2] * xlsx2htmlFactor; /* For the merged cell wordwrap trick. */ xlsx.setColumnWidth(26, sum + 1); xlsx.setColumnHidden(26, true); int row = 1; html << "<html><meta http-equiv=\"Content-Type\" content=\"text/html;charset=UTF-8\">" "<body><table border=\"0\" style='width:\"" << HTML_WIDTH << "px\";border-collapse:collapse'>"; html << "<tr><th width=\"" << col1Width << "px\"</th>"; html << "<th width=\"" << col2Width << "px\"</th>"; html << "<th width=\"" << col3Width << "px\"</th>"; const QString title = mTitle.isEmpty() ? DEFAULT_TITLE : mTitle; // Set the title of the Questionaire xlsx.write(row++, 1, title, mTitleFmt); html << mTitleStyle.arg(title.toHtmlEscaped()); xlsx.mergeCells("A1:C1"); xlsx.setRowHeight(1, TITLE_ROW_HEIGHT); const QString input = instream.readAll(); QRegularExpression questionEx(QUESTION_PATTERN); QRegularExpression choiceEx(CHOICE_PATTERN); QRegularExpression freetxtEx (FREETXT_PATTERN); QRegularExpressionMatch match = questionEx.match(input); bool foundSomething = false; int cursor = match.capturedEnd(); while (match.hasMatch() && cursor != -1) { /* We've matched a question pattern. With the answer line */ if (!match.lastCapturedIndex() == 2) { /* Should not happen without misconfiguration. */ mErrors << "Internal parser error."; return; } foundSomething = true; const QString question = match.captured(1).trimmed(); const QString answerLine = match.captured(2).trimmed(); xlsx.write(row, 2, QString(" "), mQuestionFmt); xlsx.write(row, 3, QString(" "), mQuestionFmt); xlsx.write(row++, 1, question, mQuestionFmt); html << mQuestionStyle.arg(question.toHtmlEscaped()); if (answerLine == QStringLiteral(CHOICE_IDENTIFIER)) { QRegularExpressionMatch choiceMatch = choiceEx.match(input, cursor); xlsx.setRowHeight(row, CHOICE_ROW_HEIGHT); xlsx.write(row++, 1, "Answer", mAnswerChoiceFmt); html << mAnswerChoiceStyle; int firstChoiceRow = row; int lastChoiceRow = row; while (choiceMatch.hasMatch() && choiceMatch.capturedStart() == cursor + 1) { /* We use the cursor here to keep track of the state. Only if an answer follows immediately behind the last answer we treat it as valid as otherwise we can't figure out when the next question begins. */ cursor = choiceMatch.capturedEnd(); /* Write the values */ QString choiceName = choiceMatch.captured(1).trimmed(); if (choiceName.startsWith("=")) { choiceName = " " + choiceName; } xlsx.write(row, 1, choiceName, mChoiceTextFmt); html << mChoiceTextStyle.arg(choiceName.toHtmlEscaped()); bool ok; double percent = choiceMatch.captured(3).toDouble(&ok); if (!ok) { mErrors << "Unparsable number in string: " + choiceMatch.captured(); } makeBar(html, percent, col2Width, doc, mFmt == Format_PDF); xlsx.write(row, 2, percent); const QString numVotesString = QString("%1% | %2 Number of votes"). arg(choiceMatch.captured(3)).arg(choiceMatch.captured(2)); html << mChoiceVotesStyle.arg(numVotesString.toHtmlEscaped()); xlsx.write(row, 3, numVotesString, mChoiceVotesFmt); xlsx.setRowHeight(row, CHOICE_ROW_HEIGHT); /* As long as we can match a choice which is either before the next question or before the end of the document */ choiceMatch = choiceEx.match(input, cursor); row++; lastChoiceRow++; } bars.addRange(QString("B%1:B%2").arg(firstChoiceRow).arg(lastChoiceRow)); // xlsx.groupRows(firstChoiceRow - 2, lastChoiceRow - 1, false); } else if (answerLine == QStringLiteral(TEXT_IDENTIFIER)) { QRegularExpressionMatch textMatch = freetxtEx.match(input, cursor); xlsx.setRowHeight(row, CHOICE_ROW_HEIGHT); xlsx.write(row++, 1, "Answer", mAnswerTextFmt); html << mAnswerTextStyle; /* To handle the workaround for quotes in answers we store * the number of rows and only afterwards create the html rows. */ int firstFreeRow = row; while (textMatch.hasMatch()) { if (textMatch.capturedStart() != cursor + 1) { /* The format allows unescaped quotes in the text. This makes a workaround neccessary. If we have an Unquoted string between the next quoted string and that Unquoted string is before the next question we append the unquoted string and the next quoted string with Quotes in the Row.*/ QRegularExpressionMatch nextQuestion = questionEx.match(input, cursor); if (nextQuestion.hasMatch() && nextQuestion.capturedStart() < textMatch.capturedEnd()) { /* The next question comes before the textMatch so we really have a new question. */ break; } const QString lastRow = xlsx.read(row - 1, 26).toString(); int unquotedLen = textMatch.capturedStart() - cursor; const QString unquoted = input.mid(cursor, unquotedLen); qDebug() << "Found inner quoted string: " << unquoted; /* Now combine */ const QString combined = QString("%1\"%2\"%3").arg(lastRow). arg(unquoted). arg(textMatch.captured(1).trimmed()); qDebug() << "Last row: " << lastRow; qDebug() << "Next Question is at: " << nextQuestion.capturedStart(); qDebug() << "Text match is: " << textMatch.captured(1).trimmed(); qDebug() << "cursor is at: " << cursor; qDebug() << "text match starts at: " << textMatch.capturedStart(); xlsx.write(row - 1, 26, combined, mFreeTextFmt); xlsx.write(row - 1, 1, combined, mFreeTextFmt); cursor = textMatch.capturedEnd(); textMatch = freetxtEx.match(input, cursor); continue; } cursor = textMatch.capturedEnd(); QString text = textMatch.captured(1).trimmed(); qDebug() << "Found free text: " << text; if (text.startsWith("=")) { text = " " + text; } /* Merge the cells */ xlsx.mergeCells(QString("A%1:C%1").arg(row), mFreeTextFmt); /* Merged cells ignore wordwrap the following trick is based on: http://excel.tips.net/T003207_Automatic_Row_Height_For_Merged_Cells_with_Text_Wrap.html */ /* Write the values */ xlsx.write(QString("Z%1").arg(row), text, mFreeTextFmt); xlsx.write(row, 1, text, mFreeTextFmt); row++; textMatch = freetxtEx.match(input, cursor); } for (int i = firstFreeRow; i < row; i++) { html << mFreeTextStyle.arg(xlsx.read(i, 1).toString().toHtmlEscaped()); } } /* Insert Empty row. */ xlsx.setRowHeight(row++, CHOICE_ROW_HEIGHT); match = questionEx.match(input, cursor); cursor = match.capturedEnd(); html << mEmptyRow; } xlsx.addConditionalFormatting(bars); if (!foundSomething) { mErrors << tr("Failed to parse input document."); } if (mFmt == Format_XLSX && !xlsx.saveAs(&output)) { mErrors << tr("Saving the XLSX document failed."); return; } html << "</table></body></html>"; if (mFmt == Format_HTML) { QTextStream outstream(&output); outstream << htmlString; return; } if (mFmt == Format_PDF) { output.close(); QPrinter printer(QPrinter::PrinterResolution); printer.setOutputFormat(QPrinter::PdfFormat); printer.setPaperSize(QPrinter::A4); printer.setOutputFileName(output.fileName()); doc.setHtml(htmlString); doc.print(&printer); } }