#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