aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--video/qam.py82
1 files changed, 72 insertions, 10 deletions
diff --git a/video/qam.py b/video/qam.py
index 1cb10dd..711fb04 100644
--- a/video/qam.py
+++ b/video/qam.py
@@ -1,6 +1,18 @@
from manimlib import *
from scipy import signal as sig
+def square(t, slew_rate=.1):
+ t = t % 1
+
+ if t < .1:
+ return t * 1 / slew_rate
+ elif t < .5 - slew_rate / 2:
+ return 1
+ elif t < .5 + slew_rate / 2:
+ return 1 - (t - (.5 - slew_rate / 2)) * 1 / slew_rate
+ else:
+ return 0
+
class TimeDependentComplexPlane(ComplexPlane):
CONFIG = {
@@ -70,11 +82,13 @@ class QamModulation(Scene):
def construct(self):
self.camera.frame.reorient(0, 0, 0)
+ # create a complex plane
cplane = TimeDependentComplexPlane()
self.play(ShowCreation(cplane))
+ # a complex number
+ dot = Dot(cplane.n2p(1 + 2j), color=GREEN)
- dot = Dot(cplane.n2p(1 + 2j))
number = DecimalNumber()
number.add_updater(lambda m: m.next_to(dot, RIGHT))
@@ -86,18 +100,36 @@ class QamModulation(Scene):
number.add_updater(update_nums)
+ # create and move complex number around
self.play(ShowCreation(dot), ShowCreation(number))
- self.play(dot.animate.move_to(cplane.n2p(-5 + 3j)), run_time=2)
- self.play(dot.animate.move_to(cplane.n2p(-1 - 3j)), run_time=2)
- self.play(dot.animate.move_to(cplane.n2p(4 - 3j)))
+ self.play(dot.animate.move_to(cplane.n2p(-5+3j)), run_time=2)
+ self.play(dot.animate.move_to(cplane.n2p(-4-3j)), run_time=2)
+
+ # move the number according to a time dependend complex function
+ def signal_func(t):
+ real = -2 * np.cos(2 * t)
+ imag = 3 * (square(t / 2) - .5)
+
+ return real + 1j * imag
+
+ signal_func_at_zero = signal_func(0)
+
+ self.play(dot.animate.move_to(cplane.n2p(signal_func_at_zero)), run_time=2)
+
+ def dot_update_func(m, dt):
+ m.move_to(cplane.n2p(signal_func(m.time)))
+ m.time = m.time + dt / 2
- self.wait(1)
+ dot.time = 0
+ dot.add_updater(dot_update_func)
+ self.wait(5)
- # Get out and show time dependence
- self.play(self.camera.frame.animate.move_to(2 * OUT))
+ # get out and show time dependence
+ self.play(self.camera.frame.animate.shift(2 * OUT))
- graph_y = cplane.get_inphase_graph(lambda t: -2 * np.cos(2 * t), color=RED)
- graph_x = cplane.get_quadrature_graph(lambda t: 1 * sig.square(4 * t), color=BLUE)
+ # show functions
+ graph_y = cplane.get_inphase_graph(lambda t: signal_func(t).real, color=RED)
+ graph_x = cplane.get_quadrature_graph(lambda t: signal_func(t).imag, color=BLUE)
self.play(self.camera.frame.animate.reorient(-90, 90, 90))
self.play(ShowCreation(graph_x))
@@ -106,8 +138,38 @@ class QamModulation(Scene):
self.play(ShowCreation(graph_y))
self.play(self.camera.frame.animate.reorient(-100, 90, 90))
+ self.play(self.camera.frame.animate.rotate(40 * DEGREES, axis = UP))
+
+ # create arrow for complex value
+ dot.remove_updater(dot_update_func)
+ self.play(dot.animate.move_to(cplane.n2p(signal_func_at_zero)), run_time=2)
+
+ arr = Arrow(cplane.n2p(0), cplane.n2p(signal_func_at_zero))
+ self.play(ShowCreation(arr))
+
+ def arr_update_func(m, dt):
+ m.set_points_by_ends(
+ cplane.n2p(0, time=m.time),
+ cplane.n2p(signal_func(m.time), time=m.time)
+ )
+ m.time = m.time + dt / 2
+
+ dot.time = 0
+ arr.time = 0
+
+ dot.add_updater(dot_update_func)
+ arr.add_updater(arr_update_func)
+ self.wait(10)
+
+ self.play(*map(FadeOut, (arr, graph_x, graph_y)))
+ self.play(
+ self.camera.frame.animate.rotate(50 * DEGREES, axis = UP),
+ dot.animate.move_to(cplane.n2p(1+1j))
+ )
+ self.play(self.camera.frame.animate.shift(-2 * OUT))
- # arrow = Arrow(
+ dot.remove_updater(dot_update_func)
+ arr.remove_updater(arr_update_func)
# open an interactive IPython shell here
# self.embed()