pulsarchart

import%20marimo%0A%0A__generated_with%20%3D%20%220.18.4%22%0Aapp%20%3D%20marimo.App(width%3D%22medium%22)%0A%0A%0A%40app.cell%0Adef%20_()%3A%0A%20%20%20%20import%20marimo%20as%20mo%0A%20%20%20%20import%20pandas%20as%20pd%0A%20%20%20%20import%20numpy%20as%20np%0A%20%20%20%20from%20wigglystuff%20import%20PulsarChart%0A%0A%20%20%20%20%23%20Generate%20Joy%20Division-style%20pulsar%20data%0A%20%20%20%20%23%20The%20original%20is%20radio%20pulses%20from%20pulsar%20CP%201919%20(PSR%20B1919%2B21)%0A%20%20%20%20np.random.seed(42)%0A%20%20%20%20n_rows%20%3D%2050%0A%20%20%20%20n_points%20%3D%20300%0A%0A%20%20%20%20rows%20%3D%20%5B%5D%0A%20%20%20%20for%20j%20in%20range(n_rows)%3A%0A%20%20%20%20%20%20%20%20%23%20Base%20line%20with%20small%20noise%0A%20%20%20%20%20%20%20%20y%20%3D%20np.random.normal(0%2C%200.02%2C%20n_points)%0A%0A%20%20%20%20%20%20%20%20%23%20Main%20pulse%20-%20a%20sharp%20peak%20that%20varies%20in%20shape%20per%20row%0A%20%20%20%20%20%20%20%20pulse_center%20%3D%20n_points%20%2F%2F%202%20%2B%20np.random.randint(-10%2C%2010)%0A%20%20%20%20%20%20%20%20pulse_width%20%3D%20np.random.uniform(15%2C%2025)%0A%20%20%20%20%20%20%20%20pulse_height%20%3D%20np.random.uniform(0.4%2C%201.2)%0A%0A%20%20%20%20%20%20%20%20%23%20Create%20pulse%20shape%20(combination%20of%20gaussians%20for%20irregular%20peaks)%0A%20%20%20%20%20%20%20%20x%20%3D%20np.arange(n_points)%0A%20%20%20%20%20%20%20%20pulse%20%3D%20pulse_height%20*%20np.exp(-((x%20-%20pulse_center)%20**%202)%20%2F%20(2%20*%20pulse_width%20**%202))%0A%0A%20%20%20%20%20%20%20%20%23%20Add%20secondary%20peaks%20for%20more%20interesting%20shape%0A%20%20%20%20%20%20%20%20for%20_%20in%20range(np.random.randint(1%2C%204))%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20offset%20%3D%20np.random.randint(-20%2C%2020)%0A%20%20%20%20%20%20%20%20%20%20%20%20width%20%3D%20np.random.uniform(5%2C%2015)%0A%20%20%20%20%20%20%20%20%20%20%20%20height%20%3D%20np.random.uniform(0.1%2C%200.5)%20*%2010%0A%20%20%20%20%20%20%20%20%20%20%20%20pulse%20%2B%3D%20height%20*%20np.exp(-((x%20-%20pulse_center%20-%20offset)%20**%202)%20%2F%20(2%20*%20width%20**%202))%0A%0A%20%20%20%20%20%20%20%20y%20%2B%3D%20pulse%0A%20%20%20%20%20%20%20%20rows.append(y)%0A%0A%20%20%20%20df%20%3D%20pd.DataFrame(rows)%0A%0A%20%20%20%20widget%20%3D%20mo.ui.anywidget(%0A%20%20%20%20%20%20%20%20PulsarChart(%0A%20%20%20%20%20%20%20%20%20%20%20%20df%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20width%3D400%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20height%3D600%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20overlap%3D0.85%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20stroke_width%3D1.0%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20peak_scale%3D2.5%2C%20%0A%20%20%20%20%20%20%20%20%20%20%20%20x_label%3D%22Pulse%20Phase%22%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20y_label%3D%22Pulse%20Number%22%2C%20%0A%20%20%20%20%20%20%20%20)%0A%20%20%20%20)%0A%20%20%20%20return%20mo%2C%20widget%0A%0A%0A%40app.cell%0Adef%20_()%3A%0A%20%20%20%20%23%20widget%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(mo%2C%20widget)%3A%0A%20%20%20%20mo.callout(%0A%20%20%20%20%20%20%20%20f%22Selected%20Index%3A%20%7Bwidget.selected_index%7D%2C%20Row%20Length%3A%20%7Blen(widget.selected_row)%20if%20widget.selected_row%20else%200%7D%22%0A%20%20%20%20)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(widget)%3A%0A%20%20%20%20import%20matplotlib.pyplot%20as%20plt%0A%0A%20%20%20%20def%20make_plot(selected_row)%3A%0A%20%20%20%20%20%20%20%20if%20not%20selected_row%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20return%20None%0A%20%20%20%20%20%20%20%20fig%2C%20ax%20%3D%20plt.subplots(figsize%3D(5%2C%205))%0A%20%20%20%20%20%20%20%20xs%20%3D%20%5Bp%5B%22x%22%5D%20for%20p%20in%20selected_row%5D%0A%20%20%20%20%20%20%20%20ys%20%3D%20%5Bp%5B%22y%22%5D%20for%20p%20in%20selected_row%5D%0A%20%20%20%20%20%20%20%20ax.plot(xs%2C%20ys)%0A%20%20%20%20%20%20%20%20ax.set_xlabel(%22x%22)%0A%20%20%20%20%20%20%20%20ax.set_ylabel(%22y%22)%0A%20%20%20%20%20%20%20%20plt.close(fig)%0A%20%20%20%20%20%20%20%20return%20fig%0A%0A%20%20%20%20plot%20%3D%20make_plot(widget.value.get(%22selected_row%22))%0A%20%20%20%20return%20(plot%2C)%0A%0A%0A%40app.cell%0Adef%20_(mo%2C%20plot%2C%20widget)%3A%0A%20%20%20%20mo.hstack(%5B%0A%20%20%20%20%20%20%20%20widget%2C%20%0A%20%20%20%20%20%20%20%20plot%0A%20%20%20%20%5D%2C%20align%3D%22start%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_()%3A%0A%20%20%20%20return%0A%0A%0Aif%20__name__%20%3D%3D%20%22__main__%22%3A%0A%20%20%20%20app.run()%0A