/[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

revision 751 by bcumming, Mon Jun 26 01:46:34 2006 UTC revision 1072 by gross, Thu Mar 29 06:44:30 2007 UTC
# Line 56  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      type_t funcspace;
65    
66    Finley_resetError();    Finley_resetError();
67    
68    if (nodes==NULL || elements==NULL) return;    if (nodes==NULL || elements==NULL) return;
69    if (S==NULL && isEmpty(F)) return;    if (S==NULL && isEmpty(F)) return;
70    
71    /* set all parameters in p*/    if (isEmpty(F) && !isEmpty(X) && !isEmpty(F)) {
72    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.");
73    if (! Finley_noError()) return;    }
74    
75    /*  this function assumes NS=NN */    if (S==NULL && !isEmpty(A) && !isEmpty(B) && !isEmpty(C) && !isEmpty(D)) {
76    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,"Finley_Assemble_PDE: for Finley_Assemble_PDE numNodes and numShapes have to be identical.");  
     return;  
   }  
   if (p.numDim!=p.numElementDim) {  
     Finley_setError(SYSTEM_ERROR,"Finley_Assemble_PDE: Finley_Assemble_PDE accepts volume elements only.");  
     return;  
77    }    }
78    /*  get a functionspace */  
79    type_t funcspace=UNKNOWN;    /*  get the functionspace for this assemblage call */
80      funcspace=UNKNOWN;
81    updateFunctionSpaceType(funcspace,A);    updateFunctionSpaceType(funcspace,A);
82    updateFunctionSpaceType(funcspace,B);    updateFunctionSpaceType(funcspace,B);
83    updateFunctionSpaceType(funcspace,C);    updateFunctionSpaceType(funcspace,C);
# Line 91  void Finley_Assemble_PDE(Finley_NodeFile Line 87  void Finley_Assemble_PDE(Finley_NodeFile
87    if (funcspace==UNKNOWN) return; /* all  data are empty */    if (funcspace==UNKNOWN) return; /* all  data are empty */
88    
89    /* check if all function spaces are the same */    /* check if all function spaces are the same */
   
90    if (! functionSpaceTypeEqual(funcspace,A) ) {    if (! functionSpaceTypeEqual(funcspace,A) ) {
91          Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: unexpected function space type for coefficient A");          Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: unexpected function space type for coefficient A");
92    }    }
# Line 110  void Finley_Assemble_PDE(Finley_NodeFile Line 105  void Finley_Assemble_PDE(Finley_NodeFile
105    if (! functionSpaceTypeEqual(funcspace,Y) ) {    if (! functionSpaceTypeEqual(funcspace,Y) ) {
106          Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: unexpected function space type for coefficient Y");          Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: unexpected function space type for coefficient Y");
107    }    }
108      if (! Finley_noError()) return;
109    
110      /* check if all function spaces are the same */
111      if (funcspace==FINLEY_ELEMENTS) {
112           reducedIntegrationOrder=FALSE;
113      } else if (funcspace==FINLEY_FACE_ELEMENTS)  {
114           reducedIntegrationOrder=FALSE;
115      } else if (funcspace==FINLEY_CONTACT_ELEMENTS_1)  {
116           reducedIntegrationOrder=FALSE;
117      } else if (funcspace==FINLEY_CONTACT_ELEMENTS_2)  {
118           reducedIntegrationOrder=FALSE;
119      } else if (funcspace==FINLEY_REDUCED_ELEMENTS) {
120           reducedIntegrationOrder=TRUE;
121      } else if (funcspace==FINLEY_REDUCED_FACE_ELEMENTS)  {
122           reducedIntegrationOrder=TRUE;
123      } else if (funcspace==FINLEY_REDUCED_CONTACT_ELEMENTS_1)  {
124           reducedIntegrationOrder=TRUE;
125      } else if (funcspace==FINLEY_REDUCED_CONTACT_ELEMENTS_2)  {
126           reducedIntegrationOrder=TRUE;
127      } else {
128           Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: assemblage failed because of illegal function space.");
129      }
130      if (! Finley_noError()) return;
131    
132      /* set all parameters in p*/
133      Assemble_getAssembleParameters(nodes,elements,S,F, reducedIntegrationOrder, &p);
134      if (! Finley_noError()) return;
135    
136    /* check if all function spaces are the same */    /* check if all function spaces are the same */
137    
# Line 238  void Finley_Assemble_PDE(Finley_NodeFile Line 260  void Finley_Assemble_PDE(Finley_NodeFile
260        }        }
261      }      }
262    }    }
   
