source: issm/trunk-jpl/src/c/analyses/SealevelriseAnalysis.cpp@ 21742

Last change on this file since 21742 was 21742, checked in by Eric.Larour, 8 years ago

CHG: shield ensureContiguous statement.
File size: 10.3 KB
Line 
1#include "./SealevelriseAnalysis.h"
2#include "../toolkits/toolkits.h"
3#include "../classes/classes.h"
4#include "../shared/shared.h"
5#include "../modules/modules.h"
6
7/*Model processing*/
8void SealevelriseAnalysis::CreateConstraints(Constraints* constraints,IoModel* iomodel){/*{{{*/
9 /*No constraints*/
10}/*}}}*/
11void SealevelriseAnalysis::CreateLoads(Loads* loads, IoModel* iomodel){/*{{{*/
12 /*No loads*/
13}/*}}}*/
14void SealevelriseAnalysis::CreateNodes(Nodes* nodes,IoModel* iomodel){/*{{{*/
15 ::CreateNodes(nodes,iomodel,SealevelriseAnalysisEnum,P1Enum);
16}/*}}}*/
17int SealevelriseAnalysis::DofsPerNode(int** doflist,int domaintype,int approximation){/*{{{*/
18 return 1;
19}/*}}}*/
20void SealevelriseAnalysis::UpdateElements(Elements* elements,IoModel* iomodel,int analysis_counter,int analysis_type){/*{{{*/
21
22 /*Update elements: */
23 int counter=0;
24 for(int i=0;i<iomodel->numberofelements;i++){
25 if(iomodel->my_elements[i]){
26 Element* element=(Element*)elements->GetObjectByOffset(counter);
27 element->Update(i,iomodel,analysis_counter,analysis_type,P1Enum);
28 counter++;
29 }
30 }
31
32 /*Create inputs: */
33 iomodel->FetchDataToInput(elements,"md.mask.ice_levelset",MaskIceLevelsetEnum);
34 iomodel->FetchDataToInput(elements,"md.mask.ocean_levelset",MaskOceanLevelsetEnum);
35 iomodel->FetchDataToInput(elements,"md.mask.land_levelset",MaskLandLevelsetEnum);
36 iomodel->FetchDataToInput(elements,"md.slr.deltathickness",SealevelriseDeltathicknessEnum);
37 iomodel->FetchDataToInput(elements,"md.slr.sealevel",SealevelEnum,0);
38
39}/*}}}*/
40void SealevelriseAnalysis::UpdateParameters(Parameters* parameters,IoModel* iomodel,int solution_enum,int analysis_enum){/*{{{*/
41
42 int nl;
43 IssmDouble* love_h=NULL;
44 IssmDouble* love_k=NULL;
45 IssmDouble* love_l=NULL;
46
47 bool elastic=false;
48 IssmDouble* G_elastic = NULL;
49 IssmDouble* G_elastic_local = NULL;
50 IssmDouble* U_elastic = NULL;
51 IssmDouble* U_elastic_local = NULL;
52 IssmDouble* H_elastic = NULL;
53 IssmDouble* H_elastic_local = NULL;
54 int M,m,lower_row,upper_row;
55 IssmDouble degacc=.01;
56
57 int numoutputs;
58 char** requestedoutputs = NULL;
59
60 /*transition vectors: */
61 IssmDouble **transitions = NULL;
62 int *transitions_M = NULL;
63 int *transitions_N = NULL;
64 int ntransitions;
65
66 /*some constant parameters: */
67 parameters->AddObject(iomodel->CopyConstantObject("md.slr.reltol",SealevelriseReltolEnum));
68 parameters->AddObject(iomodel->CopyConstantObject("md.slr.abstol",SealevelriseAbstolEnum));
69 parameters->AddObject(iomodel->CopyConstantObject("md.slr.maxiter",SealevelriseMaxiterEnum));
70 parameters->AddObject(iomodel->CopyConstantObject("md.slr.rigid",SealevelriseRigidEnum));
71 parameters->AddObject(iomodel->CopyConstantObject("md.slr.elastic",SealevelriseElasticEnum));
72 parameters->AddObject(iomodel->CopyConstantObject("md.slr.rotation",SealevelriseRotationEnum));
73 parameters->AddObject(iomodel->CopyConstantObject("md.slr.tide_love_h",SealevelriseTidalLoveHEnum));
74 parameters->AddObject(iomodel->CopyConstantObject("md.slr.tide_love_k",SealevelriseTidalLoveKEnum));
75 parameters->AddObject(iomodel->CopyConstantObject("md.slr.fluid_love",SealevelriseFluidLoveEnum));
76 parameters->AddObject(iomodel->CopyConstantObject("md.slr.equatorial_moi",SealevelriseEquatorialMoiEnum));
77 parameters->AddObject(iomodel->CopyConstantObject("md.slr.polar_moi",SealevelrisePolarMoiEnum));
78 parameters->AddObject(iomodel->CopyConstantObject("md.slr.angular_velocity",SealevelriseAngularVelocityEnum));
79 parameters->AddObject(iomodel->CopyConstantObject("md.slr.ocean_area_scaling",SealevelriseOceanAreaScalingEnum));
80
81 iomodel->FetchData(&elastic,"md.slr.elastic");
82 if(elastic){
83
84 /*love numbers: */
85 iomodel->FetchData(&love_h,&nl,NULL,"md.slr.love_h");
86 iomodel->FetchData(&love_k,&nl,NULL,"md.slr.love_k");
87 iomodel->FetchData(&love_l,&nl,NULL,"md.slr.love_l");
88
89 /*compute elastic green function for a range of angles*/
90 iomodel->FetchData(&degacc,"md.slr.degacc");
91 M=reCast<int,IssmDouble>(180./degacc+1.);
92
93 // AD performance is sensitive to calls to ensurecontiguous.
94 // // Providing "t" will cause ensurecontiguous to be called.
95 #ifdef _HAVE_AD_
96 G_elastic=xNew<IssmDouble>(M,"t");
97 U_elastic=xNew<IssmDouble>(M,"t");
98 H_elastic=xNew<IssmDouble>(M,"t");
99 #else
100 G_elastic=xNew<IssmDouble>(M);
101 U_elastic=xNew<IssmDouble>(M);
102 H_elastic=xNew<IssmDouble>(M);
103 #endif
104
105
106 /*compute combined legendre + love number (elastic green function:*/
107 m=DetermineLocalSize(M,IssmComm::GetComm());
108 GetOwnershipBoundariesFromRange(&lower_row,&upper_row,m,IssmComm::GetComm());
109 // AD performance is sensitive to calls to ensurecontiguous.
110 // // Providing "t" will cause ensurecontiguous to be called.
111 #ifdef _HAVE_AD_
112 G_elastic_local=xNew<IssmDouble>(m,"t");
113 U_elastic_local=xNew<IssmDouble>(m,"t");
114 H_elastic_local=xNew<IssmDouble>(m,"t");
115 #else
116 G_elastic_local=xNew<IssmDouble>(m);
117 U_elastic_local=xNew<IssmDouble>(m);
118 H_elastic_local=xNew<IssmDouble>(m);
119 #endif
120
121 for(int i=lower_row;i<upper_row;i++){
122 IssmDouble alpha,x;
123 alpha= reCast<IssmDouble>(i)*degacc * PI / 180.0;
124
125 G_elastic_local[i-lower_row]= (love_k[nl-1]-love_h[nl-1])/2.0/sin(alpha/2.0);
126 U_elastic_local[i-lower_row]= (love_h[nl-1])/2.0/sin(alpha/2.0);
127 H_elastic_local[i-lower_row]= 0;
128 IssmDouble Pn,Pn1,Pn2;
129 IssmDouble Pn_p,Pn_p1,Pn_p2;
130 for (int n=0;n<nl;n++) {
131 IssmDouble deltalove_G;
132 IssmDouble deltalove_U;
133
134 deltalove_G = (love_k[n]-love_k[nl-1]-love_h[n]+love_h[nl-1]);
135 deltalove_U = (love_h[n]-love_h[nl-1]);
136
137 /*compute legendre polynomials: P_n(cos\theta) & d P_n(cos\theta)/ d\theta: */
138 if(n==0){
139 Pn=1;
140 Pn_p=0;
141 }
142 else if(n==1){
143 Pn = cos(alpha);
144 Pn_p = 1;
145 }
146 else{
147 Pn = ( (2*n-1)*cos(alpha)*Pn1 - (n-1)*Pn2 ) /n;
148 Pn_p = ( (2*n-1)*(Pn1+cos(alpha)*Pn_p1) - (n-1)*Pn_p2 ) /n;
149 }
150 Pn2=Pn1; Pn1=Pn;
151 Pn_p2=Pn_p1; Pn_p1=Pn_p;
152
153 G_elastic_local[i-lower_row] += deltalove_G*Pn; // gravitational potential
154 U_elastic_local[i-lower_row] += deltalove_U*Pn; // vertical (up) displacement
155 H_elastic_local[i-lower_row] += sin(alpha)*love_l[n]*Pn_p; // horizontal displacements
156 }
157 }
158
159 /*merge G_elastic_local into G_elastic; U_elastic_local into U_elastic; H_elastic_local to H_elastic:{{{*/
160 int* recvcounts=xNew<int>(IssmComm::GetSize());
161 int* displs=xNew<int>(IssmComm::GetSize());
162
163 //recvcounts:
164 ISSM_MPI_Allgather(&m,1,ISSM_MPI_INT,recvcounts,1,ISSM_MPI_INT,IssmComm::GetComm());
165
166 /*displs: */
167 ISSM_MPI_Allgather(&lower_row,1,ISSM_MPI_INT,displs,1,ISSM_MPI_INT,IssmComm::GetComm());
168
169 /*All gather:*/
170 ISSM_MPI_Allgatherv(G_elastic_local, m, ISSM_MPI_DOUBLE, G_elastic, recvcounts, displs, ISSM_MPI_DOUBLE,IssmComm::GetComm());
171 ISSM_MPI_Allgatherv(U_elastic_local, m, ISSM_MPI_DOUBLE, U_elastic, recvcounts, displs, ISSM_MPI_DOUBLE,IssmComm::GetComm());
172 ISSM_MPI_Allgatherv(H_elastic_local, m, ISSM_MPI_DOUBLE, H_elastic, recvcounts, displs, ISSM_MPI_DOUBLE,IssmComm::GetComm());
173 /*free ressources: */
174 xDelete<int>(recvcounts);
175 xDelete<int>(displs);
176
177 /*}}}*/
178
179 /*Avoid singularity at 0: */
180 G_elastic[0]=G_elastic[1];
181 parameters->AddObject(new DoubleVecParam(SealevelriseGElasticEnum,G_elastic,M));
182 U_elastic[0]=U_elastic[1];
183 parameters->AddObject(new DoubleVecParam(SealevelriseUElasticEnum,U_elastic,M));
184 H_elastic[0]=H_elastic[1];
185 parameters->AddObject(new DoubleVecParam(SealevelriseHElasticEnum,H_elastic,M));
186
187 /*free ressources: */
188 xDelete<IssmDouble>(love_h);
189 xDelete<IssmDouble>(love_k);
190 xDelete<IssmDouble>(love_l);
191 xDelete<IssmDouble>(G_elastic);
192 xDelete<IssmDouble>(G_elastic_local);
193 xDelete<IssmDouble>(U_elastic);
194 xDelete<IssmDouble>(U_elastic_local);
195 xDelete<IssmDouble>(H_elastic);
196 xDelete<IssmDouble>(H_elastic_local);
197 }
198
199 /*Transitions: */
200 iomodel->FetchData(&transitions,&transitions_M,&transitions_N,&ntransitions,"md.slr.transitions");
201 if(transitions){
202 parameters->AddObject(new DoubleMatArrayParam(SealevelriseTransitionsEnum,transitions,ntransitions,transitions_M,transitions_N));
203
204 for(int i=0;i<ntransitions;i++){
205 IssmDouble* transition=transitions[i];
206 xDelete<IssmDouble>(transition);
207 }
208 xDelete<IssmDouble*>(transitions);
209 xDelete<int>(transitions_M);
210 xDelete<int>(transitions_N);
211 }
212
213 /*Requested outputs*/
214 iomodel->FindConstant(&requestedoutputs,&numoutputs,"md.slr.requested_outputs");
215 parameters->AddObject(new IntParam(SealevelriseNumRequestedOutputsEnum,numoutputs));
216 if(numoutputs)parameters->AddObject(new StringArrayParam(SealevelriseRequestedOutputsEnum,requestedoutputs,numoutputs));
217 iomodel->DeleteData(&requestedoutputs,numoutputs,"md.slr.requested_outputs");
218
219
220}/*}}}*/
221
222/*Finite Element Analysis*/
223void SealevelriseAnalysis::Core(FemModel* femmodel){/*{{{*/
224 _error_("not implemented");
225}/*}}}*/
226ElementVector* SealevelriseAnalysis::CreateDVector(Element* element){/*{{{*/
227 /*Default, return NULL*/
228 return NULL;
229}/*}}}*/
230ElementMatrix* SealevelriseAnalysis::CreateJacobianMatrix(Element* element){/*{{{*/
231_error_("Not implemented");
232}/*}}}*/
233ElementMatrix* SealevelriseAnalysis::CreateKMatrix(Element* element){/*{{{*/
234 _error_("not implemented yet");
235}/*}}}*/
236ElementVector* SealevelriseAnalysis::CreatePVector(Element* element){/*{{{*/
237_error_("not implemented yet");
238}/*}}}*/
239void SealevelriseAnalysis::GetSolutionFromInputs(Vector<IssmDouble>* solution,Element* element){/*{{{*/
240 _error_("not implemented yet");
241}/*}}}*/
242void SealevelriseAnalysis::GradientJ(Vector<IssmDouble>* gradient,Element* element,int control_type,int control_index){/*{{{*/
243 _error_("Not implemented yet");
244}/*}}}*/
245void SealevelriseAnalysis::InputUpdateFromSolution(IssmDouble* solution,Element* element){/*{{{*/
246
247 IssmDouble *deltaS = NULL;
248 IssmDouble *S = NULL;
249 int* sidlist = NULL;
250 int numvertices;
251
252 numvertices= element->GetNumberOfVertices();
253 sidlist=xNew<int>(numvertices);
254
255 element->GetVerticesSidList(sidlist);
256
257 deltaS = xNew<IssmDouble>(numvertices);
258 for(int i=0;i<numvertices;i++){
259 deltaS[i]=solution[sidlist[i]];
260 }
261
262 S = xNew<IssmDouble>(numvertices);
263 element->GetInputListOnVertices(S,SealevelEnum,0);
264
265 /*Add deltaS to S:*/
266 for (int i=0;i<numvertices;i++)S[i]+=deltaS[i];
267
268 /*Add S back into inputs: */
269 element->AddInput(SealevelEnum,S,P1Enum);
270
271 /*Free ressources:*/
272 xDelete<int>(sidlist);
273 xDelete<IssmDouble>(deltaS);
274 xDelete<IssmDouble>(S);
275
276}/*}}}*/
277void SealevelriseAnalysis::UpdateConstraints(FemModel* femmodel){/*{{{*/
278 /*Default, do nothing*/
279 return;
280}/*}}}*/
Note: See TracBrowser for help on using the repository browser.