Changeset 3788 in Sophya for trunk/Cosmo/RadioBeam/lobe.cc

Timestamp:

Jun 25, 2010, 7:35:17 PM (15 years ago)

Author:

ansari

Message:

Ajout classe de soustraction d'avant plans, Reza 25/06/2010

File:

• trunk/Cosmo/RadioBeam/lobe.cc (modified) (2 diffs)

trunk/Cosmo/RadioBeam/lobe.cc

@@ -17 +17 @@
 }
 
-/* --Methode-- */
-void BeamEffect::ApplyLobeK2D(TArray< complex<TF> >& fourAmp, double dkx, double dky, double lambda)
-//  dx, dy en radians, lambda en metres 
-{
-  fresp_.setLambda(lambda);
-  double kxx, kyy;
-  for(sa_size_t ky=0; ky<fourAmp.SizeY(); ky++) {
-    kyy =  (ky>fourAmp.SizeY()/2) ? -(double)(fourAmp.SizeY()-ky)*dky : (double)ky*dky; 
-    for(sa_size_t kx=0; kx<fourAmp.SizeX(); kx++) {
-      kxx=(double)kx*dkx;
-      fourAmp(kx, ky) *= complex<TF>(fresp_(kxx, kyy), 0.);     
-    }
-  }
-  return;
-}
 
 
@@ -52 +37 @@
     ffts.FFTForward(slice, fourAmp);
     conv.setFrequency(f0+kz*df);
-    ApplyLobeK2D(fourAmp, dkx, dky, conv.getLambda());
+    fresp_.setLambda(conv.getLambda());
+    ApplyLobeK2D(fresp_, fourAmp, dkx, dky);
     ffts.FFTBackward(fourAmp, slice, true);
-    if (kz%10==0)  cout << "BeamEffect::ApplyLobe3D() done kz=" << kz << " / a.SizeZ()=" << a.SizeZ() << endl; 
+    if (kz%20==0)  cout << "BeamEffect::ApplyLobe3D() done kz=" << kz << " / a.SizeZ()=" << a.SizeZ() << endl; 
+  }
+  return;
+}
+
+/* --Methode-- */
+void BeamEffect::Correct2RefLobe(Four2DResponse& rep, TArray< TF >& a, double dx, double dy, double f0, double df)
+// dx, dy en radioans, f0, df en MHz
+{
+  Timer tm("BeamEffect::Correct2RefLobe");
+  FFTWServer ffts(true);                     
+  ffts.setNormalize(true); 
+  
+  H21Conversions conv;
+  
+  TArray< complex<TF> > fourAmp;
+  double dkx = DeuxPI/(double)a.SizeX()/dx;
+  double dky = DeuxPI/(double)a.SizeY()/dy;
+  
+  for(sa_size_t kz=0; kz<a.SizeZ(); kz++) {
+    TArray< TF > slice( a(Range::all(), Range::all(), kz) );
+    ffts.FFTForward(slice, fourAmp);
+    conv.setFrequency(f0+kz*df);
+    fresp_.setLambda(conv.getLambda());
+    Four2DRespRatio rratio(fresp_, rep);
+    ApplyLobeK2D(rratio, fourAmp, dkx, dky);
+    ffts.FFTBackward(fourAmp, slice, true);
+    if (kz%20==0)  cout << "BeamEffect::Correct2RefLobe() done kz=" << kz << " / a.SizeZ()=" << a.SizeZ() << endl; 
+  }
+  return;
+}
+
+/* --Methode-- */
+void BeamEffect::ApplyLobeK2D(Four2DResponse& rep, TArray< complex<TF> >& fourAmp, double dkx, double dky)
+//  dx, dy en radians, lambda en metres 
+{
+  double kxx, kyy;
+  for(sa_size_t ky=0; ky<fourAmp.SizeY(); ky++) {
+    kyy =  (ky>fourAmp.SizeY()/2) ? -(double)(fourAmp.SizeY()-ky)*dky : (double)ky*dky; 
+    for(sa_size_t kx=0; kx<fourAmp.SizeX(); kx++) {
+      kxx=(double)kx*dkx;
+      fourAmp(kx, ky) *= complex<TF>(rep(kxx, kyy), 0.);        
+    }
   }
   return;