Add pygenn

2 files changed, 20 insertions, 5 deletions

diff --git a/.SRCINFO b/.SRCINFO
index bf7f37f9dd59..ff0335ea3be7 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,13 +1,18 @@
 pkgbase = genn
 	pkgdesc = GeNN: GPU-enhanced neural networks (version 4)
 	pkgver = 4.0.0_RC1
-	pkgrel = 5
+	pkgrel = 6
 	url = https://github.com/genn-team/genn
 	install = genn.install
 	arch = x86_64
 	license = GPL
 	makedepends = doxygen
+	makedepends = python
+	makedepends = python-numpy
+	makedepends = swig
 	depends = cuda
+	optdepends = python: for pygenn
+	optdepends = python-numpy: for pygenn
 	options = staticlibs
 	options = !emptydirs
 	source = https://github.com/genn-team/genn/archive/4.0.0-RC1.tar.gz
diff --git a/PKGBUILD b/PKGBUILD
index d29d59246c23..5b6b78063b61 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -1,18 +1,21 @@
 # Maintainer: Alex Dewar <a.dewar@sussex.ac.uk>
 pkgname=genn
 pkgver=4.0.0_RC1
-pkgrel=5
+pkgrel=6
 pkgdesc="GeNN: GPU-enhanced neural networks (version 4)"
 arch=(x86_64)
 url="https://github.com/genn-team/genn"
 license=('GPL')
 depends=(cuda)
-makedepends=(doxygen)
+makedepends=(doxygen python python-numpy swig)
+optdepends=("python: for pygenn" "python-numpy: for pygenn")
 options=(staticlibs !emptydirs)
 source=("$url/archive/${pkgver//_/-}.tar.gz")
 sha256sums=('b94c0b8c05d8525987f9d4279478dd41f68e8d9824cb5344adb5fbe257b38e99')
 install="${pkgname}.install"
 
+export CUDA_PATH=/opt/cuda
+
 prepare() {
 	cd genn-${pkgver//_/-}
 
@@ -32,23 +35,30 @@ build() {
 
 	# Build libgenn.a etc.
 	make
+	make cuda
+
+	# Build pygenn
+	make DYNAMIC=1 LIBRARY_DIRECTORY=`pwd`/pygenn/genn_wrapper/
+	python setup.py build
 }
 
 package() {
 	cd genn-${pkgver//_/-}
 
 	# Install libs and headers
-	CUDA_PATH=/opt/cuda make cuda
 	PREFIX="$pkgdir"/usr/ make install
 
 	# Install documentation
 	mkdir -p "$pkgdir"/usr/share/genn/documentation
 	cp -rf documentation/html/* "$pkgdir"/usr/share/genn/documentation
 
-	# Copy userproject
+	# Copy userproject folder
 	cp -R userproject "$pkgdir"/usr/src/genn
 
 	# Automatically set CUDA_PATH environment variable
     install -d "${pkgdir}"/etc/profile.d
     echo export CUDA_PATH=/opt/cuda > "${pkgdir}"/etc/profile.d/cuda_path.sh
+
+	# Install pygenn
+	python setup.py install --prefix=/usr --root="$pkgdir"
 }