Simulators/Factory.h

Go to the documentation of this file.

 
#pragma once
 
#ifndef _SIMULATORS_FACTORY_H_
#define _SIMULATORS_FACTORY_H_
 
#include "GpuLibMPSSim.h"
#include "GpuLibStateVectorSim.h"
#include "GpuLibrary.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 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::shared_ptr<GpuLibrary> GetGpuLibrary() {
    if (!gpuLibrary || !gpuLibrary->IsValid())
      return nullptr;
    return gpuLibrary;
  }
 
  static bool IsGpuLibraryAvailable() {
    return gpuLibrary && gpuLibrary->IsValid();
  }
 
  static bool InitGpuLibrary();
  static bool InitGpuLibraryWithMute();
 
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_