Changeset 3356 in Sophya for trunk/SophyaPI/DemoPIApp/fft2d.cc

Timestamp:

Oct 23, 2007, 12:17:19 PM (18 years ago)

Author:

ansari

Message:

adaptation fft2d.cc pour faire du r_4 ET r_8 (au choix defini par define TF) - Reza 23/10/2007

File:

• trunk/SophyaPI/DemoPIApp/fft2d.cc (modified) (3 diffs)

trunk/SophyaPI/DemoPIApp/fft2d.cc

@@ +1 @@
+#define TF r_8
+
 int nlig = 128; int ncol = 128;      
-TMatrix<r_8> in(nlig,ncol);          
+TMatrix<TF> in(nlig,ncol);          
 int l,c;                             
 for(l=40; l<80; l++)                 
   for(c=40; c<80; c++) {             
     double r2 = (l-60.5)*(l-60.5)+(c-60.5)*(c-60.5);   
-    in(l,c) = 30.*exp(-r2/9.);        
+    in(l,c) = (TF)30.*exp(-r2/9.);        
 }                                
 //  Le dernier argument de RandomSequence est le sigma du bruit   
 //  Si on veut rajouter du bruit, decommenter les 3 lignes suivantes  
-  TMatrix<r_8> noise(nlig,ncol);                                
+  TMatrix<TF> noise(nlig,ncol);                                
   noise = RandomSequence(RandomSequence::Gaussian, 0., 0.5);     
   in += noise;                                                  
-TMatrix< complex<r_8> > fftin;       
+TMatrix< complex<TF> > fftin;       
 FFTWServer ffts;                     
 ffts.setNormalize(true);             
@@ -25 +27 @@
 //  dans la seconde moitie du tableau en numero de ligne        
 int kx, ky;                                     
-TMatrix< complex<r_8> > fftfb(fftin.NRows(), fftin.NCols());           
-TMatrix< complex<r_8> > fftfd(fftin.NRows(), fftin.NCols());           
+TMatrix< complex<TF> > fftfb(fftin.NRows(), fftin.NCols());           
+TMatrix< complex<TF> > fftfd(fftin.NRows(), fftin.NCols());           
 // fb : Filtrage brutal , fd :  Filtrage doux 
 for(ky=0; ky<fftin.NRows(); ky++)             
@@ -34 +36 @@
     if (ky > fftin.NRows()/2) kyy = fftin.NRows()-ky;
     double kr = sqrt((double)kx*kx+kyy*kyy);              
-    complex<double> att;
+    complex<TF> att;
     // Filtrage brutal  fonction en escalier 
-    if (kr < 10)  att = complex<double>(0.5, 0.);
+    if (kr < 10)  att = complex<TF>((TF)0.5, (TF)0.);
     else att = complex<double>(1., 0.);
     fftfb(ky, kx) = fftin(ky, kx) * att ;   
     // Filtrage doux sous forme de 1-exp(-alpha k_r)
-    att = complex<double>(1.-exp(-kr*0.1), 0.);   
+    att = complex<TF>((TF)(1.-exp(-kr*0.1)), (TF)0.);   
     fftfd(ky, kx) = fftin(ky, kx) * att ;   
   }
-TMatrix<r_8> infb(nlig,ncol);          
-TMatrix<r_8> infd(nlig,ncol);          
+TMatrix<TF> infb(nlig,ncol);          
+TMatrix<TF> infd(nlig,ncol);          
 // FFTBackward -- Attention, le tableau des coefficient de Fourier 
 // est modifie - On en fait une copie 
-TMatrix< complex<r_8> > fftf(fftin.NRows(), fftin.NCols());   
+TMatrix< complex<TF> > fftf(fftin.NRows(), fftin.NCols());   
 // Pour le filtre brutal
 fftf = fftfb;