Studio/ShapeWorksMONAI/MonaiLabelJob.h

Namespaces

Name
shapeworks User usage reporting (telemetry)
monailabel

Classes

	Name
class	monailabel::MonaiLabelJob

Source code

```cpp

pragma once

include

namespace py = pybind11;

namespace shapeworks { class ShapeWorksStudioApp; class Session; class Job; } // namespace shapeworks

namespace monailabel {

using shapeworks::Job; using shapeworks::ProjectHandle; using shapeworks::Session; using shapeworks::ShapeWorksStudioApp;

pragma GCC diagnostic push

pragma GCC diagnostic ignored "-Wattributes"

class MonaiLabelJob : public Job { Q_OBJECT;

public: const static std::string MONAI_RESULT_EXTENSION; const static std::string MONAI_RESULT_DTYPE;

MonaiLabelJob(QSharedPointer session, const std::string& server_url, const std::string& client_id, const std::string& strategy, const std::string& model_type); ~MonaiLabelJob(); void setServer(const std::string& server_url); void setModelType(const std::string& model_type); inline const std::string& getServer() { return server_url_; } void setClientId(const std::string& client_id = ""); inline const std::string& getClientId() { return client_id_; }; void initializeClient(); inline std::shared_ptr getClient() const { return monai_client_; };

py::dict getInfo(); std::string getModelName(std::string modelType); std::vector getModelNames(const std::string& model_type); std::string getSessionId(); py::dict getParamsFromConfig(std::string section, std::string name);

// MONAI Client callers py::dict nextSample(std::string strategy, py::dict params); py::tuple infer(std::string model, std::string image_in, py::dict params, std::string label_in, std::string file, std::string session_id); py::dict saveLabel(std::string image_in, std::string label_in, py::dict params); py::dict uploadImage(std::string image_in, std::string image_id);

void updateShapes();

void runSegmentationModel(); void run() override; QString name() override; void python_message(std::string str); void setCurrentSampleNumber(int n);

public Q_SLOTS: void onUploadSampleClicked(); // Triggered when upload Sample button is clicked void onRunSegmentationClicked(); // Triggered when Run Segmentation is clicked void onSubmitLabelClicked(); // Triggered when Submit Label is clicked

Q_SIGNALS: void triggerUpdateView(); void triggerClientInitialized(bool success); void triggerUploadSampleCompleted(); void triggerSegmentationCompleted(); void triggerSubmitLabelCompleted();

private: std::string tmp_dir_; std::string server_url_; std::string client_id_; // TODO: implement remote server session // std::string username_; // std::string password_; // std::string auth_token_; std::string strategy_; std::string model_type_ = ""; std::string model_name_ = ""; std::unordered_map> models_available_; // TODO: add functionality to interchange between models from UI std::shared_ptr monai_client_;

// QT states bool waitingForUpload; bool waitingForSegmentation; bool waitingForLabelSubmission;

// Current subjects int sample_number_; std::string currentSampleId_; std::string currentSegmentationPath_; std::vector currentSegmentationPaths_; std::vector organNames_;

QSharedPointer session_; ProjectHandle project_;

bool error_occurred_ = false; };

pragma GCC diagnostic pop

} // namespace monailabel ```

Updated on 2026-03-31 at 16:02:11 +0000