Compression_Example.zip

# Created: 27/08/2018 # Author: Callum Bruce # Script for plotting contacts in a single timestep from edempy import Deck import numpy as np import matplotlib.pyplot as plt from mpl_toolkits.mplot3d import Axes3D deck = Deck('Compression_Example.dem') tstep = 1 # timestep to plot contacts for nGeoms = len(deck.geometryNames) # Step 1: Get contact data ids = deck.timestep[tstep].contact.surfSurf.getIds() normalForce = deck.timestep[tstep].contact.surfSurf.getNormalForce() P1 = np.zeros([len(ids),3]) P2 = np.zeros([len(ids),3]) normalForceMag = np.zeros(len(ids)) # Step 2: Setup figure fig = plt.figure() ax = fig.gca(projection='3d') ax.set_xlabel('X') ax.set_ylabel('Y') ax.set_zlabel('Z') ax.set_title('Compression at ' + str(round(deck.timestep[tstep].timestep,4)) + ' s') ax.set_xlim(min(deck.domainMin),max(deck.domainMax)) ax.set_ylim(min(deck.domainMin),max(deck.domainMax)) ax.set_zlim(min(deck.domainMin),max(deck.domainMax)) ax.axis('off') # Step 3: Calculate "scale" used to set lineweight when plotting contacts for i in range(0,len(ids)): normalForceMag[i] = np.sqrt(normalForce[i,0]**2 + normalForce[i,1]**2 + normalForce[i,2]**2) scale = normalForceMag/max(normalForceMag) # Step 4: Plot contacts for i in range(0,len(ids)): P1[i] = deck.timestep[tstep].particle[0].getPositions(ids[i,0]) P2[i] = deck.timestep[tstep].particle[0].getPositions(ids[i,1]) ax.plot3D([P1[i,0], P2[i,0]],[P1[i,1], P2[i,1]],[P1[i,2], P2[i,2]], c='b', lw=scale[i]) # Step 5: Plot stationary geometry for i in range(0,2): x = deck.creatorData.geometry[i].getCoords()[:,0] y = deck.creatorData.geometry[i].getCoords()[:,1] z = deck.creatorData.geometry[i].getCoords()[:,2] tri = deck.creatorData.geometry[i].getTriangleNodes() ax.plot_trisurf(x, y, z, linewidth=0.2, antialiased=True, triangles=tri, alpha=0.2, color='lightgrey') # Step 6: Plot moving geometry x = deck.timestep[tstep].geometry[2].getCoords()[:,0] y = deck.timestep[tstep].geometry[2].getCoords()[:,1] z = deck.timestep[tstep].geometry[2].getCoords()[:,2] tri = deck.creatorData.geometry[2].getTriangleNodes() ax.plot_trisurf(x, y, z, linewidth=0.2, antialiased=True, triangles=tri, alpha=0.2, color='lightgrey') plt.show()