# -*- 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()