def func(f1, b, set):
"""
This function Perform the gate application on the initial state of qubit and then the state tomography,
at the same time compute the analytical bloch state.
INCOME:
f1="string", b="string"
fig_data="svg",fig_data2="svg"
"""
#### Create the bloch-sphere on qutip
b1 = Bloch()
b2 = Bloch()
## create the analyical quantum state and perform tranformation
## Add states to the bloch sphere and plot it
b1.add_states(analytic(f1, b))
if not set:
states = qtomography(f1, b)
b2.add_vectors(states)
else:
### implements for
states = qtomography_s(f1, b)
for i in states:
b2.add_points(i)
b2.add_vectors(np.average(states, axis=0))
###
b2.render()
fig_data = print_figure(b2.fig, 'svg')
b1.render()
fig_data2 = print_figure(b1.fig, 'svg')
#b1.show()
#b2.show()
b1.clear()
b2.clear()
return fig_data, fig_data2
def bloch_plot(self, filename, points=None, save=False):
"""
Plot the current state on the Bloch sphere using
qutip.
"""
if points is None:
# convert current state into density operator
rho = np.outer(self.state.H, self.state)
# get Bloch vector representation
points = self.get_bloch_vec(rho)
# Can only plot systems of dimension 2 at this time
assert len(
points
) == 3, "System dimension must be spin 1/2 for Bloch sphere plot"
# create instance of 3d plot
bloch = Bloch(fig=1, figsize=[9, 9], view=[190, 10])
# add state
bloch.add_points(points)
bloch.render()
if save is True:
print('Bloch plot saved to Sim Results folder')
path = 'C:/Users/Boundsy/Desktop/Uni Work/PHS2360/Sim Results/' + str(
filename) + '.png'
bloch.fig.savefig(path, dpi=800, transparent=True)
def bloch_plot(self, points=None):
"""
Plot the current state on the Bloch sphere using
qutip.
"""
if points is None:
# convert current state into density operator
rho = np.outer(self.state.H, self.state)
# get Bloch vector representation
points = self.get_bloch_vec(rho)
# Can only plot systems of dimension 2 at this time
assert len(
points
) == 3, "System dimension must be spin 1/2 for Bloch sphere plot"
# create instance of 3d plot
bloch = Bloch(figsize=[9, 9])
# add state
bloch.add_points(points)
bloch.add_vectors([0, 0, 1])
bloch.render(bloch.fig, bloch.axes)
bloch.fig.savefig("bloch.png", dpi=600, transparent=True)
bloch.show()
def maj_vid(self, points):
#takes screenshots of majorana stars over time to produce vid
if points is None:
points = [getStars(self.state)]
i = 0
for p in points:
bloch = Bloch(figsize=[9, 9])
bloch.xlabel = ['$<F_x>$', '']
bloch.ylabel = ['$<F_y>$', '']
bloch.zlabel = ['$<F_z>$', '']
bloch.point_color = ['g', 'r',
'b'] #ensures point and line are same colour
bloch.point_marker = ['o', 'd', 'o']
bloch.add_points([p[0][0], p[1][0], p[2][0]])
bloch.add_points([p[0][1], p[1][1], p[2][1]])
bloch.add_points(p, meth='l')
bloch.render(bloch.fig, bloch.axes)
bloch.fig.savefig(
"bloch" + str(i).zfill(int(np.ceil(np.log10(len(points))))) +
".png",
dpi=600,
transparent=False)
i += 1
def bloch_plot2(self,
filename,
save=False,
vecList=[],
vecColour=[],
view=[190, 10],
points=None,
folder=False,
fig=False,
ax=False):
"""
Plot the current state on the Bloch sphere using
qutip.
"""
if points is None:
# convert current state into density operator
rho = np.outer(self.state.H, self.state)
# get Bloch vector representation
points = self.get_bloch_vec(rho)
# Can only plot systems of dimension 2 at this time
assert len(
points
) == 3, "System dimension must be spin 1/2 for Bloch sphere plot"
# create instance of 3d plot
if not fig or not ax:
bloch = Bloch(figsize=[9, 9], view=view)
else:
bloch = Bloch(fig=fig, axes=ax, view=view)
# add state
bloch.add_points(points)
# bloch.zlabel = [r'$\left|+z\right>$',r'$\left|-z\right>$']
bloch.zlabel = [r'$\ket{+_z}$', r'$\ket{-_z}$']
# bloch.ylabel = [r'$\ket{+_y}$',r'$\ket{-_y}$']
# bloch.ylabel = [r'$\ket{+_y}$',r'$\ket{-_y}$']
# print(vecList.shape)
# add vectors
if vecList.shape[1] == 3:
if len(vecColour) >= vecList.shape[0] and len(vecColour) > 0:
bloch.vector_color = vecColour
else:
bloch.vector_color = ['#CC6600', 'royalblue', 'r', 'm', 'g']
bloch.add_vectors(vecList)
# add field vector
# bloch.add_vectors([1,0,0.15])
# bloch.add_vectors([0,0,1])
# bloch.add_vectors([1,0,0])
# render bloch sphere
if not fig or not ax:
bloch.render()
else:
# bloch.render(fig = fig, axes = ax)
bloch.render(fig=fig)
# save output
if save is True:
if not folder:
folder = 'C:/Users/Boundsy/Desktop/Uni Work/PHS2360/Sim Results/'
print('Bloch plot saved to ' + str(folder))
path1 = folder + str(filename) + '.png'
path2 = folder + str(filename) + '.pdf'
bloch.fig.savefig(path1, dpi=800, transparent=True)
bloch.fig.savefig(path2, dpi=800, transparent=True)
# return axes for annotations
return bloch.fig, bloch.axes
