Work on plotting script

demos/newamplitudes/easyinterface.py

 import nifty5 as ift
+import numpy as np
+np.random.seed(42)
 
 sspace = ift.RGSpace((128, 128), (0.2, 0.2))
 hspace = sspace.get_default_codomain()
 target = ift.PowerSpace(hspace)
 
-vol = hspace.scalar_dvol**-0.5
-ft = ift.HartleyOperator(hspace, sspace).scale(vol)
-
 A = ift.NormalizedAmplitude(target, 16, 1, 1, -3, 1, 0, 1, 0, 1)
+A = ift.WPAmplitude(target, [0, -2, 0], [1E-5, 1, 1], 1, 0.99,
+                    ['rest', 'smooth', 'wienersigma'])
 
 avgA = ift.full(A.target, 0.)
 n = 1000
 for _ in range(n):
     avgA = avgA + A(ift.from_random('normal', A.domain))
 avgA = avgA/n
-
 corfldfixA = ift.CorrelatedField(sspace, avgA)
 corfld = ift.CorrelatedField(sspace, A)
 
-p = ift.Plot()
-p1 = ift.Plot()
+corfld = ift.CorrelatedFieldNormAmplitude(sspace, A, 0, 1)
+corfldfixA = ift.CorrelatedFieldNormAmplitude(sspace, avgA, 0, 1)
+
+
+cstpos = ift.from_random('normal', corfld.domain)
+p, p1, p2 = [ift.Plot() for _ in range(3)]
 lst, lst1 = [avgA**2], []
 for _ in range(8):
     pos = ift.from_random('normal', corfld.domain)
 
     skyfixA = corfldfixA.force(pos)
     p.add(skyfixA)
+    ft = ift.HartleyOperator(hspace, sspace).scale(hspace.scalar_dvol**-0.5)
     lst.append(ift.power_analyze(ft.inverse(skyfixA)))
 
+    foo = ift.MultiField.union([cstpos, pos.extract(A.domain)])
+    p2.add(corfld(foo))
+
     sky = corfld(pos)
     p1.add(sky)
     lst1.append(A.force(pos))
 p.add(lst)
 p1.add(lst1)
-p.output(name='mean_power_spectrum.png', xsize=15, ysize=15)
-p1.output(name='full.png', xsize=15, ysize=15)
+p2.add(lst1)
+p1.output(name='full.png')
+p.output(name='A_fixed.png')
+p2.output(name='xi_fixed.png')