Import video pipeline as it was

1 files changed, 160 insertions, 0 deletions

diff --git a/scenes/v16_bayes.py b/scenes/v16_bayes.py
new file mode 100644
index 0000000..311dd01
--- /dev/null
+++ b/scenes/v16_bayes.py
@@ -0,0 +1,160 @@
+# -*- coding: utf-8 -*-
+import sys, os
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+from manim import *
+from tiktok import *
+
+COLS, ROWS = 25, 40
+PX, PY = 0.27, 0.175
+GY = 1.65                       # centro vertical de la grilla
+RAD = 0.062
+APAGADO = "#2B3A5C"
+ENFERMA = 437                   # fila 17, columna 12 (queda centrada)
+FALSOS = [63, 147, 232, 318, 389, 476, 551, 668, 742, 861]
+
+
+def pos(i):
+    r, c = divmod(i, COLS)
+    return np.array([(c - (COLS - 1) / 2) * PX,
+                     GY + ((ROWS - 1) / 2 - r) * PY, 0.0])
+
+
+class Bayes(TikTok):
+    NAME = "bayes"
+
+    def construct(self):
+        self.prepare()
+        chip = self.chip("Probabilidad", CELESTE)
+
+        # --- la tarjeta del test ---------------------------------------------
+        marco = RoundedRectangle(corner_radius=0.32, width=6.4, height=3.4, stroke_width=5,
+                                 stroke_color=CELESTE, fill_color="#121C33",
+                                 fill_opacity=1).move_to(UP * 3.5)
+        et = Text("TEST", font=FONT, font_size=40, weight=BOLD, color=GRIS)
+        num = Text("99%", font=FONT, font_size=118, weight=BOLD, color=AMBAR)
+        pie = Text("de precisión", font=FONT, font_size=38, weight=BOLD, color=WHITE)
+        VGroup(et, num, pie).arrange(DOWN, buff=0.18).move_to(marco)
+        tarjeta = VGroup(marco, et, num, pie)
+
+        with self.beat(0) as b:
+            b.play(FadeIn(chip, shift=DOWN * 0.3), run_time=0.4)
+            b.play(GrowFromCenter(tarjeta), run_time=0.8)
+            b.play(Flash(num, color=AMBAR, line_length=0.35, num_lines=14,
+                         flash_radius=1.7), run_time=0.9)
+
+        sello = VGroup(
+            RoundedRectangle(corner_radius=0.18, width=4.7, height=1.25, stroke_width=8,
+                             stroke_color=ROSA, fill_color="#2A0E18", fill_opacity=0.94),
+            Text("POSITIVO", font=FONT, font_size=60, weight=BOLD, color=ROSA))
+        sello.rotate(-0.09).move_to(marco.get_bottom() + DOWN * 0.05)
+
+        with self.beat(1) as b:
+            b.play(FadeIn(sello, scale=2.4), run_time=0.55)
+            b.play(Wiggle(sello, scale_value=1.07), run_time=1.0)
+
+        preg = Text("La mayoría contesta:", font=FONT, font_size=40,
+                    weight=BOLD, color=GRIS).move_to(UP * 0.45)
+        mal = Text("99%", font=FONT, font_size=106, weight=BOLD, color=GRIS)
+        mal.next_to(preg, DOWN, buff=0.28)
+
+        with self.beat(2) as b:
+            b.play(FadeIn(preg, shift=UP * 0.2), run_time=0.5)
+            b.play(FadeIn(mal, scale=0.7), run_time=0.7)
+
+        tacha = Line(mal.get_left() + LEFT * 0.28, mal.get_right() + RIGHT * 0.28,
+                     stroke_width=11, color=ROSA)
+        bien = Text("9%", font=FONT, font_size=150, weight=BOLD,
+                    color=VERDE).move_to(DOWN * 2.05)
+
+        with self.beat(3) as b:
+            b.play(Create(tacha), run_time=0.45)
+            b.play(FadeIn(bien, scale=0.6), run_time=0.7)
+            b.play(Flash(bien, color=VERDE, flash_radius=1.7, num_lines=16), run_time=0.9)
+
+        # --- las mil personas -------------------------------------------------
+        puntos = VGroup(*[Dot(radius=RAD, color=APAGADO, fill_opacity=1).move_to(pos(i))
+                          for i in range(COLS * ROWS)])
+        rotulo = Text("1.000 personas", font=FONT, font_size=40, weight=BOLD,
+                      color=WHITE).move_to(UP * 5.55)
+
+        with self.beat(4) as b:
+            b.play(FadeOut(VGroup(tarjeta, sello, preg, mal, tacha, bien)), run_time=0.5)
+            b.play(FadeIn(puntos, lag_ratio=0), FadeIn(rotulo, shift=DOWN * 0.2),
+                   run_time=1.1)
+
+        enf = puntos[ENFERMA]
+        anillo = Circle(radius=0.30, stroke_width=6, color=ROSA).move_to(pos(ENFERMA))
+        cajita = RoundedRectangle(corner_radius=0.14, width=2.45, height=0.7, stroke_width=0,
+                                  fill_color="#0B0F1A", fill_opacity=0.92)
+        txt = Text("1 enferma", font=FONT, font_size=34, weight=BOLD, color=ROSA)
+        etiq = VGroup(cajita, txt).move_to(pos(ENFERMA) + RIGHT * 1.95)
+        guia = Line(anillo.get_right(), etiq.get_left(), stroke_width=4, color=ROSA)
+
+        with self.beat(5) as b:
+            b.play(enf.animate.set_color(ROSA).scale(2.2), run_time=0.5)
+            b.play(Create(anillo), run_time=0.4)
+            b.play(Create(guia), FadeIn(etiq), run_time=0.5)
+
+        # --- el barrido del test ----------------------------------------------
+        barrido = Line(LEFT * 3.6, RIGHT * 3.6, stroke_width=6, color=CELESTE)
+        barrido.move_to([0, pos(0)[1] + 0.22, 0])
+        ley1 = VGroup(Dot(radius=0.12, color=ROSA),
+                      Text("1 positivo verdadero", font=FONT, font_size=36,
+                           weight=BOLD, color=WHITE)).arrange(RIGHT, buff=0.26)
+        ley1.move_to(DOWN * 2.15)
+
+        with self.beat(6) as b:
+            b.play(FadeIn(barrido), run_time=0.25)
+            b.play(barrido.animate.move_to([0, pos(COLS * ROWS - 1)[1] - 0.22, 0]),
+                   run_time=max(0.9, b.floor - 1.5), rate_func=linear)
+            b.play(FadeOut(barrido),
+                   Flash(anillo, color=VERDE, flash_radius=0.65, num_lines=12),
+                   run_time=0.5)
+            b.add(ley1)
+
+        # --- los falsos positivos ---------------------------------------------
+        falsos = VGroup(*[puntos[i] for i in FALSOS])
+
+        with self.beat(7) as b:
+            b.play(LaggedStart(*[p.animate.set_color(AMBAR).scale(2.2) for p in falsos],
+                               lag_ratio=0.2), run_time=max(1.3, b.floor - 0.7))
+
+        ley2 = VGroup(Dot(radius=0.12, color=AMBAR),
+                      Text("10 falsos positivos", font=FONT, font_size=36,
+                           weight=BOLD, color=WHITE)).arrange(RIGHT, buff=0.26)
+        ley2.move_to(DOWN * 2.78)
+
+        with self.beat(8) as b:
+            b.play(FadeIn(ley2, shift=UP * 0.2), run_time=0.45)
+            b.play(*[Indicate(p, color=AMBAR, scale_factor=1.7) for p in falsos],
+                   run_time=1.1)
+
+        # --- los once positivos, en fila --------------------------------------
+        marcados = set(FALSOS) | {ENFERMA}
+        resto = VGroup(*[p for j, p in enumerate(puntos) if j not in marcados])
+        elegidos = [puntos[ENFERMA]] + [puntos[i] for i in FALSOS]
+        cuenta = Text("11 positivos", font=FONT, font_size=52, weight=BOLD,
+                      color=WHITE).move_to(UP * 2.2)
+        frac = Text("1 de 11 = 9%", font=FONT, font_size=90, weight=BOLD,
+                    color=VERDE).move_to(UP * 0.7)
+
+        with self.beat(9) as b:
+            b.play(resto.animate.set_opacity(0.12),
+                   FadeOut(VGroup(anillo, guia, etiq, rotulo, ley1, ley2)), run_time=0.5)
+            b.play(*[p.animate.move_to([(k - 5) * 0.62, 3.35, 0]).scale(2.4)
+                     for k, p in enumerate(elegidos)], run_time=0.9)
+            b.play(FadeIn(cuenta, shift=UP * 0.2), run_time=0.4)
+            b.play(FadeIn(frac, scale=0.7), run_time=0.6)
+
+        cierre = VGroup(
+            Text("Si la enfermedad es rara,", font=FONT, font_size=42, weight=BOLD, color=GRIS),
+            Text("un positivo suelto", font=FONT, font_size=50, weight=BOLD, color=WHITE),
+            Text("casi siempre es falso", font=FONT, font_size=50, weight=BOLD, color=AMBAR),
+        ).arrange(DOWN, buff=0.32).move_to(DOWN * 1.75)
+
+        with self.beat(10) as b:
+            b.play(FadeOut(resto), run_time=0.4)
+            b.play(FadeIn(cierre, shift=UP * 0.25), run_time=0.8)
+            b.play(Flash(frac, color=VERDE, flash_radius=2.2, num_lines=18), run_time=0.9)
+
+        self.finish()