/[escript]/trunk/finley/src/Mesh_saveVTK.c
ViewVC logotype

Diff of /trunk/finley/src/Mesh_saveVTK.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

trunk/esys2/finley/src/finleyC/Mesh_saveVTK.c revision 147 by jgs, Fri Aug 12 01:45:47 2005 UTC temp_trunk_copy/finley/src/Mesh_saveVTK.c revision 1384 by phornby, Fri Jan 11 02:29:38 2008 UTC
# Line 1  Line 1 
1    
2  /* $Id$ */  /* $Id$ */
3    
4    /*******************************************************
5     *
6     *           Copyright 2003-2007 by ACceSS MNRF
7     *       Copyright 2007 by University of Queensland
8     *
9     *                http://esscc.uq.edu.au
10     *        Primary Business: Queensland, Australia
11     *  Licensed under the Open Software License version 3.0
12     *     http://www.opensource.org/licenses/osl-3.0.php
13     *
14     *******************************************************/
15    
16  /**************************************************************/  /**************************************************************/
17    
18  /*   writes data and mesh in a vtk file */  /*   writes data and mesh in a vtk file */
19    /*   nodal data needs to be given on FINLEY_NODES or FINLEY_REDUCED_NODES */
20    
21  /**************************************************************/  /**************************************************************/
22    
 /*   Copyrights by ACcESS, Australia 2004 */  
 /*   Author: Paul Cochrane, cochrane@esscc.uq.edu.au */  
23    
 /**************************************************************/  
   
 #include "Finley.h"  
 #include "Common.h"  
