JBIG2Document.h

/*
 * Copyright © 2016-2018 Stefan Kebekus <stefan.kebekus@math.uni-freiburg.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 3 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, see <http://www.gnu.org/licenses/>.
 */
 
 
#ifndef JBIG2DOCUMENT
#define JBIG2DOCUMENT 1
 
#include <QSet>
 
#include "imageInfo.h"
#include "JBIG2Segment.h"
 
 
class JBIG2Document
{
 public:
  inline JBIG2Document() {clear();};
  
  explicit inline JBIG2Document(QIODevice *device) {read(device);};
  
  explicit inline JBIG2Document(QString fileName) {read(fileName);};
  
  void clear();
  
  void read(QIODevice *device);
 
  
  void read(const QString& fileName);
  
  
  inline bool isEmpty() const
  {
    QListIterator<JBIG2Segment> i(segments);
    while (i.hasNext()) {
      quint32 number = i.next().pageAssociation();
      if (number != 0)
       return false;
    }
    return true;
  };
  
  inline bool hasError() const {return !_error.isEmpty();};
  
  inline QString error() const {return _error; };
 
  inline bool hasWarnings() const {return !_warnings.isEmpty();};
 
  inline QSet<QString> warnings() const {return _warnings;};
 
  QList<quint32> pageNumbers() const;
 
 
  QString info() const;
 
  
  QByteArray getPDFDataChunk(quint32 pageNumber) const;
  
 
  imageInfo pageInfo(quint32 pageNumber) const;
 
  QImage operator[](quint32 pageNumber) const;
  
 private:
  // Checks for fixable errors, fixes them, and sets '_warnings' accordingly
  void check_n_fix();
  
  // Helper function for image rendering
  static void myImageCleanupHandler(void *info);
  
  // Returns list of numbers of those segements that refer to given
  // segmentNumber
  QList<quint32> referrers(quint32 segmentNumber);
 
  // Warnings
  QSet<QString> _warnings;
 
  // Error
  QString _error;
 
  // List of segments that make up the JBIG2 file
  QList<JBIG2Segment> segments;
};
 
#endif