mitknnInteractiveModel.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef mitknnInteractiveModel_h
#define mitknnInteractiveModel_h
 
#include <MitkPythonSegmentationUIExports.h>
 
#include <string>
#include <vector>
 
class QWidget;
 
namespace mitk
{
  namespace nnInteractive
  {
    struct ModelInfo
    {
      std::string Id;          
      std::string DisplayName; 
      bool IsDefault = false;  
      bool Downloaded = false; 
    };
 
    enum class ModelUpdateStatus
    {
      Unknown,        
      UpToDate,       
      UpdateAvailable 
    };
 
    struct ModelCheckResult
    {
      ModelUpdateStatus Status = ModelUpdateStatus::Unknown;
      std::string CurrentId;     
      std::string RecommendedId; 
    };
 
    enum class ModelListStatus
    {
      Ok,     
      Empty,  
      Failed  
    };
 
    MITKPYTHONSEGMENTATIONUI_EXPORT std::vector<ModelInfo> ListModels(const std::string& venvName, ModelListStatus* status = nullptr, QWidget* parent = nullptr);
 
    MITKPYTHONSEGMENTATIONUI_EXPORT ModelCheckResult CheckModelUpdate(const std::string& venvName, const std::string& selectedModelId, QWidget* parent = nullptr);
  }
}
 
#endif