Merge pull request #205 from BioSTEAMDevelopmentGroup/QoL_STR_changes

Add parameter N_reactors to StirredTankReactor and enable passing RGBn tuples as arguments for color_wheel in plot_spearman_2d

biosteam/plots/plots.py

@@ -662,6 +662,8 @@ def plot_spearman_2d(rhos, top=None, name=None, color_wheel=None, index=None,
  Spearman's rank correlation coefficients to be plotted.
  top=None : float, optional
  Number of parameters to plot (from highest values).
+ color_wheel=None: Iterable, optional
+ Iterable of colorpalette.Color objects or RGBn values.
  
  Returns
  -------
@@ -687,7 +689,9 @@ def plot_spearman_2d(rhos, top=None, name=None, color_wheel=None, index=None,
  if not color_wheel: color_wheel = CABBI_colors.wheel()
  fig, ax = plt.subplots()
  for i, rho in enumerate(rhos):
- plot_spearman_1d(rho, color=color_wheel[N - i - 1].RGBn, s=s, offset=i,
+ color = color_wheel[N - i - 1]
+ if hasattr(color, 'RGBn'): color = color.RGBn
+ plot_spearman_1d(rho, color=color, s=s, offset=i,
  fig=fig, ax=ax, style=False, sort=False, top=None)
  # Plot central line
  yranges = [(s/2 + s*i - 1., 1.) for i in range(len(rhos[0]))]

biosteam/units/stirred_tank_reactor.py

@@ -65,7 +65,8 @@ class StirredTankReactor(PressureVessel, Unit, isabstract=True):
  Defaults to `continuous`.
  tau_0 : 
  Cleaning and unloading time (if batch mode). Defaults to 3 hr.
- 
+ N_reactors :
+ Number of reactors.
  Notes
  -----
  The recirculation loop takes into account the required flow rate needed to
@@ -274,7 +275,6 @@ def _design(self):
  P_pascal = (self.P if self.P else self.outs[0].P)
  P_psi = P_pascal * 0.000145038 # Pa to psi
  length_to_diameter = self.length_to_diameter
-
  if self.batch:
  v_0 = ins_F_vol
  tau = self.tau
@@ -297,7 +297,7 @@ def _design(self):
  else:
  V_reactor = V_total / N
  Design['Reactor volume'] = V_reactor
-  
+ self.N_reactors = N
  D = cylinder_diameter_from_volume(V_reactor, self.length_to_diameter)
  D *= 3.28084 # Convert from m to ft
  L = D * length_to_diameter