Skip to content

Base Gradient Reference

BaseGradient

Bases: JupyterMixin

Base class for rendering color gradients in the terminal using Rich.

This class applies a smoothly interpolated gradient of foreground and/or background colors across supplied renderable content.

Attributes:

Name Type Description
console Console

Console instance used for rendering.

Source code in src/rich_gradient/_base_gradient.py
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
class BaseGradient(JupyterMixin):
    """
    Base class for rendering color gradients in the terminal using Rich.

    This class applies a smoothly interpolated gradient of foreground and/or
    background colors across supplied renderable content.

    Attributes:
        console: Console instance used for rendering.
    """

    def __init__(
        self,
        renderables: ConsoleRenderable | List[ConsoleRenderable],
        colors: Optional[List[ColorType]] = None,
        bg_colors: Optional[List[ColorType]] = None,
        console: Optional[Console] = None,
        hues: int = 5,
        rainbow: bool = False,
        expand: bool = False,
        justify: AlignMethod = "left",
        vertical_justify: VerticalAlignMethod = "top",
        show_quit_panel: bool = False,
        repeat_scale: float = 2.0,
        background: bool = False,
    ) -> None:
        """
        Initialize a BaseGradient instance.

        Args:
            renderables: A single renderable or list of renderable objects to
                which the gradient will be applied.
            colors: Optional list of colors (strings, Color, or
                ColorTriplet) for the gradient foreground. If omitted and
                rainbow is False, a spectrum of `hues` colors is used.
            bg_colors: Optional list of colors for the gradient
                background. If omitted, no background gradient is applied.
            console: Optional Rich Console to render to. Defaults to
                `rich.get_console()`.
            hues: Number of hues to generate if no explicit colors are given.
            rainbow: If True, ignore `colors` and use a full rainbow.
            expand: Whether to expand renderables to the full console width.
            justify: Horizontal alignment: 'left', 'center', or 'right'.
            vertical_justify: Vertical alignment: 'top', 'center', or 'bottom'.
            show_quit_panel: If True, displays a quit instruction panel.
            repeat_scale: Scale factor controlling gradient repeat span.
        """
        self.console: Console = console or get_console()
        self.hues: int = max(hues, 2)
        self.rainbow: bool = rainbow
        self.repeat_scale: float = repeat_scale
        self.phase: float = 0.0
        self.expand: bool = expand
        self.justify = justify  # setter will validate
        self.vertical_justify = vertical_justify  # setter will validate
        self.show_quit_panel = show_quit_panel  # setter via property
        self.background: bool = bool(background)
        if renderables is None:
            renderables = []
        self.renderables = renderables
        self.colors = colors or []
        self.bg_colors = bg_colors or []
        self._active_stops = self._initialize_color_stops()

    @property
    def renderables(self) -> List[ConsoleRenderable]:
        """List of renderable objects to which the gradient is applied."""
        return self._renderables

    @renderables.setter
    def renderables(self, value: ConsoleRenderable | List[ConsoleRenderable]) -> None:
        """Set and normalize the list of renderables."""
        render_list = value if isinstance(value, list) else [value]
        normalized: List[ConsoleRenderable] = []
        for item in render_list:
            if isinstance(item, str):
                normalized.append(RichText.from_markup(item))
            else:
                normalized.append(item)
        self._renderables = normalized

    @property
    def colors(self) -> List[ColorTriplet]:
        """List of parsed ColorTriplet objects for gradient foreground."""
        return self._foreground_colors

    @colors.setter
    def colors(self, colors: List[ColorType]) -> None:
        """
        Parse and set the foreground color stops.

        Args:
            colors: List of color strings, Color, or ColorTriplet.
        """
        if self.rainbow:
            triplets = Spectrum().triplets
        elif not colors:
            triplets = Spectrum(self.hues).triplets
        else:
            triplets = self._to_color_triplets(colors)

        # Loop smoothly by appending reversed middle stops
        if len(triplets) > 2:
            # Append reversed stops excluding final stop so gradient wraps smoothly
            triplets += list(reversed(triplets[:-1]))
        self._foreground_colors = triplets

    @property
    def bg_colors(self) -> List[ColorTriplet]:
        """List of parsed ColorTriplet objects for gradient background."""
        return self._background_colors

    @bg_colors.setter
    def bg_colors(self, colors: Optional[List[ColorType]]) -> None:
        """
        Parse and set the background color stops.

        Args:
            colors: Optional list of color strings, Color, or ColorTriplet.
        """
        if not colors:
            self._background_colors = []
            return

        if len(colors) == 1:
            triplet = Color.parse(colors[0]).get_truecolor()
            # repeat single color across hues
            self._background_colors = [triplet] * self.hues
        else:
            triplets = self._to_color_triplets(colors)
            self._background_colors = triplets

    @property
    def justify(self) -> AlignMethod:
        """Horizontal alignment method."""
        return self._justify  # type: ignore

    @justify.setter
    def justify(self, method: AlignMethod) -> None:
        """
        Validate and set horizontal alignment.

        Args:
            method: 'left', 'center', or 'right'.

        Raises:
            ValueError: If method is invalid.
        """
        if isinstance(method, str) and method.lower() in {"left", "center", "right"}:
            self._justify = method.lower()  # type: ignore
        else:
            raise ValueError(f"Invalid justify method: {method}")

    @property
    def vertical_justify(self) -> VerticalAlignMethod:
        """Vertical alignment method."""
        return self._vertical_justify  # type: ignore

    @vertical_justify.setter
    def vertical_justify(self, method: VerticalAlignMethod) -> None:
        """
        Validate and set vertical alignment.

        Args:
            method: 'top', 'center', or 'bottom'.

        Raises:
            ValueError: If method is invalid.
        """
        if isinstance(method, str) and method.lower() in {"top", "center", "bottom"}:
            self._vertical_justify = method.lower()  # type: ignore
        else:
            raise ValueError(f"Invalid vertical justify method: {method}")

    @property
    def show_quit_panel(self) -> bool:
        """Whether to display the quit instructions panel."""
        return self._show_quit_panel  # type: ignore

    @show_quit_panel.setter
    def show_quit_panel(self, value: bool) -> None:
        """
        Set whether to display the quit instructions panel.

        Args:
            show: True to display, False to hide.
        """
        self._show_quit_panel = bool(value)

    @staticmethod
    def _to_color_triplets(colors: List[ColorType]) -> List[ColorTriplet]:
        """
        Convert a list of color specifications to ColorTriplet instances.

        Args:
            colors: List of color strings, Color, or ColorTriplet.

        Returns:
            List of ColorTriplet.

        Raises:
            TypeError: If unsupported color type encountered.
            ColorParseError: If a color string fails to parse.
        """
        triplets: List[ColorTriplet] = []
        for c in colors:
            if isinstance(c, ColorTriplet):
                triplets.append(c)
            elif isinstance(c, Color):
                triplets.append(c.get_truecolor())
            elif isinstance(c, str):
                triplets.append(Color.parse(c).get_truecolor())
            else:
                raise ColorParseError(
                    f"Unsupported color type: {type(c)}\n\tCould not parse color: {c}"
                )
        return triplets

    def __rich_measure__(
        self, console: Console, options: ConsoleOptions
    ) -> Measurement:
        """
        Measure the minimum and maximum width for the gradient content.

        Args:
            console: Console for measurement.
            options: Rendering options.

        Returns:
            Measurement: Combined width constraints.
        """
        measurements = [Measurement.get(console, options, r) for r in self.renderables]
        if not measurements:
            # No renderables — return a reasonable default measurement.
            # Min width is 0; max width is the available maximum from options.
            return Measurement(0, options.max_width or 0)

        min_width = min(m.minimum for m in measurements)
        max_width = max(m.maximum for m in measurements)
        return Measurement(min_width, max_width)

    def __rich_console__(
        self, console: Console, options: ConsoleOptions
    ) -> RenderResult:
        """
        Render the gradient by applying interpolated colors to each segment.

        Args:
            console: Console to render to.
            options: Rendering options.

        Yields:
            Segment: Colored text segments for gradient effect.
        """
        width = options.max_width
        content = Group(*self.renderables)
        if self.show_quit_panel:
            # Use a Rich Text renderable so the bracketed markup tags remain literal in the output
            panel = Panel(RichText("Press [bold]Ctrl+C[/bold] to stop."), expand=False)
            content = Group(content, Align(panel, align="right"))

        lines = console.render_lines(content, options, pad=True, new_lines=False)
        for line_idx, segments in enumerate(lines):
            col = 0
            for seg in segments:
                text = seg.text
                base_style = seg.style or Style()
                cluster = ""
                cluster_width = 0
                for ch in text:
                    w = get_character_cell_size(ch)
                    if w <= 0:
                        cluster += ch
                        continue
                    if cluster:
                        style = self._get_style_at_position(
                            col - cluster_width, cluster_width, width
                        )
                        yield Segment(cluster, self._merge_styles(base_style, style))
                        cluster = ""
                        cluster_width = 0
                    cluster = ch
                    cluster_width = w
                    col += w
                if cluster:
                    style = self._get_style_at_position(
                        col - cluster_width, cluster_width, width
                    )
                    yield Segment(cluster, self._merge_styles(base_style, style))
            if line_idx < len(lines) - 1:
                yield Segment.line()

    def _get_style_at_position(self, position: int, width: int, span: int) -> Style:
        """
        Compute the Rich Style for a character cluster at a given position.

        Args:
            position: Starting cell index of the cluster.
            width: Cell width of the cluster.
            span: Total available width for gradient calculation.

        Returns:
            Style with appropriate foreground and/or background colors.
        """
        frac = self._compute_fraction(position, width, span)
        # If background mode is enabled, apply gradient to background only.
        if self.background:
            active = self.bg_colors if self.bg_colors else self.colors
            if not active:
                return Style()
            r, g, b = self._interpolate_color(frac, active)
            bg_style = f"#{int(r):02x}{int(g):02x}{int(b):02x}"
            return Style(bgcolor=bg_style)

        # Default: apply gradient to foreground; background uses bg_colors if provided.
        fg_style = ""
        bg_style = ""
        if self.colors:
            r, g, b = self._interpolate_color(frac, self.colors)
            fg_style = f"#{int(r):02x}{int(g):02x}{int(b):02x}"
        if self.bg_colors:
            r, g, b = self._interpolate_color(frac, self.bg_colors)
            bg_style = f"#{int(r):02x}{int(g):02x}{int(b):02x}"

        return Style(color=fg_style or None, bgcolor=bg_style or None)

    def _compute_fraction(self, position: int, width: int, span: float) -> float:
        """
        Compute fractional position for gradient interpolation, including phase.

        Args:
            position: Starting cell index.
            width: Cell width.
            span: Total span for gradient.

        Returns:
            Fraction between 0.0 and 1.0.
        """
        total_width = (span or 0) * (self.repeat_scale or 1.0)
        if total_width <= 0:
            # Avoid division by zero; return phase-only fraction.
            return self.phase % 1.0

        base = (position + width / 2) / total_width
        return (base + self.phase) % 1.0

    def _interpolate_color(
        self, frac: float, color_stops: list[ColorTriplet]
    ) -> tuple[float, float, float]:
        """
        Interpolate color in linear light space with gamma correction.

        Args:
            frac: Fractional position between 0.0 and 1.0.
            color_stops: List of ColorTriplet stops.

        Returns:
            Tuple of (r, g, b) in sRGB space.
        """
        if frac <= 0:
            return color_stops[0]
        if frac >= 1:
            return color_stops[-1]

        # Determine segment and local position
        segment_count = len(color_stops) - 1
        pos = frac * segment_count
        idx = int(pos)
        t = pos - idx

        r0, g0, b0 = color_stops[idx]
        r1, g1, b1 = color_stops[min(idx + 1, segment_count)]

        def to_linear(c: float) -> float:
            return (c / 255.0) ** _GAMMA_CORRECTION

        def to_srgb(x: float) -> float:
            return (x ** (1.0 / _GAMMA_CORRECTION)) * 255.0

        lr0, lg0, lb0 = to_linear(r0), to_linear(g0), to_linear(b0)
        lr1, lg1, lb1 = to_linear(r1), to_linear(g1), to_linear(b1)

        lr = lr0 + (lr1 - lr0) * t
        lg = lg0 + (lg1 - lg0) * t
        lb = lb0 + (lb1 - lb0) * t

        return to_srgb(lr), to_srgb(lg), to_srgb(lb)

    @staticmethod
    def _merge_styles(original: Style, gradient_style: Style) -> Style:
        """
        Merge original Style with gradient Style, preserving original attributes.

        Args:
            original: The existing Rich Style.
            gradient_style: Style with gradient colors.

        Returns:
            Combined Style.
        """
        return original + gradient_style if original else gradient_style

    # -----------------
    # Test helper parity
    # -----------------
    def _initialize_color_stops(self) -> List[ColorTriplet]:
        """Initialize the active color stops based on mode and provided stops.

        If only one stop is provided, duplicate it to create a smooth segment pair.
        """
        source = self.bg_colors if self.background else self.colors
        if not source:
            return []
        return [source[0], source[0]] if len(source) == 1 else source

    def _color_at(self, pos: int, width: int, span: int) -> str:
        """Return the hex color at a given position (for tests)."""
        stops = self._active_stops
        if not stops:
            return "#000000"
        frac = self._compute_fraction(pos, width, span)
        r, g, b = self._interpolate_color(frac, stops)
        return f"#{int(r):02x}{int(g):02x}{int(b):02x}"

    def _styled(self, original: Style, color: str) -> Style:
        """Return a Style with the given color or bgcolor, preserving original (for tests)."""
        return (
            original + Style(bgcolor=color)
            if self.background
            else original + Style(color=color)
        )

    def _interpolated_color(self, frac: float, stops: list, n: int):
        """Return the interpolated color at a fraction (for tests)."""
        return self._interpolate_color(frac, stops)

