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: /*-------------------------------------------------------------*/