Merge pull request #787 from sanjayankur31/fix/786

fix(imports): only import matplotlib where needed and if gui is enabled

netpyne/analysis/csd.py

@@ -20,9 +20,6 @@
 import numpy as np
 import scipy
 from numbers import Number
-import matplotlib
-from matplotlib import pyplot as plt
-from matplotlib import ticker as ticker
 import json
 import sys
 import os

netpyne/plotting/plotCSD.py

@@ -1,10 +1,13 @@
 # PLOTTING CSD
 
+from netpyne import __gui__
+if __gui__:
+    import matplotlib
+    from matplotlib import pyplot as plt
+
 from ..analysis.utils import exception, _showFigure
 import numpy as np
 import scipy
-import matplotlib
-from matplotlib import pyplot as plt
 
 
 @exception

netpyne/plotting/plotLFPLocations.py

@@ -1,7 +1,5 @@
 # Generate plots of LFP (local field potentials) and related analyses
 
-import matplotlib.patches as mpatches
-import matplotlib.pyplot as plt
 import math
 from ..analysis.utils import exception  # , loadData
 from ..analysis.tools import loadData

netpyne/plotting/plotLFPSpectrogram.py

@@ -1,7 +1,10 @@
 # Generate plots of LFP (local field potentials) and related analyses
 
-import matplotlib.patches as mpatches
-import matplotlib.pyplot as plt
+from netpyne import __gui__
+
+if __gui__:
+    import matplotlib.pyplot as plt
+
 import math
 from ..analysis.utils import exception  # , loadData
 from ..analysis.tools import loadData

netpyne/plotting/plotRaster.py

@@ -1,6 +1,9 @@
 # Generate a raster plot of spiking
 
-import matplotlib.patches as mpatches
+from netpyne import __gui__
+
+if __gui__:
+    import matplotlib.patches as mpatches
 from ..analysis.utils import exception  # , loadData
 from ..analysis.tools import loadData
 from .plotter import ScatterPlotter

netpyne/plotting/plotSpikeFreq.py

@@ -1,7 +1,9 @@
 # Generate a spike frequency plot
 
+from netpyne import __gui__
+if __gui__:
+    import matplotlib.patches as mpatches
 import numpy as np
-import matplotlib.patches as mpatches
 from numbers import Number
 from ..analysis.utils import exception, _smooth1d
 from ..analysis.tools import loadData

netpyne/plotting/plotSpikeHist.py

@@ -1,7 +1,10 @@
 # Generate a spike histogram
 
+from netpyne import __gui__
+
+if __gui__:
+    import matplotlib.patches as mpatches
 import numpy as np
-import matplotlib.patches as mpatches
 from ..analysis.utils import exception
 from ..analysis.tools import loadData
 from .plotter import HistPlotter

netpyne/plotting/plotTimeSeries.py

@@ -1,7 +1,5 @@
 # Generate plots of LFP (local field potentials) and related analyses
 
-import matplotlib.patches as mpatches
-import matplotlib.pyplot as plt
 import math
 from ..analysis.utils import exception  # , loadData
 from ..analysis.tools import loadData

netpyne/plotting/plotTimeSeriesPSD.py

@@ -1,7 +1,5 @@
 # Generate plots of LFP (local field potentials) and related analyses
 
-import matplotlib.patches as mpatches
-import matplotlib.pyplot as plt
 import math
 from ..analysis.utils import exception  # , loadData
 from ..analysis.tools import loadData

netpyne/plotting/plotter.py

@@ -3,13 +3,17 @@
 
 """
 
-import matplotlib as mpl
-import matplotlib.pyplot as plt
+from netpyne import __gui__
+
+if __gui__:
+    import matplotlib as mpl
+    import matplotlib.pyplot as plt
+    from matplotlib.offsetbox import AnchoredOffsetbox
+
 import numpy as np
 from copy import deepcopy
 import pickle, json
 import os
-from matplotlib.offsetbox import AnchoredOffsetbox
 
 try:
     basestring
@@ -1244,64 +1248,67 @@ def plot(self, **kwargs):
         return self.fig
 
 
-class _AnchoredScaleBar(AnchoredOffsetbox):
-    """
-    A class used for adding scale bars to plots
-
-    Modified from here: https://gist.github.com/dmeliza/3251476
-    """
-
-    def __init__(
-        self,
-        axis,
-        sizex=0,
-        sizey=0,
-        labelx=None,
-        labely=None,
-        loc=4,
-        pad=0.1,
-        borderpad=0.1,
-        sep=2,
-        prop=None,
-        barcolor="black",
-        barwidth=None,
-        **kwargs
-    ):
+try:
+    class _AnchoredScaleBar(AnchoredOffsetbox):
         """