bg_colors property writable

List of parsed ColorTriplet objects for gradient background.

colors property writable

List of parsed ColorTriplet objects for gradient foreground.

justify property writable

Horizontal alignment method.

renderables property writable

List of renderable objects to which the gradient is applied.

show_quit_panel property writable

Whether to display the quit instructions panel.

vertical_justify property writable

Vertical alignment method.

__init__(renderables, colors=None, bg_colors=None, console=None, hues=5, rainbow=False, expand=False, justify='left', vertical_justify='top', show_quit_panel=False, repeat_scale=2.0, background=False)

Initialize a BaseGradient instance.

Parameters:

Name Type Description Default
renderables ConsoleRenderable | List[ConsoleRenderable]

A single renderable or list of renderable objects to which the gradient will be applied.

required
colors Optional[List[ColorType]]

Optional list of colors (strings, Color, or ColorTriplet) for the gradient foreground. If omitted and rainbow is False, a spectrum of hues colors is used.

None
bg_colors Optional[List[ColorType]]

Optional list of colors for the gradient background. If omitted, no background gradient is applied.

None
console Optional[Console]

Optional Rich Console to render to. Defaults to rich.get_console().

None
hues int

Number of hues to generate if no explicit colors are given.

5
rainbow bool

If True, ignore colors and use a full rainbow.

