added checks for new indexed objects
[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 symmetry_check(void)
107 {
108         // check symmetric/antisymmetric objects
109
110         unsigned result = 0;
111
112         symbol s_i("i"), s_j("j"), s_k("k");
113         idx i(s_i, 3), j(s_j, 3), k(s_k, 3);
114         symbol A("A");
115         ex e, e1, e2;
116
117         result += check_equal(indexed(A, indexed::symmetric, i, j), indexed(A, indexed::symmetric, j, i));
118         result += check_equal(indexed(A, indexed::antisymmetric, i, j) + indexed(A, indexed::antisymmetric, j, i), 0);
119         result += check_equal(indexed(A, indexed::antisymmetric, i, j, k) - indexed(A, indexed::antisymmetric, j, k, i), 0);
120
121         return result;
122 }
123
124 static unsigned edyn_check(void)
125 {
126         // relativistic electrodynamics: check transformation laws of electric
127         // and magnetic fields by applying a Lorentz boost to the field tensor
128
129         unsigned result = 0;
130
131         symbol beta("beta");
132         ex gamma = 1 / sqrt(1 - pow(beta, 2));
133         symbol Ex("Ex"), Ey("Ey"), Ez("Ez");
134         symbol Bx("Bx"), By("By"), Bz("Bz");
135
136         // Lorentz transformation matrix (boost along x axis)
137         matrix L(4, 4);
138         L.set(0, 0, gamma);
139         L.set(0, 1, -beta*gamma);
140         L.set(1, 0, -beta*gamma);
141         L.set(1, 1, gamma);
142         L.set(2, 2, 1);
143         L.set(3, 3, 1);
144
145         // Electromagnetic field tensor
146         matrix F(4, 4);
147         F.set(0, 1, -Ex);
148         F.set(1, 0, Ex);
149         F.set(0, 2, -Ey);
150         F.set(2, 0, Ey);
151         F.set(0, 3, -Ez);
152         F.set(3, 0, Ez);
153         F.set(1, 2, -Bz);
154         F.set(2, 1, Bz);
155         F.set(1, 3, By);
156         F.set(3, 1, -By);
157         F.set(2, 3, -Bx);
158         F.set(3, 2, Bx);
159
160         // Indices
161         symbol s_mu("mu"), s_nu("nu"), s_rho("rho"), s_sigma("sigma");
162         varidx mu(s_mu, 4), nu(s_nu, 4), rho(s_rho, 4), sigma(s_sigma, 4);
163
164         // Apply transformation law of second rank tensor
165         ex e = (indexed(L, mu, rho.toggle_variance())
166            * indexed(L, nu, sigma.toggle_variance())
167            * indexed(F, rho, sigma)).simplify_indexed();
168
169         // Extract transformed electric and magnetic fields
170         ex Ex_p = e.subs(lst(mu == 1, nu == 0)).normal();
171         ex Ey_p = e.subs(lst(mu == 2, nu == 0)).normal();
172         ex Ez_p = e.subs(lst(mu == 3, nu == 0)).normal();
173         ex Bx_p = e.subs(lst(mu == 3, nu == 2)).normal();
174         ex By_p = e.subs(lst(mu == 1, nu == 3)).normal();
175         ex Bz_p = e.subs(lst(mu == 2, nu == 1)).normal();
176
177         // Check results
178         result += check_equal(Ex_p, Ex);
179         result += check_equal(Ey_p, gamma * (Ey - beta * Bz));
180         result += check_equal(Ez_p, gamma * (Ez + beta * By));
181         result += check_equal(Bx_p, Bx);
182         result += check_equal(By_p, gamma * (By + beta * Ez));
183         result += check_equal(Bz_p, gamma * (Bz - beta * Ey));
184
185         return result;
186 }
187
188 unsigned exam_indexed(void)
189 {
190         unsigned result = 0;
191         
192         cout << "examining indexed objects" << flush;
193         clog << "----------indexed objects:" << endl;
194
195         result += delta_check();  cout << '.' << flush;
196         result += metric_check();  cout << '.' << flush;
197         result += symmetry_check();  cout << '.' << flush;
198         result += edyn_check();  cout << '.' << flush;
199         
200         if (!result) {
201                 cout << " passed " << endl;
202                 clog << "(no output)" << endl;
203         } else {
204                 cout << " failed " << endl;
205         }
206         
207         return result;
208 }