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:
        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;
 
        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