.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "11_demos\itom\plots\demo_drawAndTransformShapes.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

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

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

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

.. _sphx_glr_11_demos_itom_plots_demo_drawAndTransformShapes.py:

Draw and transform shapes
============================

In this demo script, multi shapes are created.
These shapes are then rotated and translated by
various angles and distances and displayed in one plot figure.

The overlay image of the plot is finally set to a masked
object, where the masked area is equal to the union of
all shapes, visible at the canvas.

.. GENERATED FROM PYTHON SOURCE LINES 12-18

.. code-block:: Python


    import numpy as np
    from itom import dataObject
    from itom import shape
    from itom import plot








.. GENERATED FROM PYTHON SOURCE LINES 20-21

Define a ``dataObject`` and plot using it as a "background".

.. GENERATED FROM PYTHON SOURCE LINES 21-34

.. code-block:: Python

    dObj = dataObject.zeros([1200, 1200], "uint8")
    dObj.axisOffsets = (600, 600)
    [i, h] = plot(
        dObj,
        properties={
            "colorMap": "viridis",
            "geometricShapesLabelsVisible": True,
            "keepAspectRatio": True,
        },
    )

    base_shapes = []








.. GENERATED FROM PYTHON SOURCE LINES 35-36

This is one option to create shapes.

.. GENERATED FROM PYTHON SOURCE LINES 36-48

.. code-block:: Python

    rect = shape(shape.Rectangle, (400, 240), (470, 270))
    rect.name = "Rect"
    ellipse = shape(shape.Ellipse, (-300, -300), (-140, -200))
    ellipse.name = "Ellipse"
    square = shape(shape.Square, (220, -300), 60)
    square.name = "Square"
    circle = shape(shape.Circle, (-250, 100), 25)
    circle.name = "Circle"
    polygons = np.array([[100, 200, 250, 200, 90], [50, 60, 100, 150, 150]])
    polygon = shape(shape.Polygon, polygons)
    polygon.name = "Polygon"








.. GENERATED FROM PYTHON SOURCE LINES 49-50

This is another possibility, using static methods

.. GENERATED FROM PYTHON SOURCE LINES 50-65

.. code-block:: Python

    rect = shape.createRectangle(corner1=(400, 240), corner2=(470, 270), name="Rect")
    ellipse = shape.createEllipse(
        corner1=(-300, -300), corner2=(-140, -200), name="Ellipse"
    )
    # rectangle and ellipses can also be created with the arguments center and size:
    ellipse = shape.createEllipse(
        center=(0.5 * (-300 - 140), 0.5 * (-300 - 200)),
        size=(-140 + 300, -200 + 300),
        name="Ellipse",
    )
    square = shape.createSquare(center=(220, -300), sideLength=60, name="Square")
    circle = shape.createCircle(center=(-250, 100), radius=25, name="Circle")
    polygons = np.array([[100, 200, 250, 200, 90], [50, 60, 100, 150, 150]])
    polygon = shape.createPolygon(polygons, name="Polygon")








.. GENERATED FROM PYTHON SOURCE LINES 66-68

Push all base shapes in the tuple base_shapes.
These base shapes can not be moved, rotated or resized (and thus not be selected).

.. GENERATED FROM PYTHON SOURCE LINES 68-75

.. code-block:: Python

    base_shapes = [rect, ellipse, square, circle, polygon]

    for b in base_shapes:
        b.flags = shape.MoveLock | shape.RotateLock | shape.ResizeLock

    all_shapes = base_shapes.copy()








.. GENERATED FROM PYTHON SOURCE LINES 76-78

Rotate all base shapes by 60 degree (around its center).
These rotated elements can only be rotated.

.. GENERATED FROM PYTHON SOURCE LINES 78-87

.. code-block:: Python

    rotated_shapes = []
    for base_shape in base_shapes:
        temp = base_shape.copy()
        temp.rotateDeg(60)
        temp.flags = shape.MoveLock | shape.ResizeLock
        temp.name += ", 60\xb0"
        rotated_shapes.append(temp)
    all_shapes += rotated_shapes








.. GENERATED FROM PYTHON SOURCE LINES 88-90

Move all rotated shapes by dx = -70, dy = 100.
These objects can be rotated, resized and moved.

.. GENERATED FROM PYTHON SOURCE LINES 90-99

.. code-block:: Python

    rotated_translated_shapes = []
    for rotated_shape in rotated_shapes:
        temp = rotated_shape.copy()
        temp.translate([-70, 100])
        temp.flags = 0
        temp.name += ",\n dx:-70, dy:100"
        rotated_translated_shapes.append(temp)
    all_shapes += rotated_translated_shapes








.. GENERATED FROM PYTHON SOURCE LINES 100-101

Display all shapes on the plot figure.

.. GENERATED FROM PYTHON SOURCE LINES 101-109

.. code-block:: Python

    h["geometricShapes"] = all_shapes

    # create a mask object (uint8, masked pixels are set to 255, the rest is 0) from all shapes
    mask = dObj.createMask(all_shapes)

    # display the mask as overlay of the plot
    h["overlayImage"] = mask








.. GENERATED FROM PYTHON SOURCE LINES 110-112

.. image:: ../../_static/demoDrawAndTransformShapes_1.png
   :width: 100%


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

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


.. _sphx_glr_download_11_demos_itom_plots_demo_drawAndTransformShapes.py:

.. only:: html

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

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

      :download:`Download Jupyter notebook: demo_drawAndTransformShapes.ipynb <demo_drawAndTransformShapes.ipynb>`

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

      :download:`Download Python source code: demo_drawAndTransformShapes.py <demo_drawAndTransformShapes.py>`

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

      :download:`Download zipped: demo_drawAndTransformShapes.zip <demo_drawAndTransformShapes.zip>`