| [2908] | 1 |  | 
|---|
|  | 2 | #include "sopnamsp.h" | 
|---|
|  | 3 | #include "toeplitzMatrix.h" | 
|---|
|  | 4 | #include "timing.h" | 
|---|
|  | 5 |  | 
|---|
|  | 6 | /*! | 
|---|
|  | 7 | \class SOPHYA::Toeplitz | 
|---|
|  | 8 | \ingroup NTools | 
|---|
|  | 9 | Class for describing a real Toeplitz or complex matrix. | 
|---|
|  | 10 | Symmetic (/hermitian) or not. | 
|---|
|  | 11 | Conjugate gradient system solver (for symmetric or hermitian matrices | 
|---|
|  | 12 | only) | 
|---|
|  | 13 | Toeplitz matrices have  constant values along negative slope diagonals. | 
|---|
|  | 14 | A general toeplitz nxn matrix is defined by 2n-1 values and a symmetric | 
|---|
|  | 15 | (/hermitian) by n values. | 
|---|
|  | 16 | General toeplitz matrix form : | 
|---|
|  | 17 | \verbatim | 
|---|
|  | 18 | a_0   b_1   b_2   ...     b_n-1 | 
|---|
|  | 19 | a1   a0   b1   b2  ... | 
|---|
|  | 20 | a2   a1   a0   b1   ... | 
|---|
|  | 21 | ..... | 
|---|
|  | 22 | ..... | 
|---|
|  | 23 | a_n-1  ....                a0 | 
|---|
|  | 24 | \endverbatim | 
|---|
|  | 25 | */ | 
|---|
|  | 26 |  | 
|---|
|  | 27 |  | 
|---|
|  | 28 | Toeplitz::Toeplitz() : hermitian_(false) | 
|---|
|  | 29 | { | 
|---|
| [3002] | 30 | fftIntfPtr_=new FFTPackServer(true);  // true -> on ne surecrit pas les input | 
|---|
| [2908] | 31 | fftIntfPtr_->setNormalize(false); | 
|---|
|  | 32 | } | 
|---|
|  | 33 |  | 
|---|
|  | 34 | // matrice de Toplitz hermitienne. La premiere ligne est conjuguee de la | 
|---|
|  | 35 | // premeire colonne donnee (vecteur firstCol) | 
|---|
|  | 36 | Toeplitz::Toeplitz(const TVector<complex<double> >& firstCol) : hermitian_(false) | 
|---|
|  | 37 | { | 
|---|
| [3002] | 38 | fftIntfPtr_=new FFTPackServer(true); // true -> on ne surecrit pas les input | 
|---|
| [2908] | 39 | fftIntfPtr_->setNormalize(false); | 
|---|
|  | 40 | setMatrix(firstCol); | 
|---|
|  | 41 | } | 
|---|
|  | 42 |  | 
|---|
|  | 43 |  | 
|---|
|  | 44 | Toeplitz::~Toeplitz(){ if (fftIntfPtr_!=NULL) delete fftIntfPtr_;}; | 
|---|
|  | 45 |  | 
|---|
|  | 46 |  | 
|---|
|  | 47 | // initialise matrice de Toeplitz hermitienne | 
|---|
|  | 48 | void  Toeplitz::setMatrix(const TVector<complex<double> >& firstCol) | 
|---|
|  | 49 | { | 
|---|
|  | 50 | hermitian_ = true; | 
|---|
|  | 51 | dimTop_ = firstCol.Size(); | 
|---|
|  | 52 | extensionACirculanteDyadique(firstCol); | 
|---|
|  | 53 | //    transformeeFourier(vecteurCirculant_, CirculanteFourier_); | 
|---|
|  | 54 | } | 
|---|
|  | 55 |  | 
|---|
|  | 56 | // intialise la matrice de Toeplitz generale complexe | 
|---|
|  | 57 | void  Toeplitz::setMatrix(const TVector<complex<double> >& firstCol, const TVector<complex<double> >& firstRow) | 
|---|
|  | 58 | { | 
|---|
|  | 59 | hermitian_ = false; | 
|---|
|  | 60 | dimTop_ = firstCol.Size(); | 
|---|
|  | 61 | if (dimTop_ != firstRow.Size()) | 
|---|
|  | 62 | { | 
|---|
|  | 63 | cout << " Toeplitz::setMatrix : nb col different de nbLignes " << endl; | 
|---|
|  | 64 | } | 
|---|
|  | 65 | extensionACirculanteDyadique(firstCol, firstRow); | 
|---|
|  | 66 | //    transformeeFourier(vecteurCirculant_, CirculanteFourier_); | 
|---|
|  | 67 | } | 
|---|
|  | 68 |  | 
|---|
|  | 69 |  | 
|---|
|  | 70 | // intialise la matrice de Toeplitz generale reelle | 
|---|
|  | 71 | void  Toeplitz::setMatrix(const TVector<double>& firstCol, const TVector<double>& firstRow) | 
|---|
|  | 72 | { | 
|---|
|  | 73 | hermitian_ = false; | 
|---|
|  | 74 | dimTop_ = firstCol.Size(); | 
|---|
|  | 75 | if (dimTop_ != firstRow.Size()) | 
|---|
|  | 76 | { | 
|---|
|  | 77 | cout << " Toeplitz::setMatrix : nb col different de nbLignes " << endl; | 
|---|
|  | 78 | } | 
|---|
|  | 79 | extensionACirculanteDyadique(firstCol, firstRow); | 
|---|
|  | 80 | //    transformeeFourier(vecteurCirculantD_, CirculanteFourier_); | 
|---|
|  | 81 | } | 
|---|
|  | 82 |  | 
|---|
|  | 83 | // intialise la matrice de Toeplitz generale reelle symetrique | 
|---|
|  | 84 | void  Toeplitz::setMatrix(const TVector<double>& firstCol) | 
|---|
|  | 85 | { | 
|---|
|  | 86 | hermitian_ = true; | 
|---|
|  | 87 | dimTop_ = firstCol.Size(); | 
|---|
|  | 88 | extensionACirculanteDyadique(firstCol); | 
|---|
|  | 89 | //    transformeeFourier(vecteurCirculantD_, CirculanteFourier_); | 
|---|
|  | 90 | } | 
|---|
|  | 91 |  | 
|---|
|  | 92 |  | 
|---|
|  | 93 | // resolution par gradient conjugue d'un systeme dont la matrice est | 
|---|
|  | 94 | // la matrice de Toeplitz | 
|---|
|  | 95 | // (le vecteur b est le second membre) | 
|---|
|  | 96 | // renvoie le nombre d'iterations (-1 si non convergence) | 
|---|
|  | 97 | int Toeplitz::gradientToeplitz(TVector<complex<double> >& b) const | 
|---|
|  | 98 | { | 
|---|
|  | 99 | // PrtTim(" entree gradient " ); | 
|---|
|  | 100 | int k; | 
|---|
|  | 101 | if (dimTop_ != b.Size()) | 
|---|
|  | 102 | { | 
|---|
|  | 103 | throw SzMismatchError("TOEPLITZMATRIX : gradientToeplitz : RHS dimensions mismatch \n"); | 
|---|
|  | 104 | } | 
|---|
|  | 105 | if (!hermitian_) | 
|---|
|  | 106 | { | 
|---|
|  | 107 | throw PException(" TOEPLITZMATRIX : conjugate gradient for hermitian matrix only\n"); | 
|---|
|  | 108 | } | 
|---|
|  | 109 | int ndyad = vecteurCirculant_.Size(); | 
|---|
|  | 110 |  | 
|---|
|  | 111 | TVector<complex<double> > a(dimTop_); | 
|---|
|  | 112 | TVector<complex<double> > r(dimTop_); | 
|---|
|  | 113 | TVector<complex<double> > q(ndyad); | 
|---|
|  | 114 |  | 
|---|
|  | 115 | a =complex<double>(0.,0.); | 
|---|
|  | 116 |  | 
|---|
|  | 117 | r = b; | 
|---|
|  | 118 | q =complex<double>(0.,0.); | 
|---|
|  | 119 | // int dimTopM1 = dimTop_-1; | 
|---|
|  | 120 | for (k=0; k<dimTop_; k++) q(k) = r(k); | 
|---|
|  | 121 | //q(Range(0,dimTopM1)) = r; | 
|---|
|  | 122 | //q.Print(cout, q.Size()); | 
|---|
|  | 123 |  | 
|---|
|  | 124 | double norm2Rk = prodScalaire(dimTop_,r,r).real(); | 
|---|
|  | 125 | double norm2R0 = norm2Rk; | 
|---|
|  | 126 | int niter =0; | 
|---|
|  | 127 | do | 
|---|
|  | 128 | { | 
|---|
|  | 129 | int k; | 
|---|
|  | 130 | TVector<complex<double> > Aq; | 
|---|
|  | 131 | produitParVecFourier(q, Aq); | 
|---|
|  | 132 | // | 
|---|
|  | 133 | double  norme2Aq = prodScalaire(dimTop_,Aq,q).real(); | 
|---|
|  | 134 | complex<double> alpha = prodScalaire(dimTop_,r,r)/norme2Aq; | 
|---|
|  | 135 | for (k=0; k<dimTop_; k++) a(k) += alpha*q(k); | 
|---|
|  | 136 | for (k=0; k<dimTop_; k++) r(k) -= alpha*Aq(k); | 
|---|
|  | 137 | double norm2Rkp1 = prodScalaire(dimTop_,r,r).real(); | 
|---|
|  | 138 | double beta = norm2Rkp1/norm2Rk; | 
|---|
|  | 139 | for (k=0; k<dimTop_; k++) q(k) = r(k) + beta*q(k); | 
|---|
|  | 140 |  | 
|---|
|  | 141 | norm2Rk = norm2Rkp1; | 
|---|
|  | 142 |  | 
|---|
|  | 143 | //  cout << " iteration " << niter+1 << " norme residu= " << norm2Rk << " critere arret " << norm2Rk/norm2R0 << endl; | 
|---|
|  | 144 | niter++; | 
|---|
|  | 145 | } | 
|---|
|  | 146 |  | 
|---|
|  | 147 | while (norm2Rk/norm2R0 > 1.e-12 && niter < 2*dimTop_); | 
|---|
|  | 148 |  | 
|---|
|  | 149 | //   PrtTim(" fin du gradient "); | 
|---|
|  | 150 |  | 
|---|
|  | 151 | // verification a supprimer | 
|---|
|  | 152 | //TMatrix<complex<double> > ttest(dimTop_, dimTop_); | 
|---|
|  | 153 | //expliciteToeplitz(ttest); | 
|---|
|  | 154 | //TVector<complex<double> > btest; | 
|---|
|  | 155 | //ttest.Show(); | 
|---|
|  | 156 | //cout << " long vec circ " << dimTop_ << " long a " << a.Size() << endl; | 
|---|
|  | 157 | //btest = ttest*a; | 
|---|
|  | 158 | //for (int k=0; k< b.Size(); k++) | 
|---|
|  | 159 | // { | 
|---|
|  | 160 | //  cout << " verif gradient donnee " << b(k) << " retrouve " << btest(k) << endl; | 
|---|
|  | 161 | //  } | 
|---|
|  | 162 | // | 
|---|
|  | 163 | b = a; | 
|---|
|  | 164 |  | 
|---|
|  | 165 | if (niter == 2*dimTop_-1) niter = -1; | 
|---|
|  | 166 | return niter; | 
|---|
|  | 167 | } | 
|---|
|  | 168 |  | 
|---|
|  | 169 |  | 
|---|
|  | 170 | // resolution par gradient conjugue d'un systeme dont la matrice est | 
|---|
|  | 171 | // la matrice de Toeplitz | 
|---|
|  | 172 | // (le vecteur b est le second membre) | 
|---|
|  | 173 | // renvoie le nombre d'iterations (-1 si non converge | 
|---|
|  | 174 | int Toeplitz::gradientToeplitz(TVector<double>& b) const | 
|---|
|  | 175 | { | 
|---|
|  | 176 | int k; | 
|---|
|  | 177 | if (dimTop_ != b.Size()) | 
|---|
|  | 178 | { | 
|---|
|  | 179 | throw SzMismatchError("TOEPLITZMATRIX : gradientToeplitz : RHS dimensions mismatch \n"); | 
|---|
|  | 180 | } | 
|---|
|  | 181 | if (!hermitian_) | 
|---|
|  | 182 | { | 
|---|
|  | 183 | throw PException(" TOEPLITZMATRIX : conjugate gradient for hermitian matrix only\n"); | 
|---|
|  | 184 | } | 
|---|
|  | 185 |  | 
|---|
|  | 186 | int ndyad = vecteurCirculantD_.Size(); | 
|---|
|  | 187 |  | 
|---|
|  | 188 | TVector<double> a(dimTop_); | 
|---|
|  | 189 | TVector<double> r(dimTop_); | 
|---|
|  | 190 | TVector<double> q(ndyad); | 
|---|
|  | 191 |  | 
|---|
|  | 192 | a =0.; | 
|---|
|  | 193 |  | 
|---|
|  | 194 | r = b; | 
|---|
|  | 195 | q =0.; | 
|---|
|  | 196 | for (k=0; k<dimTop_; k++) q(k) = r(k); | 
|---|
|  | 197 |  | 
|---|
|  | 198 |  | 
|---|
|  | 199 | double norm2Rk = prodScalaire(dimTop_,r,r); | 
|---|
|  | 200 | double norm2R0 = norm2Rk; | 
|---|
|  | 201 | int niter =0; | 
|---|
|  | 202 | do | 
|---|
|  | 203 | { | 
|---|
|  | 204 | int k; | 
|---|
|  | 205 | TVector<double> Aq; | 
|---|
|  | 206 | produitParVecFourier(q, Aq); | 
|---|
|  | 207 |  | 
|---|
|  | 208 | // | 
|---|
|  | 209 | double  norme2Aq = prodScalaire(dimTop_,Aq,q); | 
|---|
|  | 210 | double alpha = prodScalaire(dimTop_,r,r)/norme2Aq; | 
|---|
|  | 211 | for (k=0; k<dimTop_; k++) | 
|---|
|  | 212 | { | 
|---|
|  | 213 | a(k) += alpha*q(k); | 
|---|
|  | 214 | r(k) -= alpha*Aq(k); | 
|---|
|  | 215 | } | 
|---|
|  | 216 | double norm2Rkp1 = prodScalaire(dimTop_,r,r); | 
|---|
|  | 217 | double beta = norm2Rkp1/norm2Rk; | 
|---|
|  | 218 | for (k=0; k<dimTop_; k++) q(k) = r(k) + beta*q(k); | 
|---|
|  | 219 |  | 
|---|
|  | 220 | norm2Rk = norm2Rkp1; | 
|---|
|  | 221 |  | 
|---|
|  | 222 | //  cout << " iteration " << niter+1 << " norme residu= " << norm2Rk << " critere arret " << norm2Rk/norm2R0 << endl; | 
|---|
|  | 223 | niter++; | 
|---|
|  | 224 | } | 
|---|
|  | 225 |  | 
|---|
|  | 226 | while (norm2Rk/norm2R0 > 1.e-12 && niter < 2*dimTop_); | 
|---|
|  | 227 | //   while (norm2Rk/norm2R0 > 1.e-12 && niter < 1); | 
|---|
|  | 228 |  | 
|---|
|  | 229 |  | 
|---|
|  | 230 |  | 
|---|
|  | 231 | // verification a supprimer | 
|---|
|  | 232 | //TMatrix<double> ttest(dimTop_, dimTop_); | 
|---|
|  | 233 | //expliciteToeplitz(ttest); | 
|---|
|  | 234 | // TVector<double> btest; | 
|---|
|  | 235 | // ttest.Show(); | 
|---|
|  | 236 | //cout << " long vec circ " << dimTop_ << " long a " << a.Size() << endl; | 
|---|
|  | 237 | // btest = ttest*a; | 
|---|
|  | 238 | // for (int k=0; k< b.Size(); k++) | 
|---|
|  | 239 | //  { | 
|---|
|  | 240 | //    cout << " verif gradient donnee " << b(k) << " retrouve " << btest(k) << endl; | 
|---|
|  | 241 | //   } | 
|---|
|  | 242 | // | 
|---|
|  | 243 | b = a; | 
|---|
|  | 244 |  | 
|---|
|  | 245 | if (niter == 2*dimTop_-1) niter = -1; | 
|---|
|  | 246 | return niter; | 
|---|
|  | 247 | } | 
|---|
|  | 248 | // resolution par Gradient Conjugue Generalise d'un systeme dont | 
|---|
|  | 249 | // la matrice est la matrice de Toeplitz | 
|---|
|  | 250 | // (le vecteur b est le second membre) | 
|---|
|  | 251 | // renvoie le nombre d'iterations (-1 si non converge | 
|---|
|  | 252 | int Toeplitz::CGSToeplitz(TVector<double>& b) const | 
|---|
|  | 253 | { | 
|---|
|  | 254 | int k; | 
|---|
|  | 255 | if (dimTop_ != b.Size()) | 
|---|
|  | 256 | { | 
|---|
|  | 257 | throw SzMismatchError("TOEPLITZMATRIX : gradientToeplitz : RHS dimensions mismatch \n"); | 
|---|
|  | 258 | } | 
|---|
|  | 259 |  | 
|---|
|  | 260 | int ndyad = vecteurCirculantD_.Size(); | 
|---|
|  | 261 |  | 
|---|
|  | 262 | TVector<double> x(dimTop_); | 
|---|
|  | 263 | TVector<double> r(dimTop_); | 
|---|
|  | 264 | TVector<double> r0(dimTop_); | 
|---|
|  | 265 | TVector<double> q(ndyad); | 
|---|
|  | 266 | TVector<double> p(ndyad); | 
|---|
|  | 267 |  | 
|---|
|  | 268 | // initialisations | 
|---|
|  | 269 | x =0.; | 
|---|
|  | 270 | r0 = b; | 
|---|
|  | 271 | r = r0; | 
|---|
|  | 272 | q =0.; | 
|---|
|  | 273 | p =0.; | 
|---|
|  | 274 | for (k=0; k<dimTop_; k++) | 
|---|
|  | 275 | { | 
|---|
|  | 276 | q(k) = r(k); | 
|---|
|  | 277 | p(k) = r(k); | 
|---|
|  | 278 | } | 
|---|
|  | 279 | int niter =0; | 
|---|
|  | 280 | double r0rk = prodScalaire(dimTop_,r0,r); | 
|---|
|  | 281 | double r0r0 = r0rk; | 
|---|
|  | 282 | double rkrk = r0r0; | 
|---|
|  | 283 | double r0rkp1 = r0rk; | 
|---|
|  | 284 | do | 
|---|
|  | 285 | { | 
|---|
|  | 286 | int k; | 
|---|
|  | 287 | TVector<double> Aq; | 
|---|
|  | 288 | produitParVecFourier(q, Aq); | 
|---|
|  | 289 | TVector<double> Ap; | 
|---|
|  | 290 | produitParVecFourier(p, Ap); | 
|---|
|  | 291 |  | 
|---|
|  | 292 | TVector<double> pMalphaAq(ndyad); | 
|---|
|  | 293 |  | 
|---|
|  | 294 |  | 
|---|
|  | 295 | // | 
|---|
|  | 296 |  | 
|---|
|  | 297 | double  r0Aqk = prodScalaire(dimTop_,r0,Aq); | 
|---|
|  | 298 | double alpha = r0rk/r0Aqk; | 
|---|
|  | 299 | pMalphaAq = p-alpha*Aq; | 
|---|
|  | 300 | TVector<double> aux(ndyad); | 
|---|
|  | 301 | aux = p + pMalphaAq; | 
|---|
|  | 302 | TVector<double> Aaux; | 
|---|
|  | 303 | produitParVecFourier(aux, Aaux); | 
|---|
|  | 304 |  | 
|---|
|  | 305 | for (k=0; k<dimTop_; k++) | 
|---|
|  | 306 | { | 
|---|
|  | 307 | x(k) += alpha*aux(k); | 
|---|
|  | 308 | r(k) -= alpha*Aaux(k); | 
|---|
|  | 309 | } | 
|---|
|  | 310 | r0rkp1 = prodScalaire(dimTop_,r0,r); | 
|---|
|  | 311 | double beta = r0rkp1/r0rk; | 
|---|
|  | 312 | for (k=0; k<dimTop_; k++) p(k) = r(k) + beta*pMalphaAq(k); | 
|---|
|  | 313 | for (k=0; k<dimTop_; k++) | 
|---|
|  | 314 | { | 
|---|
|  | 315 | q(k)  = p(k) + beta*(beta*q(k) + pMalphaAq(k)); | 
|---|
|  | 316 | } | 
|---|
|  | 317 | r0rk  = r0rkp1; | 
|---|
|  | 318 | rkrk = prodScalaire(dimTop_,r,r); | 
|---|
|  | 319 | //    cout << " iteration " << niter+1 << " rkrk= " << rkrk << " critere arret " << rkrk/r0r0 << endl; | 
|---|
|  | 320 | niter++; | 
|---|
|  | 321 | } | 
|---|
|  | 322 |  | 
|---|
|  | 323 | while (rkrk/r0r0 > 1.e-12 && niter < 2*dimTop_); | 
|---|
|  | 324 | // while ( niter < 10*dimTop_); | 
|---|
|  | 325 |  | 
|---|
|  | 326 | // verification a supprimer | 
|---|
|  | 327 | //TMatrix<double> ttest(dimTop_, dimTop_); | 
|---|
|  | 328 | //expliciteToeplitz(ttest); | 
|---|
|  | 329 | //TVector<double> btest; | 
|---|
|  | 330 | // ttest.Show(); | 
|---|
|  | 331 | //cout << " long vec circ " << dimTop_ << " long a " << a.Size() << endl; | 
|---|
|  | 332 | //btest = ttest*x; | 
|---|
|  | 333 | // for (int k=0; k< b.Size(); k++) | 
|---|
|  | 334 | // { | 
|---|
|  | 335 | //    cout << " verif gradient donnee " << b(k) << " retrouve " << btest(k) << endl; | 
|---|
|  | 336 | //  } | 
|---|
|  | 337 | // | 
|---|
|  | 338 | b = x; | 
|---|
|  | 339 |  | 
|---|
|  | 340 | if (niter == 2*dimTop_) niter = -1; | 
|---|
|  | 341 | return niter; | 
|---|
|  | 342 | } | 
|---|
|  | 343 | // resolution par Double Gradient Conjugue d'un systeme dont | 
|---|
|  | 344 | // la matrice est la matrice de Toeplitz | 
|---|
|  | 345 | // (le vecteur b est le second membre) | 
|---|
|  | 346 | // renvoie le nombre d'iterations (-1 si non converge | 
|---|
|  | 347 | int Toeplitz::DCGToeplitz(TVector<double>& b) | 
|---|
|  | 348 | { | 
|---|
|  | 349 | int k; | 
|---|
|  | 350 | if (dimTop_ != b.Size()) | 
|---|
|  | 351 | { | 
|---|
|  | 352 | throw SzMismatchError("TOEPLITZMATRIX : gradientToeplitz : RHS dimensions mismatch \n"); | 
|---|
|  | 353 | } | 
|---|
|  | 354 |  | 
|---|
|  | 355 | // on a besoin des produit (matrice trasposee)Xvectteur | 
|---|
|  | 356 | // on initialise eventuellement la TF de Toeplitz transposee | 
|---|
|  | 357 | if (CirculanteTransposeeFourier_.Size() == 0) initTFTransposeeToeplitzReelle(); | 
|---|
|  | 358 |  | 
|---|
|  | 359 |  | 
|---|
|  | 360 | int ndyad = vecteurCirculantD_.Size(); | 
|---|
|  | 361 |  | 
|---|
|  | 362 | TVector<double> x(dimTop_); | 
|---|
|  | 363 | TVector<double> r1(dimTop_); | 
|---|
|  | 364 | TVector<double> r2(dimTop_); | 
|---|
|  | 365 | TVector<double> d1(ndyad); | 
|---|
|  | 366 | TVector<double> d2(ndyad); | 
|---|
|  | 367 |  | 
|---|
|  | 368 | // initialisations | 
|---|
|  | 369 | x =0.; | 
|---|
|  | 370 | r1 = b; | 
|---|
|  | 371 | r2 = r1; | 
|---|
|  | 372 | d1 =0.; | 
|---|
|  | 373 | d2 =0.; | 
|---|
|  | 374 | for (k=0; k<dimTop_; k++) | 
|---|
|  | 375 | { | 
|---|
|  | 376 | d1(k) = r1(k); | 
|---|
|  | 377 | d2(k) = r2(k); | 
|---|
|  | 378 | } | 
|---|
|  | 379 | int niter =0; | 
|---|
|  | 380 | double r1r2k = prodScalaire(dimTop_,r1,r2); | 
|---|
|  | 381 | double r1r2ini = r1r2k; | 
|---|
|  | 382 | do | 
|---|
|  | 383 | { | 
|---|
|  | 384 | int k; | 
|---|
|  | 385 | TVector<double> Ad1; | 
|---|
|  | 386 | produitParVecFourier(d1, Ad1); | 
|---|
|  | 387 | TVector<double> tAd2; | 
|---|
|  | 388 | produitTransposeeParVecFourier(d2, tAd2); | 
|---|
|  | 389 |  | 
|---|
|  | 390 | // | 
|---|
|  | 391 | double  d2Ad1 = prodScalaire(dimTop_,d2,Ad1); | 
|---|
|  | 392 | double alpha = r1r2k/d2Ad1; | 
|---|
|  | 393 |  | 
|---|
|  | 394 | for (k=0; k<dimTop_; k++) | 
|---|
|  | 395 | { | 
|---|
|  | 396 | x(k) += alpha*d1(k); | 
|---|
|  | 397 | r1(k) -= alpha*Ad1(k); | 
|---|
|  | 398 | r2(k) -= alpha*tAd2(k); | 
|---|
|  | 399 | } | 
|---|
|  | 400 | prodScalaire(dimTop_,r1,r2); | 
|---|
|  | 401 | double beta = prodScalaire(dimTop_,r1,r2)/r1r2k; | 
|---|
|  | 402 | for (k=0; k<dimTop_; k++) | 
|---|
|  | 403 | { | 
|---|
|  | 404 | d1(k)  = r1(k) + beta*d1(k); | 
|---|
|  | 405 | d2(k)  = r2(k) + beta*d2(k); | 
|---|
|  | 406 | } | 
|---|
|  | 407 | r1r2k = prodScalaire(dimTop_,r1,r2); | 
|---|
|  | 408 | //      cout << " iteration " << niter+1 << " r1r2= " << r1r2k << " critere arret " << r1r2k/r1r2ini << endl; | 
|---|
|  | 409 | niter++; | 
|---|
|  | 410 | } | 
|---|
|  | 411 |  | 
|---|
|  | 412 | while ( fabs(r1r2k/r1r2ini) > 1.e-12 && niter < 2*dimTop_); | 
|---|
|  | 413 |  | 
|---|
|  | 414 | // verification a supprimer | 
|---|
|  | 415 | //TMatrix<double> ttest(dimTop_, dimTop_); | 
|---|
|  | 416 | //expliciteToeplitz(ttest); | 
|---|
|  | 417 | //TVector<double> btest; | 
|---|
|  | 418 | // ttest.Show(); | 
|---|
|  | 419 | //cout << " long vec circ " << dimTop_ << " long a " << a.Size() << endl; | 
|---|
|  | 420 | //btest = ttest*x; | 
|---|
|  | 421 | //for (int k=0; k< b.Size(); k++) | 
|---|
|  | 422 | // { | 
|---|
|  | 423 | //    cout << " verif gradient donnee " << b(k) << " retrouve " << btest(k) << endl; | 
|---|
|  | 424 | //   } | 
|---|
|  | 425 | // | 
|---|
|  | 426 | b = x; | 
|---|
|  | 427 |  | 
|---|
|  | 428 | if (niter == 2*dimTop_) niter = -1; | 
|---|
|  | 429 | return niter; | 
|---|
|  | 430 | } | 
|---|
|  | 431 |  | 
|---|
|  | 432 |  | 
|---|
|  | 433 | void  Toeplitz::expliciteCirculante(TMatrix<complex<double> >& m) const | 
|---|
|  | 434 | { | 
|---|
|  | 435 | int k; | 
|---|
|  | 436 | int n= vecteurCirculant_.Size(); | 
|---|
|  | 437 | m.ReSize(n,n); | 
|---|
|  | 438 | int decal=0; | 
|---|
|  | 439 | for (k=0; k<n; k++) | 
|---|
|  | 440 | { | 
|---|
|  | 441 | int j; | 
|---|
|  | 442 | int courant = 0; | 
|---|
|  | 443 | for (j=decal; j< n; j++) | 
|---|
|  | 444 | { | 
|---|
|  | 445 | m(j, k) = vecteurCirculant_(courant++); | 
|---|
|  | 446 | } | 
|---|
|  | 447 | for (j=0; j<decal; j++) | 
|---|
|  | 448 | { | 
|---|
|  | 449 | m(j,k)= vecteurCirculant_(courant++); | 
|---|
|  | 450 | } | 
|---|
|  | 451 | decal++; | 
|---|
|  | 452 | } | 
|---|
|  | 453 | } | 
|---|
|  | 454 | void  Toeplitz::expliciteToeplitz(TMatrix<complex<double> >& m) const | 
|---|
|  | 455 | { | 
|---|
|  | 456 | int k; | 
|---|
|  | 457 | int ndyad = vecteurCirculant_.Size(); | 
|---|
|  | 458 | m.ReSize(dimTop_, dimTop_); | 
|---|
|  | 459 | for (k=0; k<dimTop_; k++) | 
|---|
|  | 460 | { | 
|---|
|  | 461 | int j; | 
|---|
|  | 462 | for (j=k; j< dimTop_; j++) | 
|---|
|  | 463 | { | 
|---|
|  | 464 | m(j, k) = vecteurCirculant_(j-k); | 
|---|
|  | 465 | } | 
|---|
|  | 466 | for (j=0; j<k; j++) | 
|---|
|  | 467 | { | 
|---|
|  | 468 | m(k-j-1,k)= vecteurCirculant_(ndyad-j-1); | 
|---|
|  | 469 | } | 
|---|
|  | 470 | } | 
|---|
|  | 471 | } | 
|---|
|  | 472 | void  Toeplitz::expliciteToeplitz(TMatrix<double>& m) const | 
|---|
|  | 473 | { | 
|---|
|  | 474 | int k; | 
|---|
|  | 475 | int ndyad = vecteurCirculantD_.Size(); | 
|---|
|  | 476 | m.ReSize(dimTop_, dimTop_); | 
|---|
|  | 477 | for (k=0; k<dimTop_; k++) | 
|---|
|  | 478 | { | 
|---|
|  | 479 | int j; | 
|---|
|  | 480 | for (j=k; j< dimTop_; j++) | 
|---|
|  | 481 | { | 
|---|
|  | 482 | m(j, k) = vecteurCirculantD_(j-k); | 
|---|
|  | 483 | } | 
|---|
|  | 484 | for (j=0; j < k; j++) | 
|---|
|  | 485 | { | 
|---|
|  | 486 | m(k-j-1,k)= vecteurCirculantD_(ndyad-j-1); | 
|---|
|  | 487 | } | 
|---|
|  | 488 | } | 
|---|
|  | 489 | } | 
|---|
|  | 490 |  | 
|---|
|  | 491 |  | 
|---|
|  | 492 | // resolution par gradient conjugue preconditionne d'un systeme | 
|---|
|  | 493 | // dont la matrice est  la matrice de Toeplitz | 
|---|
|  | 494 | // (le vecteur b est le second membre) | 
|---|
|  | 495 | // renvoie le nombre d'iterations (-1 si non converge | 
|---|
|  | 496 | int Toeplitz::gradientToeplitzPreconTChang(TVector<complex<double> >& b) const | 
|---|
|  | 497 | { | 
|---|
|  | 498 | int k; | 
|---|
|  | 499 | if (dimTop_ != b.Size()) | 
|---|
|  | 500 | { | 
|---|
|  | 501 | throw SzMismatchError("TOEPLITZMATRIX : gradientToeplitz : RHS dimensions mismatch \n"); | 
|---|
|  | 502 | } | 
|---|
|  | 503 | if (!hermitian_) | 
|---|
|  | 504 | { | 
|---|
|  | 505 | throw PException(" TOEPLITZMATRIX : conjugate gradient for hermitian matrix only\n"); | 
|---|
|  | 506 | } | 
|---|
|  | 507 | int ndyad = vecteurCirculant_.Size(); | 
|---|
|  | 508 | // preconditionneur (en Fourier) | 
|---|
|  | 509 | TVector<complex<double> > TFourierC; | 
|---|
|  | 510 | fabricationTChangPreconHerm(TFourierC); | 
|---|
|  | 511 |  | 
|---|
|  | 512 |  | 
|---|
|  | 513 | TVector<complex<double> > a(dimTop_); | 
|---|
|  | 514 | TVector<complex<double> > r(dimTop_); | 
|---|
|  | 515 | TVector<complex<double> > z(dimTop_); | 
|---|
|  | 516 | TVector<complex<double> > q(ndyad); | 
|---|
|  | 517 |  | 
|---|
|  | 518 | a =complex<double>(0.,0.); | 
|---|
|  | 519 | r = b; | 
|---|
|  | 520 | inverseSystemeCirculantFourier(TFourierC, r, z); | 
|---|
|  | 521 | q =complex<double>(0.,0.); | 
|---|
|  | 522 | for (k=0; k<dimTop_; k++) q(k) = z(k); | 
|---|
|  | 523 |  | 
|---|
|  | 524 | complex<double> norm2zRk = prodScalaire(dimTop_,z,r); | 
|---|
|  | 525 | complex<double> norm2zR0 = norm2zRk; | 
|---|
|  | 526 | int niter =0; | 
|---|
|  | 527 | double critere = 1.e+8; | 
|---|
|  | 528 | do | 
|---|
|  | 529 | { | 
|---|
|  | 530 | int k; | 
|---|
|  | 531 | TVector<complex<double> > Aq; | 
|---|
|  | 532 | produitParVecFourier(q, Aq); | 
|---|
|  | 533 |  | 
|---|
|  | 534 | // matrice hermitienne | 
|---|
|  | 535 | double  norme2Aq = prodScalaire(dimTop_,Aq,q).real(); | 
|---|
|  | 536 | complex<double> alpha = norm2zRk/norme2Aq; | 
|---|
|  | 537 | for (k=0; k<dimTop_; k++) a(k) += alpha*q(k); | 
|---|
|  | 538 | for (k=0; k<dimTop_; k++) r(k) -= alpha*Aq(k); | 
|---|
|  | 539 |  | 
|---|
|  | 540 | inverseSystemeCirculantFourier(TFourierC, r, z); | 
|---|
|  | 541 |  | 
|---|
|  | 542 | complex<double> norm2zRkp1 = prodScalaire(dimTop_,z,r); | 
|---|
|  | 543 | complex<double> beta = norm2zRkp1/norm2zRk; | 
|---|
|  | 544 | for (k=0; k<dimTop_; k++) q(k) = z(k) + beta*q(k); | 
|---|
|  | 545 |  | 
|---|
|  | 546 | norm2zRk = norm2zRkp1; | 
|---|
|  | 547 | complex<double> rapport = norm2zRk/norm2zR0; | 
|---|
|  | 548 | critere = fabs(rapport.real())+fabs(rapport.imag()); | 
|---|
|  | 549 | //   cout << " norme residu= " << norm2zRk << " critere arret " << critere << endl; | 
|---|
|  | 550 | niter++; | 
|---|
|  | 551 | } | 
|---|
|  | 552 |  | 
|---|
|  | 553 | while ( critere > 1.e-12 && niter < 2*dimTop_); | 
|---|
|  | 554 | b = a; | 
|---|
|  | 555 | if (niter == 2*dimTop_-1) niter = -1; | 
|---|
|  | 556 | return niter; | 
|---|
|  | 557 | } | 
|---|
|  | 558 |  | 
|---|
|  | 559 | // extension de la matrice Toeplitz a une matrice circulante, avec zeros | 
|---|
|  | 560 | // additionnels : pour pouvoir utiliser les FFT | 
|---|
|  | 561 | void Toeplitz::extensionACirculanteDyadique(const TVector<complex<double> >& firstCol, const TVector<complex<double> >& firstRow) | 
|---|
|  | 562 | { | 
|---|
|  | 563 | // la matrice de Toeplitz doit etre etendue a (2N-1)x(2N_1) pour | 
|---|
|  | 564 | // la rendre circulante | 
|---|
|  | 565 |  | 
|---|
|  | 566 | int k; | 
|---|
|  | 567 | int N = dimTop_; | 
|---|
|  | 568 | int nEtendu = 2*N-1; | 
|---|
|  | 569 | // on va completer avec des zeros, pour avoir une puissance de 2 (pour fft) | 
|---|
|  | 570 | int ndyad=1; | 
|---|
|  | 571 | while ( nEtendu >  ndyad) | 
|---|
|  | 572 | { | 
|---|
|  | 573 | ndyad *=2; | 
|---|
|  | 574 | } | 
|---|
|  | 575 | //  cout << " extensionToeplitz: taille etendue de la matrice " << nEtendu << endl; | 
|---|
|  | 576 | //  cout << " puissance de 2 trouvee " << ndyad << endl; | 
|---|
|  | 577 |  | 
|---|
|  | 578 |  | 
|---|
|  | 579 | // extension de la matrice | 
|---|
|  | 580 | vecteurCirculant_.ReSize(ndyad); | 
|---|
|  | 581 | vecteurCirculant_ = complex<double>(0.,0.); | 
|---|
|  | 582 | vecteurCirculant_(0) = firstCol(0); | 
|---|
|  | 583 | for (k=1; k< N; k++) | 
|---|
|  | 584 | { | 
|---|
|  | 585 | vecteurCirculant_(k) = firstCol(k); | 
|---|
|  | 586 | vecteurCirculant_(ndyad-k) = firstRow(k); | 
|---|
|  | 587 | } | 
|---|
|  | 588 | transformeeFourier(vecteurCirculant_, CirculanteFourier_); | 
|---|
|  | 589 | } | 
|---|
|  | 590 |  | 
|---|
|  | 591 |  | 
|---|
|  | 592 | // extension de la matrice Toeplitz a une matrice circulante, avec zeros | 
|---|
|  | 593 | // additionnels : pour pouvoir utiliser les FFT | 
|---|
|  | 594 | void Toeplitz::extensionACirculanteDyadique(const TVector<complex<double> >& firstCol) | 
|---|
|  | 595 | { | 
|---|
|  | 596 | // la matrice de Toeplitz doit etre etendue a (2N-1)x(2N_1) pour | 
|---|
|  | 597 | // la rendre circulante | 
|---|
|  | 598 |  | 
|---|
|  | 599 | int k; | 
|---|
|  | 600 | int N = dimTop_; | 
|---|
|  | 601 | int nEtendu = 2*N-1; | 
|---|
|  | 602 | // on va completer avec des zeros, pour avoir une puissance de 2 (pour fft) | 
|---|
|  | 603 | int ndyad=1; | 
|---|
|  | 604 | while ( nEtendu >  ndyad) | 
|---|
|  | 605 | { | 
|---|
|  | 606 | ndyad *=2; | 
|---|
|  | 607 | } | 
|---|
|  | 608 | //  cout << " extensionToeplitz: taille etendue de la matrice " << nEtendu << endl; | 
|---|
|  | 609 | //  cout << " puissance de 2 trouvee " << ndyad << endl; | 
|---|
|  | 610 |  | 
|---|
|  | 611 |  | 
|---|
|  | 612 | // extension de la matrice | 
|---|
|  | 613 | vecteurCirculant_.ReSize(ndyad); | 
|---|
|  | 614 | vecteurCirculant_ = complex<double>(0.,0.); | 
|---|
|  | 615 | vecteurCirculant_(0) = firstCol(0); | 
|---|
|  | 616 | for (k=1; k< N; k++) | 
|---|
|  | 617 | { | 
|---|
|  | 618 | complex<double> aux = firstCol(k); | 
|---|
|  | 619 | vecteurCirculant_(k) = aux; | 
|---|
|  | 620 | vecteurCirculant_(ndyad-k) = conj(aux); | 
|---|
|  | 621 | } | 
|---|
|  | 622 | transformeeFourier(vecteurCirculant_, CirculanteFourier_); | 
|---|
|  | 623 | } | 
|---|
|  | 624 |  | 
|---|
|  | 625 |  | 
|---|
|  | 626 | // extension de la matrice Toeplitz a une matrice circulante, avec zeros | 
|---|
|  | 627 | // additionnels : pour pouvoir utiliser les FFT | 
|---|
|  | 628 | void Toeplitz::extensionACirculanteDyadique(const TVector<double>& firstCol, const TVector<double>& firstRow) | 
|---|
|  | 629 | { | 
|---|
|  | 630 | // la matrice de Toeplitz doit etre etendue a (2N-1)x(2N_1) pour | 
|---|
|  | 631 | // la rendre circulante | 
|---|
|  | 632 |  | 
|---|
|  | 633 | int k; | 
|---|
|  | 634 | int N = dimTop_; | 
|---|
|  | 635 | int nEtendu = 2*N-1; | 
|---|
|  | 636 | // on va completer avec des zeros, pour avoir une puissance de 2 (pour fft) | 
|---|
|  | 637 | int ndyad=1; | 
|---|
|  | 638 | while ( nEtendu >  ndyad) | 
|---|
|  | 639 | { | 
|---|
|  | 640 | ndyad *=2; | 
|---|
|  | 641 | } | 
|---|
|  | 642 | //  cout << " extensionToeplitz: taille etendue de la matrice " << nEtendu << endl; | 
|---|
|  | 643 | //  cout << " puissance de 2 trouvee " << ndyad << endl; | 
|---|
|  | 644 |  | 
|---|
|  | 645 |  | 
|---|
|  | 646 | // extension de la matrice | 
|---|
|  | 647 | vecteurCirculantD_.ReSize(ndyad); | 
|---|
|  | 648 | vecteurCirculantD_ = 0.; | 
|---|
|  | 649 |  | 
|---|
|  | 650 | vecteurCirculantD_(0) = firstCol(0); | 
|---|
|  | 651 | for (k=1; k< N; k++) | 
|---|
|  | 652 | { | 
|---|
|  | 653 | vecteurCirculantD_(k) = firstCol(k); | 
|---|
|  | 654 | vecteurCirculantD_(ndyad-k) = firstRow(k); | 
|---|
|  | 655 | } | 
|---|
|  | 656 | transformeeFourier(vecteurCirculantD_, CirculanteFourier_); | 
|---|
|  | 657 | } | 
|---|
|  | 658 |  | 
|---|
|  | 659 |  | 
|---|
|  | 660 |  | 
|---|
|  | 661 | // extension de la matrice Toeplitz a une matrice circulante, avec zeros | 
|---|
|  | 662 | // additionnels : pour pouvoir utiliser les FFT | 
|---|
|  | 663 | void Toeplitz::extensionACirculanteDyadique(const TVector<double>& firstCol) | 
|---|
|  | 664 | { | 
|---|
|  | 665 | // la matrice de Toeplitz doit etre etendue a (2N-1)x(2N_1) pour | 
|---|
|  | 666 | // la rendre circulante | 
|---|
|  | 667 |  | 
|---|
|  | 668 |  | 
|---|
|  | 669 | int k; | 
|---|
|  | 670 | int N = dimTop_; | 
|---|
|  | 671 | int nEtendu = 2*N-1; | 
|---|
|  | 672 | // on va completer avec des zeros, pour avoir une puissance de 2 (pour fft) | 
|---|
|  | 673 | int ndyad=1; | 
|---|
|  | 674 | while ( nEtendu >  ndyad) | 
|---|
|  | 675 | { | 
|---|
|  | 676 | ndyad *=2; | 
|---|
|  | 677 | } | 
|---|
|  | 678 | //  cout << " extensionToeplitz: taille etendue de la matrice " << nEtendu << endl; | 
|---|
|  | 679 | //  cout << " puissance de 2 trouvee " << ndyad << endl; | 
|---|
|  | 680 |  | 
|---|
|  | 681 |  | 
|---|
|  | 682 | // extension de la matrice | 
|---|
|  | 683 | vecteurCirculantD_.ReSize(ndyad); | 
|---|
|  | 684 | vecteurCirculantD_ = 0.; | 
|---|
|  | 685 | vecteurCirculantD_(0) = firstCol(0); | 
|---|
|  | 686 | for (k=1; k< N; k++) | 
|---|
|  | 687 | { | 
|---|
|  | 688 | double aux = firstCol(k); | 
|---|
|  | 689 | vecteurCirculantD_(k) = aux; | 
|---|
|  | 690 | vecteurCirculantD_(ndyad-k) = aux; | 
|---|
|  | 691 | } | 
|---|
|  | 692 | transformeeFourier(vecteurCirculantD_, CirculanteFourier_); | 
|---|
|  | 693 | } | 
|---|
|  | 694 |  | 
|---|
|  | 695 |  | 
|---|
|  | 696 | // produit de la matrice de toeplitz par un vecteur (par transformee de | 
|---|
|  | 697 | // fourier) resultat dans Tq | 
|---|
|  | 698 | void Toeplitz::produitParVecFourier(const TVector<complex<double> >& q, TVector<complex<double> >& Tq) const | 
|---|
|  | 699 | { | 
|---|
|  | 700 | int k; | 
|---|
|  | 701 | int n = q.Size(); | 
|---|
|  | 702 | if (n != vecteurCirculant_.Size() ) | 
|---|
|  | 703 | { | 
|---|
|  | 704 | cout << " produitToepVecFourier: dimensions matrice vecteur incompatibles " << endl; | 
|---|
|  | 705 | return; | 
|---|
|  | 706 | } | 
|---|
|  | 707 | TVector<complex<double> > qFourier; | 
|---|
|  | 708 | transformeeFourier(q, qFourier); | 
|---|
|  | 709 |  | 
|---|
|  | 710 |  | 
|---|
|  | 711 | TVector<complex<double> > produitFourier(n); | 
|---|
|  | 712 | for (k=0; k<qFourier.Size(); k++) | 
|---|
|  | 713 | { | 
|---|
|  | 714 | produitFourier(k) = CirculanteFourier_(k)*qFourier(k); | 
|---|
|  | 715 | } | 
|---|
|  | 716 | transformeeInverseFourier(produitFourier,Tq); | 
|---|
|  | 717 | } | 
|---|
|  | 718 |  | 
|---|
|  | 719 |  | 
|---|
|  | 720 |  | 
|---|
|  | 721 |  | 
|---|
|  | 722 | // produit de la matrice de toeplitz par un vecteur (par transformee de | 
|---|
|  | 723 | // fourier) resultat dans Tq | 
|---|
|  | 724 | void Toeplitz::produitParVecFourier(const TVector<double>& q, TVector<double>& Tq) const | 
|---|
|  | 725 | { | 
|---|
|  | 726 | int k; | 
|---|
|  | 727 | int n = q.Size(); | 
|---|
|  | 728 | if (n != vecteurCirculantD_.Size() ) | 
|---|
|  | 729 | { | 
|---|
|  | 730 | cout << " produitParVecFourier: dimensions matrice vecteur incompatibles " << "vecteur  " << n << " matrice "<< vecteurCirculantD_.Size()  << endl; | 
|---|
|  | 731 | return; | 
|---|
|  | 732 | } | 
|---|
|  | 733 | // cout << " le vecteur " << endl; | 
|---|
|  | 734 | TVector<complex<double> > qFourier; | 
|---|
|  | 735 | transformeeFourier(q, qFourier); | 
|---|
|  | 736 | int  M = qFourier.Size(); | 
|---|
|  | 737 | TVector<complex<double> > produitFourier(M); | 
|---|
|  | 738 | for (k=0; k< M; k++) | 
|---|
|  | 739 | { | 
|---|
|  | 740 | produitFourier(k) = CirculanteFourier_(k)*qFourier(k); | 
|---|
|  | 741 | } | 
|---|
|  | 742 | transformeeInverseFourier(produitFourier,Tq); | 
|---|
|  | 743 | } | 
|---|
|  | 744 | // produit de la transposee de matrice de toeplitz par un vecteur | 
|---|
|  | 745 | // (par transformee de fourier) resultat dans Tq | 
|---|
|  | 746 | void Toeplitz::produitTransposeeParVecFourier(const TVector<double>& q, TVector<double>& Tq) const | 
|---|
|  | 747 | { | 
|---|
|  | 748 | int k; | 
|---|
|  | 749 | int n = q.Size(); | 
|---|
|  | 750 | if (n != vecteurCirculantD_.Size() ) | 
|---|
|  | 751 | { | 
|---|
|  | 752 | cout << " produitTransposeeParVecFourier: dimensions matrice vecteur incompatibles " << "vecteur  " << n << " matrice "<< vecteurCirculantD_.Size()  << endl; | 
|---|
|  | 753 | return; | 
|---|
|  | 754 | } | 
|---|
|  | 755 | // cout << " le vecteur " << endl; | 
|---|
|  | 756 | TVector<complex<double> > qFourier; | 
|---|
|  | 757 | transformeeFourier(q, qFourier); | 
|---|
|  | 758 | int  M = qFourier.Size(); | 
|---|
|  | 759 | TVector<complex<double> > produitFourier(M); | 
|---|
|  | 760 | for (k=0; k< M; k++) | 
|---|
|  | 761 | { | 
|---|
|  | 762 | produitFourier(k) = CirculanteTransposeeFourier_(k)*qFourier(k); | 
|---|
|  | 763 | } | 
|---|
|  | 764 | transformeeInverseFourier(produitFourier,Tq); | 
|---|
|  | 765 | } | 
|---|
|  | 766 |  | 
|---|
|  | 767 |  | 
|---|
|  | 768 | void Toeplitz::fabricationTChangPreconHerm(TVector<complex<double> >& TFourierC) const | 
|---|
|  | 769 | { | 
|---|
|  | 770 | int k; | 
|---|
|  | 771 | TVector<complex<double> > C(dimTop_); | 
|---|
|  | 772 | C(0) = vecteurCirculant_(0); | 
|---|
|  | 773 | for (k=1; k<dimTop_; k++) | 
|---|
|  | 774 | { | 
|---|
|  | 775 | C(k) = (double(dimTop_-k)*vecteurCirculant_(k) + double(k)*conj(vecteurCirculant_(dimTop_-k)))/double(dimTop_); | 
|---|
|  | 776 | } | 
|---|
|  | 777 | transformeeFourier(C, TFourierC); | 
|---|
|  | 778 | // verifier qu'il n'y a pas de terme nul (pour pouvoir inverser) | 
|---|
|  | 779 | int index =0; | 
|---|
|  | 780 | for (k=0; k<dimTop_; k++) | 
|---|
|  | 781 | { | 
|---|
|  | 782 | if ( fabs(TFourierC(k).real())== 0. &&  fabs(TFourierC(k).imag()) == 0.) | 
|---|
|  | 783 | { | 
|---|
|  | 784 | index = 1; | 
|---|
|  | 785 | break; | 
|---|
|  | 786 | } | 
|---|
|  | 787 | } | 
|---|
|  | 788 | if (index == 1) | 
|---|
|  | 789 | { | 
|---|
|  | 790 | cout << " fabricationTChangPreconHerm  ; ATTENTION : un terme de la TF du preconditionneur est nul! " << endl; | 
|---|
|  | 791 | } | 
|---|
|  | 792 |  | 
|---|
|  | 793 | } | 
|---|
|  | 794 | void Toeplitz::inverseSystemeCirculantFourier(const TVector<complex<double> >& TFourierC, const TVector<complex<double> >& secondMembre, TVector<complex<double> >& resul) const | 
|---|
|  | 795 | { | 
|---|
|  | 796 | int k; | 
|---|
|  | 797 | int n = TFourierC.Size(); | 
|---|
|  | 798 | if (n != secondMembre.Size() ) | 
|---|
|  | 799 | { | 
|---|
|  | 800 | cout << " inverseSystemeCirculantFourier: dimensions matrice vecteur incompatibles " << endl; | 
|---|
|  | 801 | return; | 
|---|
|  | 802 | } | 
|---|
|  | 803 | TVector<complex<double> > SmFourier; | 
|---|
|  | 804 | transformeeFourier(secondMembre, SmFourier); | 
|---|
|  | 805 | TVector<complex<double> > quotientFourier(n); | 
|---|
|  | 806 | for (k=0; k<n; k++) | 
|---|
|  | 807 | { | 
|---|
|  | 808 | quotientFourier(k) = SmFourier(k)/TFourierC(k); | 
|---|
|  | 809 | } | 
|---|
|  | 810 | transformeeInverseFourier(quotientFourier,resul); | 
|---|
|  | 811 | } | 
|---|