From ade261fc3af28760935f38256f0d2638139992b7 Mon Sep 17 00:00:00 2001
From: Alejandro Saucedo <axsauze@gmail.com>
Date: Sat, 20 Feb 2021 19:37:46 +0000
Subject: [PATCH] Added sources function to python

---
 python/src/main.cpp | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/python/src/main.cpp b/python/src/main.cpp
index b79ca3e9a..978a35943 100644
--- a/python/src/main.cpp
+++ b/python/src/main.cpp
@@ -40,7 +40,17 @@ PYBIND11_MODULE(kp, m) {
                 return py::bytes((const char*)spirv.data(), spirv.size() * sizeof(uint32_t));
             },
             "Compiles string source provided and returns the value in bytes",
-            py::arg("source"), py::arg("entryPoint") = "main", py::arg("definitions") = std::vector<std::pair<std::string,std::string>>() );
+            py::arg("source"), py::arg("entryPoint") = "main", py::arg("definitions") = std::vector<std::pair<std::string,std::string>>() )
+        .def_static("compile_sources", [](
+                                    const std::vector<std::string>& source,
+                                    const std::vector<std::string>& files,
+                                    const std::string& entryPoint,
+                                    const std::vector<std::pair<std::string,std::string>>& definitions) {
+                std::vector<uint32_t> spirv = kp::Shader::compile_sources(source, files, entryPoint, definitions);
+                return py::bytes((const char*)spirv.data(), spirv.size() * sizeof(uint32_t));
+            },
+            "Compiles sources provided with file names and returns the value in bytes",
+            py::arg("sources"), py::arg("files") = std::vector<std::string>(), py::arg("entryPoint") = "main", py::arg("definitions") = std::vector<std::pair<std::string,std::string>>() );
 
     py::class_<kp::Tensor, std::shared_ptr<kp::Tensor>>(m, "Tensor", DOC(kp, Tensor))
         .def(py::init(