from __future__ import division
import numpy as np
from matplotlib import pyplot as plt
import matplotlib as mpl
from sfig import rcParams
rcParams.update({'figure.figsize':(5, 2.5)})
import matplotlib.ticker as mtick
def sci_notation(val):
"""The two args are the value and tick position"""
if val==0:
return '$0$'
ancilla, exponent = '{:1.0e}'.format(val).split('e')
if exponent[0]=='+':
exponent = exponent.lstrip('+0')
if exponent[0]=='-':
exponent = '-'+exponent.lstrip('-0')
if exponent=='':
exponent='0'
return '${} \cdot 10^{{ {} }}$'.format(ancilla, exponent)
mpl.rcParams['axes.color_cycle'] = list(reversed(['k', 'k', 'b', 'b', 'r', 'r', 'g', 'g']))
TEBD_data_location = "S:/Data/FBI/L3/itebd/"
from SimplePEPS.tools import fileio
from SimplePEPS.tools.containers import OrderedDict
b_es = OrderedDict()
#import matplotlib
#matplotlib.use('PDF')
from matplotlib.backends.backend_pdf import PdfPages
from DataAnalysis import data
from DataAnalysis import PEPS_plots as pp
import numpy as np
from matplotlib import pyplot as plt
from PEPS import IO_helper as IO
from PEPS import spectra_analyzer as spec
from PEPS.CylinderPEPS import run
from matplotlib.offsetbox import AnchoredOffsetbox, TextArea, DrawingArea, HPacker
from sfig import ticker, rcParams
makelegend = False
if makelegend:
rcParams.update({'figure.figsize':(6.25, 2.5)})
else:
rcParams.update({'figure.figsize':(3.75, 2.5)})
def go(L):
#with PdfPages('L_10_all_mom_10.pdf') as pdf:
###########################################################
#get state
i=10
stri = str(i)
IO.go_to_data_parent('interpolatedboson/a'+stri)
IO.add_to_path()
state = IO.get_state()
Ls = [L]
L_min, L_max = min(Ls), max(Ls)
shift_func = lambda L: 0
