Embed pdf inside mscz

Author: benwaffle

src/engraving/infrastructure/mscwriter.cpp

@@ -183,6 +183,11 @@ void MscWriter::writeThumbnailFile(const ByteArray& data)
     addFileData(u"Thumbnails/thumbnail.png", data);
 }
 
+void MscWriter::writePdfFile(const ByteArray& data)
+{
+    addFileData(u"preview.pdf", data);
+}
+
 void MscWriter::addImageFile(const String& fileName, const ByteArray& data)
 {
     addFileData(u"Pictures/" + fileName, data);

src/engraving/infrastructure/mscwriter.h

@@ -63,6 +63,7 @@ class MscWriter
     void addExcerptFile(const muse::String& excerptFileName, const muse::ByteArray& data);
     void writeChordListFile(const muse::ByteArray& data);
     void writeThumbnailFile(const muse::ByteArray& data);
+    void writePdfFile(const muse::ByteArray& data);
     void addImageFile(const muse::String& fileName, const muse::ByteArray& data);
     void writeAudioFile(const muse::ByteArray& data);
     void writeAudioSettingsJsonFile(const muse::ByteArray& data, const muse::io::path_t& pathPrefix = "");

src/engraving/rw/mscsaver.cpp

@@ -28,17 +28,69 @@
 #include "dom/imageStore.h"
 #include "dom/audio.h"
 
+#include "draw/painter.h"
+
 #include "rwregister.h"
 #include "inoutdata.h"
 
 #include "log.h"
 
+#ifndef NO_QT_SUPPORT
+#include <QPdfWriter>
+#endif
+
 using namespace mu;
 using namespace muse;
 using namespace muse::io;
+using namespace draw;
 using namespace mu::engraving;
 using namespace mu::engraving::rw;
 
+#ifndef NO_QT_SUPPORT
+static void writePdfPreview(MasterScore* score, io::IODevice* pdfBuf)
+{
+    QByteArray qdata;
+    QBuffer buf(&qdata);
+    buf.open(QIODevice::WriteOnly);
+
+    QPdfWriter pdfWriter(&buf);
+
+    pdfWriter.setResolution(mu::engraving::DPI);
+    pdfWriter.setCreator(QString("MuseScore Studio Version: ") + QString(score->appVersion()));
+    pdfWriter.setPageMargins(QMarginsF());
+
+    SizeF size = score->renderer()->pageSizeInch(score);
+    pdfWriter.setPageLayout(QPageLayout(QPageSize(size.toQSizeF(), QPageSize::Inch), QPageLayout::Orientation::Portrait, QMarginsF()));
+
+    QString title = score->metaTag(u"workTitle");
+    if (title.isEmpty()) {
+        title = score->name();
+    }
+    pdfWriter.setTitle(title);
+
+    Painter painter(&pdfWriter, "pdfwriter");
+
+    rendering::IScoreRenderer::PaintOptions opt;
+    opt.deviceDpi = pdfWriter.logicalDpiX();
+    opt.onNewPage = [&pdfWriter]() { pdfWriter.newPage(); };
+    opt.printPageBackground = true;
+    opt.isSetViewport = true;
+    opt.isMultiPage = false;
+    opt.isPrinting = true;
+    opt.onPaintPageSheet = [](Painter* painter, const Page* page, const RectF& pageRect) {
+        UNUSED(page);
+        painter->fillRect(pageRect, Color::WHITE);
+    };
+
+    score->renderer()->paintScore(&painter, score, opt);
+
+    painter.endDraw();
+
+    pdfBuf->write(qdata);
+}
+
+#endif
+
 bool MscSaver::writeMscz(MasterScore* score, MscWriter& mscWriter, bool onlySelection, bool doCreateThumbnail)
 {
     TRACEFUNC;
@@ -147,6 +199,18 @@ bool MscSaver::writeMscz(MasterScore* score, MscWriter& mscWriter, bool onlySele
         }
     }
 
+    // Write PDF file, used for macOS Quick Look
+    // TODO: do we always want this? e.g. when is doCreateThumbnail false?
+    #ifndef NO_QT_SUPPORT
+    {
+        ByteArray pdfData;
+        Buffer pdfBuf(&pdfData);
+        pdfBuf.open(IODevice::WriteOnly);
+        writePdfPreview(score, &pdfBuf);
+        mscWriter.writePdfFile(pdfData);
+    }
+    #endif
+
     // Write audio
     {
         if (score->audio()) {