False
expand bool

Whether to expand renderables to the full console width.

False
justify AlignMethod

Horizontal alignment: 'left', 'center', or 'right'.

'left'
vertical_justify VerticalAlignMethod

Vertical alignment: 'top', 'center', or 'bottom'.

'top'
show_quit_panel bool

If True, displays a quit instruction panel.

False
repeat_scale float

Scale factor controlling gradient repeat span.

2.0
Source code in src/rich_gradient/_base_gradient.py
def __init__(
    self,
    renderables: ConsoleRenderable | List[ConsoleRenderable],
    colors: Optional[List[ColorType]] = None,
    bg_colors: Optional[List[ColorType]] = None,
    console: Optional[Console] = None,
    hues: int = 5,
    rainbow: bool = False,
    expand: bool = False,
    justify: AlignMethod = "left",
    vertical_justify: VerticalAlignMethod = "top",
    show_quit_panel: bool = False,
    repeat_scale: float = 2.0,
    background: bool = False,
) -> None:
    """
    Initialize a BaseGradient instance.

    Args:
        renderables: A single renderable or list of renderable objects to
            which the gradient will be applied.
        colors: Optional list of colors (strings, Color, or
            ColorTriplet) for the gradient foreground. If omitted and
            rainbow is False, a spectrum of `hues` colors is used.
        bg_colors: Optional list of colors for the gradient
            background. If omitted, no background gradient is applied.
        console: Optional Rich Console to render to. Defaults to
            `rich.get_console()`.
        hues: Number of hues to generate if no explicit colors are given.
        rainbow: If True, ignore `colors` and use a full rainbow.
        expand: Whether to expand renderables to the full console width.
        justify: Horizontal alignment: 'left', 'center', or 'right'.
        vertical_justify: Vertical alignment: 'top', 'center', or 'bottom'.
        show_quit_panel: If True, displays a quit instruction panel.
        repeat_scale: Scale factor controlling gradient repeat span.
    """
    self.console: Console = console or get_console()
    self.hues: int = max(hues, 2)
    self.rainbow: bool = rainbow
    self.repeat_scale: float = repeat_scale
    self.phase: float = 0.0
    self.expand: bool = expand
    self.justify = justify  # setter will validate
    self.vertical_justify = vertical_justify  # setter will validate
    self.show_quit_panel = show_quit_panel  # setter via property
    self.background: bool = bool(background)
    if renderables is None:
        renderables = []
    self.renderables = renderables
    self.colors = colors or []
    self.bg_colors = bg_colors or []
    self._active_stops = self._initialize_color_stops()

