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: PetscInt size,size_int,id;
37: PetscDataTypeGetSize(type,&size);
38: F90GetID(type,&id);
39: ptr->addr = array;
40: ptr->id = id;
41: ptr->cookie = F90_COOKIE;
42: ptr->sd = size;
43: ptr->ndim = 1;
44: ptr->dim[0].extent = len;
45: ptr->dim[0].mult = size;
46: ptr->dim[0].lower = start;
47: ptr->sum_d = -(ptr->dim[0].lower*ptr->dim[0].mult);
50: return(0);
51: }
55: PetscErrorCode F90Array2dCreate(void *array,PetscDataType type,PetscInt start1,PetscInt len1,PetscInt start2,PetscInt len2,F90Array2d *ptr)
56: {
58: PetscInt size,size_int,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;
70: ptr->ndim = 2;
72: /* Note dims are in the reverse order for IBM */
73: ptr->dim[1].extent = len1;
74: ptr->dim[1].mult = size;
75: ptr->dim[1].lower = start1;
76: ptr->dim[0].extent = len2;
77: ptr->dim[0].mult = len1*size;
78: ptr->dim[0].lower = start2;
79: ptr->sum_d = -(ptr->dim[0].lower*ptr->dim[0].mult+ptr->dim[1].lower*ptr->dim[1].mult);
81: return(0);
82: }
87: PetscErrorCode F90Array3dCreate(void *array,PetscDataType type,PetscInt start1,PetscInt len1,PetscInt start2,PetscInt len2,PetscInt start3,PetscInt len3,F90Array3d *ptr)
88: {
90: PetscInt size,size_int,id;
96: PetscDataTypeGetSize(type,&size);
97: F90GetID(type,&id);
98: ptr->addr = array;
99: ptr->id = id;
100: ptr->cookie = F90_COOKIE;
101: ptr->sd = size;
102: ptr->ndim = 3;
104: /* Note dims are in the reverse order for IBM */
105: ptr->dim[2].extent = len1;
106: ptr->dim[2].mult = size;
107: ptr->dim[2].lower = start1;
108: ptr->dim[1].extent = len2;
109: ptr->dim[1].mult = len1*size;
110: ptr->dim[1].lower = start2;
111: ptr->dim[0].extent = len3;
112: ptr->dim[0].mult = len2*len1*size;
113: ptr->dim[0].lower = start3;
114: ptr->sum_d = -(ptr->dim[0].lower*ptr->dim[0].mult+
115: ptr->dim[1].lower*ptr->dim[1].mult+
116: ptr->dim[2].lower*ptr->dim[2].mult);
118: return(0);
119: }
123: PetscErrorCode F90Array4dCreate(void *array,PetscDataType type,PetscInt start1,PetscInt len1,PetscInt start2,PetscInt len2,PetscInt start3,PetscInt len3,PetscInt start4,PetscInt len4,F90Array4d *ptr)
124: {
126: PetscInt size,size_int,id;
132: PetscDataTypeGetSize(type,&size);
133: F90GetID(type,&id);
134: ptr->addr = array;
135: ptr->id = id;
136: ptr->cookie = F90_COOKIE;
137: ptr->sd = size;
138: ptr->ndim = 4;
140: /* Note dims are in the reverse order for IBM */
141: ptr->dim[3].extent = len1;
142: ptr->dim[3].mult = size;
143: ptr->dim[3].lower = start1;
144: ptr->dim[2].extent = len2;
145: ptr->dim[2].mult = len1*size;
146: ptr->dim[2].lower = start2;
147: ptr->dim[1].extent = len3;
148: ptr->dim[1].mult = len2*len1*size;
149: ptr->dim[1].lower = start3;
150: ptr->dim[0].extent = len4;
151: ptr->dim[0].mult = len4*len2*len1*size;
152: ptr->dim[0].lower = start4;
153: ptr->sum_d = -(ptr->dim[0].lower*ptr->dim[0].mult+
154: ptr->dim[1].lower*ptr->dim[1].mult+
155: ptr->dim[2].lower*ptr->dim[2].mult+
156: ptr->dim[3].lower*ptr->dim[3].mult);
158: return(0);
159: }
161: /*-------------------------------------------------------------*/