dfufile.hh

#ifndef DFUFILE_HH
#define DFUFILE_HH
 
#include <QFile>
#include <QVector>
#include <QByteArray>
#include <QString>
#include <QTextStream>
 
#include "addressmap.hh"
#include "errorstack.hh"
 
class CRC32;

class DFUFile: public QObject
{
    Q_OBJECT
 
public:
  class Element {
  public:
    Element();
    Element(uint32_t addr, uint32_t size);
    Element(const Element &other);
    Element &operator= (const Element &other);

    uint32_t address() const;
    void setAddress(uint32_t addr);
    uint32_t size() const;
    uint32_t memSize() const;
    bool isAligned(unsigned blocksize) const;
    const QByteArray &data() const;
    QByteArray &data();

    bool read(QFile &file, CRC32 &crc, QString &errorMessage);
    bool write(QFile &file, CRC32 &crc, QString &errorMessage) const;

    void dump(QTextStream &stream) const;
 
  protected:
    uint32_t _address;
    QByteArray _data;
  };

  class Image
  {
  public:
    typedef QVector<Element>::iterator iterator;
 
  public:
    Image();
    Image(const QString &name, uint8_t altSettings=0);
    Image(const Image &other);
    virtual ~Image();
    Image &operator=(const Image &other);

    uint8_t alternateSettings() const;
    void setAlternateSettings(uint8_t s);

    bool isNamed() const;
    const QString &name() const;
    void setName(const QString &name);
    uint32_t size() const;
    uint32_t memSize() const;
    int numElements() const;
    const Element &element(int i) const;
    Element &element(int i);
    void addElement(uint32_t addr, uint32_t size, int index=-1);
    void addElement(const Element &element);
    void remElement(int i);
    bool isAligned(unsigned blocksize) const;

    iterator begin();
    iterator end();

    bool read(QFile &file, CRC32 &crc, QString &errorMessage);
    bool write(QFile &file, CRC32 &crc, QString &errorMessage) const;

    void dump(QTextStream &stream) const;

    virtual bool isAllocated(uint32_t offset) const;

    virtual unsigned char *data(uint32_t offset);
    virtual const unsigned char *data(uint32_t offset) const;

    void sort();
 
  protected:
    uint8_t  _alternate_settings;
    QString _name;
    QVector<Element> _elements;
    AddressMap _addressmap;
  };
 
public:
    DFUFile(QObject *parent=nullptr);

    uint32_t size() const;
  uint32_t memSize() const;

    int numImages() const;
    const Image &image(int i) const;
    Image &image(int i);
    void addImage(const QString &name, uint8_t altSettings=1);
    void addImage(const Image &img);
    void remImage(int i);

  bool isAligned(unsigned blocksize) const;

  bool read(const QString &filename, const ErrorStack &err=ErrorStack());
  bool read(QFile &file, const ErrorStack &err=ErrorStack());

  bool write(const QString &filename, const ErrorStack &err=ErrorStack());
  bool write(QFile &file, const ErrorStack &err=ErrorStack());

    void dump(QTextStream &stream) const;

  virtual bool isAllocated(uint32_t offset, uint32_t img=0) const;

  virtual unsigned char *data(uint32_t offset, uint32_t img=0);
  virtual const unsigned char *data(uint32_t offset, uint32_t img=0) const;
 
protected:
    QVector<Image> _images;
};
 
#endif // DFUFILE_HH