Skip to content
Snippets Groups Projects
Commit 13976f18 authored by Julian Lenz's avatar Julian Lenz
Browse files

Added setDefaults to FancyAxes

parent 2d717749
No related branches found
No related tags found
No related merge requests found
......@@ -62,11 +62,30 @@ class FancyAxes(mpl.axes.Axes):
mCyclerErrorbar=None
lCyclerErrorbar=None
defaults=None
def __init__(self,*args,**kwargs):
self.resetCyclers()
super().__init__(*args,**kwargs)
self.setDefaults(reset=True)
return None
def setDefaults(self,**kwargs):
if kwargs.pop('reset',False):
self.defaults=dict(
callOrig=False,
withLinesErrorbar=True,
withLinesPlot=False,
advanceCycler=True,
)
if kwargs.pop('checkForKeys',True):
if set(kwargs.keys()).difference(self.defaults.keys())!=set():
raise Exception()
self.defaults.update(kwargs)
return self
def resetCyclers(self):
self.cCyclerPlot=Cycler(colors)
self.mCyclerPlot=Cycler(markers)
......@@ -86,11 +105,11 @@ class FancyAxes(mpl.axes.Axes):
of the original plot().
'''
if kwargs.pop('callOrig',False):
if kwargs.pop('callOrig',self.defaults['callOrig']):
return super().plot(*args,**kwargs)
# extract the new kwarg
withLines=kwargs.pop('withLines',False)
withLines=kwargs.pop('withLines',self.defaults['withLinesPlot'])
if isinstance(withLines,bool):
# set the default transparency if withLines is True
if withLines:
......@@ -113,7 +132,7 @@ class FancyAxes(mpl.axes.Axes):
args=args[:-1]
advance=kwargs.pop('advanceCycler',True)
advance=kwargs.pop('advanceCycler',self.defaults['advanceCycler'])
lCycler=self.lCyclerPlot
mCycler=self.mCyclerPlot
......@@ -174,13 +193,13 @@ class FancyAxes(mpl.axes.Axes):
'''
if kwargs.pop('callOrig',False):
if kwargs.pop('callOrig',self.defaults['callOrig']):
return super().errorbar(*args,**kwargs)
if len(args)==0:
return None
# extract the new kwarg
withLines=kwargs.pop('withLines',True)
withLines=kwargs.pop('withLines',self.defaults['withLinesErrorbar'])
if isinstance(withLines,bool):
# set the default transparency if withLines is True
if withLines:
......@@ -214,7 +233,7 @@ class FancyAxes(mpl.axes.Axes):
# filter out label
label=kwargs.pop('label',None)
advance=kwargs.pop('advanceCycler',True)
advance=kwargs.pop('advanceCycler',self.defaults['advanceCycler'])
lCycler=self.lCyclerErrorbar
mCycler=self.mCyclerErrorbar
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment