format files

Author: ytdHuang

benchmarks/correlations_and_spectrum.jl

@@ -14,13 +14,14 @@ function benchmark_correlations_and_spectrum!(SUITE)
     function _calculate_fft_spectrum(H, tlist, c_ops, A, B)
         corr = correlation_2op_1t(H, nothing, tlist, c_ops, A, B; progress_bar = Val(false))
         ωlist, spec = spectrum_correlation_fft(tlist, corr)
-        nothing
+        return nothing
     end
 
     SUITE["Correlations and Spectrum"]["FFT Correlation"] =
         @benchmarkable _calculate_fft_spectrum($H, $t_l, $c_ops, $(a'), $a)
 
-    SUITE["Correlations and Spectrum"]["Spectrum"]["Exponential Series"] = @benchmarkable spectrum($H, $ω_l, $c_ops, $(a'), $a)
+    SUITE["Correlations and Spectrum"]["Spectrum"]["Exponential Series"] =
+        @benchmarkable spectrum($H, $ω_l, $c_ops, $(a'), $a)
 
     return nothing
 end

src/correlations.jl

@@ -39,7 +39,8 @@ function correlation_3op_2t(
         ψ0 = steadystate(L)
     end
 
-    allequal((L.dims, ψ0.dims, A.dims, B.dims, C.dims)) || throw(DimensionMismatch("The quantum objects are not of the same Hilbert dimension."))
+    allequal((L.dims, ψ0.dims, A.dims, B.dims, C.dims)) ||
+        throw(DimensionMismatch("The quantum objects are not of the same Hilbert dimension."))
 
     kwargs2 = merge((saveat = collect(tlist),), (; kwargs...))
     ρt = mesolve(L, ψ0, tlist; kwargs2...).states

src/deprecated.jl

@@ -16,4 +16,6 @@ end
 
 export FFTCorrelation
 
-FFTCorrelation() = error("`FFTCorrelation` has been deprecated and will be removed in next major release, please use `spectrum_correlation_fft` to calculate the spectrum with FFT method instead.")
+FFTCorrelation() = error(
+    "`FFTCorrelation` has been deprecated and will be removed in next major release, please use `spectrum_correlation_fft` to calculate the spectrum with FFT method instead.",
+)

src/spectrum.jl

@@ -41,12 +41,7 @@ function spectrum(
     B::QuantumObject{<:AbstractArray{T3},OperatorQuantumObject};
     solver::SpectrumSolver = ExponentialSeries(),
     kwargs...,
-) where {
-    MT1<:AbstractMatrix,
-    T2,
-    T3,
-    HOpType<:Union{OperatorQuantumObject,SuperOperatorQuantumObject},
-}
+) where {MT1<:AbstractMatrix,T2,T3,HOpType<:Union{OperatorQuantumObject,SuperOperatorQuantumObject}}
     return _spectrum(liouvillian(H, c_ops), ωlist, A, B, solver; kwargs...)
 end
 
@@ -77,7 +72,8 @@ function _spectrum(
     solver::ExponentialSeries;
     kwargs...,
 ) where {T1,T2,T3}
-    allequal((L.dims, A.dims, B.dims)) || throw(DimensionMismatch("The quantum objects are not of the same Hilbert dimension."))
+    allequal((L.dims, A.dims, B.dims)) ||
+        throw(DimensionMismatch("The quantum objects are not of the same Hilbert dimension."))
 
     rates, vecs, ρss = _spectrum_get_rates_vecs_ss(L, solver)
 
@@ -122,11 +118,11 @@ Calculate the power spectrum corresponding to a two-time correlation function us
 - `ωlist`: the list of angular frequencies ``\omega``.
 - `Slist`: the list of the power spectrum corresponding to the angular frequencies in `ωlist`.
 """
-function spectrum_correlation_fft(tlist::AbstractVector, corr::AbstractVector; inverse::Bool=false)
+function spectrum_correlation_fft(tlist::AbstractVector, corr::AbstractVector; inverse::Bool = false)
     N = length(tlist)
     dt_list = diff(tlist)
     dt = dt_list[1]
-    
+
     all(≈(dt), dt_list) || ArgumentError("tlist must be equally spaced for FFT.")
 
     # power spectrum list
@@ -137,4 +133,4 @@ function spectrum_correlation_fft(tlist::AbstractVector, corr::AbstractVector; i
     ωlist = 2 * π * fftshift(fftfreq(N, 1 / dt))
 
     return ωlist, Slist
-end
+end

test/core-test/correlations_and_spectrum.jl

@@ -6,7 +6,7 @@
     t_l = range(0, 333 * π, length = 1000)
     corr = correlation_2op_1t(H, nothing, t_l, c_ops, a', a; progress_bar = Val(false))
     ω_l1, spec1 = spectrum_correlation_fft(t_l, corr)
-    
+
     ω_l2 = range(0, 3, length = 1000)
     spec2 = spectrum(H, ω_l2, c_ops, a', a)