def go(self): pi = self.progress.indicator nstates = len(self.state_labels) if self.interface == 'matplotlib': with pi: if 'state_pop_evolution' in self.stateprobs_file: pi.new_operation('plotting populations', nstates) for istate in range(nstates): self.plot_pop(istate) pi.progress += 1 if 'color_prob_evolution' in self.stateprobs_file: pi.new_operation('plotting ensemble populations', nstates) for istate in range(nstates): self.plot_color(istate) pi.progress += 1 else: print('population evolution not available') else: plotter = Plotter(self.stateprobs_file, 'state_pop_evolution', iteration=-1, interface='text') for istate in range(nstates): plotter.plot(istate) plotter = Plotter(self.stateprobs_file, 'color_prob_evolution', iteration=-1, interface='text') for istate in range(nstates): plotter.plot(istate)
def go(self): pi = self.progress.indicator nstates = len(self.state_labels) if self.interface == 'matplotlib': with pi: # if --evolution-mode wasn't specified, neither of these exist: if 'target_flux_evolution' in self.kinavg_file: pi.new_operation('plotting fluxes', nstates) for istate in range(nstates): self.plot_flux(istate) pi.progress += 1 # if --evolution-mode wasn't specified, we won't get this either if 'rate_evolution' in self.kinavg_file: pi.new_operation('plotting rates', nstates * nstates) for istate in range(nstates): for jstate in range(nstates): self.plot_rate(istate, jstate) pi.progress += 1 else: print('rate evolution not available') else: plotter = Plotter(self.kinavg_file, 'rate_evolution', iteration=-1, interface='text') for istate in range(nstates): for jstate in range(nstates): if istate != jstate: plotter.plot(istate, jstate) plotter = Plotter(self.kinavg_file, 'conditional_flux_evolution', iteration=-1, interface='text') for istate in range(nstates): for jstate in range(nstates): if istate != jstate: plotter.plot(istate, jstate)