# HG changeset patch
# User calpe  <miguel.calpe.linares@gmail.com>
# Date 1543824764 -3600
#      Mon Dec 03 09:12:44 2018 +0100
# Node ID 54ca04b2829727172c8440a72f3c8ba11f3ec30b
# Parent  ead4b74699ffc6069529b651e09be574274ff775
Start working module spectra_multidim.py.

diff --git a/fluidsim/solvers/ns2d/strat/output/spectra_multidim.py b/fluidsim/solvers/ns2d/strat/output/spectra_multidim.py
--- a/fluidsim/solvers/ns2d/strat/output/spectra_multidim.py
+++ b/fluidsim/solvers/ns2d/strat/output/spectra_multidim.py
@@ -7,8 +7,6 @@
 
 """
 
-from __future__ import print_function
-
 import h5py
 import numpy as np
 
@@ -20,21 +18,40 @@
 
     def compute(self):
         """Computes multidimensional spectra at one time."""
+
+        # Get variables
         energyK_fft, energyA_fft = self.output.compute_energies_fft()
         energy_fft = energyK_fft + energyA_fft
 
+        ap_fft = self.sim.state.compute("ap_fft")
+        am_fft = self.sim.state.compute("am_fft")
+
         # Computes multidimensional spectra
         spectrumkykx_E = self.oper.compute_spectrum_kykx(energy_fft)
         spectrumkykx_EK = self.oper.compute_spectrum_kykx(energyK_fft)
         spectrumkykx_EA = self.oper.compute_spectrum_kykx(energyA_fft)
 
+        # The function compute_spectrum_kykx does not supports complex variable...
+        # Only works for the energy!
+
+        # spectrumkykx_ap_fft = self.oper.compute_spectrum_kykx(ap_fft)
+        # spectrumkykx_am_fft = self.oper.compute_spectrum_kykx(am_fft)
+
         # Saves dictionary
         dict_spectra = {
             "spectrumkykx_E": spectrumkykx_E,
             "spectrumkykx_EK": spectrumkykx_EK,
-            "spectrumkykx_EA": spectrumkykx_EA,
+            "spectrumkykx_EA": spectrumkykx_EA
         }
 
+        # dict_spectra = {
+        #     "spectrumkykx_E": spectrumkykx_E,
+        #     "spectrumkykx_EK": spectrumkykx_EK,
+        #     "spectrumkykx_EA": spectrumkykx_EA,
+        #     "spectrumkykx_ap_fft": spectrumkykx_ap_fft,
+        #     "spectrumkykx_am_fft": spectrumkykx_am_fft
+        # }
+
         return dict_spectra
 
     def _online_plot_saving(self, dict_spectra):