From a8a98f0aa6bc6628338aed08372f1b90892cdf7b Mon Sep 17 00:00:00 2001
From: Paul Romano <paul.k.romano@gmail.com>
Date: Sat, 23 May 2015 11:16:12 +0700
Subject: [PATCH] Expand setup.py to include required and optional dependencies

---
 setup.py | 30 ++++++++++++++++++++++++++----
 1 file changed, 26 insertions(+), 4 deletions(-)

diff --git a/setup.py b/setup.py
index 86047cc55..830d54215 100644
--- a/setup.py
+++ b/setup.py
@@ -1,13 +1,35 @@
 #!/usr/bin/env python
 
+import glob
 import os
-from distutils.core import setup
+from setuptools import setup, find_packages
 
 setup(name='openmc',
       version='0.6.2',
-      description='OpenMC Python API',
+      packages=find_packages(),
+      scripts=glob.glob('scripts/openmc-*'),
+
+      # Required dependencies
+      install_requires=['numpy', 'scipy', 'h5py', 'matplotlib'],
+
+      # Optional dependencies
+      extras_require={
+          'vtk': ['vtk', 'silomesh'],
+          'validate': ['lxml'],
+      },
+
+      # Metadata
       author='Will Boyd',
       author_email='wbinventor@gmail.com',
+      description='OpenMC Python API',
       url='https://github.com/mit-crpg/openmc',
-      packages=['openmc'],
-      scripts=[os.path.join('scripts', f) for f in os.listdir('scripts')])
+      classifiers=[
+          'Intended Audience :: Developers',
+          'Intended Audience :: End Users/Desktop',
+          'Intended Audience :: Science/Research',
+          'License :: OSI Approved :: MIT License',
+          'Natural Language :: English',
+          'Programming Language :: Python',
+          'Topic :: Scientific/Engineering'
+      ]
+)