/[escript]/branches/domexper/dudley/src/Util.c
ViewVC logotype

Diff of /branches/domexper/dudley/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 147 by jgs, Fri Aug 12 01:45:47 2005 UTC
# Line 21  Line 21 
21    
22  /**************************************************************/  /**************************************************************/
23    
24    /*   returns true if any of the values in the short array values is not equalt to Zero */
25    
26    bool_t Finley_Util_anyNonZeroDouble(dim_t N, double* values) {
27       dim_t q;
28       for (q=0;q<N;++q) if (ABS(values[q])>0) return TRUE;
29       return FALSE;
30    }
31    /**************************************************************/
32    
33  /*   gathers double values out from in by index: */  /*   gathers double values out from in by index: */
34    
35  /*        out(1:numData,1:len)=in(1:numData,index(1:len)) */  /*        out(1:numData,1:len)=in(1:numData,index(1:len)) */
36    
37  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){
38      int s,i;      dim_t s,i;
39      for (s=0;s<len;s++) {      for (s=0;s<len;s++) {
40         for (i=0;i<numData;i++) {         for (i=0;i<numData;i++) {
41            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 50  void Finley_Util_Gather_double(int len,m
50    
51  /*        out(1:numData,1:len)=in(1:numData,index(1:len)) */  /*        out(1:numData,1:len)=in(1:numData,index(1:len)) */
52    
53  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){
54      int s,i;      dim_t s,i;
55      for (s=0;s<len;s++) {      for (s=0;s<len;s++) {
56         for (i=0;i<numData;i++) {         for (i=0;i<numData;i++) {
57            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 65  void Finley_Util_Gather_int(int len,mayb
65    
66  /*        out(1:numData,index(1:len))+=in(1:numData,1:len) */  /*        out(1:numData,index(1:len))+=in(1:numData,1:len) */
67    
68  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){
69     int i,s;     dim_t i,s;
70     for (s=0;s<len;s++) {     for (s=0;s<len;s++) {
71         for(i=0;i<numData;i++) {         for(i=0;i<numData;i++) {
72            #pragma omp atomic            #pragma omp atomic
# Line 70  void Finley_Util_AddScatter(int len,mayb Line 79  void Finley_Util_AddScatter(int len,mayb
79    
80  /*          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) */
81    
82  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) {
83      int i,j,s;      dim_t i,j,s;
84      for (i=0;i<A1*A2;i++) A[i]=0;      for (i=0;i<A1*A2;i++) A[i]=0;
85         for (i=0;i<A1;i++) {         for (i=0;i<A1;i++) {
86            for (j=0;j<A2;j++) {            for (j=0;j<A2;j++) {
# Line 86  void Finley_Util_SmallMatMult(int A1,int Line 95  void Finley_Util_SmallMatMult(int A1,int
95    
96  /*        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 */
97    
98  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) {
99      int q,i,j,s;      dim_t q,i,j,s;
100      for (i=0;i<A1*A2*len;i++) A[i]=0;      for (i=0;i<A1*A2*len;i++) A[i]=0;
101      for (q=0;q<len;q++) {      for (q=0;q<len;q++) {
102         for (i=0;i<A1;i++) {         for (i=0;i<A1;i++) {
# Line 99  void Finley_Util_SmallMatSetMult(int len Line 108  void Finley_Util_SmallMatSetMult(int len
108         }         }
109      }      }
110  }  }
   
 /* 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;  
 }  
111  /*    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 */
112  /*    the determinante is returned. */  /*    the determinante is returned. */
113    
114  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){
115     int q;     dim_t q;
116     double D,A11,A12,A13,A21,A22,A23,A31,A32,A33;     register double D,A11,A12,A13,A21,A22,A23,A31,A32,A33;
117    
118     switch(dim) {     switch(dim) {
119        case 1:        case 1:
120           for (q=0;q<len;q++) {           for (q=0;q<len;q++) {
121              D=A[INDEX3(0,0,q,dim,dim)];              D=A[q];
122              if (ABS(D) > 0 ){              if (ABS(D) > 0 ){
123                 det[q]=D;                 det[q]=D;
124                 D=1./D;                 D=1./D;
125                 invA[INDEX3(0,0,q,dim,dim)]=D;                 invA[q]=D;
126              } else {              } else {
127                 Finley_ErrorCode=ZERO_DIVISION_ERROR;                 Finley_ErrorCode=ZERO_DIVISION_ERROR;
128                 sprintf(Finley_ErrorMsg,"Non-regular matrix");                 sprintf(Finley_ErrorMsg,"Non-regular matrix");
# Line 221  void Finley_Util_InvertSmallMat(int len, Line 133  void Finley_Util_InvertSmallMat(int len,
133    
134        case 2:        case 2:
135           for (q=0;q<len;q++) {           for (q=0;q<len;q++) {
136              A11=A[INDEX3(0,0,q,dim,dim)];              A11=A[INDEX3(0,0,q,2,2)];
137              A12=A[INDEX3(0,1,q,dim,dim)];              A12=A[INDEX3(0,1,q,2,2)];
138              A21=A[INDEX3(1,0,q,dim,dim)];              A21=A[INDEX3(1,0,q,2,2)];
139              A22=A[INDEX3(1,1,q,dim,dim)];              A22=A[INDEX3(1,1,q,2,2)];
140    
141              D = A11*A22-A12*A21;              D = A11*A22-A12*A21;
142              if (ABS(D) > 0 ){              if (ABS(D) > 0 ){
143                 det[q]=D;                 det[q]=D;
144                 D=1./D;                 D=1./D;
145                 invA[INDEX3(0,0,q,dim,dim)]= A22*D;                 invA[INDEX3(0,0,q,2,2)]= A22*D;
146                 invA[INDEX3(1,0,q,dim,dim)]=-A21*D;                 invA[INDEX3(1,0,q,2,2)]=-A21*D;
147                 invA[INDEX3(0,1,q,dim,dim)]=-A12*D;                 invA[INDEX3(0,1,q,2,2)]=-A12*D;
148                 invA[INDEX3(1,1,q,dim,dim)]= A11*D;                 invA[INDEX3(1,1,q,2,2)]= A11*D;
149              } else {              } else {
150                 Finley_ErrorCode=ZERO_DIVISION_ERROR;                 Finley_ErrorCode=ZERO_DIVISION_ERROR;
151                 sprintf(Finley_ErrorMsg,"Non-regular matrix");                 sprintf(Finley_ErrorMsg,"Non-regular matrix");
# Line 244  void Finley_Util_InvertSmallMat(int len, Line 156  void Finley_Util_InvertSmallMat(int len,
156    
157        case 3:        case 3:
158       for (q=0;q<len;q++) {       for (q=0;q<len;q++) {
159              A11=A[INDEX3(0,0,q,dim,dim)];              A11=A[INDEX3(0,0,q,3,3)];
160              A21=A[INDEX3(1,0,q,dim,dim)];              A21=A[INDEX3(1,0,q,3,3)];
161              A31=A[INDEX3(2,0,q,dim,dim)];              A31=A[INDEX3(2,0,q,3,3)];
162              A12=A[INDEX3(0,1,q,dim,dim)];              A12=A[INDEX3(0,1,q,3,3)];
163              A22=A[INDEX3(1,1,q,dim,dim)];              A22=A[INDEX3(1,1,q,3,3)];
164              A32=A[INDEX3(2,1,q,dim,dim)];              A32=A[INDEX3(2,1,q,3,3)];
165              A13=A[INDEX3(0,2,q,dim,dim)];              A13=A[INDEX3(0,2,q,3,3)];
166              A23=A[INDEX3(1,2,q,dim,dim)];              A23=A[INDEX3(1,2,q,3,3)];
167              A33=A[INDEX3(2,2,q,dim,dim)];              A33=A[INDEX3(2,2,q,3,3)];
168    
169              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);
170              if (ABS(D) > 0 ){              if (ABS(D) > 0 ){
171                 det[q]    =D;                 det[q]    =D;
172                 D=1./D;                 D=1./D;
173                 invA[INDEX3(0,0,q,dim,dim)]=(A22*A33-A23*A32)*D;                 invA[INDEX3(0,0,q,3,3)]=(A22*A33-A23*A32)*D;
174                 invA[INDEX3(1,0,q,dim,dim)]=(A31*A23-A21*A33)*D;                 invA[INDEX3(1,0,q,3,3)]=(A31*A23-A21*A33)*D;
175                 invA[INDEX3(2,0,q,dim,dim)]=(A21*A32-A31*A22)*D;                 invA[INDEX3(2,0,q,3,3)]=(A21*A32-A31*A22)*D;
176                 invA[INDEX3(0,1,q,dim,dim)]=(A13*A32-A12*A33)*D;                 invA[INDEX3(0,1,q,3,3)]=(A13*A32-A12*A33)*D;
177                 invA[INDEX3(1,1,q,dim,dim)]=(A11*A33-A31*A13)*D;                 invA[INDEX3(1,1,q,3,3)]=(A11*A33-A31*A13)*D;
178                 invA[INDEX3(2,1,q,dim,dim)]=(A12*A31-A11*A32)*D;                 invA[INDEX3(2,1,q,3,3)]=(A12*A31-A11*A32)*D;
179                 invA[INDEX3(0,2,q,dim,dim)]=(A12*A23-A13*A22)*D;                 invA[INDEX3(0,2,q,3,3)]=(A12*A23-A13*A22)*D;
180                 invA[INDEX3(1,2,q,dim,dim)]=(A13*A21-A11*A23)*D;                 invA[INDEX3(1,2,q,3,3)]=(A13*A21-A11*A23)*D;
181                 invA[INDEX3(2,2,q,dim,dim)]=(A11*A22-A12*A21)*D;                 invA[INDEX3(2,2,q,3,3)]=(A11*A22-A12*A21)*D;
182              } else {              } else {
183                 Finley_ErrorCode=ZERO_DIVISION_ERROR;                 Finley_ErrorCode=ZERO_DIVISION_ERROR;
184                 sprintf(Finley_ErrorMsg,"Non-regular matrix");                 sprintf(Finley_ErrorMsg,"Non-regular matrix");
# Line 281  void Finley_Util_InvertSmallMat(int len, Line 193  void Finley_Util_InvertSmallMat(int len,
193    
194  /*    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 */
195    
196  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){
197     int q;     dim_t q;
198     double A11,A12,A13,A21,A22,A23,A31,A32,A33;     register double A11,A12,A13,A21,A22,A23,A31,A32,A33;
199    
200     switch(dim) {     switch(dim) {
201        case 1:        case 1:
202           for (q=0;q<len;q++) {           for (q=0;q<len;q++) {
203              det[q]=A[INDEX3(0,0,q,dim,dim)];              det[q]=A[q];
204           }           }
205           break;           break;
206    
207        case 2:        case 2:
208           for (q=0;q<len;q++) {           for (q=0;q<len;q++) {
209              A11=A[INDEX3(0,0,q,dim,dim)];              A11=A[INDEX3(0,0,q,2,2)];
210              A12=A[INDEX3(0,1,q,dim,dim)];              A12=A[INDEX3(0,1,q,2,2)];
211              A21=A[INDEX3(1,0,q,dim,dim)];              A21=A[INDEX3(1,0,q,2,2)];
212              A22=A[INDEX3(1,1,q,dim,dim)];              A22=A[INDEX3(1,1,q,2,2)];
213    
214              det[q] = A11*A22-A12*A21;              det[q] = A11*A22-A12*A21;
215           }           }
# Line 305  void Finley_Util_DetOfSmallMat(int len,i Line 217  void Finley_Util_DetOfSmallMat(int len,i
217    
218        case 3:        case 3:
219       for (q=0;q<len;q++) {       for (q=0;q<len;q++) {
220              A11=A[INDEX3(0,0,q,dim,dim)];              A11=A[INDEX3(0,0,q,3,3)];
221              A21=A[INDEX3(1,0,q,dim,dim)];              A21=A[INDEX3(1,0,q,3,3)];
222              A31=A[INDEX3(2,0,q,dim,dim)];              A31=A[INDEX3(2,0,q,3,3)];
223              A12=A[INDEX3(0,1,q,dim,dim)];              A12=A[INDEX3(0,1,q,3,3)];
224              A22=A[INDEX3(1,1,q,dim,dim)];              A22=A[INDEX3(1,1,q,3,3)];
225              A32=A[INDEX3(2,1,q,dim,dim)];              A32=A[INDEX3(2,1,q,3,3)];
226              A13=A[INDEX3(0,2,q,dim,dim)];              A13=A[INDEX3(0,2,q,3,3)];
227              A23=A[INDEX3(1,2,q,dim,dim)];              A23=A[INDEX3(1,2,q,3,3)];
228              A33=A[INDEX3(2,2,q,dim,dim)];              A33=A[INDEX3(2,2,q,3,3)];
229    
230              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);
231           }           }
# Line 325  void Finley_Util_DetOfSmallMat(int len,i Line 237  void Finley_Util_DetOfSmallMat(int len,i
237  /*    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  */
238  /*    or the vector A(:,0,q) in the case of dim=2                                             */  /*    or the vector A(:,0,q) in the case of dim=2                                             */
239    
240  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) {
241     int q;     dim_t q;
242     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;
243    
244     switch(dim) {     switch(dim) {
245        case 1:        case 1:
246           for (q=0;q<len;q++) Normal[INDEX1(q)]    =1;           for (q=0;q<len;q++) Normal[q]    =1;
247           break;           break;
248        case 2:        case 2:
249           for (q=0;q<len;q++) {           for (q=0;q<len;q++) {
250              A11=A[INDEX3(0,0,q,dim,dim1)];              A11=A[INDEX3(0,0,q,2,dim1)];
251              A21=A[INDEX3(1,0,q,dim,dim1)];              A21=A[INDEX3(1,0,q,2,dim1)];
252              length = sqrt(A11*A11+A21*A21);              length = sqrt(A11*A11+A21*A21);
253              if (! length>0) {              if (! length>0) {
254                 Finley_ErrorCode=ZERO_DIVISION_ERROR;                 Finley_ErrorCode=ZERO_DIVISION_ERROR;
# Line 344  void  Finley_NormalVector(int len, int d Line 256  void  Finley_NormalVector(int len, int d
256                 return;                 return;
257              } else {              } else {
258                 invlength=1./length;                 invlength=1./length;
259                 Normal[INDEX2(0,q,dim)]=A21*invlength;                 Normal[INDEX2(0,q,2)]=A21*invlength;
260                 Normal[INDEX2(1,q,dim)]=-A11*invlength;                 Normal[INDEX2(1,q,2)]=-A11*invlength;
261              }              }
262           }           }
263           break;           break;
264        case 3:        case 3:
265           for (q=0;q<len;q++) {           for (q=0;q<len;q++) {
266              A11=A[INDEX3(0,0,q,dim,dim1)];              A11=A[INDEX3(0,0,q,3,dim1)];
267              A21=A[INDEX3(1,0,q,dim,dim1)];              A21=A[INDEX3(1,0,q,3,dim1)];
268              A31=A[INDEX3(2,0,q,dim,dim1)];              A31=A[INDEX3(2,0,q,3,dim1)];
269              A12=A[INDEX3(0,1,q,dim,dim1)];              A12=A[INDEX3(0,1,q,3,dim1)];
270              A22=A[INDEX3(1,1,q,dim,dim1)];              A22=A[INDEX3(1,1,q,3,dim1)];
271              A32=A[INDEX3(2,1,q,dim,dim1)];              A32=A[INDEX3(2,1,q,3,dim1)];
272              CO_A13=A21*A32-A31*A22;              CO_A13=A21*A32-A31*A22;
273              CO_A23=A31*A12-A11*A32;              CO_A23=A31*A12-A11*A32;
274              CO_A33=A11*A22-A21*A12;              CO_A33=A11*A22-A21*A12;
# Line 367  void  Finley_NormalVector(int len, int d Line 279  void  Finley_NormalVector(int len, int d
279                 return;                 return;
280              } else {              } else {
281                 invlength=1./length;                 invlength=1./length;
282                 Normal[INDEX2(0,q,dim)]=CO_A13*invlength;                 Normal[INDEX2(0,q,3)]=CO_A13*invlength;
283                 Normal[INDEX2(1,q,dim)]=CO_A23*invlength;                 Normal[INDEX2(1,q,3)]=CO_A23*invlength;
284                 Normal[INDEX2(2,q,dim)]=CO_A33*invlength;                 Normal[INDEX2(2,q,3)]=CO_A33*invlength;
285             }             }
286                            
287        }        }
# Line 382  void  Finley_NormalVector(int len, int d Line 294  void  Finley_NormalVector(int len, int d
294  /*    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 */
295  /*    or the vector A(:,0,q) in the case of dim=2                                                                   */  /*    or the vector A(:,0,q) in the case of dim=2                                                                   */
296    
297  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) {
298     int q;     dim_t q;
299     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;
300    
301     switch(dim) {     switch(dim) {
302        case 1:        case 1:
303           for (q=0;q<len;q++) length[INDEX1(q)]    =1;           for (q=0;q<len;q++) length[q]    =1;
304           break;           break;
305        case 2:        case 2:
306           for (q=0;q<len;q++) {           for (q=0;q<len;q++) {
307              A11=A[INDEX3(0,0,q,dim,dim1)];              A11=A[INDEX3(0,0,q,2,dim1)];
308              A21=A[INDEX3(1,0,q,dim,dim1)];              A21=A[INDEX3(1,0,q,2,dim1)];
309              length[q] = sqrt(A11*A11+A21*A21);              length[q] = sqrt(A11*A11+A21*A21);
310           }           }
311           break;           break;
312        case 3:        case 3:
313           for (q=0;q<len;q++) {           for (q=0;q<len;q++) {
314              A11=A[INDEX3(0,0,q,dim,dim1)];              A11=A[INDEX3(0,0,q,3,dim1)];
315              A21=A[INDEX3(1,0,q,dim,dim1)];              A21=A[INDEX3(1,0,q,3,dim1)];
316              A31=A[INDEX3(2,0,q,dim,dim1)];              A31=A[INDEX3(2,0,q,3,dim1)];
317              A12=A[INDEX3(0,1,q,dim,dim1)];              A12=A[INDEX3(0,1,q,3,dim1)];
318              A22=A[INDEX3(1,1,q,dim,dim1)];              A22=A[INDEX3(1,1,q,3,dim1)];
319              A32=A[INDEX3(2,1,q,dim,dim1)];              A32=A[INDEX3(2,1,q,3,dim1)];
320              CO_A13=A21*A32-A31*A22;              CO_A13=A21*A32-A31*A22;
321              CO_A23=A31*A12-A11*A32;              CO_A23=A31*A12-A11*A32;
322              CO_A33=A11*A22-A21*A12;              CO_A33=A11*A22-A21*A12;
# Line 420  void  Finley_LengthOfNormalVector(int le Line 332  void  Finley_LengthOfNormalVector(int le
332  /* there is no range checking! */  /* there is no range checking! */
333  /* at output Map[invMap[i]]=i for i=0:lenInvMap */  /* at output Map[invMap[i]]=i for i=0:lenInvMap */
334    
335  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) {
336     int i;     dim_t i;
337     for (i=0;i<lenInvMap;i++) invMap[i]=0;     for (i=0;i<lenInvMap;i++) invMap[i]=0;
338     for (i=0;i<lenMap;i++) {     for (i=0;i<lenMap;i++) {
339        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 351  int Finley_Util_ValueAndIndex_compar(con
351     if (e1->value > e2->value) return  1;     if (e1->value > e2->value) return  1;
352     return 0;     return 0;
353  }  }
354  void Finley_Util_sortValueAndIndex(int n,Finley_Util_ValueAndIndex* array) {  void Finley_Util_sortValueAndIndex(dim_t n,Finley_Util_ValueAndIndex* array) {
355       /* OMP : needs parallelization !*/       /* OMP : needs parallelization !*/
356       qsort(array,n,sizeof(Finley_Util_ValueAndIndex),Finley_Util_ValueAndIndex_compar);       qsort(array,n,sizeof(Finley_Util_ValueAndIndex),Finley_Util_ValueAndIndex_compar);
357  }  }
# Line 449  void Finley_Util_sortValueAndIndex(int n Line 361  void Finley_Util_sortValueAndIndex(int n
361    
362  /* calculates the minimum value from a dim X N integer array */  /* calculates the minimum value from a dim X N integer array */
363    
364  maybelong Finley_Util_getMinInt(int dim,int N,maybelong* values) {  index_t Finley_Util_getMinInt(dim_t dim,dim_t N,index_t* values) {
365     maybelong i,j,out;     dim_t i,j;
366     out=MAYBELONG_MAX;     index_t out,out_local;
367       out=INDEX_T_MAX;
368     if (values!=NULL && dim*N>0 ) {     if (values!=NULL && dim*N>0 ) {
      /* OMP */  
369       out=values[0];       out=values[0];
370       for (j=0;j<N;j++) {       #pragma omp parallel private(out_local)
371         for (i=0;i<dim;i++) out=MIN(out,values[INDEX2(i,j,dim)]);       {
372             out_local=out;
373             #pragma omp for private(i,j) schedule(static)
374             for (j=0;j<N;j++) {
375               for (i=0;i<dim;i++) out_local=MIN(out_local,values[INDEX2(i,j,dim)]);
376             }
377             #pragma omp critical
378             out=MIN(out_local,out);
379       }       }
380     }     }
381     return out;     return out;
# Line 464  maybelong Finley_Util_getMinInt(int dim, Line 383  maybelong Finley_Util_getMinInt(int dim,
383                                                                                                                                                                                                                                                                                                        
384  /* calculates the maximum value from a dim X N integer array */  /* calculates the maximum value from a dim X N integer array */
385    
386  maybelong Finley_Util_getMaxInt(int dim,int N,maybelong* values) {  index_t Finley_Util_getMaxInt(dim_t dim,dim_t N,index_t* values) {
387     maybelong i,j,out;     dim_t i,j;
388     out=-MAYBELONG_MAX;     index_t out,out_local;
389       out=-INDEX_T_MAX;
390     if (values!=NULL && dim*N>0 ) {     if (values!=NULL && dim*N>0 ) {
      /* OMP */  
391       out=values[0];       out=values[0];
392       for (j=0;j<N;j++) {       #pragma omp parallel private(out_local)
393         for (i=0;i<dim;i++) out=MAX(out,values[INDEX2(i,j,dim)]);       {
394       }           out_local=out;
395             #pragma omp for private(i,j) schedule(static)
396             for (j=0;j<N;j++) {
397                 for (i=0;i<dim;i++) out_local=MAX(out_local,values[INDEX2(i,j,dim)]);
398             }
399             #pragma omp critical
400             out=MAX(out_local,out);
401          }
402     }     }
403     return out;     return out;
404  }  }
405    
406  /* 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. */
407    
408  maybelong Finley_Util_packMask(maybelong N,maybelong* mask,maybelong* index) {  dim_t Finley_Util_packMask(dim_t N,index_t* mask,index_t* index) {
409        maybelong out,k;        dim_t out,k;
410        out=0;        out=0;
411        /*OMP */        /*OMP */
412        for (k=0;k<N;k++) {        for (k=0;k<N;k++) {
# Line 493  maybelong Finley_Util_packMask(maybelong Line 419  maybelong Finley_Util_packMask(maybelong
419  }  }
420    
421  /* returns true if array contains value */  /* returns true if array contains value */
422  int Finley_Util_isAny(maybelong N,maybelong* array,maybelong value) {  bool_t Finley_Util_isAny(dim_t N,index_t* array,index_t value) {
423     int out=FALSE;     bool_t out=FALSE;
424     maybelong i;     dim_t i;
425     #pragma omp parallel for private(i) schedule(static) reduction(||:out)     #pragma omp parallel for private(i) schedule(static) reduction(||:out)
426     for (i=0;i<N;i++) out=out || (array[i]==value);     for (i=0;i<N;i++) out = out || (array[i]==value);
427     return out;     return out;
428  }  }
429  /* calculates the cummultative sum in array and returns the total sum */  /* calculates the cummultative sum in array and returns the total sum */
430  maybelong Finley_Util_cumsum(maybelong N,maybelong* array) {  index_t Finley_Util_cumsum(dim_t N,index_t* array) {
431     maybelong out=0,tmp,i;     index_t out=0,tmp;
432       dim_t i;
433     #ifdef _OPENMP     #ifdef _OPENMP
434        maybelong partial_sums[omp_get_max_threads()],sum;        index_t partial_sums[omp_get_max_threads()],sum;
435        #pragma omp parallel private(sum,i,tmp)        #pragma omp parallel private(sum,i,tmp)
436        {        {
437          sum=0;          sum=0;
438          #pragma omp for          #pragma omp for schedule(static)
439          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  
440          partial_sums[omp_get_thread_num()]=sum;          partial_sums[omp_get_thread_num()]=sum;
441            #pragma omp barrier
442          #pragma omp master          #pragma omp master
443          {          {
444            out=0;            out=0;
# Line 525  maybelong Finley_Util_cumsum(maybelong N Line 448  maybelong Finley_Util_cumsum(maybelong N
448               partial_sums[i]=tmp;               partial_sums[i]=tmp;
449             }             }
450          }          }
451            #pragma omp barrier
452          sum=partial_sums[omp_get_thread_num()];          sum=partial_sums[omp_get_thread_num()];
453          #pragma omp for          #pragma omp for schedule(static)
454          for (i=0;i<N;++i) array[i]+=sum;          for (i=0;i<N;++i) {
455              tmp=sum;
456              sum+=array[i];
457              array[i]=tmp;
458            }
459        }        }
460     #else     #else
461        for (i=0;i<N;++i) {        for (i=0;i<N;++i) {
# Line 539  maybelong Finley_Util_cumsum(maybelong N Line 467  maybelong Finley_Util_cumsum(maybelong N
467     return out;     return out;
468  }  }
469    
470  void Finley_copyDouble(int n,double* source, double* target) {  void Finley_copyDouble(dim_t n,double* source, double* target) {
471    int i;    dim_t i;
472    for (i=0;i<n;i++) target[i]=source[i];    for (i=0;i<n;i++) target[i]=source[i];
473  }  }
474    
475    /*
476     * $Log$
477     * Revision 1.8  2005/08/12 01:45:43  jgs
478     * erge of development branch dev-02 back to main trunk on 2005-08-12
479     *
480     * Revision 1.7.2.1  2005/08/04 22:41:11  gross
481     * some extra routines for finley that might speed-up RHS assembling in some cases (not actived right now)
482     *
483     * Revision 1.7  2005/07/08 04:07:59  jgs
484     * Merge of development branch back to main trunk on 2005-07-08
485     *
486     * Revision 1.1.1.1.2.4  2005/06/29 02:34:57  gross
487     * some changes towards 64 integers in finley
488     *
489     * Revision 1.1.1.1.2.3  2005/03/02 23:35:06  gross
490     * reimplementation of the ILU in Finley. block size>1 still needs some testing
491     *
492     * Revision 1.1.1.1.2.2  2005/02/18 02:27:31  gross
493     * two function that will be used for a reimplementation of the ILU preconditioner
494     *
495     * Revision 1.1.1.1.2.1  2004/11/12 06:58:19  gross
496     * 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
497     *
498     * Revision 1.1.1.1  2004/10/26 06:53:57  jgs
499     * initial import of project esys2
500     *
501     * Revision 1.3  2004/08/26 12:03:52  gross
502     * Some other bug in Finley_Assemble_gradient fixed.
503     *
504     * Revision 1.2  2004/07/02 04:21:13  gross
505     * Finley C code has been included
506     *
507     * Revision 1.1.1.1  2004/06/24 04:00:40  johng
508     * Initial version of eys using boost-python.
509     *
510     *
511     */

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

  ViewVC Help
Powered by ViewVC 1.1.26