[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 | }
|
---|