Source code for data_morph.shapes.lines.horizontal_lines

"""Horizontal lines shape."""

import numpy as np

from ...data.dataset import Dataset
from ..bases.line_collection import LineCollection




class HorizontalLines(LineCollection):
    """
    Class for the horizontal lines shape.

    .. plot::
       :scale: 75
       :caption:
            This shape is generated using the panda dataset.

        from data_morph.data.loader import DataLoader
        from data_morph.shapes.lines import HorizontalLines

        _ = HorizontalLines(DataLoader.load_dataset('panda')).plot()

    Parameters
    ----------
    dataset : Dataset
        The starting dataset to morph into other shapes.
    """

    name = 'h_lines'

    def __init__(self, dataset: Dataset) -> None:
        x_bounds = dataset.data_bounds.x_bounds
        y_bounds = dataset.data_bounds.y_bounds

        super().__init__(
            *[
                [[x_bounds[0], y], [x_bounds[1], y]]
                for y in np.linspace(y_bounds[0], y_bounds[1], 5)
            ]
        )