Commit a79362db authored by jansen31's avatar jansen31
Browse files

refactor

parent 27d2d6f6
......@@ -18,7 +18,7 @@ class DegenerateOrbital(PlanarMolecule):
'constraint': 'sparse real',
'sparsity_parameter': 40,
'use_sparsity_with_support': True,
'threshold_for_support': 0.05,
'threshold_for_support': 0.1,
'support_filename': None,
'Nx': None,
'Ny': None,
......@@ -117,7 +117,7 @@ class DegenerateOrbital(PlanarMolecule):
self.output['u1'] = self.algorithm.prox1.eval(self.algorithm.u)
self.output['u2'] = self.algorithm.prox2.eval(self.algorithm.u)
figsize = kwargs.pop("figsize", (12,3))
figsize = kwargs.pop("figsize", (12, 3))
for i, operator in enumerate(self.algorithm.proxOperators):
operator_name = self.proxOperators[i].__name__
f = self.plot_guess(operator.eval(self.algorithm.u),
......@@ -141,7 +141,7 @@ class DegenerateOrbital(PlanarMolecule):
# def saveOutput(self, **kwargs):
# super(PlanarMolecule, self).saveOutput(**kwargs)
def plot_guess(self, u, name=None, show=True, interpolate_and_zoom=False, figsize=(12,3)):
def plot_guess(self, u, name=None, show=True, interpolate_and_zoom=False, figsize=(12, 3)):
""""Given a list of fields, plot the individual fields and the combined intensity"""
prop = self.propagator(self) # This is not a string but the indicated class itself, to be instanciated
u_hat = prop.eval(u)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment