mirror of
https://github.com/jomjol/AI-on-the-edge-device.git
synced 2026-01-27 21:00:42 +03:00
93 lines
2.7 KiB
C++
93 lines
2.7 KiB
C++
#pragma once
|
|
|
|
#ifndef CIMAGEBASIS_H
|
|
#define CIMAGEBASIS_H
|
|
|
|
#include "defines.h"
|
|
|
|
#include <string>
|
|
#include <math.h>
|
|
#include <stdint.h>
|
|
|
|
#include <esp_http_server.h>
|
|
|
|
#include "../stb/stb_image.h"
|
|
#include "../stb/stb_image_write.h"
|
|
#include "../stb/deprecated/stb_image_resize.h"
|
|
// #include "../stb/stb_image_resize2.h"
|
|
|
|
#include "esp_heap_caps.h"
|
|
|
|
struct ImageData
|
|
{
|
|
uint8_t data[MAX_JPG_SIZE];
|
|
size_t size = 0;
|
|
};
|
|
|
|
class CImageBasis
|
|
{
|
|
protected:
|
|
bool externalImage;
|
|
std::string filename;
|
|
std::string name; // Just used for diagnostics
|
|
int memsize = 0;
|
|
|
|
void memCopy(uint8_t *_source, uint8_t *_target, int _size);
|
|
bool isInImage(int x, int y);
|
|
|
|
bool islocked;
|
|
|
|
public:
|
|
uint8_t *rgb_image = NULL;
|
|
int channels;
|
|
int width, height, bpp;
|
|
|
|
uint8_t *RGBImageLock(int _waitmaxsec = 60);
|
|
void RGBImageRelease();
|
|
uint8_t *RGBImageGet();
|
|
|
|
int getWidth() { return this->width; };
|
|
int getHeight() { return this->height; };
|
|
int getChannels() { return this->channels; };
|
|
void drawRect(int x, int y, int dx, int dy, int r = 255, int g = 255, int b = 255, int thickness = 1);
|
|
void drawLine(int x1, int y1, int x2, int y2, int r, int g, int b, int thickness = 1);
|
|
void drawCircle(int x1, int y1, int rad, int r, int g, int b, int thickness = 1);
|
|
void drawEllipse(int x1, int y1, int radx, int rady, int r, int g, int b, int thickness = 1);
|
|
|
|
void setPixelColor(int x, int y, int r, int g, int b);
|
|
void Negative(void);
|
|
void Contrast(float _contrast);
|
|
bool ImageOkay();
|
|
bool CopyFromMemory(uint8_t *_source, int _size);
|
|
|
|
void SetIndepended() { externalImage = false; };
|
|
|
|
void CreateEmptyImage(int _width, int _height, int _channels);
|
|
void EmptyImage();
|
|
|
|
CImageBasis(std::string name);
|
|
CImageBasis(std::string name, std::string _image);
|
|
CImageBasis(std::string name, uint8_t *_rgb_image, int _channels, int _width, int _height, int _bpp);
|
|
CImageBasis(std::string name, int _width, int _height, int _channels);
|
|
CImageBasis(std::string name, CImageBasis *_copyfrom);
|
|
|
|
void Resize(int _new_dx, int _new_dy);
|
|
void Resize(int _new_dx, int _new_dy, CImageBasis *_target);
|
|
void crop_image(unsigned short cropLeft, unsigned short cropRight, unsigned short cropTop, unsigned short cropBottom);
|
|
|
|
void LoadFromMemory(stbi_uc *_buffer, int len);
|
|
|
|
ImageData *writeToMemoryAsJPG(const int quality = 90);
|
|
void writeToMemoryAsJPG(ImageData *ii, const int quality = 90);
|
|
|
|
esp_err_t SendJPGtoHTTP(httpd_req_t *req, const int quality = 90);
|
|
|
|
uint8_t GetPixelColor(int x, int y, int ch);
|
|
|
|
~CImageBasis();
|
|
|
|
void SaveToFile(std::string _imageout);
|
|
};
|
|
|
|
#endif // CIMAGEBASIS_H
|