2023-02-28 01:15:26 +00:00
|
|
|
// AUTOGENERATED COPYRIGHT HEADER START
|
|
|
|
// Copyright (C) 2020-2023 Michael Fabian 'Xaymar' Dirks <info@xaymar.com>
|
|
|
|
// Copyright (C) 2022 lainon <GermanAizek@yandex.ru>
|
|
|
|
// AUTOGENERATED COPYRIGHT HEADER END
|
2020-03-20 22:10:19 +00:00
|
|
|
|
|
|
|
#pragma once
|
|
|
|
#include "nvidia-cuda.hpp"
|
|
|
|
|
2022-08-29 10:29:44 +00:00
|
|
|
#include "warning-disable.hpp"
|
|
|
|
#include <memory>
|
|
|
|
#include "warning-enable.hpp"
|
|
|
|
|
2021-06-16 10:16:35 +00:00
|
|
|
namespace streamfx::nvidia::cuda {
|
2020-11-08 05:47:47 +00:00
|
|
|
class context_stack;
|
|
|
|
|
2021-06-16 10:16:35 +00:00
|
|
|
class context : public std::enable_shared_from_this<::streamfx::nvidia::cuda::context> {
|
|
|
|
std::shared_ptr<::streamfx::nvidia::cuda::cuda> _cuda;
|
|
|
|
::streamfx::nvidia::cuda::context_t _ctx;
|
|
|
|
bool _has_device;
|
|
|
|
::streamfx::nvidia::cuda::device_t _device;
|
2020-03-20 22:10:19 +00:00
|
|
|
|
2020-11-08 05:47:47 +00:00
|
|
|
public:
|
|
|
|
~context();
|
|
|
|
|
2020-03-20 22:10:19 +00:00
|
|
|
private:
|
2020-11-05 05:39:01 +00:00
|
|
|
context();
|
2020-03-20 22:10:19 +00:00
|
|
|
|
|
|
|
public:
|
|
|
|
#ifdef WIN32
|
2020-11-05 05:39:01 +00:00
|
|
|
context(ID3D11Device* device);
|
2020-03-20 22:10:19 +00:00
|
|
|
#endif
|
|
|
|
|
2021-06-16 10:16:35 +00:00
|
|
|
::streamfx::nvidia::cuda::context_t get();
|
2020-11-08 05:47:47 +00:00
|
|
|
|
|
|
|
void push();
|
|
|
|
void pop();
|
|
|
|
|
|
|
|
void synchronize();
|
|
|
|
|
|
|
|
public:
|
2021-06-16 10:16:35 +00:00
|
|
|
std::shared_ptr<::streamfx::nvidia::cuda::context_stack> enter();
|
2020-11-08 05:47:47 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
class context_stack {
|
2021-06-16 10:16:35 +00:00
|
|
|
std::shared_ptr<::streamfx::nvidia::cuda::context> _ctx;
|
2020-11-08 05:47:47 +00:00
|
|
|
|
|
|
|
public:
|
|
|
|
inline ~context_stack()
|
|
|
|
{
|
|
|
|
_ctx->pop();
|
|
|
|
}
|
2022-07-21 11:09:10 +00:00
|
|
|
inline context_stack(std::shared_ptr<::streamfx::nvidia::cuda::context> ctx) : _ctx(std::move(ctx))
|
2020-11-08 05:47:47 +00:00
|
|
|
{
|
|
|
|
_ctx->push();
|
|
|
|
}
|
2020-03-20 22:10:19 +00:00
|
|
|
};
|
2021-06-16 10:16:35 +00:00
|
|
|
} // namespace streamfx::nvidia::cuda
|