1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97
|
/*!
* \file
* \brief Implementation of an Orthogonal Frequency Division Multiplex
* (OFDM) class
* \author Pal Frenger, Anders Persson and Tony Ottosson
*
* -------------------------------------------------------------------------
*
* IT++ - C++ library of mathematical, signal processing, speech processing,
* and communications classes and functions
*
* Copyright (C) 1995-2008 (see AUTHORS file for a list of contributors)
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*
* -------------------------------------------------------------------------
*/
#include <itpp/comm/ofdm.h>
#include <itpp/base/specmat.h>
#include <itpp/base/operators.h>
#include <itpp/signal/transforms.h>
namespace itpp {
OFDM::OFDM(int inNfft, int inNcp, int inNupsample)
{
set_parameters(inNfft, inNcp, inNupsample);
}
void OFDM::set_parameters(const int inNfft, const int inNcp, const int inNupsample){
it_assert(inNfft >= 2, "OFDM: Nfft must be >=2.");
it_assert(inNcp >= 0 && inNcp <= inNfft, "OFDM: Ncp must be >=0 and <=Nfft.");
it_assert(inNupsample >= 1 && inNupsample <= 100, "OFDM: Ncp must be >=1 and <=100.");
Nfft = inNfft;
Ncp = inNcp;
Nupsample = inNupsample;
norm_factor = std::sqrt(static_cast<double>(Nupsample*Nfft*Nfft) / (Nfft+Ncp));
setup_done = true;
}
void OFDM::modulate(const cvec &input, cvec &output){
it_assert(setup_done == true, "OFDM::modulate: You must set the length of the FFT and the cyclic prefix!");
const int N = input.length() / Nfft;
it_assert(N*Nfft == input.length(), "OFDM::modulate: Length of input vector is not a multiple of Nfft.");
output.set_length(Nupsample*N*(Nfft+Ncp));
cvec outtemp(Nfft);
for (int i = 0; i < N; i++) {
outtemp = ifft(concat(input.mid(i*Nfft, Nfft/2), zeros_c(Nfft*(Nupsample-1)),
input.mid(i*Nfft+Nfft/2, Nfft/2))) * norm_factor;
output.replace_mid(Nupsample*(Nfft+Ncp)*i, concat(outtemp.right(Nupsample*Ncp), outtemp));
}
}
cvec OFDM::modulate(const cvec &input)
{
cvec output;
modulate(input, output);
return output;
}
void OFDM::demodulate(const cvec& input, cvec &output){
it_assert(setup_done == true, "OFDM::demodulate: You must set the length of the FFT and the cyclic prefix!");
const int N = input.length() / (Nfft+Ncp) / Nupsample;
it_assert(Nupsample*N*(Nfft+Ncp) == input.length(), "OFDM: Length of input vector is not a multiple of Nfft+Ncp.");
output.set_length(N*Nfft);
// normalize also taking the energy loss into the cyclic prefix into account
for (int i = 0; i < N; i++) {
cvec x = fft(input.mid(Nupsample*(i*(Nfft+Ncp)+Ncp), Nupsample*Nfft));
output.replace_mid(Nfft*i, concat(x.left(Nfft/2), x.right(Nfft/2)) / norm_factor);
}
}
cvec OFDM::demodulate(const cvec &input){
cvec output;
demodulate(input, output);
return output;
}
} //namespace itpp
|