Skip to content

Commit

Permalink
point_selector_fin
Browse files Browse the repository at this point in the history
  • Loading branch information
Jose Alonso Solis Lemus committed Nov 26, 2024
1 parent 2b6eb90 commit 4282857
Show file tree
Hide file tree
Showing 5 changed files with 141 additions and 38 deletions.
5 changes: 4 additions & 1 deletion CemrgApp/Modules/CemrgAppModule/include/CemrgMeshPointSelector.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -84,6 +84,10 @@ class MITKCEMRGAPPMODULE_EXPORT CemrgMeshPointSelector {
void AddPointFromSurface(mitk::Surface::Pointer surface, int pickedSeedId, int label);
void AddPointFromUnstructuredGrid(mitk::UnstructuredGrid::Pointer grid, int pickedSeedId, int label);

// both functions go through the set pointDatas and update the vtkId based on the coordinates
void UpdateVtkIdFromSurface(mitk::Surface::Pointer surface, QStringList names);
void UpdateVtkIdFromUnstructuredGrid(mitk::UnstructuredGrid::Pointer grid, QStringList names);

int CleanupLastPoint();
int UnsetLastPoint(); // returns las label
void Clear();
Expand Down Expand Up @@ -113,7 +117,6 @@ class MITKCEMRGAPPMODULE_EXPORT CemrgMeshPointSelector {
vtkSmartPointer<vtkPolyData> lineSeeds;
std::vector<PointLabelData> pointsData; // Consolidated data for points

// QStringList saveFiles;
};

#endif // CemrgMeshPointSelector_h
25 changes: 23 additions & 2 deletions CemrgApp/Modules/CemrgAppModule/src/CemrgMeshPointSelector.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -170,6 +170,27 @@ void CemrgMeshPointSelector::AddPointFromUnstructuredGrid(mitk::UnstructuredGrid
pointsData[whichIndex].index = GetLastLabelIndex() + 1;
}

void CemrgMeshPointSelector::UpdateVtkIdFromSurface(mitk::Surface::Pointer surface, QStringList names) {
for (int ix = 0; ix < static_cast<int>(pointsData.size()); ix++) {
if (pointsData[ix].labelSet && names.contains(pointsData[ix].pointName)) {
vtkIdType vtkId = surface->GetVtkPolyData()->FindPoint(pointsData[ix].coordinates.data());
pointsData[ix].vtkId = vtkId;
}
}
}

void CemrgMeshPointSelector::UpdateVtkIdFromUnstructuredGrid(mitk::UnstructuredGrid::Pointer grid, QStringList names) {
for (int ix = 0; ix < static_cast<int>(pointsData.size()); ix++) {
if (pointsData[ix].labelSet && names.contains(pointsData[ix].pointName)) {
double *pt = pointsData[ix].coordinates.data();
vtkIdType vtkId = grid->GetVtkUnstructuredGrid()->FindPoint(pointsData[ix].coordinates.data());

std::cout << "Point: (" << pt[0] << " " << pt[1] << " " << pt[2] << ") Old ID:" << pointsData[ix].vtkId <<", New ID: " << vtkId << std::endl;
pointsData[ix].vtkId = vtkId;
}
}
}

int CemrgMeshPointSelector::CleanupLastPoint() {
if (pointsData.empty()) {
return -1;
Expand Down Expand Up @@ -224,7 +245,7 @@ PointLabelData CemrgMeshPointSelector::GetPointData(QString name) {
PointLabelData pd;
for (const auto& point : pointsData) {
if (point.pointName == name) {
pd = point;
pd = point;
break;
}
}
Expand Down Expand Up @@ -318,4 +339,4 @@ int CemrgMeshPointSelector::GetLastLabelIndex() {
}

return lastLabelIndex;
}
}
110 changes: 93 additions & 17 deletions CemrgApp/Plugins/kcl.cemrgapp.fourchamber/src/internal/FourChamberGuidePointsView.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -99,15 +99,16 @@ QString FourChamberGuidePointsView::whichAtrium;
const std::string FourChamberGuidePointsView::VIEW_ID = "org.mitk.views.fourchamberguidepointsview";

void FourChamberGuidePointsView::CreateQtPartControl(QWidget *parent) {
MITK_INFO << "FourChamberGuidePointsView";

// create GUI widgets from the Qt Designer's .ui file
m_Controls.setupUi(parent);
connect(m_Controls.button_guide1, SIGNAL(clicked()), this, SLOT(Help()));
connect(m_Controls.radio_load_la, SIGNAL(toggled(bool)), this, SLOT(LeftAtriumReactToToggle()));
connect(m_Controls.radio_load_ra, SIGNAL(toggled(bool)), this, SLOT(RightAtriumReactToToggle()));
connect(m_Controls.check_show_all, SIGNAL(stateChanged(int)), this, SLOT(CheckBoxShowAll(int)));
connect(m_Controls.button_save, SIGNAL(clicked()), this, SLOT(Save()));
connect(m_Controls.alpha_slider, SIGNAL(valueChanged(int)), this, SLOT(ChangeAlpha(int)));
connect(m_Controls.button_opacity, SIGNAL(clicked()), this, SLOT(ConfirmAlpha()));

inputsSelector = new QDialog(0, Qt::Window);
m_Selector.setupUi(inputsSelector);
Expand All @@ -131,7 +132,7 @@ void FourChamberGuidePointsView::CreateQtPartControl(QWidget *parent) {

vtkSmartPointer<vtkGenericOpenGLRenderWindow> renderWindow =
vtkSmartPointer<vtkGenericOpenGLRenderWindow>::New();
m_Controls.widget_1->SetRenderWindow(renderWindow);
m_Controls.widget_1->setRenderWindow(renderWindow);
m_Controls.widget_1->renderWindow()->AddRenderer(renderer);

//Setup keyboard interactor
Expand All @@ -151,8 +152,8 @@ void FourChamberGuidePointsView::CreateQtPartControl(QWidget *parent) {
Visualiser();
}

m_Controls.button_save->setEnabled(false);
Help(true);
alpha = 1.0;
pluginLoaded = true;
}

Expand All @@ -173,10 +174,14 @@ void FourChamberGuidePointsView::Help(bool firstTime){
std::string msg = "";
if(firstTime){
msg = "HELP\n";
} else if(!m_Controls.radio_load_la->isChecked()) {
msg = "LEFT ATRIUM POINTS\n";
msg += "1. Check the boxes to load the left or right atrium.\n";
msg += "2. Use the slider to adjust the opacity of the loaded atrium.\n";
msg += "3. Hover on top of the surface and press SPACE to select points.\n";
} else {
msg = "RIGHT ATRIUM POINTS\n";
msg = "HELP\n";
msg += "1. Check the boxes to load the left or right atrium.\n";
msg += "2. Use the slider to adjust the opacity of the loaded atrium.\n";
msg += "3. Hover on top of the surface and press SPACE to select points.\n";
}
QMessageBox::information(NULL, "Help", msg.c_str());
}
Expand All @@ -197,32 +202,36 @@ void FourChamberGuidePointsView::SetSubdirs(){
}

bool FourChamberGuidePointsView::CreateVisualisationMesh(QString dir, QString visName, QString originalName) {
MITK_INFO << "[INFO] Creating visualisation mesh";
bool success = false;
QString path = dir + "/" + visName + ".surfmesh.vtk";
QFileInfo fi(path);
if (!fi.exists()) {
MITK_INFO << "[INFO] Visualisation mesh does not exist. Creating...";
QString originalPath = dir + "/" + originalName;
if (!QFile::exists(originalPath)) {
MITK_WARN << "File" << orginalPath.toStdString() << "does not exist!";
return;
MITK_WARN << "File" << originalPath.toStdString() << "does not exist!";
return false;
}

std::unique_ptr<CemrgFourChamberCmd> fourch = std::make_unique<CemrgFourChamberCmd>();
QStringList arguments;
arguments << "-msh=" + originalPath;
arguments << "-msh=" + originalName;
arguments << "-surf=" + visName;
arguments << "-ofmt=vtk_polydata";
QString path = success = fourch->DockerMeshtoolGeneric(dir, "extract", "surface", arguments, visName);
QString path = fourch->DockerMeshtoolGeneric(dir, "extract", "surface", arguments, visName);

success = (QFile::exists(path));
} else {
MITK_INFO << "[INFO] Visualisation mesh already exists!";
success = true;
}

return success;
}

void FourChamberGuidePointsView::iniPreSurf() {
MITK_INFO << "[INFO] Initialising pre-surface visualisation";
SetSubdirs();
bool laVisSuccess = CreateVisualisationMesh(surfaces_uvc_la, QString("la/la_vis"), QString("la/la.vtk"));
bool raVisSuccess = CreateVisualisationMesh(surfaces_uvc_ra, QString("ra/ra_vis"), QString("ra/ra.vtk"));
Expand Down Expand Up @@ -250,7 +259,9 @@ void FourChamberGuidePointsView::Visualiser(double opacity){
double max_scalar = 4; // RA
double min_scalar = 3; // LA

SphereSourceVisualiser(pickedPointsHandler->GetLineSeeds(), "0.4,0.1,0.0", 0.1);
renderer->RemoveAllViewProps();

SphereSourceVisualiser(pickedPointsHandler->GetLineSeeds(), "0.4,0.1,0.0", 0.025);

//Create a mapper and actor for surface
vtkSmartPointer<vtkPolyDataMapper> surfMapper = vtkSmartPointer<vtkPolyDataMapper>::New();
Expand All @@ -272,7 +283,33 @@ void FourChamberGuidePointsView::Visualiser(double opacity){
renderer->AddActor(surfActor);
}

// void FourChamberGuidePointsView::LoadMeshes(int stateChange) {
// std::cout << "LoadMeshes: State: " << stateChange << std::endl;
// MITK_INFO << "[INFO] Loading meshes";
// if (!pluginLoaded) return;

// renderer->RemoveAllViewProps();

// if (m_Controls.radio_load_la->isChecked()) {
// MITK_INFO << "[INFO] Loading LA: " << path_to_la.toStdString();
// mitk::Surface::Pointer shell = mitk::IOUtil::Load<mitk::Surface>(path_to_la.toStdString());
// if (shell.IsNotNull()) {
// surface = shell;
// Visualiser(alpha); // Adjust opacity as needed
// }
// }

// if (m_Controls.radio_load_ra->isChecked()) {
// MITK_INFO << "[INFO] Loading RA: " << path_to_ra.toStdString();
// mitk::Surface::Pointer shell = mitk::IOUtil::Load<mitk::Surface>(path_to_ra.toStdString());
// if (shell.IsNotNull()) {
// surface = shell;
// Visualiser(alpha); // Adjust opacity as needed
// }
// }
// }
void FourChamberGuidePointsView::SphereSourceVisualiser(vtkSmartPointer<vtkPolyData> pointSources, QString colour, double scaleFactor){
MITK_INFO << "[INFO] Sphere source visualiser";
// e.g colour = "0.4,0.1,0.0" - values for r,g, and b separated by commas.
double r, g, b;
bool okr, okg, okb;
Expand Down Expand Up @@ -346,6 +383,16 @@ void FourChamberGuidePointsView::KeyCallBackFunc(vtkObject*, long unsigned int,
self = reinterpret_cast<FourChamberGuidePointsView*>(ClientData);
std::string key = self->interactor->GetKeySym();

if (!self->m_Controls.radio_load_la->isChecked() && !self->m_Controls.radio_load_ra->isChecked()) {
QMessageBox::information(NULL, "Info", "Please load a mesh first!");
return;
}

if (self->m_Controls.radio_load_la->isChecked() && self->m_Controls.radio_load_ra->isChecked()) {
QMessageBox::information(NULL, "Info", "Please load only one mesh at a time!");
return;
}

if (key == "space") {
//Ask the labels
self->UserSelectPvLabel();
Expand Down Expand Up @@ -411,18 +458,47 @@ void FourChamberGuidePointsView::RightAtriumReactToToggle() {
}
}

void FourChamberGuidePointsView::CheckBoxShowAll(int checkedState) {
std::cout << "Show all atria: " << checkedState << std::endl;
QMessageBox::information(NULL, "Info", "Showing both atria");
void FourChamberGuidePointsView::ChangeAlpha(int alphaChange) {
alpha = alphaChange / 100.0;
}

void FourChamberGuidePointsView::ChangeAlpha(int alpha) {
double opacity = alpha / 100.0;
std::cout << "Opacity: " << opacity << std::endl;
void FourChamberGuidePointsView::ConfirmAlpha() {
// LoadMeshes(2);
Visualiser(alpha);
}

void FourChamberGuidePointsView::Save() {
QMessageBox::information(NULL, "Info", "Saving points to files");
QStringList saveTypes = {"vtx", "coord"};
QStringList laPoints = {"la.lvapex.vtx", "la.rvsept_pt.vtx"}; // LA points
QStringList raPoints = {"ra.lvapex.vtx", "ra.rvsept_pt.vtx", "raa_apex.txt"}; // RA points

std::string la_path = surfaces_uvc_la.toStdString() + "/la/la.vtk";
mitk::UnstructuredGrid::Pointer grid = mitk::IOUtil::Load<mitk::UnstructuredGrid>(la_path);
if (grid.IsNull()) {
MITK_WARN << "LA Grid is null!";
return;
}
pickedPointsHandler->UpdateVtkIdFromUnstructuredGrid(grid, laPoints);

std::string ra_path = surfaces_uvc_ra.toStdString() + "/ra/ra.vtk";
grid = mitk::IOUtil::Load<mitk::UnstructuredGrid>(ra_path);
if (grid.IsNull()) {
MITK_WARN << "RA Grid is null!";
return;
}
pickedPointsHandler->UpdateVtkIdFromUnstructuredGrid(grid, raPoints);

QStringList points = laPoints + raPoints;

// Save points
for (auto& point : points) {
QString savetype = pickedPointsHandler->GetPointData(point).pointName.contains("vtx") ? "vtx" : "coord";
pickedPointsHandler->SaveToFile(APEX_SEPTUM(point), (QStringList(point)), savetype);
}

QMessageBox::information(NULL, "Info", "Points saved, please close this window.");
m_Controls.button_save->setEnabled(false);
}

void FourChamberGuidePointsView::UserSelectPvLabel(){
Expand Down
9 changes: 6 additions & 3 deletions CemrgApp/Plugins/kcl.cemrgapp.fourchamber/src/internal/FourChamberGuidePointsView.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -73,10 +73,11 @@ protected slots:

/// \brief Called when the user clicks the GUI button
void Help(bool firstTime=false);
void LeftAtriumReactToToggle(); //
void RightAtriumReactToToggle(); //
void CheckBoxShowAll(int checkedState); //
// void LoadMeshes(int stateChange); //
void LeftAtriumReactToToggle();
void RightAtriumReactToToggle();
void ChangeAlpha(int alpha); //
void ConfirmAlpha();
void Save();

protected:
Expand All @@ -103,6 +104,8 @@ protected slots:

void UserSelectPvLabel();

inline QString APEX_SEPTUM(QString file) { return apex_septum_files + "/" + file; };

QString parfiles;
QString apex_septum_files;
QString surfaces_uvc_la;
Expand Down
30 changes: 15 additions & 15 deletions CemrgApp/Plugins/kcl.cemrgapp.fourchamber/src/internal/FourChamberGuidePointsViewControls.ui
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,13 +16,6 @@
<layout class="QVBoxLayout" name="verticalLayout">
<item>
<layout class="QGridLayout" name="gridLayout">
<item row="1" column="1">
<widget class="QRadioButton" name="radio_load_la">
<property name="text">
<string>Load Left Atrium</string>
</property>
</widget>
</item>
<item row="0" column="0">
<widget class="QLabel" name="label_1">
<property name="text">
Expand All @@ -37,6 +30,13 @@
</property>
</widget>
</item>
<item row="1" column="1">
<widget class="QRadioButton" name="radio_load_la">
<property name="text">
<string>Load Left Atrium</string>
</property>
</widget>
</item>
<item row="1" column="2">
<widget class="QRadioButton" name="radio_load_ra">
<property name="text">
Expand All @@ -58,13 +58,6 @@
</item>
<item>
<layout class="QHBoxLayout" name="horizontalLayout">
<item>
<widget class="QCheckBox" name="check_show_all">
<property name="text">
<string>Show Both Atria</string>
</property>
</widget>
</item>
<item>
<widget class="QSlider" name="alpha_slider">
<property name="minimum">
Expand All @@ -84,10 +77,17 @@
</property>
</widget>
</item>
<item>
<widget class="QPushButton" name="button_opacity">
<property name="text">
<string>Confirm Transparency</string>
</property>
</widget>
</item>
<item>
<widget class="QPushButton" name="button_save">
<property name="text">
<string>Save</string>
<string>Save Points</string>
</property>
</widget>
</item>
Expand Down

0 comments on commit 4282857

Please sign in to comment.