Actual source code: f90_t3e.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*8;
 44:   ptr->ndim          = 1;
 45:   ptr->dim[0].extent = len;
 46:   ptr->dim[0].mult   = 1;
 47:   ptr->dim[0].lower  = start;

 49:   return(0);
 50: }

 54: PetscErrorCode F90Array2dCreate(void *array,PetscDataType type,PetscInt start1,PetscInt len1,PetscInt start2,PetscInt len2,F90Array2d *ptr)
 55: {

 57:   size_t size;
 58:   PetscInt id;

 64:   PetscDataTypeGetSize(type,&size);
 65:   F90GetID(type,&id);
 66:   ptr->addr          = array;
 67:   ptr->id            = id;
 68:   ptr->cookie        = F90_COOKIE;
 69:   ptr->sd            = size*8;
 70:   ptr->ndim          = 2;
 71:   ptr->dim[0].extent = len1;
 72:   ptr->dim[0].mult   = 1;
 73:   ptr->dim[0].lower  = start1;
 74:   ptr->dim[1].extent = len2;
 75:   ptr->dim[1].mult   = len1*1;
 76:   ptr->dim[1].lower  = start2;

 78:   return(0);
 79: }


 84: PetscErrorCode F90Array3dCreate(void *array,PetscDataType type,PetscInt start1,PetscInt len1,PetscInt start2,PetscInt len2,PetscInt start3,PetscInt len3,F90Array3d *ptr)
 85: {

 87:   size_t size;
 88:   PetscInt id;

 94:   PetscDataTypeGetSize(type,&size);
 95:   F90GetID(type,&id);
 96:   ptr->addr          = array;
 97:   ptr->id            = id;
 98:   ptr->cookie        = F90_COOKIE;
 99:   ptr->sd            = size*8;
100:   ptr->ndim          = 3;
101:   ptr->dim[0].extent = len1;
102:   ptr->dim[0].mult   = 1;
103:   ptr->dim[0].lower  = start1;
104:   ptr->dim[1].extent = len2;
105:   ptr->dim[1].mult   = len1*1;
106:   ptr->dim[1].lower  = start2;
107:   ptr->dim[2].extent = len3;
108:   ptr->dim[2].mult   = len2*len1*1;
109:   ptr->dim[2].lower  = start3;

111:   return(0);
112: }

116: PetscErrorCode F90Array4dCreate(void *array,PetscDataType type,PetscInt start1,PetscInt len1,PetscInt start2,PetscInt len2,PetscInt start3,PetscInt len3,PetscInt start4,PetscInt len4,F90Array4d *ptr)
117: {

119:   size_t size;
120:   PetscInt id;

126:   PetscDataTypeGetSize(type,&size);
127:   F90GetID(type,&id);
128:   ptr->addr          = array;
129:   ptr->id            = id;
130:   ptr->cookie        = F90_COOKIE;
131:   ptr->sd            = size*8;
132:   ptr->ndim          = 4;
133:   ptr->dim[0].extent = len1;
134:   ptr->dim[0].mult   = 1;
135:   ptr->dim[0].lower  = start1;
136:   ptr->dim[1].extent = len2;
137:   ptr->dim[1].mult   = len1*1;
138:   ptr->dim[1].lower  = start2;
139:   ptr->dim[2].extent = len3;
140:   ptr->dim[2].mult   = len2*len1*1;
141:   ptr->dim[2].lower  = start3;
142:   ptr->dim[3].extent = len4;
143:   ptr->dim[3].mult   = len3*len2*len1*1;
144:   ptr->dim[3].lower  = start4;

146:   return(0);
147: }
148: /*-------------------------------------------------------------*/