Skip to content

feat(metadata): add numerical metadata import functionality #8

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

Open
lstachowicz wants to merge 1 commit into
base: master
Choose a base branch
from
Open

feat(metadata): add numerical metadata import functionality #8

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
5 changes: 5 additions & 0 deletions CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -79,6 +79,8 @@ set(POLYSEG_SOURCES
src/shortcutssettingstab.cpp
src/aipluginmanager.cpp
src/pluginwizard.cpp
src/metadataimporter.cpp
src/metadataimportsettingsdialog.cpp
src/wizardpages/welcomepage.cpp
src/wizardpages/pluginselectionpage.cpp
src/wizardpages/custompluginpage.cpp
Expand Down Expand Up @@ -112,6 +114,8 @@ set(POLYSEG_HEADERS
src/shortcutssettingstab.h
src/aipluginmanager.h
src/pluginwizard.h
src/metadataimporter.h
src/metadataimportsettingsdialog.h
src/wizardpages/welcomepage.h
src/wizardpages/pluginselectionpage.h
src/wizardpages/custompluginpage.h
Expand All @@ -138,6 +142,7 @@ set(POLYSEG_UI_FILES
src/shortcuteditdialog.ui
src/modelcomparisondialog.ui
src/modelregistrationdialog.ui
src/metadataimportsettingsdialog.ui
src/wizardpages/welcomepage.ui
src/wizardpages/pluginselectionpage.ui
src/wizardpages/custompluginpage.ui
Expand Down
204 changes: 203 additions & 1 deletion src/mainwindow.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,8 @@
#include <iostream>

#include "aipluginmanager.h"
#include "metadataimporter.h"
#include "metadataimportsettingsdialog.h"
#include "pluginwizard.h"
#include "polygoncanvas.h"
#include "settingsdialog.h"
Expand Down Expand Up @@ -91,8 +93,8 @@ MainWindow::MainWindow(QWidget* parent)

connect(ui->actionNewProject, &QAction::triggered, this, &MainWindow::CreateNewProject);
connect(ui->actionOpenProject, &QAction::triggered, this, &MainWindow::OpenProject);
connect(ui->actionOpenImage, &QAction::triggered, this, &MainWindow::Load);
connect(ui->actionAddImages, &QAction::triggered, this, &MainWindow::AddImagesToProject);
connect(ui->actionImportDataAsImage, &QAction::triggered, this, &MainWindow::ImportDataAsImage);
connect(ui->actionSave, &QAction::triggered, this, &MainWindow::Save);
connect(ui->actionExit, &QAction::triggered, this, &QMainWindow::close);

Expand Down Expand Up @@ -1836,4 +1838,204 @@ void MainWindow::LoadLastProject()
}
}

void MainWindow::ImportDataAsImage()
{
// Select metadata file for import
QString filepath = QFileDialog::getOpenFileName(
this, "Import Data as Image", QDir::homePath(),
"Data Files (*.txt *.dat *.meta);;All Files (*)");

if (filepath.isEmpty())
{
return;
}

// Parse header to get dimensions with detailed error reporting
int width, height;
MetadataImporter::ImportError parse_error;
if (!MetadataImporter::ParseHeaderWithError(filepath, width, height, parse_error))
{
QString error_title;
switch (parse_error.type)
{
case MetadataImporter::ImportError::FILE_NOT_FOUND:
error_title = "File Access Error";
break;
case MetadataImporter::ImportError::INVALID_HEADER_FORMAT:
error_title = "Invalid File Format";
break;
case MetadataImporter::ImportError::INVALID_DIMENSIONS:
error_title = "Invalid Dimensions";
break;
default:
error_title = "Parse Error";
break;
}
QMessageBox::critical(this, error_title, parse_error.message);
return;
}

// Show import settings dialog
MetadataImportSettingsDialog dialog(filepath, width, height, this);
if (dialog.exec() != QDialog::Accepted)
{
return;
}

// Get import settings from dialog
MetadataImporter::ImportSettings settings = dialog.GetSettings();

// Process metadata file with detailed error reporting
statusBar()->showMessage("Processing metadata file...", 5000);
QApplication::processEvents();

MetadataImporter::ImportError import_error;
QImage image = MetadataImporter::ImportMetadataFileWithError(filepath, settings, import_error);
if (image.isNull())
{
QString error_title;
switch (import_error.type)
{
case MetadataImporter::ImportError::DATA_MISMATCH:
error_title = "Data Mismatch";
break;
case MetadataImporter::ImportError::INVALID_NUMERIC_DATA:
error_title = "File Format Error";
break;
case MetadataImporter::ImportError::CROP_BOUNDARY_ERROR:
error_title = "Invalid Crop Region";
break;
case MetadataImporter::ImportError::FILE_NOT_FOUND:
error_title = "File Access Error";
break;
default:
error_title = "Import Failed";
break;
}
QMessageBox::critical(this, error_title, import_error.message);
statusBar()->showMessage("Import failed", 3000);
return;
}

// Generate output filename
QFileInfo file_info(filepath);
QString base_name = file_info.baseName();
QString temp_filename = QString("metadata_%1_%2x%3.png")
.arg(base_name)
.arg(image.width())
.arg(image.height());

QString temp_path;
bool save_to_project = false;

if (!project_directory_.isEmpty())
{
// Save to project if one is loaded
QString images_dir = project_directory_ + "/images";
QDir dir;
if (!dir.exists(images_dir))
{
dir.mkpath(images_dir);
}

temp_path = images_dir + "/" + temp_filename;
save_to_project = true;
}
else
{
// Save to temporary location if no project is loaded
QString temp_dir = QStandardPaths::writableLocation(QStandardPaths::TempLocation);
temp_path = temp_dir + "/" + temp_filename;
}

// Save processed image
if (!image.save(temp_path, "PNG"))
{
QMessageBox::critical(this, "Save Failed",
"Failed to save processed image.\n\n"
"Path: " + temp_path);
statusBar()->showMessage("Save failed", 3000);
return;
}

// Apply project-level crop if enabled and saving to project
if (save_to_project && project_config_.IsCropEnabled())
{
const CropConfig& crop = project_config_.GetCropConfig();
QImage loaded_image(temp_path);

if (!loaded_image.isNull())
{
int crop_width = crop.width > 0 ? crop.width : loaded_image.width() - crop.x;
int crop_height = crop.height > 0 ? crop.height : loaded_image.height() - crop.y;

// Validate crop bounds
if (crop.x >= 0 && crop.y >= 0 &&
crop.x + crop_width <= loaded_image.width() &&
crop.y + crop_height <= loaded_image.height())
{
QImage cropped = loaded_image.copy(crop.x, crop.y, crop_width, crop_height);

// Save cropped image (overwrite the original)
if (!cropped.save(temp_path))
{
QMessageBox::warning(this, "Crop Failed",
"Failed to apply project crop settings.\n\n"
"Using original processed image.");
}
}
}
}

if (save_to_project)
{
// Rescan project images and save config
ScanProjectImages();
SaveProjectConfig();

// Load the new image
int image_index = image_list_.indexOf(temp_filename);
if (image_index >= 0)
{
LoadImageAtIndex(image_index);
}
else if (current_image_path_.isEmpty() && !image_list_.isEmpty())
{
LoadImageAtIndex(0);
}

statusBar()->showMessage(QString("Metadata imported as: %1").arg(temp_filename), 5000);
QMessageBox::information(this, "Import Complete",
QString("Metadata successfully imported as grayscale image.\n\n"
"File: %1\n"
"Dimensions: %2 x %3\n"
"Added to project images.")
.arg(temp_filename)
.arg(image.width())
.arg(image.height()));
}
else
{
// Load image directly if no project
auto pixmap = QPixmap(temp_path);
ui->label->setPixmap(pixmap);
ui->label->setFixedSize(pixmap.size());
current_image_path_ = temp_path;
current_image_index_ = -1;

UpdateWindowTitle();
UpdateStatusBar();

statusBar()->showMessage(QString("Metadata imported: %1").arg(temp_filename), 5000);
QMessageBox::information(this, "Import Complete",
QString("Metadata successfully imported as grayscale image.\n\n"
"File: %1\n"
"Dimensions: %2 x %3\n"
"No project loaded - image loaded directly.")
.arg(temp_filename)
.arg(image.width())
.arg(image.height()));
}
}


1 change: 1 addition & 0 deletions src/mainwindow.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,7 @@ class MainWindow : public QMainWindow
void PreviousClass();
void SelectClassByNumber(int number);
void AddImagesToProject();
void ImportDataAsImage();

// Project Management
void CreateNewProject();
Expand Down
19 changes: 8 additions & 11 deletions src/mainwindow.ui
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -56,8 +56,8 @@
<addaction name="separator"/>
<addaction name="actionProjectSettings"/>
<addaction name="separator"/>
<addaction name="actionOpenImage"/>
<addaction name="actionAddImages"/>
<addaction name="actionImportDataAsImage"/>
<addaction name="separator"/>
<addaction name="actionSave"/>
<addaction name="separator"/>
Expand Down Expand Up @@ -129,26 +129,23 @@
<addaction name="menuHelp"/>
</widget>

<action name="actionOpenImage">
<action name="actionAddImages">
<property name="text">
<string>Open Image</string>
<string>Import Images</string>
</property>
<property name="toolTip">
<string>Open Image (Ctrl+O) - Load a single image</string>
<string>Import Images - Add images to existing project</string>
</property>
<property name="shortcut">
<string>Ctrl+O</string>
<string>Ctrl+Shift+A</string>
</property>
</action>
<action name="actionAddImages">
<action name="actionImportDataAsImage">
<property name="text">
<string>Add Images to Project...</string>
<string>Import Data as Image</string>
</property>
<property name="toolTip">
<string>Add Images to Project - Add more images to existing project</string>
</property>
<property name="shortcut">
<string>Ctrl+Shift+A</string>
<string>Import numerical data file as grayscale image</string>
</property>
</action>
<action name="actionSave">
Expand Down
Loading
Loading