resources.h 591 Bytes
Newer Older
Nianchen Deng's avatar
sync    
Nianchen Deng committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
#include "resource.h"
#include <map>
#include <vector>

namespace utils::cuda {
    class Resources : public std::map<std::string, Resource &> {
    public:
        std::vector<cudaGraphicsResource_t> getGraphicsResourceHandlers() {
            std::vector<cudaGraphicsResource_t> handlers;
            for (auto &&item : *this) {
                auto gres = dynamic_cast<GraphicsResource *>(&item.second);
                if (gres != nullptr)
                    handlers.push_back(gres->getHandler());
            }
            return handlers;
        }
    };
} // namespace utils::cuda