From 687c5e244e46e51786afad77f5015cae9abad129 Mon Sep 17 00:00:00 2001
From: Willem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl>
Date: Wed, 6 Jan 2016 15:10:34 +0100
Subject: Add multi-GPU support to CompositeGeometryManager

---
 include/astra/CompositeGeometryManager.h | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

(limited to 'include/astra')

diff --git a/include/astra/CompositeGeometryManager.h b/include/astra/CompositeGeometryManager.h
index 6610151..49d02a7 100644
--- a/include/astra/CompositeGeometryManager.h
+++ b/include/astra/CompositeGeometryManager.h
@@ -50,9 +50,16 @@ class CProjectionGeometry3D;
 class CProjector3D;
 
 
+struct SGPUParams {
+	std::vector<int> GPUIndices;
+	size_t memory;
+};
+
 
 class _AstraExport CCompositeGeometryManager {
 public:
+	CCompositeGeometryManager();
+
 	class CPart;
 	typedef std::list<boost::shared_ptr<CPart> > TPartList;
 	class CPart {
@@ -139,10 +146,19 @@ public:
 	bool doFP(CProjector3D *pProjector, const std::vector<CFloat32VolumeData3DMemory *>& volData, const std::vector<CFloat32ProjectionData3DMemory *>& projData);
 	bool doBP(CProjector3D *pProjector, const std::vector<CFloat32VolumeData3DMemory *>& volData, const std::vector<CFloat32ProjectionData3DMemory *>& projData);
 
+	void setGPUIndices(const std::vector<int>& GPUIndices);
+
+	static void setGlobalGPUParams(const SGPUParams& params);
+
 protected:
 
 	bool splitJobs(TJobSet &jobs, size_t maxSize, int div, TJobSet &split);
 
+	std::vector<int> m_GPUIndices;
+	size_t m_iMaxSize;
+
+
+	static SGPUParams* s_params;
 };
 
 }
-- 
cgit v1.2.3