jpeggeometryprobe.h

Go to the documentation of this file.

 
#pragma once
 
 
#include <promeki/config.h>
#if PROMEKI_ENABLE_PROAV
#include <cstdint>
#include <promeki/buffer.h>
#include <promeki/imagedesc.h>
#include <promeki/namespace.h>
#include <promeki/pixelformat.h>
#include <promeki/size2d.h>
#include <promeki/string.h>
 
PROMEKI_NAMESPACE_BEGIN
 
class JpegGeometryProbe {
        public:
                struct SofData {
                                uint32_t width = 0;
                                uint32_t height = 0;
                                int nf = 0;
                                uint8_t ySf = 0;
 
                                bool isValid() const { return width > 0 && height > 0; }
                };
 
                struct Result {
                                bool valid = false;
                                Size2Du32 size;
                                PixelFormat pixelFormat;
                                bool is420 = false;
                                bool isRgb = false;
 
                                ImageDesc imageDesc() const {
                                        return valid ? ImageDesc(size, pixelFormat) : ImageDesc();
                                }
                };
 
                JpegGeometryProbe() = default;
 
                static SofData parseSof(const uint8_t *data, size_t size);
 
                void reset();
 
                bool hasGeometry() const { return _last.valid; }
 
                const Result &lastResult() const { return _last; }
 
                const Result &probe(const Buffer &reassembled, uint8_t rfc2435Type,
                                    const String &fmtp);
 
        private:
                Result _last;
                // Cache keys — used to skip the full PixelFormat
                // resolution when nothing the resolution depends on
                // has changed since the last probe call.
                uint32_t _cachedWidth = 0;
                uint32_t _cachedHeight = 0;
                int _cachedNf = 0;
                uint8_t _cachedYsf = 0;
                uint8_t _cachedRfc2435Type = 0xFFu;
                String _cachedFmtp;
                bool _hasCacheKey = false;
};
 
PROMEKI_NAMESPACE_END
 
#endif // PROMEKI_ENABLE_PROAV