Actual source code: f90_rs6000.c

  2: /*-------------------------------------------------------------*/

  6: PetscErrorCode F90GetID(PetscDataType type,PetscInt *id)
  7: {
  9:   if (type == PETSC_INT) {
 10:     *id = F90_INT_ID;
 11:   } else if (type == PETSC_DOUBLE) {
 12:     *id = F90_DOUBLE_ID;
 13: #if defined(PETSC_USE_COMPLEX)
 14:   } else if (type == PETSC_COMPLEX) {
 15:     *id = F90_COMPLEX_ID;
 16: #endif
 17:   } else if (type == PETSC_LONG) {
 18:     *id = F90_INT_ID;
 19:   } else if (type == PETSC_CHAR) {
 20:     *id = F90_CHAR_ID;
 21:   } else {
 22:     SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"Unknown PETSc datatype");
 23:   }
 24:   return(0);
 25: }

 29: PetscErrorCode F90Array1dCreate(void *array,PetscDataType type,PetscInt start,PetscInt len,F90Array1d *ptr)
 30: {
 31:   size_t size;
 32:   PetscInt id;

 38:   PetscDataTypeGetSize(type,&size);
 39:   F90GetID(type,&id);
 40:   ptr->addr          = array;
 41:   ptr->id            = id;
 42:   ptr->cookie        = F90_COOKIE;
 43:   ptr->sd            = size;
 44:   ptr->ndim          = 1;
 45:   ptr->dim[0].extent = len;
 46:   ptr->dim[0].mult   = size;
 47:   ptr->dim[0].lower  = start;
 48:   ptr->sum_d         = -(ptr->dim[0].lower*ptr->dim[0].mult);


 51:   return(0);
 52: }

 56: PetscErrorCode F90Array2dCreate(void *array,PetscDataType type,PetscInt start1,PetscInt len1,PetscInt start2,PetscInt len2,F90Array2d *ptr)
 57: {
 58:   size_t size;
 59:   PetscInt id;

 65:   PetscDataTypeGetSize(type,&size);
 66:   F90GetID(type,&id);
 67:   ptr->addr          = array;
 68:   ptr->id            = id;
 69:   ptr->cookie        = F90_COOKIE;
 70:   ptr->sd            = size;
 71:   ptr->ndim          = 2;

 73:   /* Note dims are in the reverse order for IBM */
 74:   ptr->dim[1].extent = len1;
 75:   ptr->dim[1].mult   = size;
 76:   ptr->dim[1].lower  = start1;
 77:   ptr->dim[0].extent = len2;
 78:   ptr->dim[0].mult   = len1*size;
 79:   ptr->dim[0].lower  = start2;
 80:   ptr->sum_d         = -(ptr->dim[0].lower*ptr->dim[0].mult+ptr->dim[1].lower*ptr->dim[1].mult);

 82:   return(0);
 83: }


 88: PetscErrorCode F90Array3dCreate(void *array,PetscDataType type,PetscInt start1,PetscInt len1,PetscInt start2,PetscInt len2,PetscInt start3,PetscInt len3,F90Array3d *ptr)
 89: {
 90:   size_t size;
 91:   PetscInt id;

 97:   PetscDataTypeGetSize(type,&size);
 98:   F90GetID(type,&id);
 99:   ptr->addr          = array;
100:   ptr->id            = id;
101:   ptr->cookie        = F90_COOKIE;
102:   ptr->sd            = size;
103:   ptr->ndim          = 3;

105:   /* Note dims are in the reverse order for IBM */
106:   ptr->dim[2].extent = len1;
107:   ptr->dim[2].mult   = size;
108:   ptr->dim[2].lower  = start1;
109:   ptr->dim[1].extent = len2;
110:   ptr->dim[1].mult   = len1*size;
111:   ptr->dim[1].lower  = start2;
112:   ptr->dim[0].extent = len3;
113:   ptr->dim[0].mult   = len2*len1*size;
114:   ptr->dim[0].lower  = start3;
115:   ptr->sum_d         = -(ptr->dim[0].lower*ptr->dim[0].mult+
116:                          ptr->dim[1].lower*ptr->dim[1].mult+
117:                          ptr->dim[2].lower*ptr->dim[2].mult);

119:   return(0);
120: }

124: PetscErrorCode F90Array4dCreate(void *array,PetscDataType type,PetscInt start1,PetscInt len1,PetscInt start2,PetscInt len2,PetscInt start3,PetscInt len3,PetscInt start4,PetscInt len4,F90Array4d *ptr)
125: {
126:   size_t size;
127:   PetscInt id;

133:   PetscDataTypeGetSize(type,&size);
134:   F90GetID(type,&id);
135:   ptr->addr          = array;
136:   ptr->id            = id;
137:   ptr->cookie        = F90_COOKIE;
138:   ptr->sd            = size;
139:   ptr->ndim          = 4;

141:   /* Note dims are in the reverse order for IBM */
142:   ptr->dim[3].extent = len1;
143:   ptr->dim[3].mult   = size;
144:   ptr->dim[3].lower  = start1;
145:   ptr->dim[2].extent = len2;
146:   ptr->dim[2].mult   = len1*size;
147:   ptr->dim[2].lower  = start2;
148:   ptr->dim[1].extent = len3;
149:   ptr->dim[1].mult   = len2*len1*size;
150:   ptr->dim[1].lower  = start3;
151:   ptr->dim[0].extent = len4;
152:   ptr->dim[0].mult   = len4*len2*len1*size;
153:   ptr->dim[0].lower  = start4;
154:   ptr->sum_d         = -(ptr->dim[0].lower*ptr->dim[0].mult+
155:                          ptr->dim[1].lower*ptr->dim[1].mult+
156:                          ptr->dim[2].lower*ptr->dim[2].mult+
157:                          ptr->dim[3].lower*ptr->dim[3].mult);

159:   return(0);
160: }

162: /*-------------------------------------------------------------*/