PySol/phononTest.py at master · humdrumcomet/PySol · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
from pyqtgraph.Qt import QtCore, QtGui
import pyqtgraph as pg
import pyqtgraph.opengl as gl
import numpy as np
import itertools as iTt

resolutionForPho = 5
bxSdLn = 4*np.pi
a = bxSdLn/(2*np.sqrt(2))
scl = 1
org = np.array([0,0,0])

xpre = np.array([0,1,0,0,1,1,0,1,0.5,0.5,1,0.5,0,0.5,0.25,0.75,0.25,0.75])
ypre = np.array([0,0,1,0,1,0,1,1,0.5,0.5,0.5,1,0.5,0,0.25,0.75,0.75,0.25])
zpre = np.array([0,0,0,1,0,1,1,1,0,1,0.5,0.5,0.5,0.5,0.25,0.25,0.75,0.75])

displacmentA = np.array([0.02,0.02,0])

frcCnstSrt = 2
frcCnstBnd = 4

m1 = 1
m2 = 1.25

at1 = np.array([0.25,0.25,0.25])
at1pp = np.array([0.5,0.5,0])
at1mm = np.array([0,0,0])
at1mp = np.array([0,0.5,0.5])
at1pm = np.array([0.5,0,0.5])

at2 = np.array([0,0,0])
at2pp = np.array([0.25,0.25,0.25])
at2mm = np.array([-0.25,-0.25,0.25])
at2mp = np.array([-0.25,0.25,-0.25])
at2pm = np.array([0.25,-0.25,-0.25])

symPts = dict(
	gamma = scl*np.add(org,np.array([bxSdLn*0.5,bxSdLn*0.5,bxSdLn*0.5])),
	L = scl*np.add(org,np.array([bxSdLn*0.5+(a*np.sqrt(2)/2),bxSdLn*0.5+(a*np.sqrt(2)/2),
		bxSdLn*0.5+(a*np.sqrt(2)/2)])),
	K = scl*np.add(org,np.array([bxSdLn-(a*np.sqrt(2)/4),bxSdLn-(a*np.sqrt(2)/4),bxSdLn*0.5])),
	W = scl*np.add(org,np.array([bxSdLn*0.5+(a*np.sqrt(2)/2),bxSdLn,bxSdLn*0.5])),
	X = scl*np.add(org,np.array([bxSdLn*0.5,bxSdLn,bxSdLn*0.5])),
	U = scl*np.add(org,np.array([bxSdLn*0.5+a*np.sqrt(2)/4,bxSdLn,bxSdLn*0.5+a/np.sqrt(8)]))
)

symLnComb = iTt.combinations(list(symPts.keys()),2)
symLns = {}
for i in symLnComb:
	a = np.array([symPts[i[0]], symPts[i[1]]]).T
	symLns[i] = []
	for indx, j in enumerate(a):
		symLns[i].append(np.linspace(j[0], j[1], num = resolutionForPho))
	symLns[i] = np.array(symLns[i])

forceMat = np.empty([6,6])
forceMat[0,0] =