# Copyright (c) 2026 JSOL CORPORATION
#
# This script is released under the MIT License.
# See the full license text at:
# https://www.jmag-international.com/scriptlibrary/jmag_script_library_mit/

app = designer.GetApplication()

def fourierTransform(targetDataSet, type, axisType):
    """Perform a Fourier transform on the dataset passed as an argument"""
    fftParam = app.CreateFFTParameter(u"Fourier transform")
    fftParam.AddDataSet(targetDataSet)
    # Line
    # 0:Specify dataset
    # 1:Specify all datasets
    fftParam.SetLineType(1)
    fftParam.SetType(type)
    fftParam.SetMin(u"0.01")
    fftParam.SetMax(u"0.015")
    # Periodicity
    # 0:None
    # 1:1/2
    fftParam.SetPeriodicity(1)
    fftParam.SetFrequencyMultiplier(u"0.5")
    fftParam.SetAxisType(axisType)
    return fftParam

study = app.GetCurrentStudy()
dataSet = study.GetDataSet(u"Circuit Current", 1)
type = u"Amplitude"
axisType = u"Time"
parameter = fourierTransform(dataSet, type, axisType)

dm = app.GetDataManager()
dm.CreateFFTWithParameter(parameter)