ImageGenerator.h

Go to the documentation of this file.

/*
 * Copyright (c) 2015-2018, EPFL/Blue Brain Project
 *
 * The Blue Brain BioExplorer is a tool for scientists to extract and analyse
 * scientific data from visualization
 *
 * This file is part of Blue Brain BioExplorer <https://github.com/BlueBrain/BioExplorer>
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License version 3.0 as published
 * by the Free Software Foundation.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
 * details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this library; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */
 
#pragma once
 
#include <platform/core/common/Types.h>
 
#include <turbojpeg.h>
 
namespace core
{
class ImageGenerator
{
public:
    ImageGenerator() = default;
    ~ImageGenerator();
 
    struct ImageBase64
    {
        std::string data;
    };
 
    ImageBase64 createImage(FrameBuffer& frameBuffer, const std::string& format,
                            uint8_t quality);
    ImageBase64 createImage(const std::vector<FrameBufferPtr>& frameBuffers,
                            const std::string& format, uint8_t quality);
 
    struct ImageJPEG
    {
        struct tjDeleter
        {
            void operator()(uint8_t* ptr) { tjFree(ptr); }
        };
        using JpegData = std::unique_ptr<uint8_t, tjDeleter>;
        JpegData data;
        unsigned long size{0};
    };
 
    ImageJPEG createJPEG(FrameBuffer& frameBuffer, uint8_t quality);
 
private:
    tjhandle _compressor{tjInitCompress()};
 
    ImageJPEG::JpegData _encodeJpeg(uint32_t width, uint32_t height,
                                    const uint8_t* rawData, int32_t pixelFormat,
                                    uint8_t quality, unsigned long& dataSize);
};
} // namespace core