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

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

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

revision 113 by jgs, Mon Feb 28 07:06:33 2005 UTC revision 123 by jgs, Fri Jul 8 04:08:13 2005 UTC
# Line 25  Line 25 
25    
26  /*        out(1:numData,1:len)=in(1:numData,index(1:len)) */  /*        out(1:numData,1:len)=in(1:numData,index(1:len)) */
27    
28  void Finley_Util_Gather_double(int len,maybelong* index,int numData,double* in, double * out){  void Finley_Util_Gather_double(dim_t len,index_t* index,dim_t numData,double* in, double * out){
29      int s,i;      dim_t s,i;
30      for (s=0;s<len;s++) {      for (s=0;s<len;s++) {
31         for (i=0;i<numData;i++) {         for (i=0;i<numData;i++) {
32            out[INDEX2(i,s,numData)]=in[INDEX2(i,index[s],numData)];            out[INDEX2(i,s,numData)]=in[INDEX2(i,index[s],numData)];
# Line 41  void Finley_Util_Gather_double(int len,m Line 41  void Finley_Util_Gather_double(int len,m
41    
42  /*        out(1:numData,1:len)=in(1:numData,index(1:len)) */  /*        out(1:numData,1:len)=in(1:numData,index(1:len)) */
43    
44  void Finley_Util_Gather_int(int len,maybelong* index,int numData, maybelong* in, maybelong * out){  void Finley_Util_Gather_int(dim_t len,index_t* index,dim_t numData, index_t* in, index_t * out){
45      int s,i;      dim_t s,i;
46      for (s=0;s<len;s++) {      for (s=0;s<len;s++) {
47         for (i=0;i<numData;i++) {         for (i=0;i<numData;i++) {
48            out[INDEX2(i,s,numData)]=in[INDEX2(i,index[s],numData)];            out[INDEX2(i,s,numData)]=in[INDEX2(i,index[s],numData)];
# Line 56  void Finley_Util_Gather_int(int len,mayb Line 56  void Finley_Util_Gather_int(int len,mayb
56    
57  /*        out(1:numData,index(1:len))+=in(1:numData,1:len) */  /*        out(1:numData,index(1:len))+=in(1:numData,1:len) */
58    
59  void Finley_Util_AddScatter(int len,maybelong* index,int numData,double* in,double * out){  void Finley_Util_AddScatter(dim_t len,index_t* index,dim_t numData,double* in,double * out){
60     int i,s;     dim_t i,s;
61     for (s=0;s<len;s++) {     for (s=0;s<len;s++) {
62         for(i=0;i<numData;i++) {         for(i=0;i<numData;i++) {
63            #pragma omp atomic            #pragma omp atomic
# Line 70  void Finley_Util_AddScatter(int len,mayb Line 70  void Finley_Util_AddScatter(int len,mayb
70    
71  /*          A(1:A1,1:A2)=B(1:A1,1:B2)*C(1:B2,1:A2) */  /*          A(1:A1,1:A2)=B(1:A1,1:B2)*C(1:B2,1:A2) */
72    
73  void Finley_Util_SmallMatMult(int A1,int A2, double* A, int B2, double*B, double* C) {  void Finley_Util_SmallMatMult(dim_t A1,dim_t A2, double* A, dim_t B2, double*B, double* C) {
74      int i,j,s;      dim_t i,j,s;
75      for (i=0;i<A1*A2;i++) A[i]=0;      for (i=0;i<A1*A2;i++) A[i]=0;
76         for (i=0;i<A1;i++) {         for (i=0;i<A1;i++) {
77            for (j=0;j<A2;j++) {            for (j=0;j<A2;j++) {
# Line 86  void Finley_Util_SmallMatMult(int A1,int Line 86  void Finley_Util_SmallMatMult(int A1,int
86    
87  /*        A(1:A1,1:A2,i)=B(1:A1,1:B2,i)*C(1:B2,1:A2,i) i=1,len */  /*        A(1:A1,1:A2,i)=B(1:A1,1:B2,i)*C(1:B2,1:A2,i) i=1,len */
88    
89  void Finley_Util_SmallMatSetMult(int len,int A1,int A2, double* A, int B2, double*B, double* C) {  void Finley_Util_SmallMatSetMult(dim_t len,dim_t A1,dim_t A2, double* A, dim_t B2, double*B, double* C) {
90      int q,i,j,s;      dim_t q,i,j,s;
91      for (i=0;i<A1*A2*len;i++) A[i]=0;      for (i=0;i<A1*A2*len;i++) A[i]=0;
92      for (q=0;q<len;q++) {      for (q=0;q<len;q++) {
93         for (i=0;i<A1;i++) {         for (i=0;i<A1;i++) {
# Line 99  void Finley_Util_SmallMatSetMult(int len Line 99  void Finley_Util_SmallMatSetMult(int len
99         }         }
100      }      }
101  }  }
   
 /* calcultes the LU factorization for a small matrix dimxdim matrix A */  
 /* TODO: use LAPACK */  
   
 int Finley_Util_SmallMatLU(int dim,double* A,double *LU,int* pivot){  
      double D,A11,A12,A13,A21,A22,A23,A31,A32,A33;  
      int info=0;  
      /* LAPACK version */  
      /* memcpy(LU,A,sizeof(douple)); */  
      /* dgetf2_(&dim,&dim,A,&dim,LU,pivot,&info); */  
      switch(dim) {  
       case 1:  
             D=A[INDEX2(0,0,dim)];  
             if (ABS(D) >0. ){  
                LU[INDEX2(0,0,dim)]=1./D;  
             } else {  
                info=2;  
             }  
          break;  
   
       case 2:  
             A11=A[INDEX2(0,0,dim)];  
             A12=A[INDEX2(0,1,dim)];  
             A21=A[INDEX2(1,0,dim)];  
             A22=A[INDEX2(1,1,dim)];  
   
             D = A11*A22-A12*A21;  
             if (ABS(D) > 0 ){  
                D=1./D;  
                LU[INDEX2(0,0,dim)]= A22*D;  
                LU[INDEX2(1,0,dim)]=-A21*D;  
                LU[INDEX2(0,1,dim)]=-A12*D;  
                LU[INDEX2(1,1,dim)]= A11*D;  
             } else {  
                info=2;  
             }  
          break;  
   
       case 3:  
             A11=A[INDEX2(0,0,dim)];  
             A21=A[INDEX2(1,0,dim)];  
             A31=A[INDEX2(2,0,dim)];  
             A12=A[INDEX2(0,1,dim)];  
             A22=A[INDEX2(1,1,dim)];  
             A32=A[INDEX2(2,1,dim)];  
             A13=A[INDEX2(0,2,dim)];  
             A23=A[INDEX2(1,2,dim)];  
             A33=A[INDEX2(2,2,dim)];  
   
             D  =  A11*(A22*A33-A23*A32)+ A12*(A31*A23-A21*A33)+A13*(A21*A32-A31*A22);  
             if (ABS(D) > 0 ){  
                D=1./D;  
                LU[INDEX2(0,0,dim)]=(A22*A33-A23*A32)*D;  
                LU[INDEX2(1,0,dim)]=(A31*A23-A21*A33)*D;  
                LU[INDEX2(2,0,dim)]=(A21*A32-A31*A22)*D;  
                LU[INDEX2(0,1,dim)]=(A13*A32-A12*A33)*D;  
                LU[INDEX2(1,1,dim)]=(A11*A33-A31*A13)*D;  
                LU[INDEX2(2,1,dim)]=(A12*A31-A11*A32)*D;  
                LU[INDEX2(0,2,dim)]=(A12*A23-A13*A22)*D;  
                LU[INDEX2(1,2,dim)]=(A13*A21-A11*A23)*D;  
                LU[INDEX2(2,2,dim)]=(A11*A22-A12*A21)*D;  
             } else {  
                info=2;  
             }  
          break;  
        default:  
             info=1;  
    }  
    return info;  
 }  
   
 /* solves LUx=b where LU is a LU factorization calculated by an Finley_Util_SmallMatLU call */  
 void Finley_Util_SmallMatForwardBackwardSolve(int dim ,int nrhs,double* LU,int* pivot,double* x,double* b) {  
      int i;  
      switch(dim) {  
       case 1:  
          for (i=0;i<nrhs;i++) {  
             x[INDEX2(0,i,dim)]=LU[0]*b[INDEX2(0,i,dim)];  
          }  
          break;  
       case 2:  
          for (i=0;i<nrhs;i++) {  
             x[INDEX2(0,i,dim)]=LU[INDEX2(0,0,dim)]*b[INDEX2(0,i,dim)]+LU[INDEX2(0,1,dim)]*b[INDEX2(1,i,dim)];  
             x[INDEX2(1,i,dim)]=LU[INDEX2(1,0,dim)]*b[INDEX2(0,i,dim)]+LU[INDEX2(1,1,dim)]*b[INDEX2(1,i,dim)];  
          }  
          break;  
   
       case 3:  
          for (i=0;i<nrhs;i++) {  
             x[INDEX2(0,i,dim)]=LU[INDEX2(0,0,dim)]*b[INDEX2(0,i,dim)]+LU[INDEX2(0,1,dim)]*b[INDEX2(1,i,dim)]+LU[INDEX2(0,2,dim)]*b[INDEX2(2,i,dim)];  
             x[INDEX2(1,i,dim)]=LU[INDEX2(1,0,dim)]*b[INDEX2(0,i,dim)]+LU[INDEX2(1,1,dim)]*b[INDEX2(1,i,dim)]+LU[INDEX2(1,2,dim)]*b[INDEX2(2,i,dim)];  
             x[INDEX2(2,i,dim)]=LU[INDEX2(2,0,dim)]*b[INDEX2(0,i,dim)]+LU[INDEX2(2,1,dim)]*b[INDEX2(1,i,dim)]+LU[INDEX2(2,2,dim)]*b[INDEX2(2,i,dim)];  
          }  
          break;  
    }  
    return;  
 }  
102  /*    inverts the set of dim x dim matrices A(:,:,1:len) with dim=1,2,3 */  /*    inverts the set of dim x dim matrices A(:,:,1:len) with dim=1,2,3 */
103  /*    the determinante is returned. */  /*    the determinante is returned. */
104    
105  void Finley_Util_InvertSmallMat(int len,int dim,double* A,double *invA, double* det){  void Finley_Util_InvertSmallMat(dim_t len,dim_t dim,double* A,double *invA, double* det){
106     int q;     dim_t q;
107     double D,A11,A12,A13,A21,A22,A23,A31,A32,A33;     register double D,A11,A12,A13,A21,A22,A23,A31,A32,A33;
108    
109     switch(dim) {     switch(dim) {
110        case 1:        case 1:
111           for (q=0;q<len;q++) {           for (q=0;q<len;q++) {
112              D=A[INDEX3(0,0,q,dim,dim)];              D=A[q];
113              if (ABS(D) > 0 ){              if (ABS(D) > 0 ){
114                 det[q]=D;                 det[q]=D;
115                 D=1./D;                 D=1./D;
116                 invA[INDEX3(0,0,q,dim,dim)]=D;                 invA[q]=D;
117              } else {              } else {
118                 Finley_ErrorCode=ZERO_DIVISION_ERROR;                 Finley_ErrorCode=ZERO_DIVISION_ERROR;
119                 sprintf(Finley_ErrorMsg,"Non-regular matrix");                 sprintf(Finley_ErrorMsg,"Non-regular matrix");
# Line 221  void Finley_Util_InvertSmallMat(int len, Line 124  void Finley_Util_InvertSmallMat(int len,
124    
125        case 2:        case 2:
126           for (q=0;q<len;q++) {           for (q=0;q<len;q++) {
127              A11=A[INDEX3(0,0,q,dim,dim)];              A11=A[INDEX3(0,0,q,2,2)];
128              A12=A[INDEX3(0,1,q,dim,dim)];              A12=A[INDEX3(0,1,q,2,2)];
129              A21=A[INDEX3(1,0,q,dim,dim)];              A21=A[INDEX3(1,0,q,2,2)];
130              A22=A[INDEX3(1,1,q,dim,dim)];              A22=A[INDEX3(1,1,q,2,2)];
131    
132              D = A11*A22-A12*A21;              D = A11*A22-A12*A21;
133              if (ABS(D) > 0 ){              if (ABS(D) > 0 ){
134                 det[q]=D;                 det[q]=D;
135                 D=1./D;                 D=1./D;
136                 invA[INDEX3(0,0,q,dim,dim)]= A22*D;                 invA[INDEX3(0,0,q,2,2)]= A22*D;
137                 invA[INDEX3(1,0,q,dim,dim)]=-A21*D;                 invA[INDEX3(1,0,q,2,2)]=-A21*D;
138                 invA[INDEX3(0,1,q,dim,dim)]=-A12*D;                 invA[INDEX3(0,1,q,2,2)]=-A12*D;
139                 invA[INDEX3(1,1,q,dim,dim)]= A11*D;                 invA[INDEX3(1,1,q,2,2)]= A11*D;
140              } else {              } else {
141                 Finley_ErrorCode=ZERO_DIVISION_ERROR;                 Finley_ErrorCode=ZERO_DIVISION_ERROR;
142                 sprintf(Finley_ErrorMsg,"Non-regular matrix");                 sprintf(Finley_ErrorMsg,"Non-regular matrix");
# Line 244  void Finley_Util_InvertSmallMat(int len, Line 147  void Finley_Util_InvertSmallMat(int len,
147    
148        case 3:        case 3:
149       for (q=0;q<len;q++) {       for (q=0;q<len;q++) {
150              A11=A[INDEX3(0,0,q,dim,dim)];              A11=A[INDEX3(0,0,q,3,3)];
151              A21=A[INDEX3(1,0,q,dim,dim)];              A21=A[INDEX3(1,0,q,3,3)];
152              A31=A[INDEX3(2,0,q,dim,dim)];              A31=A[INDEX3(2,0,q,3,3)];
153              A12=A[INDEX3(0,1,q,dim,dim)];              A12=A[INDEX3(0,1,q,3,3)];
154              A22=A[INDEX3(1,1,q,dim,dim)];              A22=A[INDEX3(1,1,q,3,3)];
155              A32=A[INDEX3(2,1,q,dim,dim)];              A32=A[INDEX3(2,1,q,3,3)];
156              A13=A[INDEX3(0,2,q,dim,dim)];              A13=A[INDEX3(0,2,q,3,3)];
157              A23=A[INDEX3(1,2,q,dim,dim)];              A23=A[INDEX3(1,2,q,3,3)];
158              A33=A[INDEX3(2,2,q,dim,dim)];              A33=A[INDEX3(2,2,q,3,3)];
159    
160              D  =  A11*(A22*A33-A23*A32)+ A12*(A31*A23-A21*A33)+A13*(A21*A32-A31*A22);              D  =  A11*(A22*A33-A23*A32)+ A12*(A31*A23-A21*A33)+A13*(A21*A32-A31*A22);
161              if (ABS(D) > 0 ){              if (ABS(D) > 0 ){
162                 det[q]    =D;                 det[q]    =D;
163                 D=1./D;                 D=1./D;
164                 invA[INDEX3(0,0,q,dim,dim)]=(A22*A33-A23*A32)*D;                 invA[INDEX3(0,0,q,3,3)]=(A22*A33-A23*A32)*D;
165                 invA[INDEX3(1,0,q,dim,dim)]=(A31*A23-A21*A33)*D;                 invA[INDEX3(1,0,q,3,3)]=(A31*A23-A21*A33)*D;
166                 invA[INDEX3(2,0,q,dim,dim)]=(A21*A32-A31*A22)*D;                 invA[INDEX3(2,0,q,3,3)]=(A21*A32-A31*A22)*D;
167                 invA[INDEX3(0,1,q,dim,dim)]=(A13*A32-A12*A33)*D;                 invA[INDEX3(0,1,q,3,3)]=(A13*A32-A12*A33)*D;
168                 invA[INDEX3(1,1,q,dim,dim)]=(A11*A33-A31*A13)*D;                 invA[INDEX3(1,1,q,3,3)]=(A11*A33-A31*A13)*D;
169                 invA[INDEX3(2,1,q,dim,dim)]=(A12*A31-A11*A32)*D;                 invA[INDEX3(2,1,q,3,3)]=(A12*A31-A11*A32)*D;
170                 invA[INDEX3(0,2,q,dim,dim)]=(A12*A23-A13*A22)*D;                 invA[INDEX3(0,2,q,3,3)]=(A12*A23-A13*A22)*D;
171                 invA[INDEX3(1,2,q,dim,dim)]=(A13*A21-A11*A23)*D;                 invA[INDEX3(1,2,q,3,3)]=(A13*A21-A11*A23)*D;
172                 invA[INDEX3(2,2,q,dim,dim)]=(A11*A22-A12*A21)*D;                 invA[INDEX3(2,2,q,3,3)]=(A11*A22-A12*A21)*D;
173              } else {              } else {
174                 Finley_ErrorCode=ZERO_DIVISION_ERROR;                 Finley_ErrorCode=ZERO_DIVISION_ERROR;
175                 sprintf(Finley_ErrorMsg,"Non-regular matrix");                 sprintf(Finley_ErrorMsg,"Non-regular matrix");
# Line 281  void Finley_Util_InvertSmallMat(int len, Line 184  void Finley_Util_InvertSmallMat(int len,
184    
185  /*    sets the derterminate of a set of dim x dim matrices A(:,:,1:len) with dim=1,2,3 */  /*    sets the derterminate of a set of dim x dim matrices A(:,:,1:len) with dim=1,2,3 */
186    
187  void Finley_Util_DetOfSmallMat(int len,int dim,double* A, double* det){  void Finley_Util_DetOfSmallMat(dim_t len,dim_t dim,double* A, double* det){
188     int q;     dim_t q;
189     double A11,A12,A13,A21,A22,A23,A31,A32,A33;     register double A11,A12,A13,A21,A22,A23,A31,A32,A33;
190    
191     switch(dim) {     switch(dim) {
192        case 1:        case 1:
193           for (q=0;q<len;q++) {           for (q=0;q<len;q++) {
194              det[q]=A[INDEX3(0,0,q,dim,dim)];              det[q]=A[q];
195           }           }
196           break;           break;
197    
198        case 2:        case 2:
199           for (q=0;q<len;q++) {           for (q=0;q<len;q++) {
200              A11=A[INDEX3(0,0,q,dim,dim)];              A11=A[INDEX3(0,0,q,2,2)];
201              A12=A[INDEX3(0,1,q,dim,dim)];              A12=A[INDEX3(0,1,q,2,2)];
202              A21=A[INDEX3(1,0,q,dim,dim)];              A21=A[INDEX3(1,0,q,2,2)];
203              A22=A[INDEX3(1,1,q,dim,dim)];              A22=A[INDEX3(1,1,q,2,2)];
204    
205              det[q] = A11*A22-A12*A21;              det[q] = A11*A22-A12*A21;
206           }           }
# Line 305  void Finley_Util_DetOfSmallMat(int len,i Line 208  void Finley_Util_DetOfSmallMat(int len,i
208    
209        case 3:        case 3:
210       for (q=0;q<len;q++) {       for (q=0;q<len;q++) {
211              A11=A[INDEX3(0,0,q,dim,dim)];              A11=A[INDEX3(0,0,q,3,3)];
212              A21=A[INDEX3(1,0,q,dim,dim)];              A21=A[INDEX3(1,0,q,3,3)];
213              A31=A[INDEX3(2,0,q,dim,dim)];              A31=A[INDEX3(2,0,q,3,3)];
214              A12=A[INDEX3(0,1,q,dim,dim)];              A12=A[INDEX3(0,1,q,3,3)];
215              A22=A[INDEX3(1,1,q,dim,dim)];              A22=A[INDEX3(1,1,q,3,3)];
216              A32=A[INDEX3(2,1,q,dim,dim)];              A32=A[INDEX3(2,1,q,3,3)];
217              A13=A[INDEX3(0,2,q,dim,dim)];              A13=A[INDEX3(0,2,q,3,3)];
218              A23=A[INDEX3(1,2,q,dim,dim)];              A23=A[INDEX3(1,2,q,3,3)];
219              A33=A[INDEX3(2,2,q,dim,dim)];              A33=A[INDEX3(2,2,q,3,3)];
220    
221              det[q]  =  A11*(A22*A33-A23*A32)+ A12*(A31*A23-A21*A33)+A13*(A21*A32-A31*A22);              det[q]  =  A11*(A22*A33-A23*A32)+ A12*(A31*A23-A21*A33)+A13*(A21*A32-A31*A22);
222           }           }
# Line 325  void Finley_Util_DetOfSmallMat(int len,i Line 228  void Finley_Util_DetOfSmallMat(int len,i
228  /*    returns the normalized vector Normal[dim,len] orthogonal to A(:,0,q) and A(:,1,q) in the case of dim=3  */  /*    returns the normalized vector Normal[dim,len] orthogonal to A(:,0,q) and A(:,1,q) in the case of dim=3  */
229  /*    or the vector A(:,0,q) in the case of dim=2                                             */  /*    or the vector A(:,0,q) in the case of dim=2                                             */
230    
231  void  Finley_NormalVector(int len, int dim, int dim1, double* A,double* Normal) {  void  Finley_NormalVector(dim_t len, dim_t dim, dim_t dim1, double* A,double* Normal) {
232     int q;     dim_t q;
233     double A11,A12,CO_A13,A21,A22,CO_A23,A31,A32,CO_A33,length,invlength;     register double A11,A12,CO_A13,A21,A22,CO_A23,A31,A32,CO_A33,length,invlength;
234    
235     switch(dim) {     switch(dim) {
236        case 1:        case 1:
237           for (q=0;q<len;q++) Normal[INDEX1(q)]    =1;           for (q=0;q<len;q++) Normal[q]    =1;
238           break;           break;
239        case 2:        case 2:
240           for (q=0;q<len;q++) {           for (q=0;q<len;q++) {
241              A11=A[INDEX3(0,0,q,dim,dim1)];              A11=A[INDEX3(0,0,q,2,dim1)];
242              A21=A[INDEX3(1,0,q,dim,dim1)];              A21=A[INDEX3(1,0,q,2,dim1)];
243              length = sqrt(A11*A11+A21*A21);              length = sqrt(A11*A11+A21*A21);
244              if (! length>0) {              if (! length>0) {
245                 Finley_ErrorCode=ZERO_DIVISION_ERROR;                 Finley_ErrorCode=ZERO_DIVISION_ERROR;
# Line 344  void  Finley_NormalVector(int len, int d Line 247  void  Finley_NormalVector(int len, int d
247                 return;                 return;
248              } else {              } else {
249                 invlength=1./length;                 invlength=1./length;
250                 Normal[INDEX2(0,q,dim)]=A21*invlength;                 Normal[INDEX2(0,q,2)]=A21*invlength;
251                 Normal[INDEX2(1,q,dim)]=-A11*invlength;                 Normal[INDEX2(1,q,2)]=-A11*invlength;
252              }              }
253           }           }
254           break;           break;
255        case 3:        case 3:
256           for (q=0;q<len;q++) {           for (q=0;q<len;q++) {
257              A11=A[INDEX3(0,0,q,dim,dim1)];              A11=A[INDEX3(0,0,q,3,dim1)];
258              A21=A[INDEX3(1,0,q,dim,dim1)];              A21=A[INDEX3(1,0,q,3,dim1)];
259              A31=A[INDEX3(2,0,q,dim,dim1)];              A31=A[INDEX3(2,0,q,3,dim1)];
260              A12=A[INDEX3(0,1,q,dim,dim1)];              A12=A[INDEX3(0,1,q,3,dim1)];
261              A22=A[INDEX3(1,1,q,dim,dim1)];              A22=A[INDEX3(1,1,q,3,dim1)];
262              A32=A[INDEX3(2,1,q,dim,dim1)];              A32=A[INDEX3(2,1,q,3,dim1)];
263              CO_A13=A21*A32-A31*A22;              CO_A13=A21*A32-A31*A22;
264              CO_A23=A31*A12-A11*A32;              CO_A23=A31*A12-A11*A32;
265              CO_A33=A11*A22-A21*A12;              CO_A33=A11*A22-A21*A12;
# Line 367  void  Finley_NormalVector(int len, int d Line 270  void  Finley_NormalVector(int len, int d
270                 return;                 return;
271              } else {              } else {
272                 invlength=1./length;                 invlength=1./length;
273                 Normal[INDEX2(0,q,dim)]=CO_A13*invlength;                 Normal[INDEX2(0,q,3)]=CO_A13*invlength;
274                 Normal[INDEX2(1,q,dim)]=CO_A23*invlength;                 Normal[INDEX2(1,q,3)]=CO_A23*invlength;
275                 Normal[INDEX2(2,q,dim)]=CO_A33*invlength;                 Normal[INDEX2(2,q,3)]=CO_A33*invlength;
276             }             }
277                            
278        }        }
# Line 382  void  Finley_NormalVector(int len, int d Line 285  void  Finley_NormalVector(int len, int d
285  /*    return the length of the vector which is orthogonal to the vectors A(:,0,q) and A(:,1,q) in the case of dim=3 */  /*    return the length of the vector which is orthogonal to the vectors A(:,0,q) and A(:,1,q) in the case of dim=3 */
286  /*    or the vector A(:,0,q) in the case of dim=2                                                                   */  /*    or the vector A(:,0,q) in the case of dim=2                                                                   */
287    
288  void  Finley_LengthOfNormalVector(int len, int dim, int dim1, double* A,double* length) {  void  Finley_LengthOfNormalVector(dim_t len, dim_t dim, dim_t dim1, double* A,double* length) {
289     int q;     dim_t q;
290     double A11,A12,CO_A13,A21,A22,CO_A23,A31,A32,CO_A33;     double A11,A12,CO_A13,A21,A22,CO_A23,A31,A32,CO_A33;
291    
292     switch(dim) {     switch(dim) {
293        case 1:        case 1:
294           for (q=0;q<len;q++) length[INDEX1(q)]    =1;           for (q=0;q<len;q++) length[q]    =1;
295           break;           break;
296        case 2:        case 2:
297           for (q=0;q<len;q++) {           for (q=0;q<len;q++) {
298              A11=A[INDEX3(0,0,q,dim,dim1)];              A11=A[INDEX3(0,0,q,2,dim1)];
299              A21=A[INDEX3(1,0,q,dim,dim1)];              A21=A[INDEX3(1,0,q,2,dim1)];
300              length[q] = sqrt(A11*A11+A21*A21);              length[q] = sqrt(A11*A11+A21*A21);
301           }           }
302           break;           break;
303        case 3:        case 3:
304           for (q=0;q<len;q++) {           for (q=0;q<len;q++) {
305              A11=A[INDEX3(0,0,q,dim,dim1)];              A11=A[INDEX3(0,0,q,3,dim1)];
306              A21=A[INDEX3(1,0,q,dim,dim1)];              A21=A[INDEX3(1,0,q,3,dim1)];
307              A31=A[INDEX3(2,0,q,dim,dim1)];              A31=A[INDEX3(2,0,q,3,dim1)];
308              A12=A[INDEX3(0,1,q,dim,dim1)];              A12=A[INDEX3(0,1,q,3,dim1)];
309              A22=A[INDEX3(1,1,q,dim,dim1)];              A22=A[INDEX3(1,1,q,3,dim1)];
310              A32=A[INDEX3(2,1,q,dim,dim1)];              A32=A[INDEX3(2,1,q,3,dim1)];
311              CO_A13=A21*A32-A31*A22;              CO_A13=A21*A32-A31*A22;
312              CO_A23=A31*A12-A11*A32;              CO_A23=A31*A12-A11*A32;
313              CO_A33=A11*A22-A21*A12;              CO_A33=A11*A22-A21*A12;
# Line 420  void  Finley_LengthOfNormalVector(int le Line 323  void  Finley_LengthOfNormalVector(int le
323  /* there is no range checking! */  /* there is no range checking! */
324  /* at output Map[invMap[i]]=i for i=0:lenInvMap */  /* at output Map[invMap[i]]=i for i=0:lenInvMap */
325    
326  void Finley_Util_InvertMap(int lenInvMap, maybelong* invMap,int lenMap, maybelong* Map) {  void Finley_Util_InvertMap(dim_t lenInvMap, index_t* invMap,dim_t lenMap, index_t* Map) {
327     int i;     dim_t i;
328     for (i=0;i<lenInvMap;i++) invMap[i]=0;     for (i=0;i<lenInvMap;i++) invMap[i]=0;
329     for (i=0;i<lenMap;i++) {     for (i=0;i<lenMap;i++) {
330        if (Map[i]>=0) invMap[Map[i]]=i;        if (Map[i]>=0) invMap[Map[i]]=i;
# Line 439  int Finley_Util_ValueAndIndex_compar(con Line 342  int Finley_Util_ValueAndIndex_compar(con
342     if (e1->value > e2->value) return  1;     if (e1->value > e2->value) return  1;
343     return 0;     return 0;
344  }  }
345  void Finley_Util_sortValueAndIndex(int n,Finley_Util_ValueAndIndex* array) {  void Finley_Util_sortValueAndIndex(dim_t n,Finley_Util_ValueAndIndex* array) {
346       /* OMP : needs parallelization !*/       /* OMP : needs parallelization !*/
347       qsort(array,n,sizeof(Finley_Util_ValueAndIndex),Finley_Util_ValueAndIndex_compar);       qsort(array,n,sizeof(Finley_Util_ValueAndIndex),Finley_Util_ValueAndIndex_compar);
348  }  }
# Line 449  void Finley_Util_sortValueAndIndex(int n Line 352  void Finley_Util_sortValueAndIndex(int n
352    
353  /* calculates the minimum value from a dim X N integer array */  /* calculates the minimum value from a dim X N integer array */
354    
355  maybelong Finley_Util_getMinInt(int dim,int N,maybelong* values) {  index_t Finley_Util_getMinInt(dim_t dim,dim_t N,index_t* values) {
356     maybelong i,j,out;     dim_t i,j;
357     out=MAYBELONG_MAX;     index_t out,out_local;
358       out=INDEX_T_MAX;
359     if (values!=NULL && dim*N>0 ) {     if (values!=NULL && dim*N>0 ) {
      /* OMP */  
360       out=values[0];       out=values[0];
361       for (j=0;j<N;j++) {       #pragma omp parallel private(out_local)
362         for (i=0;i<dim;i++) out=MIN(out,values[INDEX2(i,j,dim)]);       {
363             out_local=out;
364             #pragma omp for private(i,j) schedule(static)
365             for (j=0;j<N;j++) {
366               for (i=0;i<dim;i++) out_local=MIN(out_local,values[INDEX2(i,j,dim)]);
367             }
368             #pragma omp critical
369             out=MIN(out_local,out);
370       }       }
371     }     }
372     return out;     return out;
# Line 464  maybelong Finley_Util_getMinInt(int dim, Line 374  maybelong Finley_Util_getMinInt(int dim,
374                                                                                                                                                                                                                                                                                                        
375  /* calculates the maximum value from a dim X N integer array */  /* calculates the maximum value from a dim X N integer array */
376    
377  maybelong Finley_Util_getMaxInt(int dim,int N,maybelong* values) {  index_t Finley_Util_getMaxInt(dim_t dim,dim_t N,index_t* values) {
378     maybelong i,j,out;     dim_t i,j;
379     out=-MAYBELONG_MAX;     index_t out,out_local;
380       out=-INDEX_T_MAX;
381     if (values!=NULL && dim*N>0 ) {     if (values!=NULL && dim*N>0 ) {
      /* OMP */  
382       out=values[0];       out=values[0];
383       for (j=0;j<N;j++) {       #pragma omp parallel private(out_local)
384         for (i=0;i<dim;i++) out=MAX(out,values[INDEX2(i,j,dim)]);       {
385       }           out_local=out;
386             #pragma omp for private(i,j) schedule(static)
387             for (j=0;j<N;j++) {
388                 for (i=0;i<dim;i++) out_local=MAX(out_local,values[INDEX2(i,j,dim)]);
389             }
390             #pragma omp critical
391             out=MAX(out_local,out);
392          }
393     }     }
394     return out;     return out;
395  }  }
396    
397  /* set the index of the positive entries in mask. The length of index is returned. */  /* set the index of the positive entries in mask. The length of index is returned. */
398    
399  maybelong Finley_Util_packMask(maybelong N,maybelong* mask,maybelong* index) {  dim_t Finley_Util_packMask(dim_t N,index_t* mask,index_t* index) {
400        maybelong out,k;        dim_t out,k;
401        out=0;        out=0;
402        /*OMP */        /*OMP */
403        for (k=0;k<N;k++) {        for (k=0;k<N;k++) {
# Line 493  maybelong Finley_Util_packMask(maybelong Line 410  maybelong Finley_Util_packMask(maybelong
410  }  }
411    
412  /* returns true if array contains value */  /* returns true if array contains value */
413  int Finley_Util_isAny(maybelong N,maybelong* array,maybelong value) {  bool_t Finley_Util_isAny(dim_t N,index_t* array,index_t value) {
414     int out=FALSE;     bool_t out=FALSE;
415     maybelong i;     dim_t i;
416     #pragma omp parallel for private(i) schedule(static) reduction(||:out)     #pragma omp parallel for private(i) schedule(static) reduction(||:out)
417     for (i=0;i<N;i++) out=out || (array[i]==value);     for (i=0;i<N;i++) out = out || (array[i]==value);
418     return out;     return out;
419  }  }
420  /* calculates the cummultative sum in array and returns the total sum */  /* calculates the cummultative sum in array and returns the total sum */
421  maybelong Finley_Util_cumsum(maybelong N,maybelong* array) {  index_t Finley_Util_cumsum(dim_t N,index_t* array) {
422     maybelong out=0,tmp,i;     index_t out=0,tmp;
423       dim_t i;
424     #ifdef _OPENMP     #ifdef _OPENMP
425        maybelong partial_sums[omp_get_max_threads()],sum;        index_t partial_sums[omp_get_max_threads()],sum;
426        #pragma omp parallel private(sum,i,tmp)        #pragma omp parallel private(sum,i,tmp)
427        {        {
428          sum=0;          sum=0;
429          #pragma omp for          #pragma omp for schedule(static)
430          for (i=0;i<N;++i) {          for (i=0;i<N;++i) sum+=array[i];
           tmp=sum;  
           sum+=array[i];  
           array[i]=tmp;  
         }  
         #pragma omp critical  
431          partial_sums[omp_get_thread_num()]=sum;          partial_sums[omp_get_thread_num()]=sum;
432            #pragma omp barrier
433          #pragma omp master          #pragma omp master
434          {          {
435            out=0;            out=0;
# Line 525  maybelong Finley_Util_cumsum(maybelong N Line 439  maybelong Finley_Util_cumsum(maybelong N
439               partial_sums[i]=tmp;               partial_sums[i]=tmp;
440             }             }
441          }          }
442            #pragma omp barrier
443          sum=partial_sums[omp_get_thread_num()];          sum=partial_sums[omp_get_thread_num()];
444          #pragma omp for          #pragma omp for schedule(static)
445          for (i=0;i<N;++i) array[i]+=sum;          for (i=0;i<N;++i) {
446              tmp=sum;
447              sum+=array[i];
448              array[i]=tmp;
449            }
450        }        }
451     #else     #else
452        for (i=0;i<N;++i) {        for (i=0;i<N;++i) {
# Line 539  maybelong Finley_Util_cumsum(maybelong N Line 458  maybelong Finley_Util_cumsum(maybelong N
458     return out;     return out;
459  }  }
460    
461  void Finley_copyDouble(int n,double* source, double* target) {  void Finley_copyDouble(dim_t n,double* source, double* target) {
462    int i;    dim_t i;
463    for (i=0;i<n;i++) target[i]=source[i];    for (i=0;i<n;i++) target[i]=source[i];
464  }  }
465    
466    /*
467     * $Log$
468     * Revision 1.7  2005/07/08 04:07:59  jgs
469     * Merge of development branch back to main trunk on 2005-07-08
470     *
471     * Revision 1.1.1.1.2.4  2005/06/29 02:34:57  gross
472     * some changes towards 64 integers in finley
473     *
474     * Revision 1.1.1.1.2.3  2005/03/02 23:35:06  gross
475     * reimplementation of the ILU in Finley. block size>1 still needs some testing
476     *
477     * Revision 1.1.1.1.2.2  2005/02/18 02:27:31  gross
478     * two function that will be used for a reimplementation of the ILU preconditioner
479     *
480     * Revision 1.1.1.1.2.1  2004/11/12 06:58:19  gross
481     * a lot of changes to get the linearPDE class running: most important change is that there is no matrix format exposed to the user anymore. the format is chosen by the Domain according to the solver and symmetry
482     *
483     * Revision 1.1.1.1  2004/10/26 06:53:57  jgs
484     * initial import of project esys2
485     *
486     * Revision 1.3  2004/08/26 12:03:52  gross
487     * Some other bug in Finley_Assemble_gradient fixed.
488     *
489     * Revision 1.2  2004/07/02 04:21:13  gross
490     * Finley C code has been included
491     *
492     * Revision 1.1.1.1  2004/06/24 04:00:40  johng
493     * Initial version of eys using boost-python.
494     *
495     *
496     */

Legend:
Removed from v.113  
changed lines
  Added in v.123

  ViewVC Help
Powered by ViewVC 1.1.26