Skip to content

[SofaMatrix.imgui] Custom style for the rendered matrix #6211

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
fredroy merged 3 commits into from
Aug 6, 2026
+116 −5
Merged

[SofaMatrix.imgui] Custom style for the rendered matrix #6211

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
fc039f2
take the style of the GUI to draw the matrix
alxbilger Jul 27, 2026
fafccb5
select colors
alxbilger Jul 27, 2026
7664c12
button to save matrix
alxbilger Jul 27, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
121 changes: 116 additions & 5 deletions ...s/plugins/SofaMatrix/extensions/imgui/src/SofaMatrix/imgui/BaseMatrixImageProxyWidget.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,10 +19,15 @@
* *
* Contact information: contact@sofa-framework.org *
******************************************************************************/
#include <SofaImGui/ImGuiDataWidget.h>
#include <imgui.h>
#include <GL/glew.h>
#include <SofaImGui/ImGuiDataWidget.h>
#include <SofaMatrix/BaseMatrixImageProxy.h>
#include <imgui.h>
#include <sofa/core/objectmodel/Base.h>
#include <sofa/helper/Utils.h>
#include <sofa/helper/io/STBImage.h>
#include <sofa/helper/system/FileRepository.h>
#include <sofa/helper/system/FileSystem.h>

#include <optional>

Expand Down Expand Up @@ -76,26 +81,66 @@ std::optional<GLuint> textureID(sofa::linearalgebra::BaseMatrix* ptr)
return mapIt->second;
}

ImVec4& getColorNonZero()
{
static ImVec4 color = ImGui::GetStyleColorVec4(ImGuiCol_Text);
return color;
}

ImVec4& getColorZero()
{
static ImVec4 color = ImGui::GetStyleColorVec4(ImGuiCol_WindowBg);
return color;
}

void updateTexture(sofa::linearalgebra::BaseMatrix* matrix, unsigned& textureIDRef,
std::vector<unsigned char>& imageDataRef)
{
const auto width = matrix->rows();
const auto height = matrix->cols();

imageDataRef.resize(width * height, 255);
imageDataRef.resize(width * height * 3, 255);

const ImVec4 colorTrue = getColorNonZero();
const ImVec4 colorFalse = getColorZero();

// Extract R, G, B components
const unsigned char rTrue = static_cast<unsigned char>(255.f * colorTrue.x);
const unsigned char gTrue = static_cast<unsigned char>(255.f * colorTrue.y);
const unsigned char bTrue = static_cast<unsigned char>(255.f * colorTrue.z);

const unsigned char rFalse = static_cast<unsigned char>(255.f * colorFalse.x);
const unsigned char gFalse = static_cast<unsigned char>(255.f * colorFalse.y);
const unsigned char bFalse = static_cast<unsigned char>(255.f * colorFalse.z);

// write the data
for (sofa::SignedIndex y = 0; y < height; ++y)
{
for (sofa::SignedIndex x = 0; x < width; ++x)
{
imageDataRef[y * width + x] = !static_cast<bool>(matrix->element(x, y)) * std::numeric_limits<unsigned char>::max();
size_t index = (y * width + x) * 3;

// Determine the color based on matrix element value
if (matrix->element(x, y))
{
// Element is TRUE: Use Window Background Color
imageDataRef[index] = rTrue; // R
imageDataRef[index + 1] = gTrue; // G
imageDataRef[index + 2] = bTrue; // B
}
else
{
// Element is FALSE: Use Text Color
imageDataRef[index] = rFalse; // R
imageDataRef[index + 1] = gFalse; // G
imageDataRef[index + 2] = bFalse; // B
}
}
}

// Update the texture
glBindTexture(GL_TEXTURE_2D, textureIDRef);
glTexImage2D(GL_TEXTURE_2D, 0, GL_LUMINANCE, width, height, 0, GL_LUMINANCE, GL_UNSIGNED_BYTE,
glTexImage2D(GL_TEXTURE_2D, 0, GL_RGB, width, height, 0, GL_RGB, GL_UNSIGNED_BYTE,
imageDataRef.data());
}

Expand Down Expand Up @@ -183,6 +228,72 @@ void DataWidget<sofa::type::BaseMatrixImageProxy>::showWidget(MyData& data)
{
*lastCounter = counter;
}

if (ImGui::ColorEdit3(("Color non-zero##" + data.getOwner()->getPathName() + data.getName()).c_str(), (float*)&sofamatrix::imgui::getColorNonZero())
||
ImGui::ColorEdit3(("Color zero##" + data.getOwner()->getPathName() + data.getName()).c_str(), (float*)&sofamatrix::imgui::getColorZero()))
{
auto textureID = sofamatrix::imgui::textureID(matrix);
if (textureID.has_value() == true)
{
auto imageData = sofamatrix::imgui::imageData(matrix);
if (imageData.has_value() == true)
{
auto& textureIDRef = textureID.value();
auto& imageDataRef = imageData.value();

sofamatrix::imgui::updateTexture(matrix, textureIDRef, imageDataRef);
}
}
}

ImGui::NewLine();
if (ImGui::Button(("Save image##" + data.getOwner()->getPathName() + data.getName()).c_str()))
{
sofa::helper::io::STBImage image;
image.init(matrix->rows(), matrix->cols(), 1, 1,
sofa::helper::io::Image::DataType::UNORM8, sofa::helper::io::Image::ChannelFormat::RGB);

auto textureID = sofamatrix::imgui::textureID(matrix);
if (textureID.has_value() == true)
{
auto imageData = sofamatrix::imgui::imageData(matrix);
if (imageData.has_value() == true)
{
auto& imageDataRef = imageData.value();
auto& textureIDRef = textureID.value();
sofamatrix::imgui::updateTexture(matrix, textureIDRef, imageDataRef);

if (!imageDataRef.empty())
{
unsigned char* pixels = image.getPixels();
for (sofa::SignedIndex y = 0; y < matrix->cols(); ++y)
{
for (sofa::SignedIndex x = 0; x < matrix->rows(); ++x)
{
for (std::size_t i = 0; i < 3; ++i)
{
pixels[y * matrix->rows() * 3 + x * 3 + i] = imageDataRef[(matrix->rows() - y) * matrix->rows() * 3 + x * 3 + i];
}
}
}
const auto filename = sofa::helper::system::FileSystem::append(sofa::helper::Utils::getExecutableDirectory(), "matrix.bmp");
if (image.save(filename))
{
msg_info("Matrix") << "Saved matrix file in '" << filename << "'";
}
else
{
msg_error(data.getOwner()) << "Error while saving matrix to file '" << filename << "'";
}
}
else
{
msg_error(data.getOwner()) << "Cannot save matrix to file";
}
}
}
}
}
else
{
Expand Down
Loading