image_preprocessor.h

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#pragma once
 
#include <cstddef>
#include <cstdint>
#include <filesystem>
#include <string>
#include <vector>
 
namespace entropic {
 
struct ImagePreprocessConfig {
    int max_width = 1280;                       
    int max_height = 1280;                      
    size_t max_file_size = 20 * 1024 * 1024;    
    bool preserve_aspect = true;                
};
 
struct PreprocessedImage {
    std::vector<uint8_t> pixel_data;  
    int width = 0;                    
    int height = 0;                   
    int channels = 3;                 
    std::string source_path;          
};
 
class ImagePreprocessor {
public:
    explicit ImagePreprocessor(const ImagePreprocessConfig& config);
 
    PreprocessedImage preprocess_file(const std::filesystem::path& path);
 
    PreprocessedImage preprocess_buffer(
        const uint8_t* data,
        size_t len,
        const std::string& source_label);
 
private:
    ImagePreprocessConfig config_;  
 
    PreprocessedImage decode(
        const uint8_t* data, size_t len,
        const std::string& source);
 
    void resize_if_needed(PreprocessedImage& img);
};
 
} // namespace entropic