-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathmake_sp_code.sh
executable file
·143 lines (113 loc) · 6.25 KB
/
make_sp_code.sh
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
#!/bin/bash
source dimensions.in
echo "#ifndef _SP_LINALG_H"
echo "#define _SP_LINALG_H"
echo "#include <math.h>"
echo "#include \"cvc_complex.h\""
echo "#include \"cvc_linalg.h\""
echo ""
echo "// spinor dimension: $N_SPINOR_DIM"
echo "// color dimension: $N_COLOR_DIM"
echo ""
# ----------------------------------------------------------------------------------------------------------------
echo "#define _sp_eq_zero(_r) {\\"
for((beta=0;beta<$N_SPINOR_DIM;beta++)); do
for((alpha=0;alpha<$N_SPINOR_DIM;alpha++)); do
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha)))] = 0.;\\"
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha+1)))] = 0.;\\"
done
done
echo -e "}\n\n"
# ----------------------------------------------------------------------------------------------------------------
echo "#define _sp_eq_sp(_r, _s) {\\"
for((beta=0;beta<$N_SPINOR_DIM;beta++)); do
for((alpha=0;alpha<$N_SPINOR_DIM;alpha++)); do
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha)))] = (_s)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha)))];\\"
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha+1)))] = (_s)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha+1)))];\\"
done
done
echo -e "}\n\n"
# ----------------------------------------------------------------------------------------------------------------
echo "#define _sp_eq_sp_transposed(_r, _s) {\\"
for((beta=0;beta<$N_SPINOR_DIM;beta++)); do
for((alpha=0;alpha<$N_SPINOR_DIM;alpha++)); do
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha)))] = (_s)[$(printf "%2d" $alpha)][$(printf "%2d" $((2*$beta)))];\\"
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha+1)))] = (_s)[$(printf "%2d" $alpha)][$(printf "%2d" $((2*$beta+1)))];\\"
done
done
echo -e "}\n\n"
# ----------------------------------------------------------------------------------------------------------------
echo "#define _sp_pl_eq_sp(_r, _s) {\\"
for((beta=0;beta<$N_SPINOR_DIM;beta++)); do
for((alpha=0;alpha<$N_SPINOR_DIM;alpha++)); do
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha)))] += (_s)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha)))];\\"
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha+1)))] += (_s)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha+1)))];\\"
done
done
echo -e "}\n\n"
# ----------------------------------------------------------------------------------------------------------------
echo "#define _sp_mi_eq_sp(_r, _s) {\\"
for((beta=0;beta<$N_SPINOR_DIM;beta++)); do
for((alpha=0;alpha<$N_SPINOR_DIM;alpha++)); do
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha)))] -= (_s)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha)))];\\"
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha+1)))] -= (_s)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha+1)))];\\"
done
done
echo -e "}\n\n"
# ----------------------------------------------------------------------------------------------------------------
echo "#define _sp_eq_sp_ti_re(_r, _s, _c) {\\"
for((beta=0;beta<$N_SPINOR_DIM;beta++)); do
for((alpha=0;alpha<$N_SPINOR_DIM;alpha++)); do
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha)))] = (_s)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha)))] * (_c);\\"
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha+1)))] = (_s)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha+1)))] * (_c);\\"
done
done
echo -e "}\n\n"
# ----------------------------------------------------------------------------------------------------------------
echo "#define _sp_eq_sp_ti_im(_r, _s, _c) {\\"
for((beta=0;beta<$N_SPINOR_DIM;beta++)); do
for((alpha=0;alpha<$N_SPINOR_DIM;alpha++)); do
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha)))] = -(_s)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha+1)))] * (_c);\\"
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha+1)))] = +(_s)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha)))] * (_c);\\"
done
done
echo -e "}\n\n"
# ----------------------------------------------------------------------------------------------------------------
echo "#define _sp_eq_sp_ti_co(_r, _s, _c) {\\"
for((beta=0;beta<$N_SPINOR_DIM;beta++)); do
for((alpha=0;alpha<$N_SPINOR_DIM;alpha++)); do
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha)))] = +(_s)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha)))] * (_c).re - (_s)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha+1)))] * (_c).im;\\"
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha+1)))] = +(_s)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha)))] * (_c).im + (_s)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha+1)))] * (_c).re;\\"
done
done
echo -e "}\n\n"
# ----------------------------------------------------------------------------------------------------------------
echo "#define _sp_eq_gamma_ti_sp(_r, _mu, _s) {\\"
echo "int __perm[$N_SPINOR_DIM], __isimag;\\"
for((beta=0;beta<$N_SPINOR_DIM;beta++)); do
echo "__perm[$(printf "%2d" $beta)] = gamma_permutation[(_mu)][$(printf "%2d" $((2*$N_COLOR_DIM*$beta)))]/$((2*N_COLOR_DIM));\\"
done
echo "__isimag = gamma_permutation[(_mu)][0]%2;\\"
for((beta=0;beta<$N_SPINOR_DIM;beta++)); do
for((alpha=0;alpha<$N_SPINOR_DIM;alpha++)); do
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha)))] = (_s)[$(printf "%2d" $beta)][2*__perm[$(printf "%2d" $alpha)] +__isimag] * gamma_sign[(_mu)][$(printf "%2d" $((2*$N_COLOR_DIM*$alpha)))];\\"
echo "(_r)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$alpha+1)))] = (_s)[$(printf "%2d" $beta)][2*__perm[$(printf "%2d" $alpha)]+1-__isimag] * gamma_sign[(_mu)][$(printf "%2d" $((2*$N_COLOR_DIM*$alpha+1)))];\\"
done
done
echo -e "}\n\n"
# ----------------------------------------------------------------------------------------------------------------
echo "#define _co_eq_tr_sp(_c, _s) {\\"
echo "(_c)->re = \\"
for((beta=0;beta<$N_SPINOR_DIM;beta++)); do
echo " +(_s)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$beta)))]\\"
done
echo ";\\"
echo "(_c)->im = \\"
for((beta=0;beta<$N_SPINOR_DIM;beta++)); do
echo " +(_s)[$(printf "%2d" $beta)][$(printf "%2d" $((2*$beta+1)))]\\"
done
echo ";\\"
echo -e "}\n\n"
# ----------------------------------------------------------------------------------------------------------------
echo "#endif"
exit 0