"""
Example 1

- calculate spectra of undulator radiation with three different observation positions
- compare the three in the post-processor and export an ASCII file
"""

import sys
isjupyter = "ipykernel" in sys.modules

import spectra 
if isjupyter:
    # start with CLI mode
    spectra.Start(mode="c") 
    import plotly.io as pio
    pio.renderers.default = "notebook" # to enable exporting as an HTML file
else:
    # start with default settings
    # Mode = interactive, Browser = Chrome, Source Files = remote
    spectra.Start() 

# open "sample.json" in the current directory
spectra.Open("sample.json") 

if isjupyter:
    # display the parameters
    from IPython.display import display, HTML
    display(HTML("<h2> Parameters and configurations"))
    spectra.DisplayParameters(fontfamily="arial", fontsize="10px")

# select calculation: "Far Field & Ideal Condition::Energy Dependence::Angular Flux Density"
spectra.SelectCalculation("far", "energy", "fdensa")

# start calculation with an output file of "./output/sample1-1.json"
spectra.StartCalculation(folder="./output", prefix="sample1", serial=1)

# check the result; plot the spectrum of angular flux density
if isjupyter:
    display(HTML("<h2> Example of undulator-radiation spectrum"))
    spectra.PostProcessCLI.Plot(item="Flux Density")
else:
    spectra.ShowPostProcessor()

# change the paramter, vertical position shifted to y=1mm
spectra.Set("config", "xyfix", [0, 1])

# start calculation with an output file of "sample1-2.json"
# same folder and prefix
# serial number automatically incremented
spectra.StartCalculation()

# horizontal position shifted to x=1mm; output file "sample1-3.json"
spectra.Set("config", "xyfix", [1, 0])
spectra.StartCalculation()

# check the result; compare with the former result
if isjupyter:
    display(HTML("<h2> Comparison between three different observation positions"))
    spectra.PostProcessCLI.Plot(data=["sample1-1", "sample1-2", "sample1-3"], item="Flux Density")
    spectra.PostProcessCLI.Plot(data=["sample1-1", "sample1-2", "sample1-3"], item="Flux Density", xrange=[5000,6500])
else:
    spectra.PostProcess.SelectData("sample1-1")
    spectra.PostProcess.ComparativePlot("sample1-2", "sample1-3")
    spectra.PostProcess.DuplicatePlot()
    # export the comparative plot as an ASCII file
    spectra.PostProcess.Export("./output/comparative.txt")

    # expand the fundamental radiation
    spectra.PostProcess.PlotRange(x=[5000,6500])

if not isjupyter:
    # wait for the user's response
    input("Completed. Press enter to exit. ")
# exit
spectra.Exit()