/[escript]/trunk/finley/src/Assemble_LumpedSystem.c
ViewVC logotype

Diff of /trunk/finley/src/Assemble_LumpedSystem.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

trunk/esys2/finley/src/finleyC/Assemble_PDE.c revision 150 by jgs, Thu Sep 15 03:44:45 2005 UTC trunk/finley/src/Assemble_PDE.c revision 969 by ksteube, Tue Feb 13 23:02:23 2007 UTC
# Line 1  Line 1 
1  /*  /*
2   ******************************************************************************   ************************************************************
3   *                                                                            *   *          Copyright 2006 by ACcESS MNRF                   *
4   *       COPYRIGHT  ACcESS 2003,2004,2005 -  All Rights Reserved              *   *                                                          *
5   *                                                                            *   *              http://www.access.edu.au                    *
6   * This software is the property of ACcESS. No part of this code              *   *       Primary Business: Queensland, Australia            *
7   * may be copied in any form or by any means without the expressed written    *   *  Licensed under the Open Software License version 3.0    *
8   * consent of ACcESS.  Copying, use or modification of this software          *   *     http://www.opensource.org/licenses/osl-3.0.php       *
9   * by any unauthorised person is illegal unless that person has a software    *   *                                                          *
10   * license agreement with ACcESS.                                             *   ************************************************************
  *                                                                            *  
  ******************************************************************************  
11  */  */
12    
13    
# Line 58  Line 56 
56  void Finley_Assemble_PDE(Finley_NodeFile* nodes,Finley_ElementFile* elements,Paso_SystemMatrix* S, escriptDataC* F,  void Finley_Assemble_PDE(Finley_NodeFile* nodes,Finley_ElementFile* elements,Paso_SystemMatrix* S, escriptDataC* F,
57               escriptDataC* A, escriptDataC* B, escriptDataC* C, escriptDataC* D, escriptDataC* X, escriptDataC* Y ) {               escriptDataC* A, escriptDataC* B, escriptDataC* C, escriptDataC* D, escriptDataC* X, escriptDataC* Y ) {
58    
59      bool_t reducedIntegrationOrder=FALSE;
60    char error_msg[LenErrorMsg_MAX];    char error_msg[LenErrorMsg_MAX];
   double *EM_S=NULL,*EM_F=NULL,*V=NULL,*dVdv=NULL,*dSdV=NULL,*Vol=NULL,*dvdV=NULL;  
   double time0;  
   dim_t dimensions[ESCRIPT_MAX_DATA_RANK],e,q;  
61    Assemble_Parameters p;    Assemble_Parameters p;
62    index_t *index_row=NULL,*index_col=NULL,color;    double time0;
63      dim_t dimensions[ESCRIPT_MAX_DATA_RANK];
64    
65    Finley_resetError();    Finley_resetError();
66    
67      {
68      int iam, numCPUs;
69    #ifdef Paso_MPI
70      iam = elements->elementDistribution->MPIInfo->rank;
71      numCPUs = elements->elementDistribution->MPIInfo->size;
72      printf("ksteube finley/src/Assemble_PDE.c : Finley_Assemble_PDE\n");
73      printf("ksteube size=%d iam=%d\n", numCPUs,  iam);
74      printf("ksteube iam=%d #elements: L=%d I=%d B=%d\n", iam, elements->elementDistribution->numLocal, elements->elementDistribution->numInternal, elements->elementDistribution->numBoundary);
75    #endif
76      }
77    
78    if (nodes==NULL || elements==NULL) return;    if (nodes==NULL || elements==NULL) return;
79    if (S==NULL && isEmpty(F)) return;    if (S==NULL && isEmpty(F)) return;
80    
81    /* set all parameters in p*/    if (isEmpty(F) && !isEmpty(X) && !isEmpty(F)) {
82    Assemble_getAssembleParameters(nodes,elements,S,F,&p);          Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: right hand side coefficients are non-zero bat no right hand side vector given.");
83    if (! Finley_noError()) return;    }
84    
85    /*  this function assumes NS=NN */    if (S==NULL && !isEmpty(A) && !isEmpty(B) && !isEmpty(C) && !isEmpty(D)) {
86    if (p.NN!=p.NS) {          Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: coefficients are non-zero but no matrix is given.");
     Finley_setError(SYSTEM_ERROR,"__FILE__: for Finley_Assemble_PDE numNodes and numShapes have to be identical.");  
     return;  
   }  
   if (p.numDim!=p.numElementDim) {  
     Finley_setError(SYSTEM_ERROR,"__FILE__: Finley_Assemble_PDE accepts volume elements only.");  
     return;  
87    }    }
88    /*  get a functionspace */  
89      /*  get the functionspace for this assemblage call */
90    type_t funcspace=UNKNOWN;    type_t funcspace=UNKNOWN;
91    updateFunctionSpaceType(funcspace,A);    updateFunctionSpaceType(funcspace,A);
92    updateFunctionSpaceType(funcspace,B);    updateFunctionSpaceType(funcspace,B);
# Line 92  void Finley_Assemble_PDE(Finley_NodeFile Line 96  void Finley_Assemble_PDE(Finley_NodeFile
96    updateFunctionSpaceType(funcspace,Y);    updateFunctionSpaceType(funcspace,Y);
97    if (funcspace==UNKNOWN) return; /* all  data are empty */    if (funcspace==UNKNOWN) return; /* all  data are empty */
98    
   /* check if all function spaces are the same */  
99    
100      /* check if all function spaces are the same */
101    if (! functionSpaceTypeEqual(funcspace,A) ) {    if (! functionSpaceTypeEqual(funcspace,A) ) {
102          Finley_setError(TYPE_ERROR,"__FILE__: unexpected function space type for coefficient A");          Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: unexpected function space type for coefficient A");
103    }    }
104    if (! functionSpaceTypeEqual(funcspace,B) ) {    if (! functionSpaceTypeEqual(funcspace,B) ) {
105          Finley_setError(TYPE_ERROR,"__FILE__: unexpected function space type for coefficient B");          Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: unexpected function space type for coefficient B");
106    }    }
107    if (! functionSpaceTypeEqual(funcspace,C) ) {    if (! functionSpaceTypeEqual(funcspace,C) ) {
108          Finley_setError(TYPE_ERROR,"__FILE__: unexpected function space type for coefficient C");          Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: unexpected function space type for coefficient C");
109    }    }
110    if (! functionSpaceTypeEqual(funcspace,D) ) {    if (! functionSpaceTypeEqual(funcspace,D) ) {
111          Finley_setError(TYPE_ERROR,"__FILE__: unexpected function space type for coefficient D");          Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: unexpected function space type for coefficient D");
112    }    }
113    if (! functionSpaceTypeEqual(funcspace,X) ) {    if (! functionSpaceTypeEqual(funcspace,X) ) {
114          Finley_setError(TYPE_ERROR,"__FILE__: unexpected function space type for coefficient X");          Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: unexpected function space type for coefficient X");
115    }    }
116    if (! functionSpaceTypeEqual(funcspace,Y) ) {    if (! functionSpaceTypeEqual(funcspace,Y) ) {
117          Finley_setError(TYPE_ERROR,"__FILE__: unexpected function space type for coefficient Y");          Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: unexpected function space type for coefficient Y");
118    }    }
119      if (! Finley_noError()) return;
120    
121      /* check if all function spaces are the same */
122      if (funcspace==FINLEY_ELEMENTS) {
123           reducedIntegrationOrder=FALSE;
124      } else if (funcspace==FINLEY_FACE_ELEMENTS)  {
125           reducedIntegrationOrder=FALSE;
126      } else if (funcspace==FINLEY_CONTACT_ELEMENTS_1)  {
127           reducedIntegrationOrder=FALSE;
128      } else if (funcspace==FINLEY_CONTACT_ELEMENTS_2)  {
129           reducedIntegrationOrder=FALSE;
130      } else {
131           Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: assemblage failed because of illegal function space.");
132      }
133      if (! Finley_noError()) return;
134    
135      /* set all parameters in p*/
136      Assemble_getAssembleParameters(nodes,elements,S,F, reducedIntegrationOrder, &p);
137      if (! Finley_noError()) return;
138    
139    /* check if all function spaces are the same */    /* check if all function spaces are the same */
140    
141    if (! numSamplesEqual(A,p.numQuad,elements->numElements) ) {    if (! numSamplesEqual(A,p.numQuad,elements->numElements) ) {
142          sprintf(error_msg,"__FILE__: sample points of coefficient A don't match (%d,%d)",p.numQuad,elements->numElements);          sprintf(error_msg,"Finley_Assemble_PDE: sample points of coefficient A don't match (%d,%d)",p.numQuad,elements->numElements);
143          Finley_setError(TYPE_ERROR,error_msg);          Finley_setError(TYPE_ERROR,error_msg);
144    }    }
145    
146    if (! numSamplesEqual(B,p.numQuad,elements->numElements) ) {    if (! numSamplesEqual(B,p.numQuad,elements->numElements) ) {
147          sprintf(error_msg,"__FILE__: sample points of coefficient B don't match (%d,%d)",p.numQuad,elements->numElements);          sprintf(error_msg,"Finley_Assemble_PDE: sample points of coefficient B don't match (%d,%d)",p.numQuad,elements->numElements);
148          Finley_setError(TYPE_ERROR,error_msg);          Finley_setError(TYPE_ERROR,error_msg);
149    }    }
150    
151    if (! numSamplesEqual(C,p.numQuad,elements->numElements) ) {    if (! numSamplesEqual(C,p.numQuad,elements->numElements) ) {
152          sprintf(error_msg,"__FILE__: sample points of coefficient C don't match (%d,%d)",p.numQuad,elements->numElements);          sprintf(error_msg,"Finley_Assemble_PDE: sample points of coefficient C don't match (%d,%d)",p.numQuad,elements->numElements);
153          Finley_setError(TYPE_ERROR,error_msg);          Finley_setError(TYPE_ERROR,error_msg);
154    }    }
155    
156    if (! numSamplesEqual(D,p.numQuad,elements->numElements) ) {    if (! numSamplesEqual(D,p.numQuad,elements->numElements) ) {
157          sprintf(error_msg,"__FILE__: sample points of coefficient D don't match (%d,%d)",p.numQuad,elements->numElements);          sprintf(error_msg,"Finley_Assemble_PDE: sample points of coefficient D don't match (%d,%d)",p.numQuad,elements->numElements);
158          Finley_setError(TYPE_ERROR,error_msg);          Finley_setError(TYPE_ERROR,error_msg);
159    }    }
160    
161    if (! numSamplesEqual(X,p.numQuad,elements->numElements) ) {    if (! numSamplesEqual(X,p.numQuad,elements->numElements) ) {
162          sprintf(error_msg,"__FILE__: sample points of coefficient X don't match (%d,%d)",p.numQuad,elements->numElements);          sprintf(error_msg,"Finley_Assemble_PDE: sample points of coefficient X don't match (%d,%d)",p.numQuad,elements->numElements);
163          Finley_setError(TYPE_ERROR,error_msg);          Finley_setError(TYPE_ERROR,error_msg);
164    }    }
165    
166    if (! numSamplesEqual(Y,p.numQuad,elements->numElements) ) {    if (! numSamplesEqual(Y,p.numQuad,elements->numElements) ) {
167          sprintf(error_msg,"__FILE__: sample points of coefficient Y don't match (%d,%d)",p.numQuad,elements->numElements);          sprintf(error_msg,"Finley_Assemble_PDE: sample points of coefficient Y don't match (%d,%d)",p.numQuad,elements->numElements);
168          Finley_setError(TYPE_ERROR,error_msg);          Finley_setError(TYPE_ERROR,error_msg);
169    }    }
170    
# Line 152  void Finley_Assemble_PDE(Finley_NodeFile Line 175  void Finley_Assemble_PDE(Finley_NodeFile
175        dimensions[0]=p.numDim;        dimensions[0]=p.numDim;
176        dimensions[1]=p.numDim;        dimensions[1]=p.numDim;
177        if (!isDataPointShapeEqual(A,2,dimensions)) {        if (!isDataPointShapeEqual(A,2,dimensions)) {
178            sprintf(error_msg,"__FILE__: coefficient A: illegal shape, expected shape (%d,%d)",dimensions[0],dimensions[1]);            sprintf(error_msg,"Finley_Assemble_PDE: coefficient A: illegal shape, expected shape (%d,%d)",dimensions[0],dimensions[1]);
179            Finley_setError(TYPE_ERROR,error_msg);            Finley_setError(TYPE_ERROR,error_msg);
180        }        }
181      }      }
182      if (!isEmpty(B)) {      if (!isEmpty(B)) {
183         dimensions[0]=p.numDim;         dimensions[0]=p.numDim;
184         if (!isDataPointShapeEqual(B,1,dimensions)) {         if (!isDataPointShapeEqual(B,1,dimensions)) {
185            sprintf(error_msg,"__FILE__: coefficient B: illegal shape (%d,)",dimensions[0]);            sprintf(error_msg,"Finley_Assemble_PDE: coefficient B: illegal shape (%d,)",dimensions[0]);
186            Finley_setError(TYPE_ERROR,error_msg);            Finley_setError(TYPE_ERROR,error_msg);
187         }         }
188      }      }
189      if (!isEmpty(C)) {      if (!isEmpty(C)) {
190         dimensions[0]=p.numDim;         dimensions[0]=p.numDim;
191         if (!isDataPointShapeEqual(C,1,dimensions)) {         if (!isDataPointShapeEqual(C,1,dimensions)) {
192            sprintf(error_msg,"__FILE__: coefficient C, expected shape (%d,)",dimensions[0]);            sprintf(error_msg,"Finley_Assemble_PDE: coefficient C, expected shape (%d,)",dimensions[0]);
193            Finley_setError(TYPE_ERROR,error_msg);            Finley_setError(TYPE_ERROR,error_msg);
194         }         }
195      }      }
196      if (!isEmpty(D)) {      if (!isEmpty(D)) {
197         if (!isDataPointShapeEqual(D,0,dimensions)) {         if (!isDataPointShapeEqual(D,0,dimensions)) {
198            Finley_setError(TYPE_ERROR,"__FILE__: coefficient D, rank 0 expected.");            Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: coefficient D, rank 0 expected.");
199         }         }
200      }      }
201      if (!isEmpty(X)) {      if (!isEmpty(X)) {
202         dimensions[0]=p.numDim;         dimensions[0]=p.numDim;
203         if (!isDataPointShapeEqual(X,1,dimensions)) {         if (!isDataPointShapeEqual(X,1,dimensions)) {
204            sprintf(error_msg,"__FILE__: coefficient X, expected shape (%d,",dimensions[0]);            sprintf(error_msg,"Finley_Assemble_PDE: coefficient X, expected shape (%d,",dimensions[0]);
205            Finley_setError(TYPE_ERROR,error_msg);            Finley_setError(TYPE_ERROR,error_msg);
206         }         }
207      }      }
208      if (!isEmpty(Y)) {      if (!isEmpty(Y)) {
209         if (!isDataPointShapeEqual(Y,0,dimensions)) {         if (!isDataPointShapeEqual(Y,0,dimensions)) {
210            Finley_setError(TYPE_ERROR,"__FILE__: coefficient Y, rank 0 expected.");            Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: coefficient Y, rank 0 expected.");
211         }         }
212      }      }
213    } else {    } else {
# Line 194  void Finley_Assemble_PDE(Finley_NodeFile Line 217  void Finley_Assemble_PDE(Finley_NodeFile
217        dimensions[2]=p.numComp;        dimensions[2]=p.numComp;
218        dimensions[3]=p.numDim;        dimensions[3]=p.numDim;
219        if (!isDataPointShapeEqual(A,4,dimensions)) {        if (!isDataPointShapeEqual(A,4,dimensions)) {
220            sprintf(error_msg,"__FILE__: coefficient A, expected shape (%d,%d,%d,%d)",dimensions[0],dimensions[1],dimensions[2],dimensions[3]);            sprintf(error_msg,"Finley_Assemble_PDE: coefficient A, expected shape (%d,%d,%d,%d)",dimensions[0],dimensions[1],dimensions[2],dimensions[3]);
221            Finley_setError(TYPE_ERROR,error_msg);            Finley_setError(TYPE_ERROR,error_msg);
222        }        }
223      }      }
# Line 203  void Finley_Assemble_PDE(Finley_NodeFile Line 226  void Finley_Assemble_PDE(Finley_NodeFile
226        dimensions[1]=p.numDim;        dimensions[1]=p.numDim;
227        dimensions[2]=p.numComp;        dimensions[2]=p.numComp;
228        if (!isDataPointShapeEqual(B,3,dimensions)) {        if (!isDataPointShapeEqual(B,3,dimensions)) {
229            sprintf(error_msg,"__FILE__: coefficient B, expected shape (%d,%d,%d)",dimensions[0],dimensions[1],dimensions[2]);            sprintf(error_msg,"Finley_Assemble_PDE: coefficient B, expected shape (%d,%d,%d)",dimensions[0],dimensions[1],dimensions[2]);
230            Finley_setError(TYPE_ERROR,error_msg);            Finley_setError(TYPE_ERROR,error_msg);
231        }        }
232      }      }
# Line 212  void Finley_Assemble_PDE(Finley_NodeFile Line 235  void Finley_Assemble_PDE(Finley_NodeFile
235        dimensions[1]=p.numComp;        dimensions[1]=p.numComp;
236        dimensions[2]=p.numDim;        dimensions[2]=p.numDim;
237        if (!isDataPointShapeEqual(C,3,dimensions)) {        if (!isDataPointShapeEqual(C,3,dimensions)) {
238            sprintf(error_msg,"__FILE__: coefficient C, expected shape (%d,%d,%d)",dimensions[0],dimensions[1],dimensions[2]);            sprintf(error_msg,"Finley_Assemble_PDE: coefficient C, expected shape (%d,%d,%d)",dimensions[0],dimensions[1],dimensions[2]);
239            Finley_setError(TYPE_ERROR,error_msg);            Finley_setError(TYPE_ERROR,error_msg);
240        }        }
241      }      }
# Line 220  void Finley_Assemble_PDE(Finley_NodeFile Line 243  void Finley_Assemble_PDE(Finley_NodeFile
243        dimensions[0]=p.numEqu;        dimensions[0]=p.numEqu;
244        dimensions[1]=p.numComp;        dimensions[1]=p.numComp;
245        if (!isDataPointShapeEqual(D,2,dimensions)) {        if (!isDataPointShapeEqual(D,2,dimensions)) {
246            sprintf(error_msg,"__FILE__: coefficient D, expected shape (%d,%d)",dimensions[0],dimensions[1]);            sprintf(error_msg,"Finley_Assemble_PDE: coefficient D, expected shape (%d,%d)",dimensions[0],dimensions[1]);
247            Finley_setError(TYPE_ERROR,error_msg);            Finley_setError(TYPE_ERROR,error_msg);
248        }        }
249      }      }
# Line 228  void Finley_Assemble_PDE(Finley_NodeFile Line 251  void Finley_Assemble_PDE(Finley_NodeFile
251        dimensions[0]=p.numEqu;        dimensions[0]=p.numEqu;
252        dimensions[1]=p.numDim;        dimensions[1]=p.numDim;
253        if (!isDataPointShapeEqual(X,2,dimensions)) {        if (!isDataPointShapeEqual(X,2,dimensions)) {
254            sprintf(error_msg,"__FILE__: coefficient X, expected shape (%d,%d)",dimensions[0],dimensions[1]);            sprintf(error_msg,"Finley_Assemble_PDE: coefficient X, expected shape (%d,%d)",dimensions[0],dimensions[1]);
255            Finley_setError(TYPE_ERROR,error_msg);            Finley_setError(TYPE_ERROR,error_msg);
256        }        }
257      }      }
258      if (!isEmpty(Y)) {      if (!isEmpty(Y)) {
259        dimensions[0]=p.numEqu;        dimensions[0]=p.numEqu;
260        if (!isDataPointShapeEqual(Y,1,dimensions)) {        if (!isDataPointShapeEqual(Y,1,dimensions)) {
261            sprintf(error_msg,"__FILE__: coefficient Y, expected shape (%d,)",dimensions[0]);            sprintf(error_msg,"Finley_Assemble_PDE: coefficient Y, expected shape (%d,)",dimensions[0]);
262            Finley_setError(TYPE_ERROR,error_msg);            Finley_setError(TYPE_ERROR,error_msg);
263        }        }
264      }      }
265    }    }
   
