diff options
Diffstat (limited to 'src/optical_signal.cpp')
| -rw-r--r-- | src/optical_signal.cpp | 157 |
1 files changed, 157 insertions, 0 deletions
diff --git a/src/optical_signal.cpp b/src/optical_signal.cpp new file mode 100644 index 0000000..704dfd6 --- /dev/null +++ b/src/optical_signal.cpp @@ -0,0 +1,157 @@ +#include <optical_signal.h> +#include <specs.h> + +using namespace std::complex_literals; + +unsigned int OpticalSignal::nextId = 0; + +OpticalSignal &OpticalSignal::operator+=(const OpticalSignal &rhs) +{ + // cout << "Summing two signals: " << endl; + // cout << "\t" << *this << endl; + // cout << "\t" << rhs << endl; + + auto this_wavelength = this->getWavelength(); + auto rhs_wavelength = rhs.getWavelength(); + + if (this_wavelength == rhs_wavelength) + m_field += rhs.m_field; + else if (isnan(rhs_wavelength)) + (void)rhs; // nothing to do + else if (isnan(this_wavelength)) + *this = rhs; + else if (specsGlobalConfig.simulation_mode == FREQUENCY_DOMAIN) + { + // TODO: move that somewhere else ??? + // Replace with rhs + if (rhs_wavelength > this_wavelength) + *this = rhs; + } + else + cerr << "Attempted to sum signals of different wavelengths, but current" + << "simulation mode doesn't allow it (" << __FUNCTION__ << ")" << endl; + // cout << "\t--> " << *this << endl; + return *this; +} + +OpticalSignal &OpticalSignal::operator-=(const OpticalSignal &rhs) +{ + // cout << "Substracting two signals: " << endl; + // cout << "\t" << *this << endl; + // cout << "\t" << rhs << endl; + + auto this_wavelength = this->getWavelength(); + auto rhs_wavelength = rhs.getWavelength(); + + if (this_wavelength == rhs_wavelength) + m_field -= rhs.m_field; + else if (isnan(rhs_wavelength)) + (void)rhs; // nothing to do + else if (isnan(this_wavelength)) + *this = rhs; + else if (specsGlobalConfig.simulation_mode == FREQUENCY_DOMAIN) + { + // TODO: move that somewhere else ??? + // Replace with rhs + if (rhs_wavelength > this_wavelength) + *this = rhs; + } + else + cerr << "Attempted to substract signals of different wavelengths, but current" + << "simulation mode doesn't allow it (" << __FUNCTION__ << ")" << endl; + // cout << "\t--> " << *this << endl; + return *this; +} + +OpticalSignal OpticalSignal::sumSignals(OpticalSignal s0, OpticalSignal s1) +{ + cerr << "Use of deprecated" << __FUNCTION__<< endl; + + // Maybe doing just the sum is faster than this + if (s0.m_field == 0.0) { + OpticalSignal os_sum = s1; + os_sum.getNewId(); + return os_sum; + } + if (s1.m_field == (complex<double>) 0) { + OpticalSignal os_sum = s0; + os_sum.getNewId(); + return os_sum; + } + + if (s0.getWavelength() != s1.getWavelength()) + { + bool ignore_wavelength_error = false; + if (ignore_wavelength_error) + { + cerr << "Signals have different wavelengths:" << std::endl; + cerr << "\t" << s0.getWavelength() << "(" << norm(s0.m_field) << ")" + << "\t" << s1.getWavelength() << "(" << norm(s1.m_field) << ")" + << std::endl; + + sc_stop(); + } else { + OpticalSignal os_sum = s0; + if (norm(s1.m_field) > norm(s0.m_field)) + os_sum = s1; + os_sum.getNewId(); + return os_sum; + } + } + + complex<double> Asum = s0.m_field + s1.m_field; + + OpticalSignal os_sum = s0; + os_sum.m_field = Asum; + os_sum.getNewId(); + + return os_sum; +} + +uint32_t OpticalSignal::getIDFromWavelength(const double &wavelength) +{ + auto it = find (specsGlobalConfig.wavelengths_vector.cbegin(), + specsGlobalConfig.wavelengths_vector.cend(), + wavelength); + + uint32_t index; + + if (it != specsGlobalConfig.wavelengths_vector.cend()) + { // Found + index = it - specsGlobalConfig.wavelengths_vector.cbegin(); + } + else + { // Didn't find, has to add it to the vector + specsGlobalConfig.wavelengths_vector.push_back(wavelength); + index = specsGlobalConfig.wavelengths_vector.size() - 1; + // -1 because of zero indexing + } + + if (index == (uint32_t)(-1)) + { + cerr << "Error: too many wavelengths being used ." << endl; + exit(1); + } + + return index; +} + +double OpticalSignal::getWavelength() const +{ + if (m_wavelength_id >= specsGlobalConfig.wavelengths_vector.size()) + { + cerr << "Wavelength not found in global vector." << endl; + exit(1); + } + return specsGlobalConfig.wavelengths_vector[m_wavelength_id]; +} + +double OpticalSignal::getWavelength(const uint32_t &wavelength_id) +{ + if (wavelength_id >= specsGlobalConfig.wavelengths_vector.size()) + { + cerr << "Wavelength not found in global vector." << endl; + exit(1); + } + return specsGlobalConfig.wavelengths_vector[wavelength_id]; +}
\ No newline at end of file |