picture.hpp

Go to the documentation of this file.

#ifndef RAPP_OBJECT_PICTURE
#define RAPP_OBJECT_PICTURE
#include "includes.ihh"
namespace rapp {
namespace object {
class picture
{
public:

    picture(const std::string filepath)
    {
        std::ifstream bytestream(filepath, 
                                 std::ios::in | std::ios::binary | std::ios::ate);
        if (!bytestream.is_open())
            throw std::runtime_error("could not open bytestream for "+filepath);
        else 
            opencb_(bytestream);
    }

    picture(std::ifstream & bytestream)
    {
        opencb_(bytestream);
    }

    picture(const picture &) = default;
    
    picture & operator=(const picture &) = default;

    bool operator==(const picture & rhs) const
    {
        return (this->bytearray_ == rhs.bytearray_);
    }

    std::vector<rapp::types::byte> bytearray() const
    {
        return bytearray_;
    }

    std::string type() const
    {
        return imgtype_;
    }

    bool save(const std::string filepath)
    {
        std::ofstream os(filepath, std::ios::out | std::ofstream::binary);
        if (os.is_open())
        {
            std::copy(bytearray_.begin(), bytearray_.end(), 
                      std::ostreambuf_iterator<rapp::types::byte>(os));
            os.close();
            return true;
        }
        else
            return false;
    }

private:
    typedef char byte;

    // Delete empty constructor    
    picture() = delete;

    // Parse the bytestream into the bytearray
    void opencb_(std::ifstream & bytestream)
    {
        // copy byte by byte
        bytestream.seekg(0, std::ios_base::end);
        std::streampos fileSize = bytestream.tellg();
        bytearray_.resize(fileSize);
        bytestream.seekg(0, std::ios_base::beg);
        bytestream.read(&bytearray_[0], fileSize);
        // Check Magic Number to find picture format
        if ((unsigned int)bytearray_[0] == 0xFFFFFF89 
            && (unsigned int)bytearray_[1] == 0x00000050)
            imgtype_ = "png";
        if ((unsigned int)bytearray_[0] == 0xFFFFFFFF 
            && (unsigned int)bytearray_[1] == 0xFFFFFFD8)
            imgtype_ = "jpg";
    }

    std::vector<rapp::types::byte> bytearray_;
    std::string imgtype_;
};
}
}
#endif