Implemented saveToMemory and saveImageToMemory.

This commit is contained in:
Arthur Caputo 2020-03-07 02:29:28 -03:00 committed by Lukas Dürrenberger
parent c11ea7792f
commit 129774877b
4 changed files with 86 additions and 0 deletions

View File

@ -155,6 +155,24 @@ public:
//////////////////////////////////////////////////////////// ////////////////////////////////////////////////////////////
bool saveToFile(const std::string& filename) const; bool saveToFile(const std::string& filename) const;
////////////////////////////////////////////////////////////
/// \brief Save the image to a buffer in memory
///
/// The format of the image must be specified.
/// The supported image formats are bmp, png, tga and jpg.
/// This function fails if the image is empty, or if
/// the format was invalid.
///
/// \param output Buffer to fill with encoded data
/// \param format Encoding format to use
///
/// \return True if saving was successful
///
/// \see create, loadFromFile, loadFromMemory, saveToFile
///
////////////////////////////////////////////////////////////
bool saveToMemory(std::vector<sf::Uint8>& output, const std::string& format) const;
//////////////////////////////////////////////////////////// ////////////////////////////////////////////////////////////
/// \brief Return the size (width and height) of the image /// \brief Return the size (width and height) of the image
/// ///

View File

@ -153,6 +153,12 @@ bool Image::saveToFile(const std::string& filename) const
return priv::ImageLoader::getInstance().saveImageToFile(filename, m_pixels, m_size); return priv::ImageLoader::getInstance().saveImageToFile(filename, m_pixels, m_size);
} }
////////////////////////////////////////////////////////////
bool Image::saveToMemory(std::vector<sf::Uint8>& output, const std::string& format) const
{
return priv::ImageLoader::getInstance().saveImageToMemory(format, output, m_pixels, m_size);
}
//////////////////////////////////////////////////////////// ////////////////////////////////////////////////////////////
Vector2u Image::getSize() const Vector2u Image::getSize() const

View File

@ -33,6 +33,7 @@
#define STB_IMAGE_WRITE_IMPLEMENTATION #define STB_IMAGE_WRITE_IMPLEMENTATION
#include <stb_image_write.h> #include <stb_image_write.h>
#include <cctype> #include <cctype>
#include <iterator>
namespace namespace
@ -61,6 +62,14 @@ namespace
sf::InputStream* stream = static_cast<sf::InputStream*>(user); sf::InputStream* stream = static_cast<sf::InputStream*>(user);
return stream->tell() >= stream->getSize(); return stream->tell() >= stream->getSize();
} }
// stb_image callback for constructing a buffer
void bufferFromCallback(void* context, void* data, int size)
{
sf::Uint8* source = static_cast<sf::Uint8*>(data);
std::vector<sf::Uint8>* dest = static_cast<std::vector<sf::Uint8>*>(context);
std::copy(source, source + size, std::back_inserter(*dest));
}
} }
@ -272,6 +281,46 @@ bool ImageLoader::saveImageToFile(const std::string& filename, const std::vector
return false; return false;
} }
////////////////////////////////////////////////////////////
bool ImageLoader::saveImageToMemory(const std::string& format, std::vector<sf::Uint8>& output, const std::vector<Uint8>& pixels, const Vector2u& size)
{
// Make sure the image is not empty
if (!pixels.empty() && (size.x > 0) && (size.y > 0))
{
// Choose function based on format
std::string specified = toLower(format);
if (specified == "bmp")
{
// BMP format
if (stbi_write_bmp_to_func(&bufferFromCallback, &output, size.x, size.y, 4, &pixels[0]))
return true;
}
else if (specified == "tga")
{
// TGA format
if (stbi_write_tga_to_func(&bufferFromCallback, &output, size.x, size.y, 4, &pixels[0]))
return true;
}
else if (specified == "png")
{
// PNG format
if (stbi_write_png_to_func(&bufferFromCallback, &output, size.x, size.y, 4, &pixels[0], 0))
return true;
}
else if (specified == "jpg" || specified == "jpeg")
{
// JPG format
if (stbi_write_jpg_to_func(&bufferFromCallback, &output, size.x, size.y, 4, &pixels[0], 90))
return true;
}
}
err() << "Failed to save image with format \"" << format << "\"" << std::endl;
return false;
}
} // namespace priv } // namespace priv
} // namespace sf } // namespace sf

View File

@ -105,6 +105,19 @@ public:
//////////////////////////////////////////////////////////// ////////////////////////////////////////////////////////////
bool saveImageToFile(const std::string& filename, const std::vector<Uint8>& pixels, const Vector2u& size); bool saveImageToFile(const std::string& filename, const std::vector<Uint8>& pixels, const Vector2u& size);
////////////////////////////////////////////////////////////
/// \brief Save an array of pixels as an encoded image buffer
///
/// \param format Must be "bmp", "png", "tga" or "jpg"/"jpeg".
/// \param output Buffer to fill with encoded data
/// \param pixels Array of pixels to save to image
/// \param size Size of image to save, in pixels
///
/// \return True if saving was successful
///
////////////////////////////////////////////////////////////
bool saveImageToMemory(const std::string& format, std::vector<sf::Uint8>& output, const std::vector<Uint8>& pixels, const Vector2u& size);
private: private:
//////////////////////////////////////////////////////////// ////////////////////////////////////////////////////////////