Relintai/pmlpp
1
0
Fork 0
mirror of https://github.com/Relintai/pmlpp.git synced 2025-01-02 16:29:35 +01:00
Code Issues Packages Projects Releases Wiki Activity

Added Image helper methods to MLPPMatrix.

Browse Source
This commit is contained in:
Relintai 2023-04-25 20:31:32 +02:00
parent 40751bf79a
commit 9b40d4b455
2 changed files with 69 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
mlpp/lin_alg/mlpp_matrix.cpp
View File

@ -1,6 +1,8 @@
#include "mlpp_matrix.h"
#include "core/io/image.h"
#include "../stat/stat.h"
#include <random>
@ -2753,6 +2755,63 @@ bool MLPPMatrix::is_equal_approx(const Ref<MLPPMatrix> &p_with, real_t tolerance
return true;
}
Ref<Image> MLPPMatrix::get_as_image() const {
Ref<Image> image;
image.instance();
get_into_image(image);
return image;
}
void MLPPMatrix::get_into_image(Ref<Image> out) const {
ERR_FAIL_COND(!out.is_valid());
if (data_size() == 0) {
out->clear();
return;
}
PoolByteArray arr;
int ds = data_size();
arr.resize(ds);
PoolByteArray::Write w = arr.write();
uint8_t *wptr = w.ptr();
for (int i = 0; i < ds; ++i) {
wptr[i] = static_cast<uint8_t>(_data[i] * 255.0);
}
out->create(_size.x, _size.y, false, Image::FORMAT_L8, arr);
}
void MLPPMatrix::set_from_image(const Ref<Image> &p_img, const int p_image_channel) {
ERR_FAIL_COND(!p_img.is_valid());
ERR_FAIL_INDEX(p_image_channel, 4);
Size2i img_size = Size2i(p_img->get_width(), p_img->get_height());
if (img_size != _size) {
resize(img_size);
}
Ref<Image> img = p_img;
img->lock();
for (int y = 0; y < _size.y; ++y) {
for (int x = 0; x < _size.x; ++x) {
Color c = img->get_pixel(x, y);
set_element(y, x, c[p_image_channel]);
}
}
img->unlock();
}
String MLPPMatrix::to_string() {
String str;
@ -2921,6 +2980,10 @@ void MLPPMatrix::_bind_methods() {
ClassDB::bind_method(D_METHOD("is_equal_approx", "with", "tolerance"), &MLPPMatrix::is_equal_approx, CMP_EPSILON);
ClassDB::bind_method(D_METHOD("get_as_image"), &MLPPMatrix::get_as_image);
ClassDB::bind_method(D_METHOD("get_into_image", "out"), &MLPPMatrix::get_into_image);
ClassDB::bind_method(D_METHOD("set_from_image", "img", "image_channel"), &MLPPMatrix::set_from_image);
ClassDB::bind_method(D_METHOD("gaussian_noise", "n", "m"), &MLPPMatrix::gaussian_noise);
ClassDB::bind_method(D_METHOD("gaussian_noise_fill"), &MLPPMatrix::gaussian_noise_fill);

6
mlpp/lin_alg/mlpp_matrix.h
View File

@ -14,6 +14,8 @@
#include "mlpp_vector.h"
class Image;
class MLPPMatrix : public Reference {
GDCLASS(MLPPMatrix, Reference);
@ -321,6 +323,10 @@ public:
bool is_equal_approx(const Ref<MLPPMatrix> &p_with, real_t tolerance = static_cast<real_t>(CMP_EPSILON)) const;
Ref<Image> get_as_image() const;
void get_into_image(Ref<Image> out) const;
void set_from_image(const Ref<Image> &p_img, const int p_image_channel);
String to_string();
MLPPMatrix();