-        Draw a horizontal and/or vertical  bar with the size in data coordinate
-        of the give axes. A label will be drawn underneath (center-aligned).
-
-        - transform : the coordinate frame (typically axes.transData)
-        - sizex,sizey : width of x,y bar, in data units. 0 to omit
-        - labelx,labely : labels for x,y bars; None to omit
-        - loc : position in containing axes
-        - pad, borderpad : padding, in fraction of the legend font size (or prop)
-        - sep : separation between labels and bars in points.
-        - **kwargs : additional arguments passed to base class constructor
+        A class used for adding scale bars to plots
+
+        Modified from here: https://gist.github.com/dmeliza/3251476
         """
-        from matplotlib.patches import Rectangle
-        from matplotlib.offsetbox import AuxTransformBox, VPacker, HPacker, TextArea, DrawingArea
-
-        bars = AuxTransformBox(axis.transData)
-        if sizex:
-            if axis.xaxis_inverted():
-                sizex = -sizex
-            bars.add_artist(Rectangle((0, 0), sizex, 0, ec=barcolor, lw=barwidth, fc="none"))
-        if sizey:
-            if axis.yaxis_inverted():
-                sizey = -sizey
-            bars.add_artist(Rectangle((0, 0), 0, sizey, ec=barcolor, lw=barwidth, fc="none"))
-
-        if sizex and labelx:
-            self.xlabel = TextArea(labelx)
-            bars = VPacker(children=[bars, self.xlabel], align="center", pad=0, sep=sep)
-        if sizey and labely:
-            self.ylabel = TextArea(labely)
-            bars = HPacker(children=[self.ylabel, bars], align="center", pad=0, sep=sep)
-
-        AnchoredOffsetbox.__init__(
-            self, loc, pad=pad, borderpad=borderpad, child=bars, prop=prop, frameon=False, **kwargs
-        )
+
+        def __init__(
+            self,
+            axis,
+            sizex=0,
+            sizey=0,
+            labelx=None,
+            labely=None,
+            loc=4,
+            pad=0.1,
+            borderpad=0.1,
+            sep=2,
+            prop=None,
+            barcolor="black",
+            barwidth=None,
+            **kwargs
+        ):
+            """
+            Draw a horizontal and/or vertical  bar with the size in data coordinate
+            of the give axes. A label will be drawn underneath (center-aligned).
+
+            - transform : the coordinate frame (typically axes.transData)
+            - sizex,sizey : width of x,y bar, in data units. 0 to omit
+            - labelx,labely : labels for x,y bars; None to omit
+            - loc : position in containing axes
+            - pad, borderpad : padding, in fraction of the legend font size (or prop)
+            - sep : separation between labels and bars in points.
+            - **kwargs : additional arguments passed to base class constructor
+            """
+            from matplotlib.patches import Rectangle
+            from matplotlib.offsetbox import AuxTransformBox, VPacker, HPacker, TextArea, DrawingArea
+
+            bars = AuxTransformBox(axis.transData)
+            if sizex:
+                if axis.xaxis_inverted():
+                    sizex = -sizex
+                bars.add_artist(Rectangle((0, 0), sizex, 0, ec=barcolor, lw=barwidth, fc="none"))
+            if sizey:
+                if axis.yaxis_inverted():
+                    sizey = -sizey
+                bars.add_artist(Rectangle((0, 0), 0, sizey, ec=barcolor, lw=barwidth, fc="none"))
+
+            if sizex and labelx:
+                self.xlabel = TextArea(labelx)
+                bars = VPacker(children=[bars, self.xlabel], align="center", pad=0, sep=sep)
+            if sizey and labely:
+                self.ylabel = TextArea(labely)
+                bars = HPacker(children=[self.ylabel, bars], align="center", pad=0, sep=sep)
+
+                AnchoredOffsetbox.__init__(
+                    self, loc, pad=pad, borderpad=borderpad, child=bars, prop=prop, frameon=False, **kwargs
+                )
+except NameError:
+    print("-nogui passed, matplotlib is unavailable")
 
 
 def _add_scalebar(

netpyne/support/morlet.py

@@ -15,9 +15,12 @@
 from __future__ import unicode_literals
 from __future__ import absolute_import
 
+from netpyne import __gui__
+if __gui__:
+    import matplotlib.pyplot as plt
+
 import numpy as np
 import scipy.signal as sps
-import matplotlib.pyplot as plt
 
 
 def index2ms(idx, sampr):

netpyne/support/morphology.py

@@ -20,7 +20,9 @@
 from builtins import object
 import numpy as np
 import pylab as plt
-from matplotlib.pyplot import cm
+from netpyne import __gui__
+if __gui__:
+    from matplotlib.pyplot import cm
 import string
 from neuron import h
 import numbers

netpyne/support/scalebar.py

@@ -16,56 +16,61 @@
 from future import standard_library
 
 standard_library.install_aliases()
-from matplotlib.offsetbox import AnchoredOffsetbox
-
-
-class AnchoredScaleBar(AnchoredOffsetbox):
-    def __init__(
-        self,
-        transform,
-        sizex=0,
-        sizey=0,
-        labelx=None,
-        labely=None,
-        loc=4,
-        pad=0.1,
-        borderpad=0.1,
-        sep=2,
-        prop=None,
-        barcolor="black",
-        barwidth=None,
-        **kwargs
-    ):
-        """
-        Draw a horizontal and/or vertical  bar with the size in data coordinate
-        of the give axes. A label will be drawn underneath (center-aligned).
-        - transform : the coordinate frame (typically axes.transData)
-        - sizex,sizey : width of x,y bar, in data units. 0 to omit
-        - labelx,labely : labels for x,y bars; None to omit
-        - loc : position in containing axes
-        - pad, borderpad : padding, in fraction of the legend font size (or prop)
-        - sep : separation between labels and bars in points.
-        - **kwargs : additional arguments passed to base class constructor
-        """
-        from matplotlib.patches import Rectangle
-        from matplotlib.offsetbox import AuxTransformBox, VPacker, HPacker, TextArea, DrawingArea
-
-        bars = AuxTransformBox(transform)
-        if sizex:
-            bars.add_artist(Rectangle((0, 0), sizex, 0, ec=barcolor, lw=barwidth, fc="none"))
-        if sizey:
-            bars.add_artist(Rectangle((0, 0), 0, sizey, ec=barcolor, lw=barwidth, fc="none"))
-
-        if sizex and labelx:
-            self.xlabel = TextArea(labelx, minimumdescent=False)
-            bars = VPacker(children=[bars, self.xlabel], align="center", pad=0, sep=sep)
-        if sizey and labely:
-            self.ylabel = TextArea(labely)
-            bars = HPacker(children=[self.ylabel, bars], align="center", pad=0, sep=sep)
-
-        AnchoredOffsetbox.__init__(
-            self, loc, pad=pad, borderpad=borderpad, child=bars, prop=prop, frameon=False, **kwargs
-        )
+from netpyne import __gui__
+
+if __gui__:
+    from matplotlib.offsetbox import AnchoredOffsetbox
+
+try:
+    class AnchoredScaleBar(AnchoredOffsetbox):
+        def __init__(
+            self,
+            transform,
+            sizex=0,
+            sizey=0,
+            labelx=None,
+            labely=None,
+            loc=4,
+            pad=0.1,
+            borderpad=0.1,
+            sep=2,
+            prop=None,
+            barcolor="black",
+            barwidth=None,
+            **kwargs
+        ):
+            """
+            Draw a horizontal and/or vertical  bar with the size in data coordinate
+            of the give axes. A label will be drawn underneath (center-aligned).
+            - transform : the coordinate frame (typically axes.transData)
+            - sizex,sizey : width of x,y bar, in data units. 0 to omit
+            - labelx,labely : labels for x,y bars; None to omit
+            - loc : position in containing axes
+            - pad, borderpad : padding, in fraction of the legend font size (or prop)
+            - sep : separation between labels and bars in points.
+            - **kwargs : additional arguments passed to base class constructor
+            """
+            from matplotlib.patches import Rectangle
+            from matplotlib.offsetbox import AuxTransformBox, VPacker, HPacker, TextArea, DrawingArea
+
+            bars = AuxTransformBox(transform)
+            if sizex:
+                bars.add_artist(Rectangle((0, 0), sizex, 0, ec=barcolor, lw=barwidth, fc="none"))
+            if sizey:
+                bars.add_artist(Rectangle((0, 0), 0, sizey, ec=barcolor, lw=barwidth, fc="none"))
+
+            if sizex and labelx:
+                self.xlabel = TextArea(labelx, minimumdescent=False)
+                bars = VPacker(children=[bars, self.xlabel], align="center", pad=0, sep=sep)
+            if sizey and labely:
+                self.ylabel = TextArea(labely)
+                bars = HPacker(children=[self.ylabel, bars], align="center", pad=0, sep=sep)
+
+            AnchoredOffsetbox.__init__(
+                self, loc, pad=pad, borderpad=borderpad, child=bars, prop=prop, frameon=False, **kwargs
+            )
+except NameError:
+    print("-nogui passed, matplotlib is unavailable")
 
 
 def add_scalebar(

netpyne/support/stackedBarGraph.py

@@ -48,7 +48,9 @@
 ###############################################################################
 
 import numpy as np
-from matplotlib import pyplot as plt
+from netpyne import __gui__
+if __gui__:
+    from matplotlib import pyplot as plt
 
 ###############################################################################
 

requirements.txt

@@ -5,7 +5,7 @@ pandas
 future
 tables
 bokeh
-pyneuroml
+pyneuroml>=1.1.5
 neuron
 pytest
 inspyred