Simulators/Factory.h

Go to the documentation of this file.

 
#pragma once
 
#ifndef _SIMULATORS_FACTORY_H_
#define _SIMULATORS_FACTORY_H_
 
#include "GpuLibStateVectorSim.h"
#include "GpuLibMPSSim.h"
#include "GpuLibTNSim.h"
#include "GpuStabilizer.h"
#include "GpuPauliPropagator.h"
 
#include "Simulator.h"
 
namespace Simulators {
 
class SimulatorsFactory {
 public:
  static std::shared_ptr<ISimulator> CreateSimulator(
      SimulatorType t = SimulatorType::kQCSim,
      SimulationType method = SimulationType::kMatrixProductState);
 
  static std::unique_ptr<ISimulator> CreateSimulatorUnique(
      SimulatorType t = SimulatorType::kQCSim,
      SimulationType method = SimulationType::kMatrixProductState);
 
#ifdef __linux__
  static bool InitGpuLibrary();
  static bool InitGpuLibraryWithMute();
 
  static bool IsGpuLibraryAvailable() {
    return gpuLibrary && gpuLibrary->IsValid();
  }
 
  static std::shared_ptr<GpuLibrary> GetGpuLibrary() {
    if (!gpuLibrary || !gpuLibrary->IsValid()) return nullptr;
    return gpuLibrary;
  }
 
  static std::unique_ptr<GpuLibStateVectorSim> CreateGpuLibStateVectorSim() {
    if (!gpuLibrary || !gpuLibrary->IsValid()) return nullptr;
 
    return std::make_unique<GpuLibStateVectorSim>(gpuLibrary);
  }
 
  static std::unique_ptr<GpuLibMPSSim> CreateGpuLibMPSSim() {
    if (!gpuLibrary || !gpuLibrary->IsValid()) return nullptr;
 
    return std::make_unique<GpuLibMPSSim>(gpuLibrary);
  }
 
  static std::unique_ptr<GpuLibTNSim> CreateGpuLibTensorNetSim() {
    if (!gpuLibrary || !gpuLibrary->IsValid()) return nullptr;
 
    return std::make_unique<GpuLibTNSim>(gpuLibrary);
  }
 
  static std::shared_ptr<GpuStabilizer> CreateGpuStabilizerSimulator() {
    if (!gpuLibrary || !gpuLibrary->IsValid()) return nullptr;
    return std::make_shared<GpuStabilizer>(gpuLibrary);
  }
 
  static std::shared_ptr<GpuPauliPropagator> CreateGpuPauliPropagatorSimulator() {
    if (!gpuLibrary || !gpuLibrary->IsValid()) return nullptr;
    return std::make_shared<GpuPauliPropagator>(gpuLibrary);
  }
 
  static std::unique_ptr<GpuPauliPropagator> CreateGpuPauliPropagatorSimulatorUnique() {
    if (!gpuLibrary || !gpuLibrary->IsValid()) return nullptr;
    return std::make_unique<GpuPauliPropagator>(gpuLibrary);
  }
 
 private:
  static std::shared_ptr<GpuLibrary> gpuLibrary;
  static std::atomic_bool firstTime;
#else
  static bool IsGpuLibraryAvailable() { return false; }
 
  static bool InitGpuLibrary() { return false; }
#endif
};
 
}  // namespace Simulators
 
#endif  // !_SIMULATORS_FACTORY_H_