Made logging optional

calculate_husimi.py

@@ -14,7 +14,7 @@ plt = plotparams.matplotlib.pyplot
 Line2D = plotparams.matplotlib.lines.Line2D
 
 
-PLOTTING = False             # create plots of how husimi functions come to be at each step
+PLOTTING = True             # create plots of how husimi functions come to be at each step
 LOGGING_STEPS = False        # log individual steps
 
 DATADIR = 'output/'
@@ -238,7 +238,7 @@ def main():
                 # PLOTS
                 
                 if PLOTTING:
-                    print('\t\t-> plotting wave function and fft')
+                    if LOGGING_STEPS: print('\t\t-> plotting wave function and fft')
                     
                     fig, axes = plt.subplots(
                         nrows=2,
@@ -348,7 +348,7 @@ def main():
                 
                 
                 if PLOTTING:
-                    print('\t\t-> plotting derivative and fft')
+                    if LOGGING_STEPS: print('\t\t-> plotting derivative and fft')
                     
                     fig, axes = plt.subplots(
                         nrows=2,
@@ -497,7 +497,7 @@ def main():
                 
                 # PLOTTING
                 if PLOTTING:
-                    print('\t\t-> plotting constituents of husimi parts')
+                    if LOGGING_STEPS: print('\t\t-> plotting constituents of husimi parts')
 
                     fig, axes = plt.subplots(
                         nrows=2, 
@@ -652,7 +652,7 @@ def main():
                     if N_CAVITIES == 1:
                         axes_cavities = [axes]
                     else:
-                        pass
+                        axes_cavities = axes
 
                     style_normal = dict(c='black', ls='solid', label=r'$h \cdot F_{0}$'.format(inout_index))
                     style_der = dict(c='black', ls='dashed', label=r"$h' \cdot i / k_0 " + r" F_{0}$".format(inout_index))