From 71fab0c2d8ccf8107c42cfaf58379ad678f7cbb7 Mon Sep 17 00:00:00 2001
From: Bocheng Zou <bocheng.zou@outlook.com>
Date: Wed, 4 Feb 2026 16:35:54 -0600
Subject: [PATCH] Expose DEMClumpBatch::SetOriQ to Python interface

---
 src/DEM/PyDEME.cpp | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/DEM/PyDEME.cpp b/src/DEM/PyDEME.cpp
index 48fdd188..54906e29 100644
--- a/src/DEM/PyDEME.cpp
+++ b/src/DEM/PyDEME.cpp
@@ -1250,6 +1250,14 @@ PYBIND11_MODULE(DEME, obj) {
                  &deme::DEMClumpBatch::AddGeometryWildcard))
         .def("AddGeometryWildcard", static_cast<void (deme::DEMClumpBatch::*)(const std::string&, float)>(
                                         &deme::DEMClumpBatch::AddGeometryWildcard))
+        .def("SetOriQ",
+              static_cast<void (deme::DEMClumpBatch::*)(const std::vector<float4>&)>(&deme::DEMClumpBatch::SetOriQ))
+        .def("SetOriQ",
+             static_cast<void (deme::DEMClumpBatch::*)(float4)>(&deme::DEMClumpBatch::SetOriQ))
+        .def("SetOriQ",
+              static_cast<void (deme::DEMClumpBatch::*)(const std::vector<float>&)>(&deme::DEMClumpBatch::SetOriQ))
+        .def("SetOriQ", static_cast<void (deme::DEMClumpBatch::*)(const std::vector<std::vector<float>>&)>(
+                            &deme::DEMClumpBatch::SetOriQ))
         .def("GetNumContacts", &deme::DEMClumpBatch::GetNumContacts);
 
     py::class_<deme::DEMExternObj, deme::DEMInitializer, std::shared_ptr<deme::DEMExternObj>>(obj, "DEMExternObj")