KADATH
domain_shell_affecte_tau_one_coef.cpp
1 /*
2  Copyright 2017 Philippe Grandclement
3 
4  This file is part of Kadath.
5 
6  Kadath is free software: you can redistribute it and/or modify
7  it under the terms of the GNU General Public License as published by
8  the Free Software Foundation, either version 3 of the License, or
9  (at your option) any later version.
10 
11  Kadath is distributed in the hope that it will be useful,
12  but WITHOUT ANY WARRANTY; without even the implied warranty of
13  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14  GNU General Public License for more details.
15 
16  You should have received a copy of the GNU General Public License
17  along with Kadath. If not, see <http://www.gnu.org/licenses/>.
18 */
19 
20 #include "headcpp.hpp"
21 
22 #include "spheric.hpp"
23 #include "scalar.hpp"
24 #include "tensor_impl.hpp"
25 #include "tensor.hpp"
26 namespace Kadath {
27 
28 void Domain_shell::affecte_tau_one_coef_val_domain_mquant (Val_domain& so, int mquant, int cc, int& conte) const {
29 
30  so.is_zero = false ;
31  so.allocate_coef() ;
32  *so.cf=0. ;
33  Index pos_cf(nbr_coefs) ;
34 
35  bool found = false ;
36 
37  // Loop on theta
38  int baset = (*so.get_base().bases_1d[1]) (0) ;
39  for (int j=0 ; j<nbr_coefs(1) ; j++) {
40  pos_cf.set(1) = j ;
41  bool true_tet = true ;
42  switch (baset) {
43  case COS_EVEN:
44  if ((j==0) && (mquant!=0))
45  true_tet = false ;
46  break ;
47  case COS_ODD:
48  if ((j==nbr_coefs(1)-1) || ((j==0) && (mquant!=0)))
49  true_tet = false ;
50  break ;
51  case SIN_EVEN:
52  if (((j==1) && (mquant>1)) || (j==0) || (j==nbr_coefs(1)-1))
53  true_tet = false ;
54  break ;
55  case SIN_ODD:
56  if (((j==0) && (mquant>1)) || (j==nbr_coefs(1)-1))
57  true_tet = false ;
58  break ;
59  default:
60  cerr << "Unknow theta basis in Domain_shell::affecte_one_coef_val_domain" << endl ;
61  abort() ;
62  }
63 
64  if (true_tet)
65  for (int i=0 ; i<nbr_coefs(0) ; i++) {
66  pos_cf.set(0) = i ;
67  if (conte==cc) {
68  so.cf->set(pos_cf) = 1;
69  found = true ;
70  // regularity ??
71  if ((baset==COS_EVEN) || (baset==COS_ODD))
72  if (mquant!=0) {
73  pos_cf.set(1) = 0 ;
74  so.cf->set(pos_cf) = -1 ;
75  }
76  if (baset==SIN_EVEN)
77  if (mquant>1) {
78  pos_cf.set(1) = 1 ;
79  so.cf->set(pos_cf) = -j ;
80  }
81  if (baset==SIN_ODD)
82  if (mquant>1) {
83  pos_cf.set(1) = 0 ;
84  so.cf->set(pos_cf) = -(2*j+1) ;
85  }
86  }
87  else {
88  so.cf->set(pos_cf) = 0. ;
89  }
90  conte ++ ;
91  }
92  }
93  // If not found put to zero :
94  if (!found)
95  so.set_zero() ;
96 }
97 
98 void Domain_shell::affecte_tau_one_coef_val_domain (Val_domain& so, int mlim, int cc, int& conte) const {
99 
100  int kmin = 2*mlim+2 ;
101 
102  so.is_zero = false ;
103  so.allocate_coef() ;
104  *so.cf=0. ;
105  Index pos_cf(nbr_coefs) ;
106 
107  bool found = false ;
108 
109  // True values
110  // Loop on phi :
111  for (int k=0 ; k<nbr_coefs(2)-1 ; k++)
112  if (k!=1) {
113  pos_cf.set(2) = k ;
114  // Loop on theta
115  int baset = (*so.get_base().bases_1d[1]) (k) ;
116  for (int j=0 ; j<nbr_coefs(1) ; j++) {
117  pos_cf.set(1) = j ;
118  bool true_tet = true ;
119  switch (baset) {
120  case COS_EVEN:
121  if ((j==0) && (k>=kmin))
122  true_tet = false ;
123  break ;
124  case COS_ODD:
125  if ((j==nbr_coefs(1)-1) || ((j==0) && (k>=kmin)))
126  true_tet = false ;
127  break ;
128  case SIN_EVEN:
129  if (((j==1)&&(k>=kmin+2))||(j==0) || (j==nbr_coefs(1)-1))
130  true_tet = false ;
131  break ;
132  case SIN_ODD:
133  if (((j==0)&&(k>=kmin+2))||(j==nbr_coefs(1)-1))
134  true_tet = false ;
135  break ;
136  default:
137  cerr << "Unknow theta basis in Domain_shell::affecte_one_coef_val_domain" << endl ;
138  abort() ;
139  }
140 
141  if (true_tet)
142  for (int i=0 ; i<nbr_coefs(0) ; i++) {
143  pos_cf.set(0) = i ;
144  if (conte==cc) {
145  so.cf->set(pos_cf) = 1;
146  found = true ;
147  // regularity ??
148  if ((baset==COS_EVEN) || (baset==COS_ODD))
149  if (k>=kmin) {
150  pos_cf.set(1) = 0 ;
151  so.cf->set(pos_cf) = -1 ;
152  }
153 
154  if (baset==SIN_EVEN)
155  if (k>=kmin+2) {
156  pos_cf.set(1) = 1 ;
157  so.cf->set(pos_cf) = -j ;
158  }
159  if (baset==SIN_ODD)
160  if (k>=kmin+2) {
161  pos_cf.set(1) = 0 ;
162  so.cf->set(pos_cf) = -(2*j+1) ;
163  }
164 
165 
166  }
167  else {
168  so.cf->set(pos_cf) = 0. ;
169  }
170  conte ++ ;
171  }
172  }
173  }
174  // If not found put to zero :
175  if (!found)
176  so.set_zero() ;
177 }
178 
179 void Domain_shell::affecte_tau_one_coef (Tensor& tt, int dom, int cc, int& pos_cf) const {
180 
181  // Check right domain
182  assert (tt.get_space().get_domain(dom)==this) ;
183 
184  int val = tt.get_valence() ;
185  switch (val) {
186  case 0 :
187  if (tt.is_m_quant_affected()) {
188  // Special case for bosonic field
189  affecte_tau_one_coef_val_domain_mquant (tt.set().set_domain(dom), tt.get_parameters().get_m_quant(), cc, pos_cf) ;
190  }
191  else
192  affecte_tau_one_coef_val_domain (tt.set().set_domain(dom), 0, cc, pos_cf) ;
193  break ;
194  case 1 : {
195  bool found = false ;
196  // Cartesian basis
197  if (tt.get_basis().get_basis(dom)==CARTESIAN_BASIS) {
198  affecte_tau_one_coef_val_domain (tt.set(1).set_domain(dom), 0, cc, pos_cf) ;
199  affecte_tau_one_coef_val_domain (tt.set(2).set_domain(dom), 0, cc, pos_cf) ;
200  affecte_tau_one_coef_val_domain (tt.set(3).set_domain(dom), 0, cc, pos_cf) ;
201  found = true ;
202  }
203  // Spherical coordinates
204  if (tt.get_basis().get_basis(dom)==SPHERICAL_BASIS) {
205  affecte_tau_one_coef_val_domain (tt.set(1).set_domain(dom), 0, cc, pos_cf) ;
206  affecte_tau_one_coef_val_domain (tt.set(2).set_domain(dom), 1, cc, pos_cf) ;
207  affecte_tau_one_coef_val_domain (tt.set(3).set_domain(dom), 1, cc, pos_cf) ;
208  found = true ;
209  }
210  // MTZ coordinates
211  if (tt.get_basis().get_basis(dom)==MTZ_BASIS) {
212  affecte_tau_one_coef_val_domain (tt.set(1).set_domain(dom), 0, cc, pos_cf) ;
213  affecte_tau_one_coef_val_domain (tt.set(2).set_domain(dom), 1, cc, pos_cf) ;
214  affecte_tau_one_coef_val_domain (tt.set(3).set_domain(dom), 1, cc, pos_cf) ;
215  found = true ;
216  }
217  if (!found) {
218  cerr << "Unknown type of vector Domain_shell::affecte_tau_one_coef" << endl ;
219  abort() ;
220  }
221  }
222  break ;
223  case 2 : {
224  bool found = false ;
225  // Cartesian basis and symetric
226  if ((tt.get_basis().get_basis(dom)==CARTESIAN_BASIS) && (tt.get_n_comp()==6)) {
227  affecte_tau_one_coef_val_domain (tt.set(1,1).set_domain(dom), 0, cc, pos_cf) ;
228  affecte_tau_one_coef_val_domain (tt.set(1,2).set_domain(dom), 0, cc, pos_cf) ;
229  affecte_tau_one_coef_val_domain (tt.set(1,3).set_domain(dom), 0, cc, pos_cf) ;
230  affecte_tau_one_coef_val_domain (tt.set(2,2).set_domain(dom), 0, cc, pos_cf) ;
231  affecte_tau_one_coef_val_domain (tt.set(2,3).set_domain(dom), 0, cc, pos_cf) ;
232  affecte_tau_one_coef_val_domain (tt.set(3,3).set_domain(dom), 0, cc, pos_cf) ;
233  found = true ;
234  }
235  // Cartesian basis and not symetric
236  if ((tt.get_basis().get_basis(dom)==CARTESIAN_BASIS) && (tt.get_n_comp()==9)) {
237  affecte_tau_one_coef_val_domain (tt.set(1,1).set_domain(dom), 0, cc, pos_cf) ;
238  affecte_tau_one_coef_val_domain (tt.set(1,2).set_domain(dom), 0, cc, pos_cf) ;
239  affecte_tau_one_coef_val_domain (tt.set(1,3).set_domain(dom), 0, cc, pos_cf) ;
240  affecte_tau_one_coef_val_domain (tt.set(2,1).set_domain(dom), 0, cc, pos_cf) ;
241  affecte_tau_one_coef_val_domain (tt.set(2,2).set_domain(dom), 0, cc, pos_cf) ;
242  affecte_tau_one_coef_val_domain (tt.set(2,3).set_domain(dom), 0, cc, pos_cf) ;
243  affecte_tau_one_coef_val_domain (tt.set(3,1).set_domain(dom), 0, cc, pos_cf) ;
244  affecte_tau_one_coef_val_domain (tt.set(3,2).set_domain(dom), 0, cc, pos_cf) ;
245  affecte_tau_one_coef_val_domain (tt.set(3,3).set_domain(dom), 0, cc, pos_cf) ;
246  found = true ;
247  }
248  // Spherical coordinates and symetric
249  if ((tt.get_basis().get_basis(dom)==SPHERICAL_BASIS) && (tt.get_n_comp()==6)) {
250  affecte_tau_one_coef_val_domain (tt.set(1,1).set_domain(dom), 0,cc ,pos_cf) ;
251  affecte_tau_one_coef_val_domain (tt.set(1,2).set_domain(dom), 1, cc, pos_cf) ;
252  affecte_tau_one_coef_val_domain (tt.set(1,3).set_domain(dom), 1, cc, pos_cf) ;
253  affecte_tau_one_coef_val_domain (tt.set(2,2).set_domain(dom), 2, cc, pos_cf) ;
254  affecte_tau_one_coef_val_domain (tt.set(2,3).set_domain(dom), 2, cc, pos_cf) ;
255  affecte_tau_one_coef_val_domain (tt.set(3,3).set_domain(dom), 2, cc, pos_cf) ;
256  found = true ;
257  }
258  // Spherical coordinates and not symetric
259  if ((tt.get_basis().get_basis(dom)==SPHERICAL_BASIS) && (tt.get_n_comp()==9)) {
260  affecte_tau_one_coef_val_domain (tt.set(1,1).set_domain(dom), 0,cc ,pos_cf) ;
261  affecte_tau_one_coef_val_domain (tt.set(1,2).set_domain(dom), 1, cc, pos_cf) ;
262  affecte_tau_one_coef_val_domain (tt.set(1,3).set_domain(dom), 1, cc, pos_cf) ;
263  affecte_tau_one_coef_val_domain (tt.set(2,1).set_domain(dom), 1, cc, pos_cf) ;
264  affecte_tau_one_coef_val_domain (tt.set(2,2).set_domain(dom), 2, cc, pos_cf) ;
265  affecte_tau_one_coef_val_domain (tt.set(2,3).set_domain(dom), 2, cc, pos_cf) ;
266  affecte_tau_one_coef_val_domain (tt.set(3,1).set_domain(dom), 1, cc, pos_cf) ;
267  affecte_tau_one_coef_val_domain (tt.set(3,2).set_domain(dom), 2, cc, pos_cf) ;
268  affecte_tau_one_coef_val_domain (tt.set(3,3).set_domain(dom), 2, cc, pos_cf) ;
269  found = true ;
270  }
271  // MTZ coordinates and symetric
272  if ((tt.get_basis().get_basis(dom)==MTZ_BASIS) && (tt.get_n_comp()==6)) {
273  affecte_tau_one_coef_val_domain (tt.set(1,1).set_domain(dom), 0,cc ,pos_cf) ;
274  affecte_tau_one_coef_val_domain (tt.set(1,2).set_domain(dom), 1, cc, pos_cf) ;
275  affecte_tau_one_coef_val_domain (tt.set(1,3).set_domain(dom), 1, cc, pos_cf) ;
276  affecte_tau_one_coef_val_domain (tt.set(2,2).set_domain(dom), 2, cc, pos_cf) ;
277  affecte_tau_one_coef_val_domain (tt.set(2,3).set_domain(dom), 2, cc, pos_cf) ;
278  affecte_tau_one_coef_val_domain (tt.set(3,3).set_domain(dom), 2, cc, pos_cf) ;
279  found = true ;
280  }
281  // MTZ coordinates and not symetric
282  if ((tt.get_basis().get_basis(dom)==MTZ_BASIS) && (tt.get_n_comp()==9)) {
283  affecte_tau_one_coef_val_domain (tt.set(1,1).set_domain(dom), 0,cc ,pos_cf) ;
284  affecte_tau_one_coef_val_domain (tt.set(1,2).set_domain(dom), 1, cc, pos_cf) ;
285  affecte_tau_one_coef_val_domain (tt.set(1,3).set_domain(dom), 1, cc, pos_cf) ;
286  affecte_tau_one_coef_val_domain (tt.set(2,1).set_domain(dom), 1, cc, pos_cf) ;
287  affecte_tau_one_coef_val_domain (tt.set(2,2).set_domain(dom), 2, cc, pos_cf) ;
288  affecte_tau_one_coef_val_domain (tt.set(2,3).set_domain(dom), 2, cc, pos_cf) ;
289  affecte_tau_one_coef_val_domain (tt.set(3,1).set_domain(dom), 1, cc, pos_cf) ;
290  affecte_tau_one_coef_val_domain (tt.set(3,2).set_domain(dom), 2, cc, pos_cf) ;
291  affecte_tau_one_coef_val_domain (tt.set(3,3).set_domain(dom), 2, cc, pos_cf) ;
292  found = true ;
293  }
294  if (!found) {
295  cerr << "Unknown type of 2-tensor Domain_shell::affecte_tau_one_coef" << endl ;
296  abort() ;
297  }
298  }
299  break ;
300  default :
301  cerr << "Valence " << val << " not implemented in Domain_shell::affecte_tau" << endl ;
302  break ;
303  }
304 }}
Kadath::Array::set
reference set(const Index &pos)
Read/write of an element.
Definition: array.hpp:186
Kadath::Base_spectral::bases_1d
Bases_container bases_1d
Arrays containing the various basis of decomposition.
Definition: base_spectral.hpp:82
Kadath::Base_tensor::get_basis
int get_basis(int nd) const
Read only the basis in a given domain.
Definition: base_tensor.hpp:93
Kadath::Domain_shell::affecte_tau_one_coef_val_domain
void affecte_tau_one_coef_val_domain(Val_domain &so, int mlim, int cc, int &pos_cf) const
Sets at most one coefficient of a Val_domain to 1.
Definition: domain_shell_affecte_tau_one_coef.cpp:98
Kadath::Domain_shell::affecte_tau_one_coef
virtual void affecte_tau_one_coef(Tensor &, int, int, int &) const
Sets at most one coefficient of a Tensor to 1.
Definition: domain_shell_affecte_tau_one_coef.cpp:179
Kadath::Domain_shell::affecte_tau_one_coef_val_domain_mquant
void affecte_tau_one_coef_val_domain_mquant(Val_domain &so, int mquant, int cc, int &pos_cf) const
Sets at most one coefficient of a Val_domain to 1.
Definition: domain_shell_affecte_tau_one_coef.cpp:28
Kadath::Domain::nbr_coefs
Dim_array nbr_coefs
Number of coefficients.
Definition: space.hpp:66
Kadath::Index
Class that gives the position inside a multi-dimensional Array.
Definition: index.hpp:38
Kadath::Index::set
int & set(int i)
Read/write of the position in a given dimension.
Definition: index.hpp:72
Kadath::Param_tensor::get_m_quant
int get_m_quant() const
Returns .
Definition: tensor.hpp:747
Kadath::Scalar::set_domain
Val_domain & set_domain(int)
Read/write of a particular Val_domain.
Definition: scalar.hpp:555
Kadath::Space::get_domain
const Domain * get_domain(int i) const
returns a pointer on the domain.
Definition: space.hpp:1385
Kadath::Tensor
Tensor handling.
Definition: tensor.hpp:149
Kadath::Tensor::get_parameters
const Param_tensor & get_parameters() const
Returns a pointer on the possible additional parameter.
Definition: tensor.hpp:311
Kadath::Tensor::set
Scalar & set(const Array< int > &ind)
Returns the value of a component (read/write version).
Definition: tensor_impl.hpp:91
Kadath::Tensor::get_basis
const Base_tensor & get_basis() const
Returns the vectorial basis (triad) on which the components are defined.
Definition: tensor.hpp:504
Kadath::Tensor::get_n_comp
int get_n_comp() const
Returns the number of stored components.
Definition: tensor.hpp:514
Kadath::Tensor::get_valence
int get_valence() const
Returns the valence.
Definition: tensor.hpp:509
Kadath::Tensor::is_m_quant_affected
bool is_m_quant_affected() const
Checks whether the additional parameter is affected (used for boson stars for instance).
Definition: tensor.hpp:326
Kadath::Tensor::get_space
const Space & get_space() const
Returns the Space.
Definition: tensor.hpp:499
Kadath::Val_domain
Class for storing the basis of decompositions of a field and its values on both the configuration and...
Definition: val_domain.hpp:69
Kadath::Val_domain::set_zero
void set_zero()
Sets the Val_domain to zero (logical state to zero and arrays destroyed).
Definition: val_domain.cpp:223
Kadath::Val_domain::allocate_coef
void allocate_coef()
Allocates the values in the coefficient space and destroys the values in the configuration space.
Definition: val_domain.cpp:216
Kadath::Val_domain::cf
Array< double > * cf
Pointer on the Array of the values in the coefficients space.
Definition: val_domain.hpp:77
Kadath::Val_domain::is_zero
bool is_zero
Indicator used for null fields (for speed issues).
Definition: val_domain.hpp:74
Kadath::Val_domain::get_base
const Base_spectral & get_base() const
Returns the basis of decomposition.
Definition: val_domain.hpp:122