pdfwriter.h

//
//  pdfwriter.h
//  Scribus
//
//  Created by Andreas Vox on 25.12.14.
//
//

#ifndef Scribus_pdfwriter_h
#define Scribus_pdfwriter_h

#include <QByteArray>
#include <QDataStream>
#include <QFile>
#include <QDateTime>
#include <QList>
#include <QRect>
#include <QString>

#include "pdfoptions.h"
#include "pdfstructs.h"
#include "scstreamfilter.h"

namespace Pdf
{
    
    
    bool isWhiteSpace(char c);
    bool isDelimiter(char c);
    bool isRegular(char c);
    
    QByteArray toPdfDocEncoding(QString s);
    
    QByteArray toUTF16(QString s);
    
    QByteArray toAscii(QString s);

    QByteArray toPdf(bool v);
    
    QByteArray toPdf(int v);
    
    QByteArray toPdf(uint v);
    
    QByteArray toPdf(qlonglong v);
    
    QByteArray toPdf(qulonglong v);
    
    QByteArray toPdf(double v);
    
    QByteArray toObjRef(PdfId id);
    
    QByteArray toLiteralString(QByteArray s);
    QByteArray toLiteralString(QString s);
    
    QByteArray toHexString(QByteArray s);
    
    QByteArray toHexString8(quint8 b);
    
    QByteArray toHexString16(quint16 b);
    
    QByteArray toHexString32(quint32 b);
    
    QByteArray toName(QByteArray s);
    QByteArray toName(QString s);

    QByteArray toDateString(QDateTime dt);
    
    QByteArray toRectangleArray(QRect r);
    QByteArray toRectangleArray(QRectF r);
    
    
    
    
class Writer
{
public:
    Writer();
    
    // file handling
    bool open (const QString& filename);
    QDataStream& getOutStream() { return outStream; }
    bool close(bool aborted);
    qint64 bytesWritten() { return Spool.pos(); }
    
    // encryption
    void setFileId(const QByteArray& id);
    void setEncryption(bool keylen16, const QByteArray& ownerKey, const QByteArray& userKey, int permissions);
    QByteArray encryptBytes(const QByteArray& in, PdfId objNum);
    
    QByteArray ComputeRC4Key(PdfId ObjNum);
private:
    void CalcOwnerKey(const QByteArray & Owner, const QByteArray & User);
    void CalcUserKey(const QByteArray & User, int Permission);
    QByteArray FitKey(const QByteArray & pass);
public:
    
    // writing
    void writeHeader(PDFOptions::PDFVersion vers);
    void writeXrefAndTrailer();
    void write(const QByteArray& bytes);
    void write(const Pdf::ResourceDictionary& dict);
    void write(const PdfFont font);

    // objects
    PdfId reserveObjects(unsigned int n);
    
    PdfId newObject() { return reserveObjects(1); }

    void startObj(PdfId id);
    
    PdfId startObj() {
        PdfId res = newObject(); startObj(res); return res;
    }
    
    void endObj(PdfId id);
    void endObjectWithStream(bool encrypted, PdfId id, const QByteArray& streamContent);
    ScStreamFilter* openStreamFilter(bool encrypted, PdfId objId);
    
    
    // private:
    PdfId CatalogObj; // 1
    PdfId InfoObj; // 2
    PdfId OutlinesObj; // 3
    PdfId PagesObj; // 4
    PdfId DestsObj; // 5
    PdfId AcroFormObj; // 6
    PdfId NamesObj; // 7
    PdfId ThreadsObj; // 8
    PdfId OCPropertiesObj; // (9)
    PdfId OutputIntentObj; // (9 or 10)
    PdfId EncryptObj; 
    PdfId MetaDataObj;
    PdfId ResourcesObj;
    
private:
    PdfId ObjCounter;
    PdfId CurrentObj;
    
    QFile Spool;
    QDataStream outStream;
    
    QList<qint64> XRef;
    
    QByteArray KeyGen;
    QByteArray OwnerKey;
    QByteArray UserKey;
    QByteArray FileID;
    QByteArray EncryKey;
    int KeyLen;

};

}
#endif