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