24  #include "Mesh.h"  #include "Mesh.h"
25  #include "escript/Data/DataC.h"  #include "Assemble.h"
26  #include "vtkCellType.h"  /* copied from vtk source directory !!! */  #include "vtkCellType.h"  /* copied from vtk source directory !!! */
27    
28  void Finley_Mesh_saveVTK(const char * filename_p, Finley_Mesh *mesh_p, escriptDataC* data_p) {  #define LEN_PRINTED_INT_FORMAT (9+1)
29    /* if there is no mesh we just return */  #define INT_FORMAT "%d "
30    if (mesh_p==NULL) return;  #define INT_NEWLINE_FORMAT "%d\n"
31    Finley_ElementFile* elements=NULL;  #define FLOAT_SCALAR_FORMAT "%12.6e\n"
32    #define FLOAT_VECTOR_FORMAT "%12.6e %12.6e %12.6e\n"
33    #define FLOAT_TENSOR_FORMAT "%12.6e %12.6e %12.6e %12.6e %12.6e %12.6e %12.6e %12.6e %12.6e\n"
34    #define LEN_PRINTED_FLOAT_SCALAR_FORMAT (12+1)
35    #define LEN_PRINTED_FLOAT_VECTOR_FORMAT (3*(12+1)+1)
36    #define LEN_PRINTED_FLOAT_TENSOR_FORMAT (9*(12+1)+1)
37    #define NEWLINE "\n"
38    #define LEN_TMP_BUFFER LEN_PRINTED_FLOAT_TENSOR_FORMAT+(MAX_numNodes*LEN_PRINTED_INT_FORMAT+1)+2
39    #define NCOMP_MAX 9
40    #define __STRCAT(dest,chunk,dest_in_use)  \
41    {                  \
42      strcpy(&dest[dest_in_use], chunk); \
43      dest_in_use+=strlen(chunk); \
44    }
45    
46    void Finley_Mesh_saveVTK(const char * filename_p,
47                             Finley_Mesh *mesh_p,
48                             const dim_t num_data,
49                             char* *names_p,
50                             escriptDataC* *data_pp)
51    {
52      char error_msg[LenErrorMsg_MAX], *txt_buffer=NULL, tmp_buffer[LEN_TMP_BUFFER];
53      double sampleAvg[NCOMP_MAX], *values, rtmp;
54      size_t len_txt_buffer, max_len_names, txt_buffer_in_use;
55      FILE * fileHandle_p = NULL;
56      int mpi_size, i_data, i,j , cellType;
57      dim_t nDim, globalNumPoints, numCells, globalNumCells, numVTKNodesPerElement, myNumPoints, numPointsPerSample, rank, nComp, nCompReqd, shape, NN, numCellFactor, myNumCells, max_name_len;
58      bool_t do_write, *isCellCentered=NULL,write_celldata=FALSE,write_pointdata=FALSE;
59      bool_t set_scalar=FALSE,set_vector=FALSE, set_tensor=FALSE;
60      index_t myFirstNode, myLastNode, *globalNodeIndex, k, *node_index, myFirstCell;
61      #ifdef PASO_MPI
62      int ierr;
63      int amode = MPI_MODE_CREATE | MPI_MODE_WRONLY |  MPI_MODE_SEQUENTIAL;
64      MPI_File mpi_fileHandle_p;
65      MPI_Status mpi_status;
66      MPI_Request mpi_req;
67      MPI_Info mpi_info=MPI_INFO_NULL;
68      #endif
69      Paso_MPI_rank my_mpi_rank;
70      int nodetype=FINLEY_NODES;
71      int elementtype=FINLEY_UNKNOWN;
72    char elemTypeStr[32];    char elemTypeStr[32];
73    int i, j, k, numVTKNodesPerElement, isCellCentered=FALSE, nodetype=FINLEY_DEGREES_OF_FREEDOM;    Finley_NodeMapping *nodeMapping=NULL;
74    double* values, rtmp;    Finley_ElementFile* elements=NULL;
75    int nDim = mesh_p->Nodes->numDim;    ElementTypeId TypeId;
76      
77    /* get a pointer to the relevant mesh component */  
78    if (isEmpty(data_p)) {    /****************************************/
79      elements=mesh_p->Elements;    /*                                      */
80    } else {    /*       tags in the vtk file           */
81      switch(getFunctionSpaceType(data_p)) {  
82      case(FINLEY_DEGREES_OF_FREEDOM):    char* tags_header="<?xml version=\"1.0\"?>\n" \
83        nodetype = FINLEY_DEGREES_OF_FREEDOM;                      "<VTKFile type=\"UnstructuredGrid\" version=\"0.1\">\n" \
84        isCellCentered = FALSE;                      "<UnstructuredGrid>\n" \
85        elements = mesh_p->Elements;                      "<Piece NumberOfPoints=\"%d\" NumberOfCells=\"%d\">\n" \
86        break;                      "<Points>\n" \
87      case(FINLEY_REDUCED_DEGREES_OF_FREEDOM):                      "<DataArray NumberOfComponents=\"%d\" type=\"Float64\" format=\"ascii\">\n";
88        Finley_ErrorCode=VALUE_ERROR;    char *tag_End_DataArray = "</DataArray>\n";
89        sprintf(Finley_ErrorMsg,    char* tag_End_PointData =  "</PointData>\n";
90            "Reduced degrees of freedom is not yet "    char* tag_End_CellData =  "</CellData>\n";
91            "implemented for saving vtk files\n");    char *footer = "</Piece>\n</UnstructuredGrid>\n</VTKFile>\n";
92        return;    char* tags_End_Points_and_Start_Conn = "</DataArray>\n</Points>\n<Cells>\n<DataArray Name=\"connectivity\" type=\"Int32\" format=\"ascii\">\n" ;
93      case(FINLEY_NODES):    char* tags_End_Conn_and_Start_Offset = "</DataArray>\n<DataArray Name=\"offsets\" type=\"Int32\" format=\"ascii\">\n";
94        nodetype=FINLEY_NODES;    char* tags_End_Offset_and_Start_Type = "</DataArray>\n<DataArray Name=\"types\" type=\"UInt8\" format=\"ascii\">\n";
95        isCellCentered=FALSE;    char* tag_Float_DataArray="<DataArray Name=\"%s\" type=\"Float64\" NumberOfComponents=\"%d\" format=\"ascii\">\n";
96        elements=mesh_p->Elements;    char *tags_End_Type_And_Cells = "</DataArray>\n</Cells>\n";
       break;  
     case(FINLEY_ELEMENTS):  
       isCellCentered=TRUE;  
       elements=mesh_p->Elements;  
       break;  
     case(FINLEY_FACE_ELEMENTS):  
       isCellCentered=TRUE;  
       elements=mesh_p->FaceElements;  
       break;  
     case(FINLEY_POINTS):  
       isCellCentered=TRUE;  
       elements=mesh_p->Points;  
       break;  
     case(FINLEY_CONTACT_ELEMENTS_1):  
     case(FINLEY_CONTACT_ELEMENTS_2):  
       isCellCentered=TRUE;  
       elements=mesh_p->ContactElements;  
       break;  
     default:  
       Finley_ErrorCode=TYPE_ERROR;  
       sprintf(Finley_ErrorMsg,  
           "Finley does not know anything about function space type %d",  
           getFunctionSpaceType(data_p));  
       return;  
     }  
   }  
97    
98    /* the number of points */    int VTK_QUADRATIC_HEXAHEDRON_INDEX[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 16, 17, 18, 19, 12, 13, 14, 15 };
99    int numPoints = mesh_p->Nodes->numNodes;    /* if there is no mesh we just return */
100      if (mesh_p==NULL) return;
101    
102    /* the number of cells */    my_mpi_rank = mesh_p->Nodes->MPIInfo->rank;
103    if (elements == NULL) {    mpi_size  = mesh_p->Nodes->MPIInfo->size;
104      Finley_ErrorCode = VALUE_ERROR;    nDim = mesh_p->Nodes->numDim;
105      sprintf(Finley_ErrorMsg,  
106          "elements object is NULL; cannot proceed");    if (! ( (nDim ==2) || (nDim == 3) ) ) {
107      return;          Finley_setError(IO_ERROR, "saveVTK: spatial dimension 2 or 3 is supported only.");
108            return;  
109    }    }
110    int numCells = elements->numElements;      /*************************************************************************************/
111      
112    /* open the file and check handle */    /* open the file and check handle */
113    FILE * fileHandle_p = fopen(filename_p, "w");  
114    if (fileHandle_p==NULL) {    if (mpi_size > 1) {
115      Finley_ErrorCode=IO_ERROR;          #ifdef PASO_MPI
116      sprintf(Finley_ErrorMsg,            /* Collective Call */
117          "File %s could not be opened for writing.", filename_p);            #ifdef MPIO_HINTS
118      return;              MPI_Info_create(&mpi_info);
119    }              /*  MPI_Info_set(mpi_info, "striping_unit",        "424288"); */
120    /* xml header */              /*  MPI_Info_set(mpi_info, "striping_factor",      "16"); */
121    fprintf(fileHandle_p, "<?xml version=\"1.0\"?>\n");              /*  MPI_Info_set(mpi_info, "collective_buffering", "true"); */
122    fprintf(fileHandle_p,              /*  MPI_Info_set(mpi_info, "cb_block_size",        "131072"); */
123        "<VTKFile type=\"UnstructuredGrid\" version=\"0.1\">\n");              /*  MPI_Info_set(mpi_info, "cb_buffer_size",       "1048567"); */
124                /*  MPI_Info_set(mpi_info, "cb_nodes",             "8"); */
125    /* finley uses an unstructured mesh, so UnstructuredGrid *should* work */              /*    MPI_Info_set(mpi_info, "access_style", "write_once, sequential"); */
126    fprintf(fileHandle_p, "<UnstructuredGrid>\n");            
127                /*XFS only */
128    /* is there only one "piece" to the data?? */              /*   MPI_Info_set(mpi_info, "direct_write",          "true"); */
129    fprintf(fileHandle_p, "<Piece "            #endif
130        "NumberOfPoints=\"%d\" "            ierr=MPI_File_open(mesh_p->Nodes->MPIInfo->comm, (char*)filename_p, amode,mpi_info, &mpi_fileHandle_p);
131        "NumberOfCells=\"%d\">\n",            if (! ierr) {
132        numPoints, numCells);            perror(filename_p);
133                  sprintf(error_msg, "saveVTK: File %s could not be opened for writing in parallel.", filename_p);
134    /* now for the points; equivalent to positions section in saveDX() */                Finley_setError(IO_ERROR,error_msg);
135    /* "The points element explicitly defines coordinates for each point            } else {
136     * individually.  It contains one DataArray element describing an array               MPI_File_set_view(mpi_fileHandle_p,MPI_DISPLACEMENT_CURRENT,MPI_CHAR, MPI_CHAR, "native" , mpi_info);
137     * with three components per value, each specifying the coordinates of one            }
138     * point" - from Vtk User's Guide          #endif
    */  
   fprintf(fileHandle_p, "<Points>\n");  
   /*  
    * the reason for this if statement is explained in the long comment below  
    */  
   if (nDim < 3) {  
     fprintf(fileHandle_p, "<DataArray "  
         "NumberOfComponents=\"3\" "  
         "type=\"Float32\" "  
         "format=\"ascii\">\n");  
139    } else {    } else {
140      fprintf(fileHandle_p, "<DataArray "          fileHandle_p = fopen(filename_p, "w");
141          "NumberOfComponents=\"%d\" "          if (fileHandle_p==NULL) {
142          "type=\"Float32\" "             sprintf(error_msg, "saveVTK: File %s could not be opened for writing.", filename_p);
143          "format=\"ascii\">\n",             Finley_setError(IO_ERROR,error_msg);
144          nDim);           }
145    }    }
146    for (i = 0; i < mesh_p->Nodes->numNodes; i++) {    if (! Paso_MPIInfo_noError(mesh_p->Nodes->MPIInfo) ) return;
147      fprintf(fileHandle_p,    /*************************************************************************************/
148          "%e", mesh_p->Nodes->Coordinates[INDEX2(0, i, nDim)]);  
149      for (j = 1; j < nDim; j++) {    /* find the mesh type to be written */
       fprintf(fileHandle_p,  
           " %f",mesh_p->Nodes->Coordinates[INDEX2(j, i, nDim)]);  
       /* vtk/mayavi doesn't like 2D data, it likes 3D data with a degenerate  
        * third dimension to handle 2D data (like a sheet of paper).  So, if  
        * nDim is 2, we have to append zeros to the array to get this third  
        * dimension, and keep the visualisers happy.  
        * Indeed, if nDim is less than 3, must pad all empty dimensions, so  
        * that the total number of dims is 3.  
        */  
       if (nDim < 3) {  
     for (k=0; k<3-nDim; k++) {  
       fprintf(fileHandle_p, " 0");  
     }  
       }  
     }  
     fprintf(fileHandle_p, "\n");  
   }  
   fprintf(fileHandle_p, "</DataArray>\n");  
   fprintf(fileHandle_p, "</Points>\n");  
   
   /* connections */  
   /* now for the cells */  
   /* "The Cells element defines cells explicitly by specifying point  
    * connectivity and cell types.  It contains three DataArray elements.  The  
    * first array specifies the point connectivity.  All cells' point lists  
    * are concatenated together.  The second array specifies th eoffset into  
    * the connectivity array for the end of each cell.  The third array  
    * specifies the type of each cell.  
    */  
   /* if no element table is present jump over the connection table */  
   int cellType;  
   if (elements!=NULL) {  
     fprintf(fileHandle_p, "<Cells>\n");  
     ElementTypeId TypeId = elements->ReferenceElement->Type->TypeId;  
     switch(TypeId) {  
     case Point1:  
       cellType = VTK_VERTEX;  
       break;  
     case Line2:  
       cellType = VTK_LINE;  
       break;  
     case Line3:  
       cellType = VTK_QUADRATIC_EDGE;  
       break;  
     case Tri3:  
       cellType = VTK_TRIANGLE;  
       break;  
     case Tri6:  
       cellType = VTK_QUADRATIC_TRIANGLE;  
       break;  
     case Rec4:  
       cellType = VTK_QUAD;  
       break;  
     case Rec8:  
       cellType = VTK_QUADRATIC_QUAD;  
       break;  
     case Tet4:  
       cellType = VTK_TETRA;  
       break;  
     case Tet10:  
       cellType = VTK_QUADRATIC_TETRA;  
       break;  
     case Hex8:  
       cellType = VTK_HEXAHEDRON;  
       break;  
     case Hex20:  
       cellType = VTK_QUADRATIC_HEXAHEDRON;  
       break;  
     case Line2Face:  
       cellType = VTK_VERTEX;  
       break;  
     case Line3Face:  
       cellType = VTK_VERTEX;  
       break;  
     case Tri3Face:  
       cellType = VTK_LINE;  
       break;  
     case Tri6Face:  
       cellType = VTK_QUADRATIC_EDGE;  
       break;  
     case Rec4Face:  
       cellType = VTK_LINE;  
       break;  
     case Rec8Face:  
       cellType = VTK_QUADRATIC_EDGE;  
       break;  
     case Tet4Face:  
       cellType = VTK_TRIANGLE;  
       break;  
     case Tet10Face:  
       cellType = VTK_QUADRATIC_TRIANGLE;  
       break;  
     case Hex8Face:  
       cellType = VTK_QUADRATIC_QUAD;  
       break;  
     case Hex20Face:  
       cellType = VTK_QUADRATIC_QUAD;  
       break;  
     case Point1_Contact:  
       cellType = VTK_VERTEX;  
       break;  
     case Line2_Contact:  
       cellType = VTK_LINE;  
       break;  
     case Line3_Contact:  
       cellType = VTK_QUADRATIC_EDGE;  
       break;  
     case Tri3_Contact:  
       cellType = VTK_TRIANGLE;  
       break;  
     case Tri6_Contact:  
       cellType = VTK_QUADRATIC_TRIANGLE;  
       break;  
     case Rec4_Contact:  
       cellType = VTK_QUAD;  
       break;  
     case Rec8_Contact:  
       cellType = VTK_QUADRATIC_QUAD;  
       break;  
     case Line2Face_Contact:  
       cellType = VTK_VERTEX;  
       break;  
     case Line3Face_Contact:  
       cellType = VTK_VERTEX;  
       break;  
     case Tri3Face_Contact:  
       cellType = VTK_LINE;  
       break;  
     case Tri6Face_Contact:  
       cellType = VTK_QUADRATIC_EDGE;  
       break;  
     case Rec4Face_Contact:  
       cellType = VTK_LINE;  
       break;  
     case Rec8Face_Contact:  
       cellType = VTK_QUADRATIC_EDGE;  
       break;  
     case Tet4Face_Contact:  
       cellType = VTK_TRIANGLE;  
       break;  
     case Tet10Face_Contact:  
       cellType = VTK_QUADRATIC_TRIANGLE;  
       break;  
     case Hex8Face_Contact:  
       cellType = VTK_QUAD;  
       break;  
     case Hex20Face_Contact:  
       cellType = VTK_QUADRATIC_QUAD;  
       break;  
     default:  
       Finley_ErrorCode=VALUE_ERROR;  
       sprintf(Finley_ErrorMsg,  
           "Element type %s is not supported by VTK",  
           elements->ReferenceElement->Type->Name);  
       return;  
     }  
   
     switch(cellType) {  
     case VTK_VERTEX:  
       numVTKNodesPerElement = 1;  
       strcpy(elemTypeStr, "VTK_VERTEX");  
       break;  
     case VTK_LINE:  
       numVTKNodesPerElement = 2;  
       strcpy(elemTypeStr, "VTK_LINE");  
       break;  
     case VTK_TRIANGLE:  
       numVTKNodesPerElement = 3;  
       strcpy(elemTypeStr, "VTK_TRIANGLE");  
       break;  
     case VTK_QUAD:  
       numVTKNodesPerElement = 4;  
       strcpy(elemTypeStr, "VTK_QUAD");  
       break;  
     case VTK_TETRA:  
       numVTKNodesPerElement = 4;  
       strcpy(elemTypeStr, "VTK_TETRA");  
       break;  
     case VTK_HEXAHEDRON:  
       numVTKNodesPerElement = 8;  
       strcpy(elemTypeStr, "VTK_HEXAHEDRON");  
       break;  
     case VTK_QUADRATIC_EDGE:  
       numVTKNodesPerElement = 3;  
       strcpy(elemTypeStr, "VTK_QUADRATIC_EDGE");  
       break;  
     case VTK_QUADRATIC_TRIANGLE:  
       numVTKNodesPerElement = 6;  
       strcpy(elemTypeStr, "VTK_QUADRATIC_TRIANGLE");  
       break;  
     case VTK_QUADRATIC_QUAD:  
       numVTKNodesPerElement = 8;  
       strcpy(elemTypeStr, "VTK_QUADRATIC_QUAD");  
       break;  
     case VTK_QUADRATIC_TETRA:  
       numVTKNodesPerElement = 10;  
       strcpy(elemTypeStr, "VTK_QUADRATIC_TETRA");  
       break;  
     case VTK_QUADRATIC_HEXAHEDRON:  
       numVTKNodesPerElement = 20;  
       strcpy(elemTypeStr, "VTK_QUADRATIC_HEXAHEDRON");  
       break;  
     default:  
       Finley_ErrorCode = VALUE_ERROR;  
       sprintf(Finley_ErrorMsg,  
           "Cell type %d is not supported by VTK", cellType);  
       return;  
     }  
150    
151      /* write out the DataArray element for the connectivity */    isCellCentered=TMPMEMALLOC(num_data,bool_t);
152      fprintf(fileHandle_p, "<DataArray "    max_len_names=0;
153          "Name=\"connectivity\" "    if (!Finley_checkPtr(isCellCentered)) {
154          "type=\"Int32\" "       nodetype=FINLEY_UNKNOWN;
155          "format=\"ascii\">\n");       elementtype=FINLEY_UNKNOWN;
156      int NN = elements->ReferenceElement->Type->numNodes;       for (i_data=0;i_data<num_data;++i_data) {
157      if (VTK_QUADRATIC_HEXAHEDRON==cellType) {         if (! isEmpty(data_pp[i_data])) {
158         for (i = 0; i < numCells; i++) {           switch(getFunctionSpaceType(data_pp[i_data]) ) {
159           fprintf(fileHandle_p,"%d %d %d %d %d %d %d %d %d %d %d %d %d %d %d %d %d %d %d %d",           case FINLEY_NODES:
160                              mesh_p->Elements->Nodes[INDEX2(0, i, NN)],             nodetype = (nodetype == FINLEY_REDUCED_NODES) ? FINLEY_REDUCED_NODES : FINLEY_NODES;
161                              mesh_p->Elements->Nodes[INDEX2(1, i, NN)],             isCellCentered[i_data]=FALSE;
162                              mesh_p->Elements->Nodes[INDEX2(2, i, NN)],             if (elementtype==FINLEY_UNKNOWN || elementtype==FINLEY_ELEMENTS) {
163                              mesh_p->Elements->Nodes[INDEX2(3, i, NN)],               elementtype=FINLEY_ELEMENTS;
164                              mesh_p->Elements->Nodes[INDEX2(4, i, NN)],             } else {
165                              mesh_p->Elements->Nodes[INDEX2(5, i, NN)],               Finley_setError(TYPE_ERROR,"saveVTK: cannot write given data in single file.");
166                              mesh_p->Elements->Nodes[INDEX2(6, i, NN)],             }
167                              mesh_p->Elements->Nodes[INDEX2(7, i, NN)],             break;
168                              mesh_p->Elements->Nodes[INDEX2(8, i, NN)],           case FINLEY_REDUCED_NODES:
169                              mesh_p->Elements->Nodes[INDEX2(9, i, NN)],             nodetype = FINLEY_REDUCED_NODES;
170                              mesh_p->Elements->Nodes[INDEX2(10, i, NN)],             isCellCentered[i_data]=FALSE;
171                              mesh_p->Elements->Nodes[INDEX2(11, i, NN)],             if (elementtype==FINLEY_UNKNOWN || elementtype==FINLEY_ELEMENTS) {
172                              mesh_p->Elements->Nodes[INDEX2(16, i, NN)],               elementtype=FINLEY_ELEMENTS;
173                              mesh_p->Elements->Nodes[INDEX2(17, i, NN)],             } else {
174                              mesh_p->Elements->Nodes[INDEX2(18, i, NN)],               Finley_setError(TYPE_ERROR,"saveVTK: cannot write given data in single file.");
175                              mesh_p->Elements->Nodes[INDEX2(19, i, NN)],             }
176                              mesh_p->Elements->Nodes[INDEX2(12, i, NN)],             break;
177                              mesh_p->Elements->Nodes[INDEX2(13, i, NN)],           case FINLEY_ELEMENTS:
178                              mesh_p->Elements->Nodes[INDEX2(14, i, NN)],           case FINLEY_REDUCED_ELEMENTS:
179                              mesh_p->Elements->Nodes[INDEX2(15, i, NN)]);             isCellCentered[i_data]=TRUE;
180           fprintf(fileHandle_p, "\n");             if (elementtype==FINLEY_UNKNOWN || elementtype==FINLEY_ELEMENTS) {
181                 elementtype=FINLEY_ELEMENTS;
182               } else {
183                 Finley_setError(TYPE_ERROR,"saveVTK: cannot write given data in single file.");
184               }
185               break;
186             case FINLEY_FACE_ELEMENTS:
187             case FINLEY_REDUCED_FACE_ELEMENTS:
188               isCellCentered[i_data]=TRUE;
189               if (elementtype==FINLEY_UNKNOWN || elementtype==FINLEY_FACE_ELEMENTS) {
190                 elementtype=FINLEY_FACE_ELEMENTS;
191               } else {
192                 Finley_setError(TYPE_ERROR,"saveVTK: cannot write given data in single file.");
193               }
194               break;
195             case FINLEY_POINTS:
196               isCellCentered[i_data]=TRUE;
197               if (elementtype==FINLEY_UNKNOWN || elementtype==FINLEY_POINTS) {
198                 elementtype=FINLEY_POINTS;
199               } else {
200                 Finley_setError(TYPE_ERROR,"saveVTK: cannot write given data in single file.");
201               }
202               break;
203             case FINLEY_CONTACT_ELEMENTS_1:
204             case FINLEY_REDUCED_CONTACT_ELEMENTS_1:
205               isCellCentered[i_data]=TRUE;
206               if (elementtype==FINLEY_UNKNOWN || elementtype==FINLEY_CONTACT_ELEMENTS_1) {
207                 elementtype=FINLEY_CONTACT_ELEMENTS_1;
208               } else {
209                 Finley_setError(TYPE_ERROR,"saveVTK: cannot write given data in single file.");
210               }
211               break;
212             case FINLEY_CONTACT_ELEMENTS_2:
213             case FINLEY_REDUCED_CONTACT_ELEMENTS_2:
214               isCellCentered[i_data]=TRUE;
215               if (elementtype==FINLEY_UNKNOWN || elementtype==FINLEY_CONTACT_ELEMENTS_1) {
216                 elementtype=FINLEY_CONTACT_ELEMENTS_1;
217               } else {
218                 Finley_setError(TYPE_ERROR,"saveVTK: cannot write given data in single file.");
219               }
220               break;
221             default:
222               sprintf(error_msg,"saveVTK: unknown function space type %d",getFunctionSpaceType(data_pp[i_data]));
223               Finley_setError(TYPE_ERROR,error_msg);
224             }
225             if (isCellCentered[i_data]) {
226               write_celldata=TRUE;
227             } else {
228               write_pointdata=TRUE;
229             }
230             max_len_names =MAX(max_len_names,strlen(names_p[i_data]));
231         }         }
232      } else {       }
233         for (i = 0; i < numCells; i++) {       nodetype = (nodetype == FINLEY_UNKNOWN) ? FINLEY_NODES : nodetype;
234           for (j = 0; j < numVTKNodesPerElement; j++) fprintf(fileHandle_p,"%d ", mesh_p->Elements->Nodes[INDEX2(j, i, NN)]);    }
235           fprintf(fileHandle_p, "\n");    if (Finley_noError()) {
236    
237         /***************************************/
238    
239         /* select number of points and the mesh component */
240      
241         if (nodetype == FINLEY_REDUCED_NODES) {
242            myFirstNode = Finley_NodeFile_getFirstReducedNode(mesh_p->Nodes);
243            myLastNode = Finley_NodeFile_getLastReducedNode(mesh_p->Nodes);
244            globalNumPoints = Finley_NodeFile_getGlobalNumReducedNodes(mesh_p->Nodes);
245            globalNodeIndex= Finley_NodeFile_borrowGlobalReducedNodesIndex(mesh_p->Nodes);
246         } else {
247            myFirstNode = Finley_NodeFile_getFirstNode(mesh_p->Nodes);
248            myLastNode = Finley_NodeFile_getLastNode(mesh_p->Nodes);
249            globalNumPoints = Finley_NodeFile_getGlobalNumNodes(mesh_p->Nodes);
250            globalNodeIndex= Finley_NodeFile_borrowGlobalNodesIndex(mesh_p->Nodes);
251         }
252         myNumPoints = myLastNode - myFirstNode;
253         if (elementtype==FINLEY_UNKNOWN) elementtype=FINLEY_ELEMENTS;
254         switch(elementtype) {
255           case FINLEY_ELEMENTS:
256              elements=mesh_p->Elements;
257              break;
258            case FINLEY_FACE_ELEMENTS:
259              elements=mesh_p->FaceElements;
260              break;
261            case FINLEY_POINTS:
262              elements=mesh_p->Points;
263              break;
264            case FINLEY_CONTACT_ELEMENTS_1:
265              elements=mesh_p->ContactElements;
266              break;
267         }
268         if (elements==NULL) {
269           Finley_setError(SYSTEM_ERROR,"saveVTK: undefined element file");
270         } else {
271           /* map finley element type to VTK element type */
272           numCells = elements->numElements;
273           globalNumCells = Finley_ElementFile_getGlobalNumElements(elements);
274           myNumCells= Finley_ElementFile_getMyNumElements(elements);
275           myFirstCell= Finley_ElementFile_getFirstElement(elements);
276           NN = elements->numNodes;
277           if (nodetype==FINLEY_REDUCED_NODES || nodetype==FINLEY_REDUCED_NODES) {
278              TypeId = elements->LinearReferenceElement->Type->TypeId;
279           } else {
280              TypeId = elements->ReferenceElement->Type->TypeId;
281         }         }
282      }         switch(TypeId) {
283      fprintf(fileHandle_p, "\n");          case Point1:
284      fprintf(fileHandle_p, "</DataArray>\n");          case Line2Face:
285            case Line3Face:
286      /* write out the DataArray element for the offsets */          case Point1_Contact:
287      fprintf(fileHandle_p, "<DataArray "          case Line2Face_Contact:
288          "Name=\"offsets\" "          case Line3Face_Contact:
289          "type=\"Int32\" "            numCellFactor=1;
290          "format=\"ascii\">\n");            cellType = VTK_VERTEX;
291      int counter = 0;  /* counter for the number of offsets written to file */            numVTKNodesPerElement = 1;
292      for (i=numVTKNodesPerElement; i<=numCells*numVTKNodesPerElement; i+=numVTKNodesPerElement) {            strcpy(elemTypeStr, "VTK_VERTEX");
293        fprintf(fileHandle_p, "%d ", i);            break;
294        counter++;        
295        /* if the counter gets too big (i.e. the line gets too long),          case Line2:
296         * then add a newline and reset */          case Tri3Face:
297        if (counter > 19) {          case Rec4Face:
298        counter = 0;          case Line2_Contact:
299        fprintf(fileHandle_p, "\n");          case Tri3_Contact:
300            case Tri3Face_Contact:
301            case Rec4Face_Contact:
302              numCellFactor=1;
303              cellType = VTK_LINE;
304              numVTKNodesPerElement = 2;
305              strcpy(elemTypeStr, "VTK_LINE");
306              break;
307          
308            case Tri3:
309            case Tet4Face:
310            case Tet4Face_Contact:
311              numCellFactor=1;
312              cellType = VTK_TRIANGLE;
313              numVTKNodesPerElement = 3;
314              strcpy(elemTypeStr, "VTK_TRIANGLE");
315              break;
316          
317            case Rec4:
318            case Hex8Face:
319            case Rec4_Contact:
320            case Hex8Face_Contact:
321              numCellFactor=1;
322              cellType = VTK_QUAD;
323              numVTKNodesPerElement = 4;
324              strcpy(elemTypeStr, "VTK_QUAD");
325              break;
326          
327            case Tet4:
328              numCellFactor=1;
329              cellType = VTK_TETRA;
330              numVTKNodesPerElement = 4;
331              strcpy(elemTypeStr, "VTK_TETRA");
332              break;
333          
334            case Hex8:
335              numCellFactor=1;
336              cellType = VTK_HEXAHEDRON;
337              numVTKNodesPerElement = 8;
338              strcpy(elemTypeStr, "VTK_HEXAHEDRON");
339              break;
340          
341            case Line3:
342            case Tri6Face:
343            case Rec8Face:
344            case Line3_Contact:
345            case Tri6Face_Contact:
346            case Rec8Face_Contact:
347              numCellFactor=1;
348              cellType = VTK_QUADRATIC_EDGE;
349              numVTKNodesPerElement = 3;
350              strcpy(elemTypeStr, "VTK_QUADRATIC_EDGE");
351              break;
352          
353            case Tri6:
354            case Tet10Face:
355            case Tri6_Contact:
356            case Tet10Face_Contact:
357              numCellFactor=1;
358              cellType = VTK_QUADRATIC_TRIANGLE;
359              numVTKNodesPerElement = 6;
360              strcpy(elemTypeStr, "VTK_QUADRATIC_TRIANGLE");
361              break;
362          
363            case Rec8:
364            case Hex20Face:
365            case Rec8_Contact:
366            case Hex20Face_Contact:
367              numCellFactor=1;
368              cellType = VTK_QUADRATIC_QUAD;
369              numVTKNodesPerElement = 8;
370              strcpy(elemTypeStr, "VTK_QUADRATIC_QUAD");
371              break;
372          
373            case Tet10:
374              numCellFactor=1;
375              cellType = VTK_QUADRATIC_TETRA;
376              numVTKNodesPerElement = 10;
377              strcpy(elemTypeStr, "VTK_QUADRATIC_TETRA");
378              break;
379          
380            case Hex20:
381              numCellFactor=1;
382              cellType = VTK_QUADRATIC_HEXAHEDRON;
383              numVTKNodesPerElement = 20;
384              strcpy(elemTypeStr, "VTK_QUADRATIC_HEXAHEDRON");
385              break;
386          
387            default:
388              sprintf(error_msg, "saveVTK: Element type %s is not supported by VTK",elements->ReferenceElement->Type->Name);
389              Finley_setError(VALUE_ERROR,error_msg);
390            }
391         }
392      }
393      /***************************************/
394    
395      /***************************************/
396      /*                                     */
397      /*   allocate text buffer              */
398      /*                                     */
399      max_name_len=0;
400      for (i_data =0 ;i_data<num_data;++i_data) max_name_len=MAX(max_name_len,strlen(names_p[i_data]));
401      len_txt_buffer= strlen(tags_header) + 3 * LEN_PRINTED_INT_FORMAT + (30+3*max_name_len); /* header */
402      if (mpi_size > 1) len_txt_buffer=MAX(len_txt_buffer, myNumPoints * LEN_TMP_BUFFER);
403      if (mpi_size > 1) len_txt_buffer=MAX(len_txt_buffer, numCellFactor*myNumCells*(LEN_PRINTED_INT_FORMAT*numVTKNodesPerElement+1));
404      len_txt_buffer=MAX(len_txt_buffer,200+3*max_len_names);
405      len_txt_buffer=MAX(len_txt_buffer, strlen(tag_Float_DataArray) + LEN_PRINTED_INT_FORMAT + max_len_names);
406      if (mpi_size > 1) len_txt_buffer=MAX(len_txt_buffer, numCellFactor*myNumCells*LEN_PRINTED_FLOAT_TENSOR_FORMAT);
407      if (mpi_size > 1) len_txt_buffer=MAX(len_txt_buffer, myNumPoints*LEN_PRINTED_FLOAT_TENSOR_FORMAT);
408      txt_buffer=TMPMEMALLOC(len_txt_buffer+1,char);
409      Finley_checkPtr(txt_buffer);
410      
411      if (Finley_noError()) {
412    
413         /* select number of points and the mesh component */
414    
415         sprintf(txt_buffer,tags_header,globalNumPoints,numCellFactor*globalNumCells,3);
416    
417          if (mpi_size > 1) {
418              if ( my_mpi_rank == 0) {
419                #ifdef PASO_MPI
420                  MPI_File_iwrite_shared(mpi_fileHandle_p,txt_buffer,strlen(txt_buffer),MPI_CHAR,&mpi_req);
421                  MPI_Wait(&mpi_req,&mpi_status);
422                #endif
423              }
424          } else {
425             fprintf(fileHandle_p,txt_buffer);
426        }        }
     }  
     fprintf(fileHandle_p, "\n");  
     fprintf(fileHandle_p, "</DataArray>\n");  
427    
428      /* write out the DataArray element for the types */        /* write the nodes */
429      counter = 0; /* counter for the number of types written to file */        
430      fprintf(fileHandle_p, "<DataArray "        if (mpi_size > 1) {
431          "Name=\"types\" "  
432          "type=\"UInt8\" "           txt_buffer[0] = '\0';
433          "format=\"ascii\">\n");           txt_buffer_in_use=0;
434      for (i=0; i<numCells; i++) {           if (nDim==2) {
435        fprintf(fileHandle_p, "%d ", cellType);              for (i = 0; i < mesh_p->Nodes->numNodes; i++) {
436        counter++;                 if ( (myFirstNode <= globalNodeIndex[i]) && (globalNodeIndex[i] < myLastNode) ) {
437        /* if the counter gets too big (i.e. the line gets too long),                   sprintf(tmp_buffer,FLOAT_VECTOR_FORMAT,
438         * then add a newline and reset */                                      mesh_p->Nodes->Coordinates[INDEX2(0, i, nDim)],
439        if (counter > 30) {                                      mesh_p->Nodes->Coordinates[INDEX2(1, i, nDim)],
440        counter = 0;                                      0.);
441        fprintf(fileHandle_p, "\n");                   __STRCAT(txt_buffer,tmp_buffer,txt_buffer_in_use);
442                   }
443                }      
444             } else {
445                for (i = 0; i < mesh_p->Nodes->numNodes; i++) {
446                   if ( (myFirstNode <= globalNodeIndex[i]) && (globalNodeIndex[i] < myLastNode) ) {
447                     sprintf(tmp_buffer,FLOAT_VECTOR_FORMAT,
448                                                      mesh_p->Nodes->Coordinates[INDEX2(0, i, nDim)],
449                                                      mesh_p->Nodes->Coordinates[INDEX2(1, i, nDim)],
450                                                      mesh_p->Nodes->Coordinates[INDEX2(2, i, nDim)]);
451                     __STRCAT(txt_buffer,tmp_buffer,txt_buffer_in_use);
452                   }
453                }    
454      
455             }
456             #ifdef PASO_MPI
457                MPI_File_write_ordered(mpi_fileHandle_p, txt_buffer,txt_buffer_in_use, MPI_CHAR, &mpi_status);
458             #endif    
459          } else {
460             if (nDim==2) {
461                for (i = 0; i < mesh_p->Nodes->numNodes; i++) {
462                   if ( (myFirstNode <= globalNodeIndex[i]) && (globalNodeIndex[i] < myLastNode) ) {
463                     fprintf(fileHandle_p,FLOAT_VECTOR_FORMAT,
464                                          mesh_p->Nodes->Coordinates[INDEX2(0, i, nDim)],
465                                          mesh_p->Nodes->Coordinates[INDEX2(1, i, nDim)],
466                                          0.);
467                   }
468                }      
469             } else {
470                for (i = 0; i < mesh_p->Nodes->numNodes; i++) {
471                   if ( (myFirstNode <= globalNodeIndex[i]) && (globalNodeIndex[i] < myLastNode) ) {
472                     fprintf(fileHandle_p,FLOAT_VECTOR_FORMAT,
473                                                  mesh_p->Nodes->Coordinates[INDEX2(0, i, nDim)],
474                                                  mesh_p->Nodes->Coordinates[INDEX2(1, i, nDim)],
475                                                  mesh_p->Nodes->Coordinates[INDEX2(2, i, nDim)]);
476                   }
477                }    
478      
479             }
480        }        }
     }  
     fprintf(fileHandle_p, "\n");  
     fprintf(fileHandle_p, "</DataArray>\n");  
481    
482      /* finish off the <Cells> element */        /* close the Points and open connectivity */
     fprintf(fileHandle_p, "</Cells>\n");  
   }  
483    
484    /* data */        if (mpi_size > 1) {
485    if (!isEmpty(data_p)) {            if ( my_mpi_rank == 0) {
486      int rank = getDataPointRank(data_p);               #ifdef PASO_MPI
487      int nComp = getDataPointSize(data_p);                  MPI_File_iwrite_shared(mpi_fileHandle_p, tags_End_Points_and_Start_Conn, strlen(tags_End_Points_and_Start_Conn), MPI_CHAR, &mpi_req);
488      /* barf if rank is greater than two */                  MPI_Wait(&mpi_req,&mpi_status);
489      if (rank > 2) {               #endif
490        Finley_ErrorCode = VALUE_ERROR;            }
491        sprintf(Finley_ErrorMsg,        } else {
492            "Vtk can't handle objects with rank greater than 2\n"           fprintf(fileHandle_p,tags_End_Points_and_Start_Conn);
493            "object rank = %d\n", rank);        }
494        return;  
495      }       /* write the cells */
496      /* if the rank == 0:   --> scalar data       if (nodetype == FINLEY_REDUCED_NODES) {
497       * if the rank == 1:   --> vector data          node_index=elements->ReferenceElement->Type->linearNodes;
498       * if the rank == 2:   --> tensor data       } else if (VTK_QUADRATIC_HEXAHEDRON==cellType) {
499       */          node_index=VTK_QUADRATIC_HEXAHEDRON_INDEX;
500      char dataNameStr[31], dataTypeStr[63];       } else if (numVTKNodesPerElement!=NN) {
501      int nCompReqd=1;   /* the number of components required by vtk */          node_index=elements->ReferenceElement->Type->geoNodes;
502      if (rank == 0) {       } else {
503        strcpy(dataNameStr, "escript_scalar_data");          node_index=NULL;
504        sprintf(dataTypeStr, "Scalars=\"%s\"", dataNameStr);       }
505        nCompReqd = 1;  
506      }       if ( mpi_size > 1) {
507      else if (rank == 1) {          txt_buffer[0] = '\0';
508        strcpy(dataNameStr, "escript_vector_data");          txt_buffer_in_use=0;
509        sprintf(dataTypeStr, "Vectors=\"%s\"", dataNameStr);          if (node_index == NULL) {
510        nCompReqd = 3;             for (i = 0; i < numCells; i++) {
511                  if (elements->Owner[i] == my_mpi_rank) {
512                     for (j = 0; j < numVTKNodesPerElement; j++) {
513                         sprintf(tmp_buffer,INT_FORMAT,globalNodeIndex[elements->Nodes[INDEX2(j, i, NN)]]);
514                         __STRCAT(txt_buffer,tmp_buffer,txt_buffer_in_use)
515                     }
516                     __STRCAT(txt_buffer,NEWLINE,txt_buffer_in_use)
517                  }
518               }
519            } else {
520               for (i = 0; i < numCells; i++) {
521                  if (elements->Owner[i] == my_mpi_rank) {
522                     for (j = 0; j < numVTKNodesPerElement; j++) {
523                         sprintf(tmp_buffer,INT_FORMAT,globalNodeIndex[elements->Nodes[INDEX2(node_index[j], i, NN)]]);
524                         __STRCAT(txt_buffer,tmp_buffer,txt_buffer_in_use)
525                     }
526                     __STRCAT(txt_buffer,NEWLINE,txt_buffer_in_use)
527                  }
528               }
529            }
530            #ifdef PASO_MPI
531               MPI_File_write_ordered(mpi_fileHandle_p,txt_buffer,txt_buffer_in_use, MPI_CHAR, &mpi_status);
532            #endif    
533         } else {
534            if (node_index == NULL) {
535               for (i = 0; i < numCells; i++) {
536                  for (j = 0; j < numVTKNodesPerElement; j++) {
537                     fprintf(fileHandle_p,INT_FORMAT,globalNodeIndex[elements->Nodes[INDEX2(j, i, NN)]]);
538                   }
539                  fprintf(fileHandle_p,NEWLINE);
540               }
541            } else {
542               for (i = 0; i < numCells; i++) {
543                  for (j = 0; j < numVTKNodesPerElement; j++) {
544                     fprintf(fileHandle_p,INT_FORMAT,globalNodeIndex[elements->Nodes[INDEX2(node_index[j], i, NN)]]);
545                   }
546                  fprintf(fileHandle_p,NEWLINE);
547               }
548            }
549    
550         }
551        
552         /* finalize the connection and start the offset section */
553         if (mpi_size > 1) {
554            if( my_mpi_rank == 0) {
555               #ifdef PASO_MPI
556                  MPI_File_iwrite_shared(mpi_fileHandle_p,tags_End_Conn_and_Start_Offset,strlen(tags_End_Conn_and_Start_Offset),MPI_CHAR,&mpi_req);
557                  MPI_Wait(&mpi_req,&mpi_status);
558               #endif
559            }
560         } else {
561            fprintf(fileHandle_p,tags_End_Conn_and_Start_Offset);
562         }
563    
564        /* write the offsets */
565          
566         if ( mpi_size > 1) {
567            txt_buffer[0] = '\0';
568            txt_buffer_in_use=0;
569            for (i=numVTKNodesPerElement*(myFirstCell*numCellFactor+1); i<=numCells*numVTKNodesPerElement*numCellFactor; i+=numVTKNodesPerElement) {
570               sprintf(tmp_buffer, INT_NEWLINE_FORMAT, i);
571               __STRCAT(txt_buffer,tmp_buffer,txt_buffer_in_use);
572             }
573             #ifdef PASO_MPI
574                MPI_File_write_ordered(mpi_fileHandle_p,txt_buffer,txt_buffer_in_use, MPI_CHAR, &mpi_status);
575             #endif    
576         } else {
577            for (i=numVTKNodesPerElement; i<=numCells*numVTKNodesPerElement*numCellFactor; i+=numVTKNodesPerElement) {
578               fprintf(fileHandle_p, INT_NEWLINE_FORMAT, i);
579            }
580        
581         }
582         /* finalize the offset section and start the type section */
583         if ( mpi_size > 1) {
584            if ( my_mpi_rank == 0) {
585               #ifdef PASO_MPI
586                  MPI_File_iwrite_shared(mpi_fileHandle_p,tags_End_Offset_and_Start_Type,strlen(tags_End_Offset_and_Start_Type),MPI_CHAR,&mpi_req);
587                  MPI_Wait(&mpi_req,&mpi_status);
588               #endif
589            }
590        } else {
591           fprintf(fileHandle_p,tags_End_Offset_and_Start_Type);
592      }      }
593      else if (rank == 2) {      
594        strcpy(dataNameStr, "escript_tensor_data");        
595        sprintf(dataTypeStr, "Tensors=\"%s\"", dataNameStr);       /* write element type */
596        nCompReqd = 9;       sprintf(tmp_buffer, INT_NEWLINE_FORMAT, cellType);
597         if ( mpi_size > 1) {
598            txt_buffer[0] = '\0';
599            txt_buffer_in_use=0;
600            for (i=0; i<numCells*numCellFactor; i++) __STRCAT(txt_buffer,tmp_buffer,txt_buffer_in_use);
601             #ifdef PASO_MPI
602                MPI_File_write_ordered(mpi_fileHandle_p,txt_buffer,txt_buffer_in_use, MPI_CHAR, &mpi_status);
603             #endif    
604         } else {
605            for (i=0; i<numCells*numCellFactor; i++) fprintf(fileHandle_p, tmp_buffer);
606         }
607         /* finalize cell information */
608         if ( mpi_size > 1) {
609            if ( my_mpi_rank == 0) {
610               #ifdef PASO_MPI
611                  MPI_File_iwrite_shared(mpi_fileHandle_p,tags_End_Type_And_Cells,strlen(tags_End_Type_And_Cells),MPI_CHAR,&mpi_req);
612                  MPI_Wait(&mpi_req,&mpi_status);
613               #endif
614            }
615        } else {
616           fprintf(fileHandle_p,tags_End_Type_And_Cells);
617      }      }
618      /* if have cell centred data then use CellData element,   }
619       * if have node centred data, then use PointData element  
620       */   /* Write cell data */
621      if (isCellCentered) {   if (write_celldata && Finley_noError()) {
622        /* now for the cell data */        /* mark the active data arrays */
623        fprintf(fileHandle_p, "<CellData %s>\n", dataTypeStr);        txt_buffer[0] = '\0';
624        fprintf(fileHandle_p,        set_scalar=FALSE,set_vector=FALSE, set_tensor=FALSE;
625            "<DataArray "        strcat(txt_buffer, "<CellData");
626            "Name=\"%s\" "        for (i_data =0 ;i_data<num_data;++i_data) {
627            "type=\"Float32\" "          if (! isEmpty(data_pp[i_data]) && isCellCentered[i_data]) {
628            "NumberOfComponents=\"%d\" "            /* if the rank == 0:   --> scalar data */
629            "format=\"ascii\">\n",            /* if the rank == 1:   --> vector data */
630            dataNameStr, nCompReqd);            /* if the rank == 2:   --> tensor data */
631        int numPointsPerSample = elements->ReferenceElement->numQuadNodes;  
632        if (numPointsPerSample) {            switch(getDataPointRank(data_pp[i_data])) {
633      for (i=0; i<numCells; i++) {            case 0:
634        values = getSampleData(data_p, i);              if (! set_scalar) {
635        double sampleAvg[nComp];                strcat(txt_buffer," Scalars=\"");
636        for (k=0; k<nComp; k++) {                strcat(txt_buffer,names_p[i_data]);
637          /* averaging over the number of points in the sample */                strcat(txt_buffer,"\"");
638          rtmp = 0.;                set_scalar=TRUE;
639          for (j=0; j<numPointsPerSample; j++) {              }
640            rtmp += values[INDEX2(k,j,nComp)];              break;
641          }            case 1:
642          sampleAvg[k] = rtmp/numPointsPerSample;              if (! set_vector) {
643        }                strcat(txt_buffer," Vectors=\"");
644        /* if the number of required components is more than the number                strcat(txt_buffer,names_p[i_data]);
645         * of actual components, pad with zeros                strcat(txt_buffer,"\"");
646         */                set_vector=TRUE;
647        /* probably only need to get shape of first element */              }
648        int shape = getDataPointShape(data_p, 0);              break;
649        if (shape > 3) {            case 2:
650          Finley_ErrorCode = VALUE_ERROR;              if (! set_tensor) {
651          sprintf(Finley_ErrorMsg,                strcat(txt_buffer," Tensors=\"");
652              "shape should be 1, 2, or 3; I got %d\n", shape);                strcat(txt_buffer,names_p[i_data]);
653          return;                strcat(txt_buffer,"\"");
654        }                set_tensor=TRUE;
655        /* write the data different ways for scalar, vector and tensor */              }
656        if (nCompReqd == 1) {              break;
657          fprintf(fileHandle_p, " %f", sampleAvg[0]);            default:
658        }              sprintf(error_msg, "saveVTK: data %s: Vtk can't handle objects with rank greater than 2.",names_p[i_data]);
659        else if (nCompReqd == 3) {              Finley_setError(VALUE_ERROR,error_msg);
660          int shape = getDataPointShape(data_p, 0);              return;
661          /* write out the data */            }
662          for (int m=0; m<shape; m++) {          }
           fprintf(fileHandle_p, " %f", sampleAvg[m]);  
         }  
         /* pad with zeros */  
         for (int m=0; m<nCompReqd-shape; m++) {  
           fprintf(fileHandle_p, " 0");  
         }  
       }  
       else if (nCompReqd == 9) {  
         /* tensor data, so have a 3x3 matrix to output as a row  
          * of 9 data points */  
         int count = 0;  
         int elems = 0;  
         for (int m=0; m<shape; m++) {  
           for (int n=0; n<shape; n++) {  
         fprintf(fileHandle_p, " %f", sampleAvg[count]);  
         count++;  
         elems++;  
           }  
           for (int n=0; n<3-shape; n++) {  
         fprintf(fileHandle_p, " 0");  
         elems++;  
           }  
         }  
         for (int m=0; m<nCompReqd-elems; m++) {  
           fprintf(fileHandle_p, " 0");  
         }  
       }  
       fprintf(fileHandle_p, "\n");  
     }  
663        }        }
664        fprintf(fileHandle_p, "</DataArray>\n");        strcat(txt_buffer, ">\n");
665        fprintf(fileHandle_p, "</CellData>\n");        if ( mpi_size > 1) {
666      } else {          if ( my_mpi_rank == 0) {
667        /* now for the point data */             #ifdef PASO_MPI
668        fprintf(fileHandle_p, "<PointData %s>\n", dataTypeStr);                MPI_File_iwrite_shared(mpi_fileHandle_p,txt_buffer,strlen(txt_buffer),MPI_CHAR,&mpi_req);
669        fprintf(fileHandle_p, "<DataArray "                MPI_Wait(&mpi_req,&mpi_status);
670            "Name=\"%s\" "             #endif
671            "type=\"Float32\" "          }
672            "NumberOfComponents=\"%d\" "        } else {
673            "format=\"ascii\">\n",            fprintf(fileHandle_p,txt_buffer);
674            dataNameStr, nCompReqd);        }
675        for (i=0; i<mesh_p->Nodes->numNodes; i++) {        /* write the arrays */
676      switch (nodetype) {        for (i_data =0 ;i_data<num_data;++i_data) {
677      case(FINLEY_DEGREES_OF_FREEDOM):           if (! isEmpty(data_pp[i_data]) && isCellCentered[i_data]) {
678        values = getSampleData(data_p,              txt_buffer[0] = '\0';
679                   mesh_p->Nodes->degreeOfFreedom[i]);              txt_buffer_in_use=0;
680        break;              numPointsPerSample=getNumDataPointsPerSample(data_pp[i_data]);
681      case(FINLEY_REDUCED_DEGREES_OF_FREEDOM):              rank = getDataPointRank(data_pp[i_data]);
682        if (mesh_p->Nodes->toReduced[i]>=0) {              nComp = getDataPointSize(data_pp[i_data]);
683          values = getSampleData(data_p,              nCompReqd=1;   /* the number of components mpi_required by vtk */
684                     mesh_p->Nodes->reducedDegreeOfFreedom[i]);              shape=0;
685        }              if (rank == 0) {
686        break;                nCompReqd = 1;
687      case(FINLEY_NODES):              } else if (rank == 1) {
688        values = getSampleData(data_p,i);                shape=getDataPointShape(data_pp[i_data], 0);
689        break;                if  (shape>3) {
690      }                  Finley_setError(VALUE_ERROR, "saveVTK: rank 1 object must have less then 4 components");
691      /* write out the data */                }
692      /* if the number of required components is more than the number                nCompReqd = 3;
693       * of actual components, pad with zeros              } else {
694       */                shape=getDataPointShape(data_pp[i_data], 0);
695      /* probably only need to get shape of first element */                if  (shape>3 || shape != getDataPointShape(data_pp[i_data], 1)) {
696      int shape = getDataPointShape(data_p, 0);                  Finley_setError(VALUE_ERROR, "saveVTK: rank 2 object must have less then 4x4 components and must have a square shape");
697      if (shape > 3) {                }
698        Finley_ErrorCode = VALUE_ERROR;                nCompReqd = 9;
699        sprintf(Finley_ErrorMsg,              }
700            "shape should be 1, 2, or 3; I got %d\n", shape);              if (Finley_noError()) {
701        return;                 sprintf(txt_buffer,tag_Float_DataArray,names_p[i_data], nCompReqd);
702      }                 if ( mpi_size > 1) {
703      /* write the data different ways for scalar, vector and tensor */                   if ( my_mpi_rank == 0) {
704      if (nCompReqd == 1) {                      #ifdef PASO_MPI
705        fprintf(fileHandle_p, " %f", values[0]);                         MPI_File_iwrite_shared(mpi_fileHandle_p,txt_buffer,strlen(txt_buffer),MPI_CHAR,&mpi_req);
706      }                         MPI_Wait(&mpi_req,&mpi_status);
707      else if (nCompReqd == 3) {                      #endif
708        int shape = getDataPointShape(data_p, 0);                   }
709        /* write out the data */                 } else {
710        for (int m=0; m<shape; m++) {                     fprintf(fileHandle_p,txt_buffer);
711          fprintf(fileHandle_p, " %f", values[m]);                 }
712        }                 for (i=0; i<numCells; i++) {
713        /* pad with zeros */                     if (elements->Owner[i] == my_mpi_rank) {
714        for (int m=0; m<nCompReqd-shape; m++) {                        values = getSampleData(data_pp[i_data], i);
715          fprintf(fileHandle_p, " 0");                        /* averaging over the number of points in the sample */
716        }                        for (k=0; k<MIN(nComp,NCOMP_MAX); k++) {
717      }                           if (isExpanded(data_pp[i_data])) {
718      else if (nCompReqd == 9) {                             rtmp = 0.;
719        /* tensor data, so have a 3x3 matrix to output as a row                             for (j=0; j<numPointsPerSample; j++) rtmp += values[INDEX2(k,j,nComp)];
720         * of 9 data points */                             sampleAvg[k] = rtmp/numPointsPerSample;
721        int count = 0;                          } else {
722        int elems = 0;                             sampleAvg[k] = values[k];
723        for (int m=0; m<shape; m++) {                          }
724          for (int n=0; n<shape; n++) {                        }
725            fprintf(fileHandle_p, " %f", values[count]);                        /* if the number of mpi_required components is more than the number
726            count++;                        * of actual components, pad with zeros
727            elems++;                        */
728          }                        /* probably only need to get shape of first element */
729          for (int n=0; n<3-shape; n++) {                        /* write the data different ways for scalar, vector and tensor */
730            fprintf(fileHandle_p, " 0");                        if (nCompReqd == 1) {
731            elems++;                          sprintf(tmp_buffer,FLOAT_SCALAR_FORMAT,sampleAvg[0]);
732          }                        } else if (nCompReqd == 3) {
733        }                          if (shape==1) {
734        for (int m=0; m<nCompReqd-elems; m++) {                           sprintf(tmp_buffer,FLOAT_VECTOR_FORMAT,sampleAvg[0],0.,0.);
735          fprintf(fileHandle_p, " 0");                          } else if (shape==2) {
736        }                           sprintf(tmp_buffer,FLOAT_VECTOR_FORMAT,sampleAvg[0],sampleAvg[1],0.);
737      }                          } else if (shape==3) {
738      fprintf(fileHandle_p, "\n");                           sprintf(tmp_buffer,FLOAT_VECTOR_FORMAT,sampleAvg[0],sampleAvg[1],sampleAvg[2]);
739                            }
740                          } else if (nCompReqd == 9) {
741                            if (shape==1) {
742                             sprintf(tmp_buffer,FLOAT_TENSOR_FORMAT,sampleAvg[0],0.,0.,
743                                                                    0.,0.,0.,
744                                                                    0.,0.,0.);
745                            } else if (shape==2) {
746                             sprintf(tmp_buffer,FLOAT_TENSOR_FORMAT,sampleAvg[0],sampleAvg[1],0.,
747                                                                    sampleAvg[2],sampleAvg[3],0.,
748                                                                    0.,0.,0.);
749                            } else if (shape==3) {
750                             sprintf(tmp_buffer,FLOAT_TENSOR_FORMAT,sampleAvg[0],sampleAvg[1],sampleAvg[2],
751                                                                    sampleAvg[3],sampleAvg[4],sampleAvg[5],
752                                                                    sampleAvg[6],sampleAvg[7],sampleAvg[8]);
753                            }
754                          }
755                          if ( mpi_size > 1) {
756                            __STRCAT(txt_buffer,tmp_buffer,txt_buffer_in_use);
757                          } else {
758                            fprintf(fileHandle_p,tmp_buffer);
759                          }
760                      }
761                   }
762                   if ( mpi_size > 1) {
763                         #ifdef PASO_MPI
764                            MPI_File_write_ordered(mpi_fileHandle_p,txt_buffer,txt_buffer_in_use, MPI_CHAR, &mpi_status);
765                         #endif    
766                         if ( my_mpi_rank == 0) {
767                            #ifdef PASO_MPI
768                               MPI_File_iwrite_shared(mpi_fileHandle_p,tag_End_DataArray,strlen(tag_End_DataArray),MPI_CHAR,&mpi_req);
769                               MPI_Wait(&mpi_req,&mpi_status);
770                            #endif
771                         }
772                   } else {
773                       fprintf(fileHandle_p,tag_End_DataArray);
774                   }
775                }
776             }
777          }
778          if ( mpi_size > 1) {
779            if ( my_mpi_rank == 0) {
780               #ifdef PASO_MPI
781                  MPI_File_iwrite_shared(mpi_fileHandle_p,tag_End_CellData,strlen(tag_End_CellData),MPI_CHAR,&mpi_req);
782                  MPI_Wait(&mpi_req,&mpi_status);
783               #endif
784            }
785          } else {
786              fprintf(fileHandle_p,tag_End_CellData);
787        }        }
       fprintf(fileHandle_p, "</DataArray>\n");  
       fprintf(fileHandle_p, "</PointData>\n");  
     }  
788    }    }
789      /* point data */
790    /* finish off the piece */    if (write_pointdata && Finley_noError()) {
791    fprintf(fileHandle_p, "</Piece>\n");        /* mark the active data arrays */
792          set_scalar=FALSE,set_vector=FALSE, set_tensor=FALSE;
793    fprintf(fileHandle_p, "</UnstructuredGrid>\n");        txt_buffer[0] = '\0';
794    /* write the xml footer */        strcat(txt_buffer, "<PointData");
795    fprintf(fileHandle_p, "</VTKFile>\n");        for (i_data =0 ;i_data<num_data;++i_data) {
796    /* close the file */          if (! isEmpty(data_pp[i_data]) && !isCellCentered[i_data]) {
797    fclose(fileHandle_p);            /* if the rank == 0:   --> scalar data */
798              /* if the rank == 1:   --> vector data */
799              /* if the rank == 2:   --> tensor data */
800    
801              switch(getDataPointRank(data_pp[i_data])) {
802              case 0:
803                if (! set_scalar) {
804                  strcat(txt_buffer," Scalars=\"");
805                  strcat(txt_buffer,names_p[i_data]);
806                  strcat(txt_buffer,"\"");
807                  set_scalar=TRUE;
808                }
809                break;
810              case 1:
811                if (! set_vector) {
812                  strcat(txt_buffer," Vectors=\"");
813                  strcat(txt_buffer,names_p[i_data]);
814                  strcat(txt_buffer,"\"");
815                  set_vector=TRUE;
816                }
817                break;
818              case 2:
819                if (! set_tensor) {
820                  strcat(txt_buffer," Tensors=\"");
821                  strcat(txt_buffer,names_p[i_data]);
822                  strcat(txt_buffer,"\"");
823                  set_tensor=TRUE;
824                }
825                break;
826              default:
827                sprintf(error_msg, "saveVTK: data %s: Vtk can't handle objects with rank greater than 2.",names_p[i_data]);
828                Finley_setError(VALUE_ERROR,error_msg);
829                return;
830              }
831            }
832          }
833          strcat(txt_buffer, ">\n");
834          if ( mpi_size > 1) {
835            if ( my_mpi_rank == 0) {
836               #ifdef PASO_MPI
837                  MPI_File_iwrite_shared(mpi_fileHandle_p,txt_buffer,strlen(txt_buffer),MPI_CHAR,&mpi_req);
838                  MPI_Wait(&mpi_req,&mpi_status);
839               #endif
840            }
841          } else {
842              fprintf(fileHandle_p,txt_buffer);
843          }
844          /* write the arrays */
845          for (i_data =0 ;i_data<num_data;++i_data) {
846             if (! isEmpty(data_pp[i_data]) && !isCellCentered[i_data]) {
847                txt_buffer[0] = '\0';
848                txt_buffer_in_use=0;
849                numPointsPerSample=getNumDataPointsPerSample(data_pp[i_data]);
850                rank = getDataPointRank(data_pp[i_data]);
851                nComp = getDataPointSize(data_pp[i_data]);
852                if (getFunctionSpaceType(data_pp[i_data]) == FINLEY_REDUCED_NODES) {
853                   nodeMapping=mesh_p->Nodes->reducedNodesMapping;
854                } else {
855                   nodeMapping=mesh_p->Nodes->nodesMapping;
856                }
857                nCompReqd=1;   /* the number of components mpi_required by vtk */
858                shape=0;
859                if (rank == 0) {
860                  nCompReqd = 1;
861                } else if (rank == 1) {
862                  shape=getDataPointShape(data_pp[i_data], 0);
863                  if  (shape>3) {
864                    Finley_setError(VALUE_ERROR, "saveVTK: rank 1 object must have less then 4 components");
865                  }
866                  nCompReqd = 3;
867                } else {
868                  shape=getDataPointShape(data_pp[i_data], 0);
869                  if  (shape>3 || shape != getDataPointShape(data_pp[i_data], 1)) {
870                    Finley_setError(VALUE_ERROR, "saveVTK: rank 2 object must have less then 4x4 components and must have a square shape");
871                  }
872                  nCompReqd = 9;
873                }
874                if (Finley_noError()) {
875                   sprintf(txt_buffer,tag_Float_DataArray,names_p[i_data], nCompReqd);
876                   if ( mpi_size > 1) {
877                     if ( my_mpi_rank == 0) {
878                        #ifdef PASO_MPI
879                           MPI_File_iwrite_shared(mpi_fileHandle_p,txt_buffer,strlen(txt_buffer),MPI_CHAR,&mpi_req);
880                           MPI_Wait(&mpi_req,&mpi_status);
881                        #endif
882                     }
883                   } else {
884                       fprintf(fileHandle_p,txt_buffer);
885                   }
886                   for (i=0; i<mesh_p->Nodes->numNodes; i++) {
887                      k=globalNodeIndex[i];
888                      if ( (myFirstNode <= k) && (k < myLastNode) ) {
889                         values = getSampleData(data_pp[i_data], nodeMapping->target[i]);
890                         /* if the number of mpi_required components is more than the number
891                         * of actual components, pad with zeros
892                         */
893                         /* probably only need to get shape of first element */
894                         /* write the data different ways for scalar, vector and tensor */
895                         if (nCompReqd == 1) {
896                           sprintf(tmp_buffer,FLOAT_SCALAR_FORMAT,values[0]);
897                         } else if (nCompReqd == 3) {
898                           if (shape==1) {
899                            sprintf(tmp_buffer,FLOAT_VECTOR_FORMAT,values[0],0.,0.);
900                           } else if (shape==2) {
901                            sprintf(tmp_buffer,FLOAT_VECTOR_FORMAT,values[0],values[1],0.);
902                           } else if (shape==3) {
903                            sprintf(tmp_buffer,FLOAT_VECTOR_FORMAT,values[0],values[1],values[2]);
904                           }
905                         } else if (nCompReqd == 9) {
906                           if (shape==1) {
907                            sprintf(tmp_buffer,FLOAT_TENSOR_FORMAT,values[0],0.,0.,
908                                                                   0.,0.,0.,
909                                                                   0.,0.,0.);
910                           } else if (shape==2) {
911                            sprintf(tmp_buffer,FLOAT_TENSOR_FORMAT,values[0],values[1],0.,
912                                                                   values[2],values[3],0.,
913                                                                   0.,0.,0.);
914                           } else if (shape==3) {
915                            sprintf(tmp_buffer,FLOAT_TENSOR_FORMAT,values[0],values[1],values[2],
916                                                                   values[3],values[4],values[5],
917                                                                   values[6],values[7],values[8]);
918                           }
919                         }
920                         if ( mpi_size > 1) {
921                           __STRCAT(txt_buffer,tmp_buffer,txt_buffer_in_use);
922                         } else {
923                           fprintf(fileHandle_p,tmp_buffer);
924                         }
925                      }
926                   }
927                   if ( mpi_size > 1) {
928                       #ifdef PASO_MPI
929                         MPI_File_write_ordered(mpi_fileHandle_p,txt_buffer,txt_buffer_in_use, MPI_CHAR, &mpi_status);
930                       #endif    
931                       if ( my_mpi_rank == 0) {
932                          #ifdef PASO_MPI
933                             MPI_File_iwrite_shared(mpi_fileHandle_p,tag_End_DataArray,strlen(tag_End_DataArray),MPI_CHAR,&mpi_req);
934                             MPI_Wait(&mpi_req,&mpi_status);
935                          #endif
936                       }
937                   } else {
938                      fprintf(fileHandle_p,tag_End_DataArray);
939                   }
940                }
941              }
942            }
943            if ( mpi_size > 1) {
944              if ( my_mpi_rank == 0) {
945                 #ifdef PASO_MPI
946                    MPI_File_iwrite_shared(mpi_fileHandle_p,tag_End_CellData,strlen(tag_End_PointData),MPI_CHAR,&mpi_req);
947                    MPI_Wait(&mpi_req,&mpi_status);
948                 #endif
949              }
950            } else {
951                fprintf(fileHandle_p,tag_End_PointData);
952            }
953      }
954      if (Finley_noError()) {
955         if ( mpi_size > 1) {
956           if ( my_mpi_rank == 0) {
957              #ifdef PASO_MPI
958                 MPI_File_iwrite_shared(mpi_fileHandle_p,footer,strlen(footer),MPI_CHAR,&mpi_req);
959                 MPI_Wait(&mpi_req,&mpi_status);
960                 #ifdef MPIO_HINTS
961                   MPI_Info_free(&mpi_info);
962                   #undef MPIO_HINTS
963                 #endif
964                 MPI_File_close(&mpi_fileHandle_p);
965              #endif
966            }
967         } else {
968             fprintf(fileHandle_p,footer);
969             fclose(fileHandle_p);
970         }
971      }
972      TMPMEMFREE(isCellCentered);
973      TMPMEMFREE(txt_buffer);
974    return;    return;
975  }  }
   
 /*  
  * $Log$  
  * Revision 1.6  2005/08/12 01:45:43  jgs  
  * erge of development branch dev-02 back to main trunk on 2005-08-12  
  *  
  * Revision 1.5.2.1  2005/08/10 06:14:37  gross  
  * QUADRATIC HEXAHEDRON elements fixed  
  *  
  * Revision 1.5  2005/07/08 04:07:55  jgs  
  * Merge of development branch back to main trunk on 2005-07-08  
  *  
  * Revision 1.4  2005/05/06 04:26:15  jgs  
  * Merge of development branch back to main trunk on 2005-05-06  
  *  
  * Revision 1.1.2.7  2005/06/29 02:34:54  gross  
  * some changes towards 64 integers in finley  
  *  
  * Revision 1.1.2.6  2005/05/06 01:17:19  cochrane  
  * Fixed incorrect reporting of number of components in PointData arrays for  
  * vector data.  
  *  
  * Revision 1.1.2.5  2005/05/05 05:38:44  cochrane  
  * Improved formatting of VTK file output.  
  *  
  * Revision 1.1.2.4  2005/02/22 10:03:54  cochrane  
  * Implementation of writing of vtk xml files from finley.  This function will  
  * require more testing, but on the cases that I have tried (and with the help  
  * of Lutz and mayavi), it looks like it's producing the correct output.  Testing  
  * with more realistic data would be good.  I'm at least confident enough to  
  * commit my changes.  
  *  
  * Revision 1.1.2.3  2005/02/17 05:53:26  gross  
  * some bug in saveDX fixed: in fact the bug was in  
  * DataC/getDataPointShape  
  *  
  * Revision 1.1.2.2  2005/02/10 01:34:22  cochrane  
  * Quick fix to make sure that saveVTK compiles so that finley is still buildable.  Apologies to those this has affected.  
  *  
  * Revision 1.1.2.1  2005/02/09 06:53:15  cochrane  
  * Initial import to repository.  This is the file to implement saving finley/escript meshes out to vtk formatted files.  It is basically just a hack of the opendx equivalent, with a lot of the opendx stuff still in the file, so it doesn't actually work just yet, but it probably needs to be added to the cvs.  
  *  
  * Revision 1.1.1.1  2004/10/26 06:53:57  jgs  
  * initial import of project esys2  
  *  
  * Revision 1.1  2004/07/27 08:27:11  gross  
  * Finley: saveDX added: now it is possible to write data on boundary and contact elements  
  *  
  */  

Legend:
Removed from v.147  
changed lines
  Added in v.1384

  ViewVC Help
Powered by ViewVC 1.1.26