#### Test PField print "Opening vtk file" body3d = pfunction.Body3D("3D.vtk") X = np.arange(0.0, 1.0, 0.001) Y = np.arange(0.0, 1.0, 0.001) X, Y = np.meshgrid(X, Y) Z = 0.0*X s = X.shape f = pfunction.PFunc_dsd("c", body3d=body3d) for i in range(0,s[0]): for j in range(0,s[1]): Z[i,j] = f.calc(pfunction.c_dbl_array([X[i,j],Y[i,j],0.2])) fig = plt.figure(fignum, figsize=(8,7)) allfig.append(fig) fignum += 1 fig.suptitle('3D Field Z=0.2', fontsize=12) p = contour(X, Y, Z, [0.15, 0.5, 0.85]) for i in range(0,s[0]): for j in range(0,s[1]): Z[i,j] = f.calc(pfunction.c_dbl_array([X[i,j],Y[i,j],0.7])) fig = plt.figure(fignum, figsize=(8,7)) allfig.append(fig) fignum += 1
fignum = 0 allfig = [] # test PSimple_dd print "test PSimple_dd" f = pfunction.PSimple_dd("Chebyshev_2") print " created" print " evaluate", max, "times" for i in range(0, max): f.calc(0.2) print " result:", f.get() f.delete() print " deleted\n" a = np.array([2.0, 3.0]) b = pfunction.c_dbl_array(a) # test Python print "test Python code" v = 0 print " var:", a print " evaluate", max, "times" for i in range(0, max): v = b[0] * (b[1] * b[1]) + (b[0] * b[0] * b[0]) + ( b[1] * b[1] * b[1]) + (b[0] * b[0]) * b[1] print " result:", v print " done\n" # test PSimple_dsd print "test PSimple_dsd" f = pfunction.PSimple_dsd("MyFunc_f")
#### Test PField print "Opening vtk file" body3d = pfunction.Body3D("3D.vtk") X = np.arange(0.0, 1.0, 0.001) Y = np.arange(0.0, 1.0, 0.001) X, Y = np.meshgrid(X, Y) Z = 0.0 * X s = X.shape f = pfunction.PFunc_dsd("c", body3d=body3d) for i in range(0, s[0]): for j in range(0, s[1]): Z[i, j] = f.calc(pfunction.c_dbl_array([X[i, j], Y[i, j], 0.2])) fig = plt.figure(fignum, figsize=(8, 7)) allfig.append(fig) fignum += 1 fig.suptitle('3D Field Z=0.2', fontsize=12) p = contour(X, Y, Z, [0.15, 0.5, 0.85]) for i in range(0, s[0]): for j in range(0, s[1]): Z[i, j] = f.calc(pfunction.c_dbl_array([X[i, j], Y[i, j], 0.7])) fig = plt.figure(fignum, figsize=(8, 7)) allfig.append(fig) fignum += 1 fig.suptitle('3D Field Z=0.7', fontsize=12)
f = pfunction.PFunc_dsd("c", body2d=body2d) #### fig = plt.figure(fignum, figsize=(8, 8)) allfig.append(fig) fignum += 1 fig.suptitle('Concentration Field', fontsize=12) ax = fig.gca(projection='3d') print "Evaluating field..." for i in range(0, s[0]): for j in range(0, s[1]): Z[i, j] = f.calc(pfunction.c_dbl_array([X[i, j], Y[i, j]])) print "Plotting..." ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm, linewidth=0, antialiased=False) # fig = plt.figure(fignum, figsize=(8, 7))
#### fig = plt.figure(fignum, figsize=(8,8)) allfig.append(fig) fignum += 1 fig.suptitle('Concentration Field', fontsize=12) ax = fig.gca(projection='3d') print "Evaluating field..." for i in range(0,s[0]): for j in range(0,s[1]): Z[i,j] = f.calc(pfunction.c_dbl_array([X[i,j],Y[i,j]])) print "Plotting..." ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm, linewidth=0, antialiased=False) # fig = plt.figure(fignum, figsize=(8,7)) allfig.append(fig) fignum += 1 fig.suptitle('Field', fontsize=12) x = np.arange(0.0,1.0, 0.00001) y = np.arange(0.0,1.0, 0.1) for i in range(0,y.size): field = np.zeros( (x.size))
allfig = [] # test PSimple_dd print "test PSimple_dd" f = pfunction.PSimple_dd("Chebyshev_2") print " created" print " evaluate", max, "times" for i in range(0,max): f.calc(0.2) print " result:", f.get() f.delete() print " deleted\n" a = np.array([2.0, 3.0]); b = pfunction.c_dbl_array(a) # test Python print "test Python code" v = 0 print " var:", a print " evaluate", max, "times" for i in range(0,max): v = b[0]*(b[1]*b[1])+(b[0]*b[0]*b[0])+(b[1]*b[1]*b[1])+(b[0]*b[0])*b[1] print " result:", v print " done\n" # test PSimple_dsd print "test PSimple_dsd"