21e75405e80f92a06646f9abb68f05a612c1850f
[ginac.git] / check / exam_indexed.cpp
1 /** @file exam_indexed.cpp
2  *
3  *  Here we test manipulations on GiNaC's indexed objects. */
4
5 /*
6  *  GiNaC Copyright (C) 1999-2001 Johannes Gutenberg University Mainz, Germany
7  *
8  *  This program is free software; you can redistribute it and/or modify
9  *  it under the terms of the GNU General Public License as published by
10  *  the Free Software Foundation; either version 2 of the License, or
11  *  (at your option) any later version.
12  *
13  *  This program is distributed in the hope that it will be useful,
14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  *  GNU General Public License for more details.
17  *
18  *  You should have received a copy of the GNU General Public License
19  *  along with this program; if not, write to the Free Software
20  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
21  */
22
23 #include "exams.h"
24
25 static unsigned check_equal(const ex &e1, const ex &e2)
26 {
27         ex e = e1 - e2;
28         if (!e.is_zero()) {
29                 clog << e1 << "-" << e2 << " erroneously returned "
30                      << e << " instead of 0" << endl;
31                 return 1;
32         }
33         return 0;
34 }
35
36 static unsigned check_equal_simplify(const ex &e1, const ex &e2)
37 {
38         ex e = simplify_indexed(e1) - e2;
39         if (!e.is_zero()) {
40                 clog << "simplify_indexed(" << e1 << ")-" << e2 << " erroneously returned "
41                      << e << " instead of 0" << endl;
42                 return 1;
43         }
44         return 0;
45 }
46
47 static unsigned delta_check(void)
48 {
49         // checks identities of the delta tensor
50
51         unsigned result = 0;
52
53         symbol s_i("i"), s_j("j"), s_k("k");
54         idx i(s_i, 3), j(s_j, 3), k(s_k, 3);
55         symbol A("A");
56
57         // symmetry
58         result += check_equal(delta_tensor(i, j), delta_tensor(j, i));
59
60         // trace = dimension of index space
61         result += check_equal(delta_tensor(i, i), 3);
62         result += check_equal_simplify(delta_tensor(i, j) * delta_tensor(i, j), 3);
63
64         // contraction with delta tensor
65         result += check_equal_simplify(delta_tensor(i, j) * indexed(A, k), delta_tensor(i, j) * indexed(A, k));
66         result += check_equal_simplify(delta_tensor(i, j) * indexed(A, j), indexed(A, i));
67         result += check_equal_simplify(delta_tensor(i, j) * indexed(A, i), indexed(A, j));
68         result += check_equal_simplify(delta_tensor(i, j) * delta_tensor(j, k) * indexed(A, i), indexed(A, k));
69
70         return result;
71 }
72
73 static unsigned metric_check(void)
74 {
75         // checks identities of the metric tensor
76
77         unsigned result = 0;
78
79         symbol s_mu("mu"), s_nu("nu"), s_rho("rho"), s_sigma("sigma");
80         varidx mu(s_mu, 4), nu(s_nu, 4), rho(s_rho, 4), sigma(s_sigma, 4);
81         symbol A("A");
82
83         // becomes delta tensor if indices have opposite variance
84         result += check_equal(metric_tensor(mu, nu.toggle_variance()), delta_tensor(mu, nu.toggle_variance()));
85
86         // scalar contraction = dimension of index space
87         result += check_equal(metric_tensor(mu, mu.toggle_variance()), 4);
88         result += check_equal_simplify(metric_tensor(mu, nu) * metric_tensor(mu.toggle_variance(), nu.toggle_variance()), 4);
89
90         // contraction with metric tensor
91         result += check_equal_simplify(metric_tensor(mu, nu) * indexed(A, nu), metric_tensor(mu, nu) * indexed(A, nu));
92         result += check_equal_simplify(metric_tensor(mu, nu) * indexed(A, nu.toggle_variance()), indexed(A, mu));
93         result += check_equal_simplify(metric_tensor(mu, nu) * indexed(A, mu.toggle_variance()), indexed(A, nu));
94         result += check_equal_simplify(metric_tensor(mu, nu) * metric_tensor(mu.toggle_variance(), rho.toggle_variance()) * indexed(A, nu.toggle_variance()), indexed(A, rho.toggle_variance()));
95         result += check_equal_simplify(metric_tensor(mu, rho) * metric_tensor(nu, sigma) * indexed(A, rho.toggle_variance(), sigma.toggle_variance()), indexed(A, mu, nu));
96         result += check_equal_simplify(indexed(A, mu.toggle_variance()) * metric_tensor(mu, nu) - indexed(A, mu.toggle_variance()) * metric_tensor(nu, mu), 0);
97         result += check_equal_simplify(indexed(A, mu.toggle_variance(), nu.toggle_variance()) * metric_tensor(nu, rho), indexed(A, mu.toggle_variance(), rho));
98
99         // contraction with delta tensor yields a metric tensor
100         result += check_equal_simplify(delta_tensor(mu, nu.toggle_variance()) * metric_tensor(nu, rho), metric_tensor(mu, rho));
101         result += check_equal_simplify(metric_tensor(mu, nu) * indexed(A, nu.toggle_variance()) * delta_tensor(mu.toggle_variance(), rho), indexed(A, rho));
102
103         return result;
104 }
105
106 static unsigned epsilon_check(void)
107 {
108         // checks identities of the epsilon tensor
109
110         unsigned result = 0;
111
112         symbol s_mu("mu"), s_nu("nu"), s_rho("rho"), s_sigma("sigma");
113         varidx mu(s_mu, 4), nu(s_nu, 4), rho(s_rho, 4), sigma(s_sigma, 4);
114
115         // antisymmetry
116         result += check_equal(lorentz_eps(mu, nu, rho, sigma) + lorentz_eps(sigma, rho, mu, nu), 0);
117
118         // convolution is zero
119         result += check_equal(lorentz_eps(mu, nu, rho, nu.toggle_variance()), 0);
120         result += check_equal(lorentz_eps(mu, nu, mu.toggle_variance(), nu.toggle_variance()), 0);
121         result += check_equal_simplify(lorentz_g(mu.toggle_variance(), nu.toggle_variance()) * lorentz_eps(mu, nu, rho, sigma), 0);
122
123         return result;
124 }
125
126 static unsigned symmetry_check(void)
127 {
128         // check symmetric/antisymmetric objects
129
130         unsigned result = 0;
131
132         symbol s_i("i"), s_j("j"), s_k("k");
133         idx i(s_i, 3), j(s_j, 3), k(s_k, 3);
134         symbol A("A");
135         ex e, e1, e2;
136
137         result += check_equal(indexed(A, indexed::symmetric, i, j), indexed(A, indexed::symmetric, j, i));
138         result += check_equal(indexed(A, indexed::antisymmetric, i, j) + indexed(A, indexed::antisymmetric, j, i), 0);
139         result += check_equal(indexed(A, indexed::antisymmetric, i, j, k) - indexed(A, indexed::antisymmetric, j, k, i), 0);
140
141         return result;
142 }
143
144 static unsigned edyn_check(void)
145 {
146         // Relativistic electrodynamics
147
148         // Test 1: check transformation laws of electric and magnetic fields by
149         // applying a Lorentz boost to the field tensor
150
151         unsigned result = 0;
152
153         symbol beta("beta");
154         ex gamma = 1 / sqrt(1 - pow(beta, 2));
155         symbol Ex("Ex"), Ey("Ey"), Ez("Ez");
156         symbol Bx("Bx"), By("By"), Bz("Bz");
157
158         // Lorentz transformation matrix (boost along x axis)
159         matrix L(4, 4);
160         L.set(0, 0, gamma);
161         L.set(0, 1, -beta*gamma);
162         L.set(1, 0, -beta*gamma);
163         L.set(1, 1, gamma);
164         L.set(2, 2, 1); L.set(3, 3, 1);
165
166         // Electromagnetic field tensor
167         matrix F(4, 4, lst(
168                  0, -Ex, -Ey, -Ez,
169                 Ex,   0, -Bz,  By,
170                 Ey,  Bz,   0, -Bx,
171                 Ez, -By,  Bx // 0
172         ));
173
174         // Indices
175         symbol s_mu("mu"), s_nu("nu"), s_rho("rho"), s_sigma("sigma");
176         varidx mu(s_mu, 4), nu(s_nu, 4), rho(s_rho, 4), sigma(s_sigma, 4);
177
178         // Apply transformation law of second rank tensor
179         ex e = (indexed(L, mu, rho.toggle_variance())
180               * indexed(L, nu, sigma.toggle_variance())
181               * indexed(F, rho, sigma)).simplify_indexed();
182
183         // Extract transformed electric and magnetic fields
184         ex Ex_p = e.subs(lst(mu == 1, nu == 0)).normal();
185         ex Ey_p = e.subs(lst(mu == 2, nu == 0)).normal();
186         ex Ez_p = e.subs(lst(mu == 3, nu == 0)).normal();
187         ex Bx_p = e.subs(lst(mu == 3, nu == 2)).normal();
188         ex By_p = e.subs(lst(mu == 1, nu == 3)).normal();
189         ex Bz_p = e.subs(lst(mu == 2, nu == 1)).normal();
190
191         // Check results
192         result += check_equal(Ex_p, Ex);
193         result += check_equal(Ey_p, gamma * (Ey - beta * Bz));
194         result += check_equal(Ez_p, gamma * (Ez + beta * By));
195         result += check_equal(Bx_p, Bx);
196         result += check_equal(By_p, gamma * (By + beta * Ez));
197         result += check_equal(Bz_p, gamma * (Bz - beta * Ey));
198
199         // Test 2: check energy density and Poynting vector of electromagnetic field
200
201         // Minkowski metric
202         ex eta = diag_matrix(lst(1, -1, -1, -1));
203
204         // Covariant field tensor
205         ex F_mu_nu = (indexed(eta, mu.toggle_variance(), rho.toggle_variance())
206                     * indexed(eta, nu.toggle_variance(), sigma.toggle_variance())
207                     * indexed(F, rho, sigma)).simplify_indexed();
208
209         // Energy-momentum tensor
210         ex T = (-indexed(eta, rho, sigma) * F_mu_nu.subs(s_nu == s_rho) 
211                 * F_mu_nu.subs(lst(s_mu == s_nu, s_nu == s_sigma))
212               + indexed(eta, mu.toggle_variance(), nu.toggle_variance())
213                 * F_mu_nu.subs(lst(s_mu == s_rho, s_nu == s_sigma))
214                 * indexed(F, rho, sigma) / 4).simplify_indexed() / (4 * Pi);
215
216         // Extract energy density and Poynting vector
217         ex E = T.subs(lst(s_mu == 0, s_nu == 0)).normal();
218         ex Px = T.subs(lst(s_mu == 0, s_nu == 1));
219         ex Py = T.subs(lst(s_mu == 0, s_nu == 2)); 
220         ex Pz = T.subs(lst(s_mu == 0, s_nu == 3));
221
222         // Check results
223         result += check_equal(E, (Ex*Ex+Ey*Ey+Ez*Ez+Bx*Bx+By*By+Bz*Bz) / (8 * Pi));
224         result += check_equal(Px, (Ez*By-Ey*Bz) / (4 * Pi));
225         result += check_equal(Py, (Ex*Bz-Ez*Bx) / (4 * Pi));
226         result += check_equal(Pz, (Ey*Bx-Ex*By) / (4 * Pi));
227
228         return result;
229 }
230
231 unsigned exam_indexed(void)
232 {
233         unsigned result = 0;
234         
235         cout << "examining indexed objects" << flush;
236         clog << "----------indexed objects:" << endl;
237
238         result += delta_check();  cout << '.' << flush;
239         result += metric_check();  cout << '.' << flush;
240         result += epsilon_check();  cout << '.' << flush;
241         result += symmetry_check();  cout << '.' << flush;
242         result += edyn_check();  cout << '.' << flush;
243         
244         if (!result) {
245                 cout << " passed " << endl;
246                 clog << "(no output)" << endl;
247         } else {
248                 cout << " failed " << endl;
249         }
250         
251         return result;
252 }