diff --git a/simphony/simulation.py b/simphony/simulation.py index a8599209..c468f046 100644 --- a/simphony/simulation.py +++ b/simphony/simulation.py @@ -240,4 +240,18 @@ def plot(self, output_port): def get_magnitude_by_frequency_thz(self, output_port): freq = np.divide(self.frequency, MathPrefixes.TERA) mag = np.power(np.absolute(self.simulated_matrix[:, output_port]), 2) - return freq, mag \ No newline at end of file + return freq, mag + + def get_magnitude_by_wavelength_nm(self, output_port): + wl = self.frequencyToWavelength(self.frequency) / MathPrefixes.NANO + mag = np.power(np.absolute(self.simulated_matrix[:, output_port]), 2) + return wl, mag + + def export_s_matrix(self): + """Returns the matrix result of the multi-input simulation. + + Returns + ------- + frequency, matrix: np.array, np.ndarray + """ + return self.frequency, self.simulated_matrix \ No newline at end of file