.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "auto_examples/01_regular_geometry.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

    .. note::
        :class: sphx-glr-download-link-note

        :ref:`Go to the end <sphx_glr_download_auto_examples_01_regular_geometry.py>`
        to download the full example code.

.. rst-class:: sphx-glr-example-title

.. _sphx_glr_auto_examples_01_regular_geometry.py:


Regular Geometry
================

The underlying geometry (a.k.a. grid) is the foundation of a block model.
Since parq-blockmodel is designed to work with regular geometries,  this example demonstrates
the underlying regular geometry by visualising it with PyVista.

.. GENERATED FROM PYTHON SOURCE LINES 9-20

.. code-block:: Python

    import tempfile

    import pandas as pd
    from pathlib import Path

    from seaborn import axes_style

    from parq_blockmodel import ParquetBlockModel, RegularGeometry

    # sphinx_gallery_thumbnail_number = -1








.. GENERATED FROM PYTHON SOURCE LINES 21-29

Create Geometry
----------------
We create a RegularGeometry object (not a ParquetBlockModel). The geometry knows nothing about attributes,
it only knows about the underlying geometry of the block model.

RegularGeometry can be created with either:
- Old style: keyword arguments (corner, block_size, shape, axis_u/v/w)
- New style: LocalGeometry and WorldFrame components

.. GENERATED FROM PYTHON SOURCE LINES 29-43

.. code-block:: Python


    corner: tuple[float, float, float] = (0.0, 0.0, 0.0)
    block_size: tuple[float, float, float] = (1.0, 1.0, 1.0)
    shape: tuple[int, int, int] = (2, 2, 2)

    # Old-style (backward compatible - recommended for most users)
    geom: RegularGeometry = RegularGeometry(
        corner=corner,
        block_size=block_size,
        shape=shape,
    )

    geom





.. rst-class:: sphx-glr-script-out

 .. code-block:: none


    <parq_blockmodel.geometry.RegularGeometry object at 0x7f98e70afc20>



.. GENERATED FROM PYTHON SOURCE LINES 44-47

Visualise in 3D
---------------
Plot using PyVista.  The geometry is a mesh, so we can use the `plot` method to visualise it.

.. GENERATED FROM PYTHON SOURCE LINES 47-64

.. code-block:: Python


    import pyvista as pv

    isometric_view = [
        (6, 5, 3),  # camera position
        (1, 1, 1),  # focal point (center of the grid)
        (0, 0, 1),  # view up direction
    ]

    # Create a PyVista plotter
    plotter = pv.Plotter()
    edges = geom.to_pyvista().extract_all_edges()
    plotter.add_mesh(edges, color="black", line_width=1)
    plotter.show_axes()
    plotter.camera_position = isometric_view
    plotter.show(title="Regular Geometry", window_size=(800, 600))




.. image-sg:: /auto_examples/images/sphx_glr_01_regular_geometry_001.png
   :alt: 01 regular geometry
   :srcset: /auto_examples/images/sphx_glr_01_regular_geometry_001.png
   :class: sphx-glr-single-img





.. GENERATED FROM PYTHON SOURCE LINES 65-69

Visualise in 2D
---------------
In most cases a 2D visualisation is sufficient to understand the underlying geometry.
We can use the `to_pyvista` method to get a PyVista mesh and then plot it in 2D.

.. GENERATED FROM PYTHON SOURCE LINES 69-96

.. code-block:: Python


    grid = geom.to_pyvista()
    # Slice at z=0.5 to intersect the first layer of cells
    slice_2d = grid.slice(normal='z', origin=(0, 0, 0.5))
    edges_2d = slice_2d.extract_all_edges()

    points = edges_2d.points
    labels = [f"({x:.1f}, {y:.1f})" for x, y, _ in points]

    centroids_mesh = slice_2d.cell_centers()
    centroids = centroids_mesh.points
    labels_centroids_2d = [f"C({x:.1f}, {y:.1f})" for x, y, _ in centroids]

    plotter = pv.Plotter()
    plotter.add_mesh(edges_2d, color="black", show_edges=True)
    plotter.add_point_labels(points, labels, font_size=12, point_color="red", point_size=10)
    plotter.add_points(centroids, color="blue", point_size=15, render_points_as_spheres=True)
    plotter.add_point_labels(centroids,
                             labels_centroids_2d,
                             font_size=12,
                             point_color="blue",
                             point_size=0,  # Hide label points
                             always_visible=True,
                             render_points_as_spheres=False)
    plotter.show_axes()
    plotter.view_xy()
    plotter.show(title="2D Slice with Corner and Centroid Coordinates", window_size=(600, 600))



.. image-sg:: /auto_examples/images/sphx_glr_01_regular_geometry_002.png
   :alt: 01 regular geometry
   :srcset: /auto_examples/images/sphx_glr_01_regular_geometry_002.png
   :class: sphx-glr-single-img






.. rst-class:: sphx-glr-timing

   **Total running time of the script:** (0 minutes 2.008 seconds)


.. _sphx_glr_download_auto_examples_01_regular_geometry.py:

.. only:: html

  .. container:: sphx-glr-footer sphx-glr-footer-example

    .. container:: sphx-glr-download sphx-glr-download-jupyter

      :download:`Download Jupyter notebook: 01_regular_geometry.ipynb <01_regular_geometry.ipynb>`

    .. container:: sphx-glr-download sphx-glr-download-python

      :download:`Download Python source code: 01_regular_geometry.py <01_regular_geometry.py>`

    .. container:: sphx-glr-download sphx-glr-download-zip

      :download:`Download zipped: 01_regular_geometry.zip <01_regular_geometry.zip>`


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.github.io>`_