OilfieldToolsNet/opm-core
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add function create_cart_grid_2d. Preliminary implementation.

Browse Source
This commit is contained in:
Jostein R. Natvig 2011-12-14 06:58:02 +01:00
parent a1c3f22dc4
commit 369a438f13
1 changed files with 234 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

234
opmcore/utility/cart_grid.c
View File

@ -28,6 +28,7 @@
*/
#include <stdlib.h>
#include <stdio.h>
#include "cart_grid.h"
void
@ -67,6 +68,12 @@ create_cart_grid(int nx, int ny, int nz)
int *fnodes, *fnodepos, *fcells, *cfaces, *cfacepos;
G = malloc(1 * sizeof *G);
if (G == NULL)
{
fprintf(stderr, "Cannot allocate space for grid.\n");
exit(EXIT_FAILURE);
}
G->dimensions = 3;
Nx = nx+1;
@ -95,6 +102,20 @@ create_cart_grid(int nx, int ny, int nz)
G->cell_centroids = malloc(G->number_of_cells * 3 * sizeof *(G->cell_centroids));
G->cell_volumes = malloc(G->number_of_cells * 1 * sizeof *(G->cell_volumes));
if ((G->face_nodes == NULL ) ||
(G->face_nodepos == NULL ) ||
(G->face_cells == NULL ) ||
(G->face_centroids == NULL ) ||
(G->face_normals == NULL ) ||
(G->face_areas == NULL ) ||
(G->cell_faces == NULL ) ||
(G->cell_facepos == NULL ) ||
(G->cell_centroids == NULL ) ||
(G->cell_volumes == NULL ) )
{
fprintf(stderr, "Cannot allocate space for grid.\n");
exit(EXIT_FAILURE);
}
cfaces = G->cell_faces;
cfacepos = G->cell_facepos;
@ -250,3 +271,216 @@ create_cart_grid(int nx, int ny, int nz)
return G;
}
grid_t *
create_cart_grid_2d(int nx, int ny)
{
int i,j;
int nxf, nyf;
int Nx, Ny;
grid_t *G;
double *coord, *ccentroids, *cvolumes;
double *fnormals, *fcentroids, *fareas;
int *fnodes, *fnodepos, *fcells, *cfaces, *cfacepos;
G = malloc(1 * sizeof *G);
if (G == NULL)
{
fprintf(stderr, "Cannot allocate space for grid.\n");
exit(EXIT_FAILURE);
}
G->dimensions = 2;
Nx = nx+1;
Ny = ny+1;
nxf = Nx*ny;
nyf = nx*Ny;
G->number_of_cells = nx*ny;
G->number_of_faces = nxf+nyf;
G->number_of_nodes = Nx*Ny;
G->node_coordinates = malloc(G->number_of_nodes * 2 * sizeof *(G->node_coordinates));
G->face_nodes = malloc(G->number_of_faces * 2 * sizeof *(G->face_nodes));
G->face_nodepos = malloc((G->number_of_faces+1) * sizeof *(G->face_nodepos));
G->face_cells = malloc(G->number_of_faces * 2 * sizeof *(G->face_cells));
G->face_centroids = malloc(G->number_of_faces * 2 * sizeof *(G->face_centroids));
G->face_normals = malloc(G->number_of_faces * 2 * sizeof *(G->face_normals));
G->face_areas = malloc(G->number_of_faces * 1 * sizeof *(G->face_areas));
G->cell_faces = malloc(G->number_of_cells * 4 * sizeof *(G->cell_faces));
G->cell_facepos = malloc((G->number_of_cells+1) * sizeof *(G->cell_facepos));
G->cell_centroids = malloc(G->number_of_cells * 2 * sizeof *(G->cell_centroids));
G->cell_volumes = malloc(G->number_of_cells * 1 * sizeof *(G->cell_volumes));
if ((G->face_nodes == NULL ) ||
(G->face_nodepos == NULL ) ||
(G->face_cells == NULL ) ||
(G->face_centroids == NULL ) ||
(G->face_normals == NULL ) ||
(G->face_areas == NULL ) ||
(G->cell_faces == NULL ) ||
(G->cell_facepos == NULL ) ||
(G->cell_centroids == NULL ) ||
(G->cell_volumes == NULL ) )
{
fprintf(stderr, "Cannot allocate space for grid.\n");
exit(EXIT_FAILURE);
}
cfaces = G->cell_faces;
cfacepos = G->cell_facepos;
ccentroids = G->cell_centroids;
cvolumes = G->cell_volumes;
cfacepos[0] = 0;
for (j=0; j<ny; ++j) {
for (i=0; i<nx; ++i) {
*cfaces++ = i+ Nx*j;
*cfaces++ = i+ nx*j +nxf;
*cfaces++ = i+1+Nx*j;
*cfaces++ = i+ nx*(j+1)+nxf;
cfacepos[1] = cfacepos[0]+4;
++cfacepos;
*ccentroids++ = i+0.5;
*ccentroids++ = j+0.5;
*cvolumes++ = 1;
}
}
fnodes = G->face_nodes;
fnodepos = G->face_nodepos;
fcells = G->face_cells;
fnormals = G->face_normals;
fcentroids = G->face_centroids;
fareas = G->face_areas;
/* Faces with x-normal */
for (j=0; j<ny; ++j) {
for (i=0; i<nx+1; ++i) {
*fnodes++ = i+Nx*j;
*fnodes++ = i+Nx*j+1;
fnodepos[1] = fnodepos[0] + 2;
++fnodepos;
if (i==0) {
*fcells++ = -1;
*fcells++ = i+nx*j;
}
else if (i == nx) {
*fcells++ = i-1+nx*j;
*fcells++ = -1;
}
else {
*fcells++ = i-1 + nx*j;
*fcells++ = i + nx*j;
}
*fnormals++ = 1;
*fnormals++ = 0;
*fcentroids++ = i;
*fcentroids++ = j+0.5;
*fareas++ = 1;
}
}
/* Faces with y-normal */
for (j=0; j<ny+1; ++j) {
for (i=0; i<nx; ++i) {
*fnodes++ = i+ Nx*j;
*fnodes++ = i+1 + Nx*j;
fnodepos[1] = fnodepos[0] + 2;
++fnodepos;
if (j==0) {
*fcells++ = -1;
*fcells++ = i+nx*j;
}
else if (j == ny) {
*fcells++ = i+nx*j;
*fcells++ = -1;
}
else {
*fcells++ = i+nx*(j-1);
*fcells++ = i+nx*j;
}
*fnormals++ = 0;
*fnormals++ = 1;
*fcentroids++ = i+0.5;
*fcentroids++ = j;
*fareas++ = 1;
}
}
coord = G->node_coordinates;
for (j=0; j<ny+1; ++j) {
for (i=0; i<nx+1; ++i) {
*coord++ = i;
*coord++ = j;
}
}
return G;
}
void
free_cart_grid(struct UnstructuredGrid *g)
{
if (g != NULL)
{
free(g->node_coordinates);
free(g->face_nodes);
free(g->face_nodepos);
free(g->face_cells);
free(g->face_centroids);
free(g->face_normals);
free(g->face_areas);
free(g->cell_faces);
free(g->cell_facepos);
free(g->cell_centroids);
free(g->cell_volumes);
}
free(g);
}
#if UNIT_TEST
int main()
{
grid_t *g = create_cart_grid_2d(2,2);
int i;
int k;
for(i=0; i<g->number_of_cells; ++i)
{
fprintf(stderr, "%d: ", i);
for (k=g->cell_facepos[i]; k<g->cell_facepos[i+1]; ++k)
{
fprintf(stderr, "%d ", g->cell_faces[k]);
}
fprintf(stderr, "\n");
}
free_cart_grid(g);
return 0;
}
#endif