263    if (Finley_noError()) {    if (Finley_noError()) {
264       time0=Finley_timer();       time0=Finley_timer();
265       #pragma omp parallel private(index_col,index_row,EM_S,EM_F,V,dVdv,dSdV,Vol,dvdV,color,q) \       if (p.numEqu == p. numComp) {
266              firstprivate(elements,nodes,S,F,A,B,C,D,X,Y)          if (p.numEqu > 1) {
267       {            /* system of PDESs */
268           EM_S=EM_F=V=dVdv=dSdV=Vol=dvdV=NULL;            if (p.numDim==3) {
269           index_row=index_col=NULL;              if (p.row_NS == p.col_NS && p.row_NS == p.row_NN && p.col_NS == p.col_NN ) {
270                   Finley_Assemble_PDE_System2_3D(p,elements,S,F,A,B,C,D,X,Y);
271           /* 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 ) {
272                   if ( !isEmpty(A) || !isEmpty(B) || !isEmpty(C) || !isEmpty(X) ) {
273           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.");
274           EM_F=(double*) THREAD_MEMALLOC(p.NN_row*p.numEqu,double);                 } else {
275           V=(double*) THREAD_MEMALLOC(p.NN*p.numDim,double);                    Finley_Assemble_PDE_System2_C(p,elements,S,F,D,Y);
276           dVdv=(double*) THREAD_MEMALLOC(p.numDim*p.numDim*p.numQuad,double);                 }
277           dvdV=(double*) THREAD_MEMALLOC(p.numDim*p.numDim*p.numQuad,double);              } else {
278           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.");
279           Vol=(double*) THREAD_MEMALLOC(p.numQuad,double);              }
280           index_col=(index_t*) THREAD_MEMALLOC(p.NN_col,index_t);            } else if (p.numDim==2) {
281           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 )) {
282                   Finley_Assemble_PDE_System2_2D(p,elements,S,F,A,B,C,D,X,Y);
283           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 ) {
284                  Finley_checkPtr(index_row) || Finley_checkPtr(dVdv) || Finley_checkPtr(dSdV) || Finley_checkPtr(Vol) ))  {                 if ( !isEmpty(A) || !isEmpty(B) || !isEmpty(C) || !isEmpty(X) ) {
285                      Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: Contact elements require A, B, C and X to be empty.");
286             /*  open loop over all colors: */                 } else {
287  #ifndef PASO_MPI                    Finley_Assemble_PDE_System2_C(p,elements,S,F,D,Y);
288             for (color=elements->minColor;color<=elements->maxColor;color++) {                 }
289                /*  open loop over all elements: */              } else {
290                #pragma omp for private(e) schedule(static)                 Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE supports numShape=NumNodes or 2*numShape=NumNodes only.");
291                for(e=0;e<elements->numElements;e++){              }
292                  if (elements->Color[e]==color) {            } else if (p.numDim==2) {
293  #else              if (p.row_NS == p.col_NS && p.row_NS == p.row_NN && p.col_NS == p.col_NN ) {
294             for(e=0;e<elements->numElements;e++){                 Finley_Assemble_PDE_System2_1D(p,elements,S,F,A,B,C,D,X,Y);
295  #endif                } else if ( p.row_NS == p.col_NS &&  2*p.row_NS == p.row_NN && 2*p.col_NS == p.col_NN ) {
296  //============================                 if ( !isEmpty(A) || !isEmpty(B) || !isEmpty(C) || !isEmpty(X) ) {
297                    for (q=0;q<p.NN_row;q++) index_row[q]=p.label_row[elements->Nodes[INDEX2(p.row_node[q],e,p.NN)]];                    Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: Contact elements require A, B, C and X to be empty.");
298                    /* gather V-coordinates of nodes into V: */                 } else {
299            Finley_Util_Gather_double(p.NN,&(elements->Nodes[INDEX2(0,e,p.NN)]),p.numDim,nodes->Coordinates,V);                    Finley_Assemble_PDE_System2_C(p,elements,S,F,D,Y);
300                    /*  calculate dVdv(i,j,q)=V(i,k)*DSDv(k,j,q) */                 }
301            Finley_Util_SmallMatMult(p.numDim,p.numDim*p.numQuad,dVdv,p.NS,V,p.referenceElement->dSdv);              } else {
302                    /*  dvdV=invert(dVdv) inplace */                 Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE supports numShape=NumNodes or 2*numShape=NumNodes only.");
303            Finley_Util_InvertSmallMat(p.numQuad,p.numDim,dVdv,dvdV,Vol);              }
304                    /*  calculate dSdV=DSDv*DvDV */            } else {
305            Finley_Util_SmallMatSetMult(p.numQuad,p.NS_row,p.numDim,dSdV,p.numDim,p.referenceElement_row->dSdv,dvdV);              Finley_setError(VALUE_ERROR,"Finley_Assemble_PDE supports spatial dimensions 1,2,3 only.");
306                    /*  scale volume: */            }
307            for (q=0;q<p.numQuad;q++) Vol[q]=ABS(Vol[q]*p.referenceElement->QuadWeights[q]);          } else {
308  //============================            /* single PDES */
309                  if (p.numDim==3) {
310                     /*   integration for the stiffness matrix: */              if (p.row_NS == p.col_NS && p.row_NS == p.row_NN && p.col_NS == p.col_NN ) {
311                     /*   in order to optimze the number of operations the case of constants coefficience needs a bit more work */                 Finley_Assemble_PDE_Single2_3D(p,elements,S,F,A,B,C,D,X,Y);
312                     /*   to make use of some symmetry. */              } else if ( p.row_NS == p.col_NS &&  2*p.row_NS == p.row_NN && 2*p.col_NS == p.col_NN ) {
313                   if ( !isEmpty(A) || !isEmpty(B) || !isEmpty(C) || !isEmpty(X) ) {
314                       if (S!=NULL) {                    Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: Contact elements require A, B, C and X to be empty.");
315                         for (q=0;q<p.NN_row*p.NN_col*p.numEqu*p.numComp;q++) EM_S[q]=0;                 } else {
316                         if (p.numEqu==1 && p.numComp==1) {                    Finley_Assemble_PDE_Single2_C(p,elements,S,F,D,Y);
317                         Finley_Assemble_PDEMatrix_Single2(p.NS_row,p.numDim,p.numQuad,                 }
318                                                                       p.referenceElement_row->S,dSdV,Vol,p.NN_row,EM_S,              } else {
319                                                                       getSampleData(A,e),isExpanded(A),                 Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE supports numShape=NumNodes or 2*numShape=NumNodes only.");
320                                                                       getSampleData(B,e),isExpanded(B),              }
321                                                                       getSampleData(C,e),isExpanded(C),            } else if (p.numDim==2) {
322                                                                       getSampleData(D,e),isExpanded(D));              if ((p.row_NS == p.col_NS) && (p.row_NS == p.row_NN) && (p.col_NS == p.col_NN )) {
323                         } else {                 Finley_Assemble_PDE_Single2_2D(p,elements,S,F,A,B,C,D,X,Y);
324                         Finley_Assemble_PDEMatrix_System2(p.NS_row,p.numDim,p.numQuad,p.numEqu,p.numComp,              } else if ( p.row_NS == p.col_NS &&  2*p.row_NS == p.row_NN && 2*p.col_NS == p.col_NN ) {
325                                                                       p.referenceElement_row->S,dSdV,Vol,p.NN_row,EM_S,                 if ( !isEmpty(A) || !isEmpty(B) || !isEmpty(C) || !isEmpty(X) ) {
326                                                                       getSampleData(A,e),isExpanded(A),                    Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: Contact elements require A, B, C and X to be empty.");
327                                                                       getSampleData(B,e),isExpanded(B),                 } else {
328                                                                       getSampleData(C,e),isExpanded(C),                    Finley_Assemble_PDE_Single2_C(p,elements,S,F,D,Y);
329                                                                       getSampleData(D,e),isExpanded(D));                 }
330                         }              } else {
331                         for (q=0;q<p.NN_col;q++) index_col[q]=p.label_col[elements->Nodes[INDEX2(p.col_node[q],e,p.NN)]];                 Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE supports numShape=NumNodes or 2*numShape=NumNodes only.");
332                         Finley_Assemble_addToSystemMatrix(S,p.NN_row,index_row,p.numEqu,p.NN_col,index_col,p.numComp,EM_S);              }
333                       }            } else if (p.numDim==2) {
334                       if (!isEmpty(F)) {              if (p.row_NS == p.col_NS && p.row_NS == p.row_NN && p.col_NS == p.col_NN ) {
335                         for (q=0;q<p.NN_row*p.numEqu;q++) EM_F[q]=0;                 Finley_Assemble_PDE_Single2_1D(p,elements,S,F,A,B,C,D,X,Y);
336                         if (p.numEqu==1) {              } else if ( p.row_NS == p.col_NS &&  2*p.row_NS == p.row_NN && 2*p.col_NS == p.col_NN ) {
337                         Finley_Assemble_RHSMatrix_Single(p.NS_row,p.numDim,p.numQuad,                 if ( !isEmpty(A) || !isEmpty(B) || !isEmpty(C) || !isEmpty(X) ) {
338                                                                   p.referenceElement_row->S,dSdV,Vol,p.NN_row,EM_F,                    Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE: Contact elements require A, B, C and X to be empty.");
339                                                                   getSampleData(X,e),isExpanded(X),                 } else {
340                                                                   getSampleData(Y,e),isExpanded(Y));                    Finley_Assemble_PDE_Single2_C(p,elements,S,F,D,Y);
341                         } else {                 }
342                         Finley_Assemble_RHSMatrix_System(p.NS_row,p.numDim,p.numQuad,              } else {
343                                                                   p.numEqu,p.referenceElement_row->S,dSdV,Vol,p.NN_row,EM_F,                 Finley_setError(TYPE_ERROR,"Finley_Assemble_PDE supports numShape=NumNodes or 2*numShape=NumNodes only.");
                                                                  getSampleData(X,e),isExpanded(X),  
                                                                  getSampleData(Y,e),isExpanded(Y));  
                        }  
                        /* add  */  
 #ifndef PASO_MPI  
                        Finley_Util_AddScatter(p.NN_row,index_row,p.numEqu,EM_F,getSampleData(F,0));  
 #else  
                        Finley_Util_AddScatter_upperBound(p.NN_row,index_row,p.numEqu,EM_F,getSampleData(F,0),p.degreeOfFreedomUpperBound);  
 #endif                                        
                     }  
                 }  
               }  
 #ifndef PASO_MPI  
344              }              }
345           }            } else {
346  #endif              Finley_setError(VALUE_ERROR,"Finley_Assemble_PDE supports spatial dimensions 1,2,3 only.");
347           /* clean up */            }
348           THREAD_MEMFREE(EM_S);          }
349           THREAD_MEMFREE(EM_F);       } else {
350           THREAD_MEMFREE(V);            Finley_setError(VALUE_ERROR,"Finley_Assemble_PDE requires number of equations == number of solutions  .");
          THREAD_MEMFREE(dVdv);  
          THREAD_MEMFREE(dvdV);  
          THREAD_MEMFREE(dSdV);  
          THREAD_MEMFREE(Vol);  
          THREAD_MEMFREE(index_col);  
          THREAD_MEMFREE(index_row);  
351       }       }
352       #ifdef Finley_TRACE       #ifdef Finley_TRACE
353       printf("timing: assemblage PDE: %.4e sec\n",Finley_timer()-time0);       printf("timing: assemblage PDE: %.4e sec\n",Finley_timer()-time0);

Legend:
Removed from v.751  
changed lines
  Added in v.1072

  ViewVC Help
Powered by ViewVC 1.1.26