73 lines
1.7 KiB
C++
73 lines
1.7 KiB
C++
#ifndef MBGL_UTIL_JPEG_READER_HPP
|
|
#define MBGL_UTIL_JPEG_READER_HPP
|
|
|
|
#include <mbgl/platform/default/image_reader.hpp>
|
|
|
|
// jpeg
|
|
extern "C"
|
|
{
|
|
#include <jpeglib.h>
|
|
}
|
|
|
|
#pragma GCC diagnostic push
|
|
#pragma GCC diagnostic ignored "-Wunknown-pragmas"
|
|
#pragma GCC diagnostic ignored "-Wunused-local-typedefs"
|
|
#pragma GCC diagnostic ignored "-Wshadow"
|
|
#include <boost/iostreams/stream.hpp>
|
|
#pragma GCC diagnostic pop
|
|
|
|
namespace mbgl { namespace util {
|
|
|
|
template <typename T>
|
|
class JpegReader : public ImageReader
|
|
{
|
|
public:
|
|
using source_type = T;
|
|
using input_stream = boost::iostreams::stream<source_type>;
|
|
const static unsigned BUF_SIZE = 4096;
|
|
private:
|
|
struct jpeg_stream_wrapper
|
|
{
|
|
jpeg_source_mgr manager;
|
|
input_stream * stream;
|
|
JOCTET buffer[BUF_SIZE];
|
|
};
|
|
|
|
struct jpeg_info_guard
|
|
{
|
|
jpeg_info_guard(jpeg_decompress_struct * cinfo)
|
|
: i_(cinfo) {}
|
|
|
|
~jpeg_info_guard()
|
|
{
|
|
jpeg_destroy_decompress(i_);
|
|
}
|
|
jpeg_decompress_struct * i_;
|
|
};
|
|
|
|
private:
|
|
source_type source_;
|
|
input_stream stream_;
|
|
unsigned width_;
|
|
unsigned height_;
|
|
public:
|
|
JpegReader(const uint8_t* data, size_t size);
|
|
~JpegReader();
|
|
unsigned width() const;
|
|
unsigned height() const;
|
|
std::unique_ptr<uint8_t[]> read();
|
|
private:
|
|
void init();
|
|
static void on_error(j_common_ptr cinfo);
|
|
static void on_error_message(j_common_ptr cinfo);
|
|
static void init_source(j_decompress_ptr cinfo);
|
|
static boolean fill_input_buffer(j_decompress_ptr cinfo);
|
|
static void skip(j_decompress_ptr cinfo, long count);
|
|
static void term(j_decompress_ptr cinfo);
|
|
static void attach_stream(j_decompress_ptr cinfo, input_stream* in);
|
|
};
|
|
|
|
}}
|
|
|
|
#endif // MBGL_UTIL_JPEG_READER_HPP
|