__rich_console__(console, options)

Render the gradient by applying interpolated colors to each segment.

Parameters:

Name Type Description Default
console Console

Console to render to.

required
options ConsoleOptions

Rendering options.

required

Yields:

Name Type Description
Segment RenderResult

Colored text segments for gradient effect.

Source code in src/rich_gradient/_base_gradient.py
def __rich_console__(
    self, console: Console, options: ConsoleOptions
) -> RenderResult:
    """
    Render the gradient by applying interpolated colors to each segment.

    Args:
        console: Console to render to.
        options: Rendering options.

    Yields:
        Segment: Colored text segments for gradient effect.
    """
    width = options.max_width
    content = Group(*self.renderables)
    if self.show_quit_panel:
        # Use a Rich Text renderable so the bracketed markup tags remain literal in the output
        panel = Panel(RichText("Press [bold]Ctrl+C[/bold] to stop."), expand=False)
        content = Group(content, Align(panel, align="right"))

    lines = console.render_lines(content, options, pad=True, new_lines=False)
    for line_idx, segments in enumerate(lines):
        col = 0
        for seg in segments:
            text = seg.text
            base_style = seg.style or Style()
            cluster = ""
            cluster_width = 0
            for ch in text:
                w = get_character_cell_size(ch)
                if w <= 0:
                    cluster += ch
                    continue
                if cluster:
                    style = self._get_style_at_position(
                        col - cluster_width, cluster_width, width
                    )
                    yield Segment(cluster, self._merge_styles(base_style, style))
                    cluster = ""
                    cluster_width = 0
                cluster = ch
                cluster_width = w
                col += w
            if cluster:
                style = self._get_style_at_position(
                    col - cluster_width, cluster_width, width
                )
                yield Segment(cluster, self._merge_styles(base_style, style))
        if line_idx < len(lines) - 1:
            yield Segment.line()

