PySDM-examples 2.119__tar.gz → 2.120__tar.gz

{pysdm_examples-2.119 → pysdm_examples-2.120}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pysdm-examples
-Version: 2.119
+Version: 2.120
 Summary: PySDM usage examples reproducing results from literature and depicting how to use PySDM from Python Jupyter notebooks
 Home-page: https://github.com/open-atmos/PySDM
 Author: https://github.com/open-atmos/PySDM/graphs/contributors
@@ -77,7 +77,7 @@ The 2D prescribed-flow framework used here can be traced back to the work of
     implementation of the <a href="https://doi.org/10.1002/fld.1071">MPDATA numerical scheme of Smolarkiewicz (e.g., 2006)</a>.
   An animation depicting PySDM simulation capturing <mark>aerosol collisional processing</mark> by warm rain is shown below:
 
-![animation](https://github.com/open-atmos/PySDM/wiki/files/kinematic_2D_example.gif)
+<source src="https://github.com/open-atmos/PySDM/releases/download/tip/docs_intro_animation_ubuntu-24.04.mp4" type="video/mp4">
 
 Example notebooks:
 - <a href="https://open-atmos.github.io/PySDM/PySDM_examples/Arabas_et_al_2015.html">Arabas_et_al_2015</a>

pysdm_examples-2.120/PySDM_examples/utils/pvanim.py

@@ -0,0 +1,304 @@
+#!/usr/bin/env pvpython
+import argparse
+from collections import namedtuple
+import pathlib
+
+from paraview import simple as pvs  # pylint: disable=import-error
+
+pvs._DisableFirstRenderCameraReset()
+
+
+def cli_using_argparse(argp):
+    argp.add_argument("product_path", help="path to pvd products file")
+    argp.add_argument("attributes_path", help=" path to pvd attributes file")
+    argp.add_argument("output_path", help="path where to write output files")
+    argp.add_argument(
+        "--multiplicity_preset",
+        default="Inferno (matplotlib)",
+        help="Preset for multiplicity",
+    )
+    argp.add_argument(
+        "--multiplicity_logscale",
+        action="store_false",
+        help="Use log scale for multiplicity",
+    )
+    argp.add_argument(
+        "--effectiveradius_preset",
+        default="Black, Blue and White",
+        help="Preset for effectiveradius",
+    )
+    argp.add_argument(
+        "--effectiveradius_logscale",
+        action="store_false",
+        help="Use log scale for effectiveradius",
+    )
+    argp.add_argument(
+        "--effectiveradius_nan_color",
+        nargs=3,
+        type=float,
+        default=[0.666, 0.333, 1.0],
+        help="Nan color in RGB format for effectiveradius",
+    )
+    argp.add_argument(
+        "--sd_products_opacity", type=float, default=0.9, help="Opacity for sd_products"
+    )
+    argp.add_argument(
+        "--calculator1_opacity",
+        type=float,
+        default=0.19,
+        help="Opacity for calculator1",
+    )
+    argp.add_argument(
+        "--sd_attributes_opacity",
+        type=float,
+        default=0.77,
+        help="Opacity for sd_attributes",
+    )
+    argp.add_argument(
+        "--animationframename",
+        type=str,
+        help="Name of the file with animation last frame",
+    )
+    argp.add_argument(
+        "--animationname",
+        type=str,
+        help="Name of the file with animation",
+    )
+    argp.add_argument(
+        "--framerate", type=int, help="Number of frame rates.", default=15
+    )
+
+
+ap = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
+cli_using_argparse(ap)
+
+args = ap.parse_args()
+sd_productspvd = pvs.OpenDataFile(args.product_path)
+sd_attributespvd = pvs.OpenDataFile(args.attributes_path)
+
+
+setup = {
+    "renderView1": pvs.GetActiveViewOrCreate("RenderView"),
+    "sd_attributespvdDisplay": pvs.GetDisplayProperties(
+        sd_attributespvd, view=pvs.GetActiveViewOrCreate("RenderView")
+    ),
+    "effectiveradiusLUT": pvs.GetColorTransferFunction("effectiveradius"),
+    "sd_productspvdDisplay": pvs.GetDisplayProperties(
+        sd_productspvd, view=pvs.GetActiveViewOrCreate("RenderView")
+    ),
+}
+
+setup = namedtuple("Setup", setup.keys())(**setup)
+
+setup.effectiveradiusLUT.RescaleTransferFunction(0.1380997175392798, 207.7063518856934)
+
+setup.renderView1.Update()
+
+
+def create_new_calculator(
+    calcinput,
+    representation,
+    function,
+    color_by1,
+    color_by2,
+    color_by3,
+    scalar_coloring=False,
+    hide=False,
+    *,
+    y,
+    registrationame,
+):
+    calculator = pvs.Calculator(registrationName=registrationame, Input=calcinput)
+    display = pvs.Show(calculator, y.renderView1, representation)
+    calculator.Function = function
+    y.renderView1.Update()
+    if scalar_coloring is True:
+        pvs.ColorBy(display, (color_by1, color_by2, color_by3))
+    if hide is True:
+        pvs.Hide(calculator, y.renderView1)
+    return y.renderView1.Update()
+
+
+def scalar_bar(name, *, y, erLUT):
+    calculator1Display = pvs.Show(
+        calculator1, y.renderView1, "UnstructuredGridRepresentation"
+    )
+    calculator1Display.SetScalarBarVisibility(y.renderView1, True)
+    scalarBar = pvs.GetScalarBar(erLUT.effectiveradiusLUT, y.renderView1)
+    scalarBar.ComponentTitle = ""
+    scalarBar.Title = name
+    y.renderView1.Update()
+
+
+def create_glyph(
+    registration_name, put, scale_array1, scale_array2, color_by=False, *, y
+):
+    glyph = pvs.Glyph(registrationName=registration_name, Input=put, GlyphType="Arrow")
+    glyphDisplay = pvs.Show(glyph, y.renderView1, "GeometryRepresentation")
+    glyphDisplay.Representation = "Surface"
+    glyph.ScaleArray = [scale_array1, scale_array2]
+    glyph.ScaleFactor = 100
+    glyphDisplay.SetScalarBarVisibility(y.renderView1, True)
+    if color_by is True:
+        glyphDisplay.ColorArrayName = ["POINTS", ""]
+        pvs.ColorBy(glyphDisplay, None)
+    y.renderView1.Update()
+
+
+def apply_presets_logscale_opacity_and_update(*, y, attdisplay, erLUT, proddisplay):
+    multiplicityLUT = pvs.GetColorTransferFunction("multiplicity")
+    multiplicityLUT.RescaleTransferFunction(19951.0, 50461190157.0)
+    calculator1Display = pvs.Show(
+        calculator1, y.renderView1, "UnstructuredGridRepresentation"
+    )
+    multiplicityLUT.ApplyPreset(args.multiplicity_preset, True)
+    if args.multiplicity_logscale:
+        multiplicityLUT.MapControlPointsToLogSpace()
+        multiplicityLUT.UseLogScale = 1
+    else:
+        multiplicityLUT.MapControlPointsToLinearSpace()
+        multiplicityLUT.UseLogScale = 0
+
+    erLUT.effectiveradiusLUT.ApplyPreset(args.effectiveradius_preset, True)
+    if args.effectiveradius_logscale:
+        erLUT.effectiveradiusLUT.MapControlPointsToLogSpace()
+        erLUT.effectiveradiusLUT.UseLogScale = 1
+    else:
+        erLUT.effectiveradiusLUT.MapControlPointsToLinearSpace()
+        erLUT.effectiveradiusLUT.UseLogScale = 0
+
+    erLUT.effectiveradiusLUT.NanColor = args.effectiveradius_nan_color
+
+    proddisplay.sd_productspvdDisplay.SetRepresentationType("Surface With Edges")
+    proddisplay.sd_productspvdDisplay.Opacity = args.sd_products_opacity
+    calculator1Display.Opacity = args.calculator1_opacity
+    attdisplay.sd_attributespvdDisplay.Opacity = args.sd_attributes_opacity
+
+    y.renderView1.Update()
+
+
+def get_layout(*, y):
+    layout1 = pvs.GetLayout()
+    layout1.SetSize(1205, 739)
+    y.renderView1.Update()
+
+
+def set_current_camera_placement(*, y):
+    y.renderView1.InteractionMode = "2D"
+    y.renderView1.CameraPosition = [
+        836.5045867211775,
+        677.8909274570431,
+        -4098.0762113533165,
+    ]
+    y.renderView1.CameraFocalPoint = [836.5045867211775, 677.8909274570431, 0.0]
+    y.renderView1.CameraViewUp = [1.0, 0.0, 0.0]
+    y.renderView1.CameraParallelScale = 1060.6601717798205
+    y.renderView1.Update()
+
+
+def axes_settings(*, view):
+    view.ViewSize = [2000, 800]
+    view.Background = [1, 1, 1]
+    view.CenterAxesVisibility = True
+    view.OrientationAxesVisibility = True
+    axesGrid = view.AxesGrid
+    axesGrid.Visibility = True
+    axesGrid.XTitle = "Z [m]"
+    axesGrid.YTitle = "X [m]"
+
+    axesGrid.XAxisUseCustomLabels = True
+    axesGrid.XAxisLabels = [300, 600, 900, 1200]
+    axesGrid.YAxisUseCustomLabels = True
+    axesGrid.YAxisLabels = [300, 600, 900, 1200]
+
+    axesGrid.XTitleFontSize = 30
+    axesGrid.XLabelFontSize = 30
+    axesGrid.YTitleFontSize = 30
+    axesGrid.YLabelFontSize = 30
+
+    axesGrid.XTitleColor = [1.0, 1.0, 1.0]
+    axesGrid.XLabelColor = [1.0, 1.0, 1.0]
+    axesGrid.YTitleColor = [1.0, 1.0, 1.0]
+    axesGrid.YLabelColor = [1.0, 1.0, 1.0]
+    axesGrid.GridColor = [0.1, 0.1, 0.1]
+    view.CenterAxesVisibility = False
+    view.Update()
+
+
+def time_annotation(*, y):
+    time = pvs.AnnotateTimeFilter(
+        guiName="AnnotateTimeFilter1", Format="Time:{time:f}s"
+    )
+    pvs.Show(time, y.renderView1)
+    y.renderView1.Update()
+
+
+def text(text_in, position_y, *, view):
+    sentence = pvs.Text()
+    sentence.Text = text_in
+    textDisplay = pvs.Show(sentence, view)
+    textDisplay.Color = [1.0, 1.0, 1.0]
+    textDisplay.WindowLocation = "Any Location"
+    textDisplay.FontSize = 16
+    textDisplay.Position = [0.01, position_y]
+
+
+def last_anim_frame(animation_frame_name):
+    time_steps = sd_productspvd.TimestepValues
+    last_time = time_steps[len(time_steps) - 1]
+    setup.renderView1.ViewTime = last_time
+    for reader in (sd_productspvd,):
+        reader.UpdatePipeline(last_time)
+        pvs.ExportView(
+            filename=str(pathlib.Path(args.output_path) / animation_frame_name),
+            view=setup.renderView1,
+            Rasterize3Dgeometry=False,
+            GL2PSdepthsortmethod="BSP sorting (slow, best)",
+        )
+    pvs.RenderAllViews()
+
+
+calculator1 = create_new_calculator(
+    sd_attributespvd,
+    "UnstructuredGridRepresentation",
+    '"relative fall velocity"*(-iHat)',
+    "None",
+    "None",
+    "None",
+    y=setup,
+    registrationame="Calculator1",
+)
+scalar_bar("effective radius [um]", y=setup, erLUT=setup)
+create_glyph("Glyph1", calculator1, "POINTS", "relative fall velocity", y=setup)
+calculator2 = create_new_calculator(
+    sd_productspvd,
+    "StructuredGridRepresentation",
+    "cx*jHat+cy*iHat",
+    "CELLS",
+    "Result",
+    "Magnitude",
+    True,
+    True,
+    y=setup,
+    registrationame="Calculator2",
+)
+apply_presets_logscale_opacity_and_update(
+    y=setup, attdisplay=setup, erLUT=setup, proddisplay=setup
+)
+create_glyph("Glyph2", calculator2, "CELLS", "Result", True, y=setup)
+get_layout(y=setup)
+set_current_camera_placement(y=setup)
+axes_settings(view=setup.renderView1)
+time_annotation(y=setup)
+text("Arrows scale with Courant number C=u*Δt/Δx,", 0.7, view=setup.renderView1)
+text("reflecting the grid spacing Δx and Δy.", 0.65, view=setup.renderView1)
+last_anim_frame(animation_frame_name=args.animationframename)
+scene = pvs.GetAnimationScene()
+scene.UpdateAnimationUsingDataTimeSteps()
+pvs.Render(setup.renderView1)
+pvs.SaveAnimation(
+    str(pathlib.Path(args.output_path) / args.animationname),
+    setup.renderView1,
+    FrameRate=args.framerate,
+)
+pvs.RenderAllViews()

{pysdm_examples-2.119 → pysdm_examples-2.120}/docs/pysdm_examples_landing.md

@@ -41,7 +41,7 @@ The 2D prescribed-flow framework used here can be traced back to the work of
     implementation of the <a href="https://doi.org/10.1002/fld.1071">MPDATA numerical scheme of Smolarkiewicz (e.g., 2006)</a>.
   An animation depicting PySDM simulation capturing <mark>aerosol collisional processing</mark> by warm rain is shown below:
 
-![animation](https://github.com/open-atmos/PySDM/wiki/files/kinematic_2D_example.gif)
+<source src="https://github.com/open-atmos/PySDM/releases/download/tip/docs_intro_animation_ubuntu-24.04.mp4" type="video/mp4">
 
 Example notebooks:
 - `PySDM_examples.Arabas_et_al_2015`

{pysdm_examples-2.119 → pysdm_examples-2.120}/pyproject.toml

@@ -4,4 +4,4 @@ local_scheme = "no-local-version"
 version_scheme = "post-release"
 
 [build-system]
-requires = ['setuptools==80.4.0', 'setuptools-scm==8.3.1']
+requires = ['setuptools==80.7.1', 'setuptools-scm==8.3.1']

{pysdm_examples-2.119 → pysdm_examples-2.120}/pysdm_examples.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pysdm-examples
-Version: 2.119
+Version: 2.120
 Summary: PySDM usage examples reproducing results from literature and depicting how to use PySDM from Python Jupyter notebooks
 Home-page: https://github.com/open-atmos/PySDM
 Author: https://github.com/open-atmos/PySDM/graphs/contributors
@@ -77,7 +77,7 @@ The 2D prescribed-flow framework used here can be traced back to the work of
     implementation of the <a href="https://doi.org/10.1002/fld.1071">MPDATA numerical scheme of Smolarkiewicz (e.g., 2006)</a>.
   An animation depicting PySDM simulation capturing <mark>aerosol collisional processing</mark> by warm rain is shown below:
 
-![animation](https://github.com/open-atmos/PySDM/wiki/files/kinematic_2D_example.gif)
+<source src="https://github.com/open-atmos/PySDM/releases/download/tip/docs_intro_animation_ubuntu-24.04.mp4" type="video/mp4">
 
 Example notebooks:
 - <a href="https://open-atmos.github.io/PySDM/PySDM_examples/Arabas_et_al_2015.html">Arabas_et_al_2015</a>

{pysdm_examples-2.119 → pysdm_examples-2.120}/pysdm_examples.egg-info/SOURCES.txt

@@ -156,6 +156,7 @@ PySDM_examples/utils/__init__.py
 PySDM_examples/utils/basic_simulation.py
 PySDM_examples/utils/dummy_controller.py
 PySDM_examples/utils/progbar_controller.py
+PySDM_examples/utils/pvanim.py
 PySDM_examples/utils/read_vtk_1d.py
 PySDM_examples/utils/widgets/__init__.py
 PySDM_examples/utils/widgets/freezer.py