Image.h

#pragma once
// ===========================================================================
// Copyright (c) 2015 Toolchefs Ltd. All rights reserved.
//
// Use of this software is subject to the terms of the Toolchefs license
// agreement provided at the time of installation or download, or which
// otherwise accompanies this software in either electronic or hard copy form.
// ===========================================================================
 
 
#include <AtomsUtils/Globals.h>
#include <vector>
#include <string>
 
namespace AtomsUtils
{
 
    class ATOMSUTILS_EXPORT Image
    {
    public:
 
        enum ImageType
        {
            kByte = 0,
            kFloat,
        };
 
        Image();
 
 
        Image(unsigned int width, unsigned int height, unsigned int channel, unsigned short imgType);
 
 
        Image(float* data, unsigned int width, unsigned int height, unsigned int channel);
 
 
        Image(unsigned char* data, unsigned int width, unsigned int height, unsigned int channel);
 
 
        Image(std::vector<float> data, unsigned int width, unsigned int height, unsigned int channel);
 
 
        Image(std::vector<unsigned char> data, unsigned int width, unsigned int height, unsigned int channel);
 
        ~Image();
 
        Image(const Image& other);
 
        Image& operator=(const Image& other);
 
        inline unsigned short imageType() const;
 
        inline unsigned int width() const;
 
        inline unsigned int height() const;
 
        inline unsigned int channels() const;
 
        inline const float* pixelsFloat() const;
 
        inline float* pixelsFloat();
 
        inline const unsigned char* pixels() const;
 
        inline unsigned char* pixels();
 
        void clear();
 
        void loadFromFile(const std::string& fileName);
 
 
        bool saveToFile(const std::string& fileName) const;
 
        void reverseV();
 
    private:
 
        union
        {
            float* m_pixelsFloat;
            unsigned char* m_pixels;
        };
 
        unsigned int m_width;
 
        unsigned int m_height;
 
        unsigned int m_channels;
 
        unsigned short m_type;
    };
}
 
#include "Image.impl.h"