ophyd device plotter

device_plotter.py

def plot_ophyd_device(device, dont_plot={'ramp_rate'}):
    '''
    Plot all of the signals of an ophyd device.
    There is probably some inaccuracy of the timestamps.
    Simple implementation.
    '''
    history = {signal: [getattr(device, signal).get()]
                     for signal in set(device.component_names) - dont_plot}
    history['timestamp'] = [time.time()]
    print(set(device.component_names) - dont_plot)

    def update(i):
        history['timestamp'].append(time.time())
        x = history['timestamp']
        plt.cla()
        for signal in set(device.component_names) - dont_plot:
            history[signal].append(getattr(device, signal).get())
            plt.plot(x, history[signal])
        plt.xlabel('time')
        plt.ylabel('value')
        plt.title(device.name)
        plt.gcf().autofmt_xdate()
        plt.tight_layout()

    ani = FuncAnimation(plt.gcf(), update, 1000)
    plt.tight_layout()
    plt.show(block=True)