:py:mod:`gnn_tracking.utils.plotting`
=====================================

.. py:module:: gnn_tracking.utils.plotting


Module Contents
---------------

Classes
~~~~~~~

.. autoapisummary::

   gnn_tracking.utils.plotting.EventPlotter
   gnn_tracking.utils.plotting.PointCloudPlotter
   gnn_tracking.utils.plotting.GraphPlotter



Functions
~~~~~~~~~

.. autoapisummary::

   gnn_tracking.utils.plotting.plot_rz
   gnn_tracking.utils.plotting.plot_3d



.. py:class:: EventPlotter(indir: str | os.PathLike)


   .. py:method:: calc_eta(r, z)


   .. py:method:: append_coordinates(hits: pandas.DataFrame, truth: pandas.DataFrame, particles: pandas.DataFrame) -> pandas.DataFrame


   .. py:method:: get_hits(evtid=None)


   .. py:method:: plot_ep_rv_uv(evtid=None, savefig=False, filename='')



.. py:class:: PointCloudPlotter(indir: str | os.PathLike, n_sectors=64)


   .. py:method:: plot_ep_rv_uv(i: int, sector: str, axs: Sequence[matplotlib.pyplot.Axes], display=True, pixel_only=False)


   .. py:method:: plot_ep_rv_uv_all_sectors(evtid: int, savefig=False, filename='', pixel_only=False)


   .. py:method:: plot_ep_rv_uv_with_boundary(evtid: int, sector: int, di, ds, ulim_low=0, ulim_high=0.035, vlim_low=-0.004, vlim_high=0.004, savefig=False, filename='', pixel_only=False)



.. py:class:: GraphPlotter(indir: str | os.PathLike = '', n_sectors=64)


   Plotter for graph data.

   :param indir: Input directory with graphs (if loading by name)
   :param n_sectors:

   .. py:method:: configure_plt(style='seaborn-paper')


   .. py:method:: plot_ep_rz_uv(*, graph: torch_geometric.data.Data, sector: int, name: str = '', filename='')

      :param graph:
      :param sector:
      :param name: If ``graph`` is not specified, load from ``self.indir / name``.
      :param filename: If specified, save figure to file

      Returns:



   .. py:method:: plot_2d(X, y, edge_index, name='', ax=None, x1_label='', x2_label='', single_particle=False)


   .. py:method:: plot_rz(graph: torch_geometric.data.Data, name='', scale=None, savefig=False, filename='', ax=None)



.. py:function:: plot_rz(X, idxs, y, savefig=False, filename='rz.png')


.. py:function:: plot_3d(X, idxs, y)