File:Wigner quasiprobability distribution of superpositioned Fock states.webm
From Wikimedia Commons, the free media repository
Revision as of 10:17, 10 September 2024 by Geek3 (talk | contribs) (Adding Category:Animations of quantum wave functions)
Size of this JPG preview of this WEBM file: 800 × 533 pixels. Other resolutions: 320 × 213 pixels | 640 × 427 pixels | 1,024 × 683 pixels | 1,280 × 853 pixels | 2,400 × 1,600 pixels.
Original file (WebM audio/video file, VP9, length 7.5 s, 2,400 × 1,600 pixels, 506 kbps overall, file size: 464 KB)
File information
Structured data
Captions
Contents
Summary
DescriptionWigner quasiprobability distribution of superpositioned Fock states.webm |
English: Wigner quasiprobability distribution of superpositioned Fock states, for .
Matplotlib codeimport matplotlib.pyplot as plt
import matplotlib as mpl
import numpy as np
from IPython.display import display
from qutip import (about, basis, coherent, coherent_dm, displace, fock, ket2dm,
plot_wigner, squeeze, thermal_dm, wigner_cmap, wigner)
import scipy.ndimage
import os
from tqdm import tqdm
def rotate_and_crop(array, angle, xvec, yvec):
rotated_array = scipy.ndimage.rotate(array, -angle, reshape=False)
rows, cols = rotated_array.shape
center_row, center_col = rows // 2, cols // 2
target_rows, target_cols = len(yvec), len(xvec)
start_row = center_row - target_rows // 2
end_row = start_row + target_rows
start_col = center_col - target_cols // 2
end_col = start_col + target_cols
return rotated_array[start_row:end_row, start_col:end_col]
def plot_wigner_marginals(W, xvec, yvec, marginal_max, resolution=200, angle=0):
wmap = wigner_cmap(W)
wlim = np.abs(W).max()
cmap = plt.colormaps['RdBu']
fig = plt.figure()
n, m = 5, 1
fig, axes = plt.subplot_mosaic(
[ ["top"] * n + ["3d"] * m ] * m + [ ["mid"] * n + ["right"] * m] * n,
figsize=(20, 20),
layout="constrained",
width_ratios=[1.05] * (n+m))
ax = axes["mid"]
norm = mpl.colors.Normalize(-wlim, wlim)
ax.contourf(xvec, yvec, W, resolution // 3, norm=norm, cmap=cmap)
ax = axes["top"]
x_marginal = np.sum(W, axis=0)
y_marginal = np.sum(W, axis=1)
ax.fill_between(xvec, x_marginal, 0, color='#938fba', alpha=0.5)
ax.plot(xvec, x_marginal, color='#4a5a90')
ax.set_xlim(min(xvec), max(xvec))
ax.set_ylim(0, marginal_max * 1.05)
ax.set_xticks([])
ax.set_yticks([])
ax = axes["right"]
ax.fill_betweenx(yvec, np.sum(W, axis=1), 0, color='#938fba', alpha=0.5)
ax.plot(y_marginal, yvec, color='#4a5a90')
ax.set_xlim(0, marginal_max * 1.05)
ax.set_ylim(min(yvec), max(yvec))
ax.set_xticks([])
ax.set_yticks([])
ax = axes["3d"]
ax.axis('off')
return fig
def plot_wigner_with_marginals(psi, **kwargs):
radius = kwargs.get('radius', 5)
resolution = kwargs.get('resolution', 500)
angles = kwargs.get('angles', np.linspace(0, 2*np.pi, 100))
dir_path = kwargs.get('dir_path', './output')
xvec_upscaled = np.linspace(-radius*1.5, radius*1.5, int(resolution*1.5))
yvec_upscaled = np.linspace(-radius*1.5, radius*1.5, int(resolution*1.5))
xvec = np.linspace(-radius, radius, int(resolution))
yvec = np.linspace(-radius, radius, int(resolution))
W_upscaled = wigner(psi, xvec_upscaled, yvec_upscaled)
marginal_max = max(max(np.sum(W_upscaled, axis=0)), max(np.sum(W_upscaled, axis=1)))
print(f"outputting to {dir_path}")
for N, angle in tqdm(enumerate(angles)):
W = rotate_and_crop(W_upscaled, angle, xvec, yvec)
fig = plot_wigner_marginals(W, xvec, yvec, marginal_max=marginal_max, resolution=resolution, angle=angle)
if not os.path.exists(dir_path):
os.makedirs(dir_path)
fig.savefig(f"{dir_path}/{N:03d}.png",bbox_inches='tight')
plt.close(fig)
mpl.use('agg')
configs = {
"N_dim" : 40,
"radius" : 3.5,
"resolution" : 500,
"angles" : [i * 2 for i in range(180)],
"dir_path" : ""
}
for n1, n2 in [(0, 0), (0, 1), (0, 2), (1, 1), (1, 2), (2, 2)]:
psi = (fock(configs["N_dim"], n1) + fock(configs["N_dim"], n2)).unit()
configs["dir_path"] = f"./fock/fock_{n1}_{n2}"
plot_wigner_with_marginals(psi, **configs)
Sh codefor dir in ./*/; do
# Extract folder name
folder_name=$(basename "$dir")
# Create output path
output_path="./${folder_name}.webm"
# Convert images to webm
echo $output_path
ffmpeg -y -framerate 24 -i "$dir"%03d.png -c:v libvpx-vp9 -b:v 0 -crf 30 -pix_fmt yuva420p "$output_path"
done
ffmpeg -y \
-i ./fock_0_0.webm \
-i ./fock_0_1.webm \
-i ./fock_0_2.webm \
-i ./fock_1_1.webm \
-i ./fock_1_2.webm \
-i ./fock_2_2.webm \
-filter_complex "
[0:v]scale=800:800[v0];
[1:v]scale=800:800[v1];
[2:v]scale=800:800[v2];
[3:v]scale=800:800[v3];
[4:v]scale=800:800[v4];
[5:v]scale=800:800[v5];
[v0][v1][v2]hstack=inputs=3[row0];
[v3][v4][v5]hstack=inputs=3[row1];
[row0][row1]vstack=inputs=2[out]
" \
-map "[out]" \
-c:v libvpx-vp9 -b:v 0 -crf 30 -pix_fmt yuva420p \
output_grid.webm
|
Date | |
Source | Own work |
Author | Cosmia Nebula |
Licensing
I, the copyright holder of this work, hereby publish it under the following license:
This file is licensed under the Creative Commons Attribution-Share Alike 4.0 International license.
- You are free:
- to share – to copy, distribute and transmit the work
- to remix – to adapt the work
- Under the following conditions:
- attribution – You must give appropriate credit, provide a link to the license, and indicate if changes were made. You may do so in any reasonable manner, but not in any way that suggests the licensor endorses you or your use.
- share alike – If you remix, transform, or build upon the material, you must distribute your contributions under the same or compatible license as the original.
File history
Click on a date/time to view the file as it appeared at that time.
Date/Time | Thumbnail | Dimensions | User | Comment | |
---|---|---|---|---|---|
current | 03:58, 31 August 2024 | 7.5 s, 2,400 × 1,600 (464 KB) | Cosmia Nebula (talk | contribs) | Uploaded while editing "Wigner quasiprobability distribution" on en.wikipedia.org |
You cannot overwrite this file.
File usage on Commons
There are no pages that use this file.
Transcode status
Update transcode statusFile usage on other wikis
The following other wikis use this file:
- Usage on en.wikipedia.org
Metadata
This file contains additional information such as Exif metadata which may have been added by the digital camera, scanner, or software program used to create or digitize it. If the file has been modified from its original state, some details such as the timestamp may not fully reflect those of the original file. The timestamp is only as accurate as the clock in the camera, and it may be completely wrong.
Software used |
---|