__rich_measure__(console, options)

Measure the minimum and maximum width for the gradient content.

Parameters:

Name Type Description Default
console Console

Console for measurement.

required
options ConsoleOptions

Rendering options.

required

Returns:

Name Type Description
Measurement Measurement

Combined width constraints.

Source code in src/rich_gradient/_base_gradient.py
def __rich_measure__(
    self, console: Console, options: ConsoleOptions
) -> Measurement:
    """
    Measure the minimum and maximum width for the gradient content.

    Args:
        console: Console for measurement.
        options: Rendering options.

    Returns:
        Measurement: Combined width constraints.
    """
    measurements = [Measurement.get(console, options, r) for r in self.renderables]
    if not measurements:
        # No renderables — return a reasonable default measurement.
        # Min width is 0; max width is the available maximum from options.
        return Measurement(0, options.max_width or 0)

    min_width = min(m.minimum for m in measurements)
    max_width = max(m.maximum for m in measurements)
    return Measurement(min_width, max_width)

_color_at(pos, width, span)

Return the hex color at a given position (for tests).

Source code in src/rich_gradient/_base_gradient.py
def _color_at(self, pos: int, width: int, span: int) -> str:
    """Return the hex color at a given position (for tests)."""
    stops = self._active_stops
    if not stops:
        return "#000000"
    frac = self._compute_fraction(pos, width, span)
    r, g, b = self._interpolate_color(frac, stops)
    return f"#{int(r):02x}{int(g):02x}{int(b):02x}"