266    if (Finley_noError()) {    if (Finley_noError()) {
267       time0=Finley_timer();       time0=Finley_timer();
268       #pragma omp parallel private(index_col,index_row,EM_S,EM_F,V,dVdv,dSdV,Vol,dvdV,color,q) \       if (p.numEqu == p. numComp) {
269              firstprivate(elements,nodes,S,F,A,B,C,D,X,Y)          if (p.numEqu > 1) {
270       {            /* system of PDESs */
271           EM_S=EM_F=V=dVdv=dSdV=Vol=dvdV=NULL;            if (p.numDim==3) {
272           index_row=index_col=NULL;              if (p.row_NS == p.col_NS && p.row_NS == p.row_NN && p.col_NS == p.col_NN ) {
273                   Finley_Assemble_PDE_System2_3D(p,elements,S,F,A,B,C,D,X,Y);
274           /* allocate work arrays: */              } else if ( p.row_NS == p.col_NS &&  2*p.row_NS == p.row_NN && 2*p.col_NS == p.col_NN ) {
275                   if ( !isEmpty(A) || !isEmpty(B) || !isEmpty(C) || !isEmpty(X) ) {
276           EM_S=(double*) THREAD_MEMALLOC(p.NN_row*p.NN_col*p.numEqu*p.numComp,double);                    Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: Contact elements require A, B, C and X to be empty.");
277           EM_F=(double*) THREAD_MEMALLOC(p.NN_row*p.numEqu,double);                 } else {
278           V=(double*) THREAD_MEMALLOC(p.NN*p.numDim,double);                    Finley_Assemble_PDE_System2_C(p,elements,S,F,D,Y);
279           dVdv=(double*) THREAD_MEMALLOC(p.numDim*p.numDim*p.numQuad,double);                 }
280           dvdV=(double*) THREAD_MEMALLOC(p.numDim*p.numDim*p.numQuad,double);              } else {
281           dSdV=(double*) THREAD_MEMALLOC(p.NS_row*p.numQuad*p.numDim,double);                 Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE supports numShape=NumNodes or 2*numShape=NumNodes only.");
282           Vol=(double*) THREAD_MEMALLOC(p.numQuad,double);              }
283           index_col=(index_t*) THREAD_MEMALLOC(p.NN_col,index_t);            } else if (p.numDim==2) {
284           index_row=(index_t*) THREAD_MEMALLOC(p.NN_row,index_t);              if ((p.row_NS == p.col_NS) && (p.row_NS == p.row_NN) && (p.col_NS == p.col_NN )) {
285                   Finley_Assemble_PDE_System2_2D(p,elements,S,F,A,B,C,D,X,Y);
286           if (! (Finley_checkPtr(EM_S) || Finley_checkPtr(EM_F) || Finley_checkPtr(V) || Finley_checkPtr(index_col) ||              } else if ( p.row_NS == p.col_NS &&  2*p.row_NS == p.row_NN && 2*p.col_NS == p.col_NN ) {
287                  Finley_checkPtr(index_row) || Finley_checkPtr(dVdv) || Finley_checkPtr(dSdV) || Finley_checkPtr(Vol) ))  {                 if ( !isEmpty(A) || !isEmpty(B) || !isEmpty(C) || !isEmpty(X) ) {
288                      Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: Contact elements require A, B, C and X to be empty.");
289             /*  open loop over all colors: */                 } else {
290             for (color=elements->minColor;color<=elements->maxColor;color++) {                    Finley_Assemble_PDE_System2_C(p,elements,S,F,D,Y);
291                /*  open loop over all elements: */                 }
292                #pragma omp for private(e) schedule(static)              } else {
293                for(e=0;e<elements->numElements;e++){                 Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE supports numShape=NumNodes or 2*numShape=NumNodes only.");
294                  if (elements->Color[e]==color) {              }
295                    for (q=0;q<p.NN_row;q++) index_row[q]=p.label_row[elements->Nodes[INDEX2(p.row_node[q],e,p.NN)]];            } else if (p.numDim==2) {
296                    /* gather V-coordinates of nodes into V: */              if (p.row_NS == p.col_NS && p.row_NS == p.row_NN && p.col_NS == p.col_NN ) {
297            Finley_Util_Gather_double(p.NN,&(elements->Nodes[INDEX2(0,e,p.NN)]),p.numDim,nodes->Coordinates,V);                 Finley_Assemble_PDE_System2_1D(p,elements,S,F,A,B,C,D,X,Y);
298                    /*  calculate dVdv(i,j,q)=V(i,k)*DSDv(k,j,q) */              } else if ( p.row_NS == p.col_NS &&  2*p.row_NS == p.row_NN && 2*p.col_NS == p.col_NN ) {
299            Finley_Util_SmallMatMult(p.numDim,p.numDim*p.numQuad,dVdv,p.NS,V,p.referenceElement->dSdv);                 if ( !isEmpty(A) || !isEmpty(B) || !isEmpty(C) || !isEmpty(X) ) {
300                    /*  dvdV=invert(dVdv) inplace */                    Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: Contact elements require A, B, C and X to be empty.");
301            Finley_Util_InvertSmallMat(p.numQuad,p.numDim,dVdv,dvdV,Vol);                 } else {
302                    /*  calculate dSdV=DSDv*DvDV */                    Finley_Assemble_PDE_System2_C(p,elements,S,F,D,Y);
303            Finley_Util_SmallMatSetMult(p.numQuad,p.NS_row,p.numDim,dSdV,p.numDim,p.referenceElement_row->dSdv,dvdV);                 }
304                    /*  scale volume: */              } else {
305            for (q=0;q<p.numQuad;q++) Vol[q]=ABS(Vol[q]*p.referenceElement->QuadWeights[q]);                 Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE supports numShape=NumNodes or 2*numShape=NumNodes only.");
306                    }
307                     /*   integration for the stiffness matrix: */            } else {
308                     /*   in order to optimze the number of operations the case of constants coefficience needs a bit more work */              Finley_setError(VALUE_ERROR,"Finley_Assemble_PDE supports spatial dimensions 1,2,3 only.");
309                     /*   to make use of some symmetry. */            }
310            } else {
311                       if (S!=NULL) {            /* single PDES */
312                         for (q=0;q<p.NN_row*p.NN_col*p.numEqu*p.numComp;q++) EM_S[q]=0;            if (p.numDim==3) {
313                         if (p.numEqu==1 && p.numComp==1) {              if (p.row_NS == p.col_NS && p.row_NS == p.row_NN && p.col_NS == p.col_NN ) {
314                         Finley_Assemble_PDEMatrix_Single2(p.NS_row,p.numDim,p.numQuad,                 Finley_Assemble_PDE_Single2_3D(p,elements,S,F,A,B,C,D,X,Y);
315                                                                       p.referenceElement_row->S,dSdV,Vol,p.NN_row,EM_S,              } else if ( p.row_NS == p.col_NS &&  2*p.row_NS == p.row_NN && 2*p.col_NS == p.col_NN ) {
316                                                                       getSampleData(A,e),isExpanded(A),                 if ( !isEmpty(A) || !isEmpty(B) || !isEmpty(C) || !isEmpty(X) ) {
317                                                                       getSampleData(B,e),isExpanded(B),                    Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: Contact elements require A, B, C and X to be empty.");
318                                                                       getSampleData(C,e),isExpanded(C),                 } else {
319                                                                       getSampleData(D,e),isExpanded(D));                    Finley_Assemble_PDE_Single2_C(p,elements,S,F,D,Y);
320                         } else {                 }
321                         Finley_Assemble_PDEMatrix_System2(p.NS_row,p.numDim,p.numQuad,p.numEqu,p.numComp,              } else {
322                                                                       p.referenceElement_row->S,dSdV,Vol,p.NN_row,EM_S,                 Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE supports numShape=NumNodes or 2*numShape=NumNodes only.");
323                                                                       getSampleData(A,e),isExpanded(A),              }
324                                                                       getSampleData(B,e),isExpanded(B),            } else if (p.numDim==2) {
325                                                                       getSampleData(C,e),isExpanded(C),              if ((p.row_NS == p.col_NS) && (p.row_NS == p.row_NN) && (p.col_NS == p.col_NN )) {
326                                                                       getSampleData(D,e),isExpanded(D));                 Finley_Assemble_PDE_Single2_2D(p,elements,S,F,A,B,C,D,X,Y);
327                         }              } else if ( p.row_NS == p.col_NS &&  2*p.row_NS == p.row_NN && 2*p.col_NS == p.col_NN ) {
328                         for (q=0;q<p.NN_col;q++) index_col[q]=p.label_col[elements->Nodes[INDEX2(p.col_node[q],e,p.NN)]];                 if ( !isEmpty(A) || !isEmpty(B) || !isEmpty(C) || !isEmpty(X) ) {
329                         Finley_Assemble_addToSystemMatrix(S,p.NN_row,index_row,p.numEqu,p.NN_col,index_col,p.numComp,EM_S);                    Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: Contact elements require A, B, C and X to be empty.");
330                       }                 } else {
331                       if (!isEmpty(F)) {                    Finley_Assemble_PDE_Single2_C(p,elements,S,F,D,Y);
332                         for (q=0;q<p.NN_row*p.numEqu;q++) EM_F[q]=0;                 }
333                         if (p.numEqu==1) {              } else {
334                         Finley_Assemble_RHSMatrix_Single(p.NS_row,p.numDim,p.numQuad,                 Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE supports numShape=NumNodes or 2*numShape=NumNodes only.");
335                                                                   p.referenceElement_row->S,dSdV,Vol,p.NN_row,EM_F,              }
336                                                                   getSampleData(X,e),isExpanded(X),            } else if (p.numDim==2) {
337                                                                   getSampleData(Y,e),isExpanded(Y));              if (p.row_NS == p.col_NS && p.row_NS == p.row_NN && p.col_NS == p.col_NN ) {
338                         } else {                 Finley_Assemble_PDE_Single2_1D(p,elements,S,F,A,B,C,D,X,Y);
339                         Finley_Assemble_RHSMatrix_System(p.NS_row,p.numDim,p.numQuad,              } else if ( p.row_NS == p.col_NS &&  2*p.row_NS == p.row_NN && 2*p.col_NS == p.col_NN ) {
340                                                                   p.numEqu,p.referenceElement_row->S,dSdV,Vol,p.NN_row,EM_F,                 if ( !isEmpty(A) || !isEmpty(B) || !isEmpty(C) || !isEmpty(X) ) {
341                                                                   getSampleData(X,e),isExpanded(X),                    Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: Contact elements require A, B, C and X to be empty.");
342                                                                   getSampleData(Y,e),isExpanded(Y));                 } else {
343                         }                    Finley_Assemble_PDE_Single2_C(p,elements,S,F,D,Y);
344                         /* add  */                 }
345                         Finley_Util_AddScatter(p.NN_row,index_row,p.numEqu,EM_F,getSampleData(F,0));              } else {
346                      }                 Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE supports numShape=NumNodes or 2*numShape=NumNodes only.");
                 }  
               }  
347              }              }
348           }            } else {
349           /* clean up */              Finley_setError(VALUE_ERROR,"Finley_Assemble_PDE supports spatial dimensions 1,2,3 only.");
350           THREAD_MEMFREE(EM_S);            }
351           THREAD_MEMFREE(EM_F);          }
352           THREAD_MEMFREE(V);       } else {
353           THREAD_MEMFREE(dVdv);            Finley_setError(VALUE_ERROR,"Finley_Assemble_PDE requires number of equations == number of solutions  .");
          THREAD_MEMFREE(dvdV);  
          THREAD_MEMFREE(dSdV);  
          THREAD_MEMFREE(Vol);  
          THREAD_MEMFREE(index_col);  
          THREAD_MEMFREE(index_row);  
354       }       }
355       #ifdef Finley_TRACE       #ifdef Finley_TRACE
356       printf("timing: assemblage PDE: %.4e sec\n",Finley_timer()-time0);       printf("timing: assemblage PDE: %.4e sec\n",Finley_timer()-time0);

Legend:
Removed from v.150  
changed lines
  Added in v.969

  ViewVC Help
Powered by ViewVC 1.1.26