62 lines
2.6 KiB
C++
62 lines
2.6 KiB
C++
// SPDX-FileCopyrightText: 2025 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#ifndef JFJOCH_COMPRESSEDIMAGE_H
|
|
#define JFJOCH_COMPRESSEDIMAGE_H
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "../compression/CompressionAlgorithmEnum.h"
|
|
#include "ColorScale.h"
|
|
|
|
enum class CompressedImageMode {Int8, Int16, Int32, Uint8, Uint16, Uint32, RGB, Float16, Float32, Float64};
|
|
|
|
CompressedImageMode CalcImageMode(size_t byte_depth, bool is_float, bool is_signed);
|
|
|
|
class CompressedImage {
|
|
size_t xpixel;
|
|
size_t ypixel;
|
|
CompressedImageMode mode;
|
|
CompressionAlgorithm algorithm;
|
|
const uint8_t *data;
|
|
size_t size; // After compression
|
|
std::string channel;
|
|
public:
|
|
CompressedImage();
|
|
CompressedImage(const void* data, size_t size,
|
|
size_t xpixel, size_t ypixel,
|
|
CompressedImageMode mode,
|
|
CompressionAlgorithm algorithm = CompressionAlgorithm::NO_COMPRESSION,
|
|
std::string channel = "default");
|
|
|
|
CompressedImage(const std::vector<rgb>& input, size_t xpixel, size_t ypixel);
|
|
CompressedImage(const std::vector<float>& input, size_t xpixel, size_t ypixel);
|
|
CompressedImage(const std::vector<uint8_t>& input, size_t xpixel, size_t ypixel,
|
|
CompressedImageMode mode = CompressedImageMode::Uint8,
|
|
CompressionAlgorithm algorithm = CompressionAlgorithm::NO_COMPRESSION);
|
|
CompressedImage(const std::vector<uint16_t>& input, size_t xpixel, size_t ypixel);
|
|
CompressedImage(const std::vector<uint32_t>& input, size_t xpixel, size_t ypixel);
|
|
CompressedImage(const std::vector<int16_t>& input, size_t xpixel, size_t ypixel);
|
|
CompressedImage(const std::vector<int32_t>& input, size_t xpixel, size_t ypixel);
|
|
CompressedImage &Channel(std::string channel);
|
|
|
|
[[nodiscard]] size_t GetUncompressedSize() const;
|
|
[[nodiscard]] bool IsSigned() const;
|
|
void GetUncompressed(std::vector<uint8_t> &buffer) const;
|
|
[[nodiscard]] const uint8_t* GetUncompressedPtr(std::vector<uint8_t> &buffer) const;
|
|
[[nodiscard]] size_t GetByteDepth() const;
|
|
[[nodiscard]] size_t GetNumChannels() const;
|
|
[[nodiscard]] std::string GetChannel() const;
|
|
[[nodiscard]] size_t GetWidth() const;
|
|
[[nodiscard]] size_t GetHeight() const;
|
|
[[nodiscard]] CompressedImageMode GetMode() const;
|
|
[[nodiscard]] CompressionAlgorithm GetCompressionAlgorithm() const;
|
|
[[nodiscard]] const uint8_t *GetCompressed() const;
|
|
[[nodiscard]] size_t GetCompressedSize() const;
|
|
};
|
|
|
|
#endif //JFJOCH_COMPRESSEDIMAGE_H
|