From 3d0b9ba1d610e68a63c2ee2920c1cf0d14d7e247 Mon Sep 17 00:00:00 2001 From: David Minton Date: Tue, 14 Apr 2020 13:17:51 -0400 Subject: [PATCH] Fixed bug in crater_profile.py in complex example --- examples/morphology_test_cases/complex/crater_profile.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/examples/morphology_test_cases/complex/crater_profile.py b/examples/morphology_test_cases/complex/crater_profile.py index e09c6069..12689d84 100644 --- a/examples/morphology_test_cases/complex/crater_profile.py +++ b/examples/morphology_test_cases/complex/crater_profile.py @@ -24,7 +24,7 @@ def image_dem(filename,gridsize,pix,CTEM_dem): # Create surface dem map solar_angle = 20.0 dem_map = np.copy(CTEM_dem) - np.roll(CTEM_dem, 1, 0) - dem_map = (0.5 * np.pi) + np.arctan2 + dem_map = (0.5 * np.pi) + np.arctan2(dem_map, pix) dem_map = dem_map - np.radians(solar_angle) * (0.5 * np.pi) np.place(dem_map, dem_map > (0.5 * np.pi), 0.5 * np.pi) dem_map = np.absolute(dem_map) @@ -121,8 +121,10 @@ def pol2cart(rho, phi): #Compare with CTEM output +gridsize = 2000 dem_file = 'surface_dem.dat' CTEM_dem = np.fromfile(dem_file , dtype = np.float64) +CTEM_dem.shape = (gridsize,gridsize) CTEM_dem *= 1e-3 pix = 0.200 gridsize = 2000 @@ -152,7 +154,7 @@ def pol2cart(rho, phi): plt.savefig(f'{crater_name}_profile.png',dpi=300,bbox_inches='tight') -#os.system(f'open {crater_name}_profile.png') +os.system(f'open {crater_name}_profile.png') imgfile = f'{crater_name}.png' image_dem(imgfile,gridsize,pix,orig_dem)