_compute_fraction(position, width, span)

Compute fractional position for gradient interpolation, including phase.

Parameters:

Name Type Description Default
position int

Starting cell index.

required
width int

Cell width.

required
span float

Total span for gradient.

required

Returns:

Type Description
float

Fraction between 0.0 and 1.0.

Source code in src/rich_gradient/_base_gradient.py
def _compute_fraction(self, position: int, width: int, span: float) -> float:
    """
    Compute fractional position for gradient interpolation, including phase.

    Args:
        position: Starting cell index.
        width: Cell width.
        span: Total span for gradient.

    Returns:
        Fraction between 0.0 and 1.0.
    """
    total_width = (span or 0) * (self.repeat_scale or 1.0)
    if total_width <= 0:
        # Avoid division by zero; return phase-only fraction.
        return self.phase % 1.0

    base = (position + width / 2) / total_width
    return (base + self.phase) % 1.0

_get_style_at_position(position, width, span)

Compute the Rich Style for a character cluster at a given position.

Parameters:

Name Type Description Default
position int

Starting cell index of the cluster.

required
width int

Cell width of the cluster.

required
span int

Total available width for gradient calculation.

required

Returns:

Type Description
Style

Style with appropriate foreground and/or background colors.

Source code in src/rich_gradient/_base_gradient.py
def _get_style_at_position(self, position: int, width: int, span: int) -> Style:
    """
    Compute the Rich Style for a character cluster at a given position.

    Args:
        position: Starting cell index of the cluster.
        width: Cell width of the cluster.
        span: Total available width for gradient calculation.

    Returns:
        Style with appropriate foreground and/or background colors.
    """
    frac = self._compute_fraction(position, width, span)
    # If background mode is enabled, apply gradient to background only.
    if self.background:
        active = self.bg_colors if self.bg_colors else self.colors
        if not active:
            return Style()
        r, g, b = self._interpolate_color(frac, active)
        bg_style = f"#{int(r):02x}{int(g):02x}{int(b):02x}"
        return Style(bgcolor=bg_style)

    # Default: apply gradient to foreground; background uses bg_colors if provided.
    fg_style = ""
    bg_style = ""
    if self.colors:
        r, g, b = self._interpolate_color(frac, self.colors)
        fg_style = f"#{int(r):02x}{int(g):02x}{int(b):02x}"
    if self.bg_colors:
        r, g, b = self._interpolate_color(frac, self.bg_colors)
        bg_style = f"#{int(r):02x}{int(g):02x}{int(b):02x}"

    return Style(color=fg_style or None, bgcolor=bg_style or None)

_initialize_color_stops()

Initialize the active color stops based on mode and provided stops.

If only one stop is provided, duplicate it to create a smooth segment pair.

Source code in src/rich_gradient/_base_gradient.py
def _initialize_color_stops(self) -> List[ColorTriplet]:
    """Initialize the active color stops based on mode and provided stops.

    If only one stop is provided, duplicate it to create a smooth segment pair.
    """
    source = self.bg_colors if self.background else self.colors
    if not source:
        return []
    return [source[0], source[0]] if len(source) == 1 else source

_interpolate_color(frac, color_stops)

Interpolate color in linear light space with gamma correction.

Parameters:

Name Type Description Default
frac float

Fractional position between 0.0 and 1.0.

required
color_stops list[ColorTriplet]

List of ColorTriplet stops.

required

Returns:

Type Description
tuple[float, float, float]

Tuple of (r, g, b) in sRGB space.

