ImageDiff.hpp

/* ***** BEGIN LICENSE BLOCK *****
 * FW4SPL - Copyright (C) IRCAD, 2017.
 * Distributed under the terms of the GNU Lesser General Public License (LGPL) as
 * published by the Free Software Foundation.
 * ****** END LICENSE BLOCK ****** */

#ifndef __FWDATATOOLS_IMAGEDIFF_HPP__
#define __FWDATATOOLS_IMAGEDIFF_HPP__

#include "fwDataTools/config.hpp"
#include "fwDataTools/helper/Image.hpp"

#include <fwData/Image.hpp>

namespace fwDataTools
{

class FWDATATOOLS_CLASS_API ImageDiff
{
public:

    struct ElementType
    {
        ::fwData::Image::IndexType m_index;
        ::fwData::Image::BufferType* m_oldValue;
        ::fwData::Image::BufferType* m_newValue;
    };

    FWDATATOOLS_API ImageDiff(const size_t imageElementSize = 0, const size_t reservedElements = 0);

    FWDATATOOLS_API ~ImageDiff();

    FWDATATOOLS_API ImageDiff(const ImageDiff& other);

    FWDATATOOLS_API ImageDiff(ImageDiff&& other);

    FWDATATOOLS_API ImageDiff& operator= (const ImageDiff& other);

    FWDATATOOLS_API ImageDiff& operator= (ImageDiff&& other);

    FWDATATOOLS_API void addDiff(const ImageDiff& diff);

    FWDATATOOLS_API void addDiff(const ::fwData::Image::IndexType index, const ::fwData::Image::BufferType* oldValue,
                                 const ::fwData::Image::BufferType* newValue);

    FWDATATOOLS_API void applyDiff(const ::fwData::Image::sptr& img) const;

    FWDATATOOLS_API void revertDiff(const ::fwData::Image::sptr& img) const;

    FWDATATOOLS_API size_t getSize() const;

    FWDATATOOLS_API size_t getNumberOfElements() const;

    FWDATATOOLS_API void clear();

    FWDATATOOLS_API void shrink();

    FWDATATOOLS_API ElementType getElement(size_t index) const;

    inline ::fwData::Image::IndexType getElementDiffIndex(size_t eltIndex) const;

private:

    void applyDiffElt(helper::Image& img, size_t eltIndex) const;

    void revertDiffElt(helper::Image& img, size_t eltIndex) const;

    size_t m_imgEltSize;

    size_t m_eltSize;

    size_t m_nbElts;

    size_t m_reservedSize;

    std::uint8_t* m_buffer;
};

//------------------------------------------------------------------------------

fwData::Image::IndexType ImageDiff::getElementDiffIndex(size_t eltIndex) const
{
    std::uint8_t* eltPtr = m_buffer + eltIndex * m_eltSize;
    return *reinterpret_cast< ::fwData::Image::IndexType* >(eltPtr);
}

} // namespace fwDataTools

#endif // __FWDATATOOLS_IMAGEDIFF_HPP__