90197c082014cafbbe70767d92aa5ff77b70b8bf
[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-2002 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 check_equal_simplify(const ex &e1, const ex &e2, const scalar_products &sp)
48 {
49         ex e = simplify_indexed(e1, sp) - e2;
50         if (!e.is_zero()) {
51                 clog << "simplify_indexed(" << e1 << ")-" << e2 << " erroneously returned "
52                      << e << " instead of 0" << endl;
53                 return 1;
54         }
55         return 0;
56 }
57
58 static unsigned delta_check(void)
59 {
60         // checks identities of the delta tensor
61
62         unsigned result = 0;
63
64         symbol s_i("i"), s_j("j"), s_k("k");
65         idx i(s_i, 3), j(s_j, 3), k(s_k, 3);
66         symbol A("A");
67
68         // symmetry
69         result += check_equal(delta_tensor(i, j), delta_tensor(j, i));
70
71         // trace = dimension of index space
72         result += check_equal(delta_tensor(i, i), 3);
73         result += check_equal_simplify(delta_tensor(i, j) * delta_tensor(i, j), 3);
74
75         // contraction with delta tensor
76         result += check_equal_simplify(delta_tensor(i, j) * indexed(A, k), delta_tensor(i, j) * indexed(A, k));
77         result += check_equal_simplify(delta_tensor(i, j) * indexed(A, j), indexed(A, i));
78         result += check_equal_simplify(delta_tensor(i, j) * indexed(A, i), indexed(A, j));
79         result += check_equal_simplify(delta_tensor(i, j) * delta_tensor(j, k) * indexed(A, i), indexed(A, k));
80
81         return result;
82 }
83
84 static unsigned metric_check(void)
85 {
86         // checks identities of the metric tensor
87
88         unsigned result = 0;
89
90         symbol s_mu("mu"), s_nu("nu"), s_rho("rho"), s_sigma("sigma");
91         varidx mu(s_mu, 4), nu(s_nu, 4), rho(s_rho, 4), sigma(s_sigma, 4);
92         symbol A("A");
93
94         // becomes delta tensor if indices have opposite variance
95         result += check_equal(metric_tensor(mu, nu.toggle_variance()), delta_tensor(mu, nu.toggle_variance()));
96
97         // scalar contraction = dimension of index space
98         result += check_equal(metric_tensor(mu, mu.toggle_variance()), 4);
99         result += check_equal_simplify(metric_tensor(mu, nu) * metric_tensor(mu.toggle_variance(), nu.toggle_variance()), 4);
100
101         // contraction with metric tensor
102         result += check_equal_simplify(metric_tensor(mu, nu) * indexed(A, nu), metric_tensor(mu, nu) * indexed(A, nu));
103         result += check_equal_simplify(metric_tensor(mu, nu) * indexed(A, nu.toggle_variance()), indexed(A, mu));
104         result += check_equal_simplify(metric_tensor(mu, nu) * indexed(A, mu.toggle_variance()), indexed(A, nu));
105         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()));
106         result += check_equal_simplify(metric_tensor(mu, rho) * metric_tensor(nu, sigma) * indexed(A, rho.toggle_variance(), sigma.toggle_variance()), indexed(A, mu, nu));
107         result += check_equal_simplify(indexed(A, mu.toggle_variance()) * metric_tensor(mu, nu) - indexed(A, mu.toggle_variance()) * metric_tensor(nu, mu), 0);
108         result += check_equal_simplify(indexed(A, mu.toggle_variance(), nu.toggle_variance()) * metric_tensor(nu, rho), indexed(A, mu.toggle_variance(), rho));
109
110         // contraction with delta tensor yields a metric tensor
111         result += check_equal_simplify(delta_tensor(mu, nu.toggle_variance()) * metric_tensor(nu, rho), metric_tensor(mu, rho));
112         result += check_equal_simplify(metric_tensor(mu, nu) * indexed(A, nu.toggle_variance()) * delta_tensor(mu.toggle_variance(), rho), indexed(A, rho));
113
114         return result;
115 }
116
117 static unsigned epsilon_check(void)
118 {
119         // checks identities of the epsilon tensor
120
121         unsigned result = 0;
122
123         symbol s_mu("mu"), s_nu("nu"), s_rho("rho"), s_sigma("sigma"), s_tau("tau");
124         symbol d("d");
125         varidx mu(s_mu, 4), nu(s_nu, 4), rho(s_rho, 4), sigma(s_sigma, 4), tau(s_tau, 4);
126         varidx mu_co(s_mu, 4, true), nu_co(s_nu, 4, true), rho_co(s_rho, 4, true), sigma_co(s_sigma, 4, true), tau_co(s_tau, 4, true);
127
128         // antisymmetry
129         result += check_equal(lorentz_eps(mu, nu, rho, sigma) + lorentz_eps(sigma, rho, mu, nu), 0);
130
131         // convolution is zero
132         result += check_equal(lorentz_eps(mu, nu, rho, nu_co), 0);
133         result += check_equal(lorentz_eps(mu, nu, mu_co, nu_co), 0);
134         result += check_equal_simplify(lorentz_g(mu_co, nu_co) * lorentz_eps(mu, nu, rho, sigma), 0);
135
136         // contraction with symmetric tensor is zero
137         result += check_equal_simplify(lorentz_eps(mu, nu, rho, sigma) * indexed(d, sy_symm(), mu_co, nu_co), 0);
138         result += check_equal_simplify(lorentz_eps(mu, nu, rho, sigma) * indexed(d, sy_symm(), nu_co, sigma_co, rho_co), 0);
139         ex e = lorentz_eps(mu, nu, rho, sigma) * indexed(d, sy_symm(), mu_co, tau);
140         result += check_equal_simplify(e, e);
141
142         // contractions of epsilon tensors
143         result += check_equal_simplify(lorentz_eps(mu, nu, rho, sigma) * lorentz_eps(mu_co, nu_co, rho_co, sigma_co), -24);
144         result += check_equal_simplify(lorentz_eps(tau, nu, rho, sigma) * lorentz_eps(mu_co, nu_co, rho_co, sigma_co), -6 * delta_tensor(tau, mu_co));
145
146         return result;
147 }
148
149 static unsigned symmetry_check(void)
150 {
151         // check symmetric/antisymmetric objects
152
153         unsigned result = 0;
154
155         idx i(symbol("i"), 3), j(symbol("j"), 3), k(symbol("k"), 3), l(symbol("l"), 3);
156         symbol A("A"), B("B");
157         ex e;
158
159         result += check_equal(indexed(A, sy_symm(), i, j), indexed(A, sy_symm(), j, i));
160         result += check_equal(indexed(A, sy_anti(), i, j) + indexed(A, sy_anti(), j, i), 0);
161         result += check_equal(indexed(A, sy_anti(), i, j, k) - indexed(A, sy_anti(), j, k, i), 0);
162         e = indexed(A, sy_symm(), i, j, k) *
163             indexed(B, sy_anti(), l, k, i);
164         result += check_equal_simplify(e, 0);
165         e = indexed(A, sy_symm(), i, i, j, j) *
166             indexed(B, sy_anti(), k, l); // GiNaC 0.8.0 had a bug here
167         result += check_equal_simplify(e, e);
168
169         symmetry R = sy_symm(sy_anti(0, 1), sy_anti(2, 3));
170         e = indexed(A, R, i, j, k, l) + indexed(A, R, j, i, k, l);
171         result += check_equal(e, 0);
172         e = indexed(A, R, i, j, k, l) + indexed(A, R, i, j, l, k);
173         result += check_equal(e, 0);
174         e = indexed(A, R, i, j, k, l) - indexed(A, R, j, i, l, k);
175         result += check_equal(e, 0);
176         e = indexed(A, R, i, j, k, l) + indexed(A, R, k, l, j, i);
177         result += check_equal(e, 0);
178
179         e = indexed(A, i, j);
180         result += check_equal(symmetrize(e) + antisymmetrize(e), e);
181         e = indexed(A, sy_symm(), i, j, k, l);
182         result += check_equal(symmetrize(e), e);
183         result += check_equal(antisymmetrize(e), 0);
184         e = indexed(A, sy_anti(), i, j, k, l);
185         result += check_equal(symmetrize(e), 0);
186         result += check_equal(antisymmetrize(e), e);
187
188         return result;
189 }
190
191 static unsigned scalar_product_check(void)
192 {
193         // check scalar product replacement
194
195         unsigned result = 0;
196
197     idx i(symbol("i"), 3), j(symbol("j"), 3);
198     symbol A("A"), B("B"), C("C");
199         ex e;
200
201     scalar_products sp;
202     sp.add(A, B, 0); // A and B are orthogonal
203     sp.add(A, C, 0); // A and C are orthogonal
204     sp.add(A, A, 4); // A^2 = 4 (A has length 2)
205
206     e = (indexed(A + B, i) * indexed(A + C, i)).expand(expand_options::expand_indexed);
207         result += check_equal_simplify(e, indexed(B, i) * indexed(C, i) + 4, sp);
208         e = indexed(A, i, i) * indexed(B, j, j); // GiNaC 0.8.0 had a bug here
209         result += check_equal_simplify(e, e, sp);
210
211         return result;
212 }
213
214 static unsigned edyn_check(void)
215 {
216         // Relativistic electrodynamics
217
218         // Test 1: check transformation laws of electric and magnetic fields by
219         // applying a Lorentz boost to the field tensor
220
221         unsigned result = 0;
222
223         symbol beta("beta");
224         ex gamma = 1 / sqrt(1 - pow(beta, 2));
225         symbol Ex("Ex"), Ey("Ey"), Ez("Ez");
226         symbol Bx("Bx"), By("By"), Bz("Bz");
227
228         // Lorentz transformation matrix (boost along x axis)
229         matrix L(4, 4);
230         L(0, 0) = gamma;
231         L(0, 1) = -beta*gamma;
232         L(1, 0) = -beta*gamma;
233         L(1, 1) = gamma;
234         L(2, 2) = 1; L(3, 3) = 1;
235
236         // Electromagnetic field tensor
237         matrix F(4, 4, lst(
238                  0, -Ex, -Ey, -Ez,
239                 Ex,   0, -Bz,  By,
240                 Ey,  Bz,   0, -Bx,
241                 Ez, -By,  Bx,   0
242         ));
243
244         // Indices
245         symbol s_mu("mu"), s_nu("nu"), s_rho("rho"), s_sigma("sigma");
246         varidx mu(s_mu, 4), nu(s_nu, 4), rho(s_rho, 4), sigma(s_sigma, 4);
247
248         // Apply transformation law of second rank tensor
249         ex e = (indexed(L, mu, rho.toggle_variance())
250               * indexed(L, nu, sigma.toggle_variance())
251               * indexed(F, rho, sigma)).simplify_indexed();
252
253         // Extract transformed electric and magnetic fields
254         ex Ex_p = e.subs(lst(mu == 1, nu == 0)).normal();
255         ex Ey_p = e.subs(lst(mu == 2, nu == 0)).normal();
256         ex Ez_p = e.subs(lst(mu == 3, nu == 0)).normal();
257         ex Bx_p = e.subs(lst(mu == 3, nu == 2)).normal();
258         ex By_p = e.subs(lst(mu == 1, nu == 3)).normal();
259         ex Bz_p = e.subs(lst(mu == 2, nu == 1)).normal();
260
261         // Check results
262         result += check_equal(Ex_p, Ex);
263         result += check_equal(Ey_p, gamma * (Ey - beta * Bz));
264         result += check_equal(Ez_p, gamma * (Ez + beta * By));
265         result += check_equal(Bx_p, Bx);
266         result += check_equal(By_p, gamma * (By + beta * Ez));
267         result += check_equal(Bz_p, gamma * (Bz - beta * Ey));
268
269         // Test 2: check energy density and Poynting vector of electromagnetic field
270
271         // Minkowski metric
272         ex eta = diag_matrix(lst(1, -1, -1, -1));
273
274         // Covariant field tensor
275         ex F_mu_nu = (indexed(eta, mu.toggle_variance(), rho.toggle_variance())
276                     * indexed(eta, nu.toggle_variance(), sigma.toggle_variance())
277                     * indexed(F, rho, sigma)).simplify_indexed();
278
279         // Energy-momentum tensor
280         ex T = (-indexed(eta, rho, sigma) * F_mu_nu.subs(s_nu == s_rho) 
281                 * F_mu_nu.subs(lst(s_mu == s_nu, s_nu == s_sigma))
282               + indexed(eta, mu.toggle_variance(), nu.toggle_variance())
283                 * F_mu_nu.subs(lst(s_mu == s_rho, s_nu == s_sigma))
284                 * indexed(F, rho, sigma) / 4).simplify_indexed() / (4 * Pi);
285
286         // Extract energy density and Poynting vector
287         ex E = T.subs(lst(s_mu == 0, s_nu == 0)).normal();
288         ex Px = T.subs(lst(s_mu == 0, s_nu == 1));
289         ex Py = T.subs(lst(s_mu == 0, s_nu == 2)); 
290         ex Pz = T.subs(lst(s_mu == 0, s_nu == 3));
291
292         // Check results
293         result += check_equal(E, (Ex*Ex+Ey*Ey+Ez*Ez+Bx*Bx+By*By+Bz*Bz) / (8 * Pi));
294         result += check_equal(Px, (Ez*By-Ey*Bz) / (4 * Pi));
295         result += check_equal(Py, (Ex*Bz-Ez*Bx) / (4 * Pi));
296         result += check_equal(Pz, (Ey*Bx-Ex*By) / (4 * Pi));
297
298         return result;
299 }
300
301 static unsigned spinor_check(void)
302 {
303         // check identities of the spinor metric
304
305         unsigned result = 0;
306
307         symbol psi("psi");
308         spinidx A(symbol("A"), 2), B(symbol("B"), 2), C(symbol("C"), 2);
309         ex A_co = A.toggle_variance(), B_co = B.toggle_variance();
310         ex e;
311
312         e = spinor_metric(A_co, B_co) * spinor_metric(A, B);
313         result += check_equal_simplify(e, 2);
314         e = spinor_metric(A_co, B_co) * spinor_metric(B, A);
315         result += check_equal_simplify(e, -2);
316         e = spinor_metric(A_co, B_co) * spinor_metric(A, C);
317         result += check_equal_simplify(e, delta_tensor(B_co, C));
318         e = spinor_metric(A_co, B_co) * spinor_metric(B, C);
319         result += check_equal_simplify(e, -delta_tensor(A_co, C));
320         e = spinor_metric(A_co, B_co) * spinor_metric(C, A);
321         result += check_equal_simplify(e, -delta_tensor(B_co, C));
322         e = spinor_metric(A, B) * indexed(psi, B_co);
323         result += check_equal_simplify(e, indexed(psi, A));
324         e = spinor_metric(A, B) * indexed(psi, A_co);
325         result += check_equal_simplify(e, -indexed(psi, B));
326         e = spinor_metric(A_co, B_co) * indexed(psi, B);
327         result += check_equal_simplify(e, -indexed(psi, A_co));
328         e = spinor_metric(A_co, B_co) * indexed(psi, A);
329         result += check_equal_simplify(e, indexed(psi, B_co));
330
331         return result;
332 }
333
334 static unsigned dummy_check(void)
335 {
336         // check dummy index renaming
337
338         unsigned result = 0;
339
340         symbol p("p"), q("q");
341         idx i(symbol("i"), 3), j(symbol("j"), 3), n(symbol("n"), 3);
342         varidx mu(symbol("mu"), 4), nu(symbol("nu"), 4);
343         ex e;
344
345         e = indexed(p, i) * indexed(q, i) - indexed(p, j) * indexed(q, j);
346         result += check_equal_simplify(e, 0);
347
348         e = indexed(p, i) * indexed(p, i) * indexed(q, j) * indexed(q, j)
349           - indexed(p, n) * indexed(p, n) * indexed(q, j) * indexed(q, j);
350         result += check_equal_simplify(e, 0);
351
352         e = indexed(p, mu, mu.toggle_variance()) - indexed(p, nu, nu.toggle_variance());
353         result += check_equal_simplify(e, 0);
354
355         return result;
356 }
357
358 unsigned exam_indexed(void)
359 {
360         unsigned result = 0;
361         
362         cout << "examining indexed objects" << flush;
363         clog << "----------indexed objects:" << endl;
364
365         result += delta_check();  cout << '.' << flush;
366         result += metric_check();  cout << '.' << flush;
367         result += epsilon_check();  cout << '.' << flush;
368         result += symmetry_check();  cout << '.' << flush;
369         result += scalar_product_check();  cout << '.' << flush;
370         result += edyn_check();  cout << '.' << flush;
371         result += spinor_check(); cout << '.' << flush;
372         result += dummy_check(); cout << '.' << flush;
373         
374         if (!result) {
375                 cout << " passed " << endl;
376                 clog << "(no output)" << endl;
377         } else {
378                 cout << " failed " << endl;
379         }
380         
381         return result;
382 }