Source code in src/rich_gradient/_base_gradient.py
def _interpolate_color(
    self, frac: float, color_stops: list[ColorTriplet]
) -> tuple[float, float, float]:
    """
    Interpolate color in linear light space with gamma correction.

    Args:
        frac: Fractional position between 0.0 and 1.0.
        color_stops: List of ColorTriplet stops.

    Returns:
        Tuple of (r, g, b) in sRGB space.
    """
    if frac <= 0:
        return color_stops[0]
    if frac >= 1:
        return color_stops[-1]

    # Determine segment and local position
    segment_count = len(color_stops) - 1
    pos = frac * segment_count
    idx = int(pos)
    t = pos - idx

    r0, g0, b0 = color_stops[idx]
    r1, g1, b1 = color_stops[min(idx + 1, segment_count)]

    def to_linear(c: float) -> float:
        return (c / 255.0) ** _GAMMA_CORRECTION

    def to_srgb(x: float) -> float:
        return (x ** (1.0 / _GAMMA_CORRECTION)) * 255.0

    lr0, lg0, lb0 = to_linear(r0), to_linear(g0), to_linear(b0)
    lr1, lg1, lb1 = to_linear(r1), to_linear(g1), to_linear(b1)

    lr = lr0 + (lr1 - lr0) * t
    lg = lg0 + (lg1 - lg0) * t
    lb = lb0 + (lb1 - lb0) * t

    return to_srgb(lr), to_srgb(lg), to_srgb(lb)

_interpolated_color(frac, stops, n)

Return the interpolated color at a fraction (for tests).

Source code in src/rich_gradient/_base_gradient.py
def _interpolated_color(self, frac: float, stops: list, n: int):
    """Return the interpolated color at a fraction (for tests)."""
    return self._interpolate_color(frac, stops)

_merge_styles(original, gradient_style) staticmethod

Merge original Style with gradient Style, preserving original attributes.

Parameters:

Name Type Description Default
original Style

The existing Rich Style.

required
gradient_style Style

Style with gradient colors.

required

Returns:

Type Description
Style

Combined Style.

Source code in src/rich_gradient/_base_gradient.py
@staticmethod
def _merge_styles(original: Style, gradient_style: Style) -> Style:
    """
    Merge original Style with gradient Style, preserving original attributes.

    Args:
        original: The existing Rich Style.
        gradient_style: Style with gradient colors.

    Returns:
        Combined Style.
    """
    return original + gradient_style if original else gradient_style

_styled(original, color)

Return a Style with the given color or bgcolor, preserving original (for tests).

Source code in src/rich_gradient/_base_gradient.py
def _styled(self, original: Style, color: str) -> Style:
    """Return a Style with the given color or bgcolor, preserving original (for tests)."""
    return (
        original + Style(bgcolor=color)
        if self.background
        else original + Style(color=color)
    )

_to_color_triplets(colors) staticmethod

Convert a list of color specifications to ColorTriplet instances.

Parameters:

Name Type Description Default
colors List[ColorType]

List of color strings, Color, or ColorTriplet.

required

Returns:

Type Description
List[ColorTriplet]

List of ColorTriplet.

Raises:

Type Description
TypeError

If unsupported color type encountered.

ColorParseError

If a color string fails to parse.

Source code in src/rich_gradient/_base_gradient.py
@staticmethod
def _to_color_triplets(colors: List[ColorType]) -> List[ColorTriplet]:
    """
    Convert a list of color specifications to ColorTriplet instances.

    Args:
        colors: List of color strings, Color, or ColorTriplet.

    Returns:
        List of ColorTriplet.

    Raises:
        TypeError: If unsupported color type encountered.
        ColorParseError: If a color string fails to parse.
    """
    triplets: List[ColorTriplet] = []
    for c in colors:
        if isinstance(c, ColorTriplet):
            triplets.append(c)
        elif isinstance(c, Color):
            triplets.append(c.get_truecolor())
        elif isinstance(c, str):
            triplets.append(Color.parse(c).get_truecolor())
        else:
            raise ColorParseError(
                f"Unsupported color type: {type(c)}\n\tCould not parse color: {c}"
            )
    return triplets