QmitkSetupVirtualEnvUtil.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.s
 
============================================================================*/
 
#ifndef QmitkSetupVirtualEnvUtil_h_Included
#define QmitkSetupVirtualEnvUtil_h_Included
 
#include "mitkLog.h"
#include "mitkProcessExecutor.h"
#include <MitkSegmentationUIExports.h>
#include <QString>
#include <functional>
 
class MITKSEGMENTATIONUI_EXPORT QmitkSetupVirtualEnvUtil
{
public:
  using CallbackType = void (*)(itk::Object *, const itk::EventObject &, void *);
 
  QmitkSetupVirtualEnvUtil(const QString& baseDir);
  QmitkSetupVirtualEnvUtil();
 
  bool SetupVirtualEnv(const QString &venvName,
                       const QStringList &packages,
                       std::function<bool()> validator,
                       CallbackType printCallback,
                       const QString &torchVersion = "");
 
  virtual QString GetVirtualEnvPath();
 
  void PipInstall(const std::string &library,
                  const std::string &workingDir,
                  CallbackType callback,
                  const std::string &command = "pip3");
 
  void PipInstall(const std::string &library, CallbackType callback,
                  const std::string &command = "pip3");
 
 
  void ExecutePython(const std::string &args,
                     const std::string &pythonPath,
                     CallbackType callback,
                     const std::string &command = "python");
 
  void ExecutePython(const std::string &args, CallbackType callback,
                     const std::string &command = "python");
 
  void InstallPytorch(const QString &workingDir, CallbackType callback, const QString &torchVersion = "");
 
  void InstallPytorch(const QString& torchVersion = "");
 
  void InstallPytorch(CallbackType callback, const QString& torchVersion = "");
 
 
  QString& GetBaseDir();
 
  QString& GetSystemPythonPath();
 
  QString& GetPythonPath();
 
  QString& GetPipPath();
 
  void SetSystemPythonPath(const QString& path);
 
  void SetPythonPath(const QString& path);
 
  void SetPipPath(const QString& path);
 
  void SetVirtualEnvPath(const QString &path);
 
  bool IsPythonPath(const QString &pythonPath);
 
  static bool IsVenvInstalled(const QString &pythonPath);
 
  static QString GetPipPackageVersion(const QString &pythonPath, const QString &packageName);
 
  static void PrintProcessEvent(itk::Object *, const itk::EventObject &e, void *);
 
  static std::pair<QString, QString> GetExactPythonPath(const QString &pyEnv);
  
  static QStringList AutoParsePythonPaths();
 
private:
  QString m_PythonPath;
  QString m_PipPath;
  QString m_BaseDir;
  QString m_venvPath;
  QString m_SysPythonPath;
};
#endif