diff --git a/demos/demo.py b/demos/demo.py index f21fc5784fead894d5ebf5a30a8446011a97bb62..1f850b71da027f5c8b5ff9a0b5dee7a396a28cb3 100644 --- a/demos/demo.py +++ b/demos/demo.py @@ -41,3 +41,12 @@ if __name__ == '__main__': #plotting on logarithmic scale plt.imsave('diffuse_component.png', Starblade.s.val, vmin=vmin, vmax=vmax) plt.imsave('pointlike_component.png', Starblade.u.val, vmin=vmin, vmax=vmax) + plt.figure() + k_lenghts = Starblade.power_spectrum.domain[0].k_lengths + plt.plot(k_lenghts, Starblade.power_spectrum.val) + plt.title('power spectrum') + plt.yscale('log') + plt.xscale('log') + plt.ylabel('power') + plt.xscale('harmonic mode') + plt.savefig('power_spectrum.png') diff --git a/demos/multichannel_demo.py b/demos/multichannel_demo.py index 79a1f3f41605980058403f1c31527659c4062640..bbe5fcf793f68d079536577c5d7bce44cbe364b1 100644 --- a/demos/multichannel_demo.py +++ b/demos/multichannel_demo.py @@ -29,9 +29,7 @@ if __name__ == '__main__': data = data.astype(float) data = data.clip(0.0001) alpha = 1.3 - MultiStarblade = sb.build_multi_starblade(data, alpha, newton_iterations=1) - # power = MultiStarblade[0].power_spectrum / np.arange(len(MultiStarblade[0].power_spectrum.val))**0 - # MultiStarblade = sb.build_multi_starblade(data,alpha,newton_iterations=1, manual_power_spectrum=power) + MultiStarblade = sb.build_multi_starblade(data, alpha=alpha) for i in range(50): MultiStarblade = sb.multi_starblade_iteration(MultiStarblade, processes=1) @@ -46,10 +44,4 @@ if __name__ == '__main__': plt.imsave('rgb_diffuse.jpg',diffuse/255.) plt.imsave('rgb_point.jpg',point/255.) - plt.figure() - plt.plot(MultiStarblade[0].power_spectrum.val) - plt.yscale('log') - plt.xscale('log') - plt.savefig('power.jpg') - plt.close()