CGNS接口API写入CGNS格式文件

 

#include <iostream>
#include <QFile>

using namespace std;
#include "cgnslib.h"

int main()
{

    int result;

    int index_file;
    result = cg_set_file_type(CG_FILE_ADF);
    if (result != CG_OK) {
        cout << "SetFileType Error : " << cg_get_error() << endl;
    }
    else {
        cout << "SetFileType result = " << result << endl;
    }
    

    QFile file("test.cgns");
    int mode = CG_MODE_READ;
    if (file.exists()) {
        mode = CG_MODE_MODIFY;
    }
    else {
        mode = CG_MODE_WRITE;
    }
    if (CG_OK != cg_open("test.cgns", mode, &index_file))
        cg_error_print();


    int index_base;
    result = cg_base_write(index_file, "BaseNode", 3, 3, &index_base);
    if (result != CG_OK) {
        cout << "Base Error : " << cg_get_error() << endl;
    }
    else {
        cout << "Base result = " << result << endl;
    }
    

    int nbases;
    result = cg_nbases(index_file, &nbases);
    if (result != CG_OK) {
        cout << "Base Error : " << cg_get_error() << endl;
    }
    else {
        cout << "nbases = " << nbases << endl;
    }

    int index_zone;

    /**
    * @brief 选择创建结构化网格还是非结构化网格，Zone节点信息的存储方式不一样
    */
    string structureType = "Unstructured";

    if (structureType == "Unstructured") {
        cgsize_t isize[3] = { 8, 1, 0 };
        result = cg_zone_write(index_file, nbases, "ZoneNode", isize, Unstructured, &index_zone);
    }
    else {
        cgsize_t isize[3][3];
        // 每个维度的顶点数
        isize[0][0] = 2;
        isize[0][1] = 2;
        isize[0][2] = 2;
        // 每个维度的单元数
        isize[1][0] = isize[0][0] - 1;
        isize[1][1] = isize[0][1] - 1;
        isize[1][2] = isize[0][2] - 1;
        // 每个维度的边界顶点数，结构化网格的边界顶点数永远为0
        isize[2][0] = 0;
        isize[2][1] = 0;
        isize[2][2] = 0;
        //create zone
        result = cg_zone_write(index_file, nbases, "ZoneNode", *isize, Structured, &index_zone);
    }


    if (result != CG_OK) {
        cout << "Zone Error : " << cg_get_error() << endl;
    }
    else {
        cout << "Zone result = " << result << endl;
    }


    int nzones;
    result = cg_nzones(index_file, nbases, &nzones);
    if (result != CG_OK) {
        cout << "Zone Error : " << cg_get_error() << endl;
    }
    else {
        cout << "Zone Num = " << nzones << endl;
    }


    /* 必须使用cg_goto定位到要写信息的节点，否则不知道该往哪个节点写信息，会报错 */
    cg_goto(index_file, index_base, "end");
    result = cg_descriptor_write("Information", "The information text!");
    if (result != CG_OK) {
        cout << "descriptor Error : " << cg_get_error() << endl;
    }
    else {
        cout << "descriptor result = " << result << endl;
    }


    int ndescriptors;
    result = cg_ndescriptors(&ndescriptors);
    if (result != CG_OK) {
        cout << "Zone Error : " << cg_get_error() << endl;
    }
    else {
        cout << "descriptor Num = " << ndescriptors << endl;
    }


    vector<vector<float>> myarray = { {1, 2, 3}, {10, 20, 30}, {100, 200, 300} };
    result = cg_user_data_write("Array Node");
    if (CG_OK == result) {
        cg_goto(index_file, index_base, "UserDefinedData_t", 1, "end");
        
        for (int i = 0; i < myarray.size(); i++)
        {
            string name = "MyArray" + to_string(i);
            cgsize_t arraysize = myarray[i].size();
            result = cg_array_write(name.c_str(), CGNS_ENUMV(RealSingle), 1, &arraysize, &myarray[i][0]);
            if (CG_OK != result) {
                cout << "WriteArray Error :" << cg_get_error() << endl;
            }
        }
    }

    double x[2][2][2], y[2][2][2], z[2][2][2];
    int index_coord;

    int ni = 2;
    int nj = 2;
    int nk = 2;
    for (int k = 0; k < nk; ++k)
    {
        for (int j = 0; j < nj; ++j)
        {
            for (int i = 0; i < ni; ++i)
            {
                x[k][j][i] = i;
                y[k][j][i] = j;
                z[k][j][i] = k;
            }
        }
    }

    result = cg_coord_write(index_file, index_base, index_zone, RealDouble, "CoordinateX", x, &index_coord);
    if (CG_OK != result) {
        cout << "Coord Error:" << cg_get_error() << endl;
    }
    result = cg_coord_write(index_file, index_base, index_zone, RealDouble, "CoordinateY", y, &index_coord);
    if (CG_OK != result) {
        cout << "Coord Error:" << cg_get_error() << endl;
    }
    result = cg_coord_write(index_file, index_base, index_zone, RealDouble, "CoordinateZ", z, &index_coord);
    if (CG_OK != result) {
        cout << "Coord Error:" << cg_get_error() << endl;
    }


    string solname_vertex = "VertexFlowSolution";
    int index_flow_vertex;
    result = cg_sol_write(index_file, index_base, index_zone, solname_vertex.c_str(), Vertex, &index_flow_vertex);
    if (CG_OK != result) {
        cout << "WriteSol Error:" << cg_get_error() << endl;
    }

    int index_field_vertex;
    float soldata_vertex[2][2][2];
    for (int k = 0; k < nk; ++k)
    {
        for (int j = 0; j < nj; ++j)
        {
            for (int i = 0; i < ni; ++i)
            {
                soldata_vertex[k][j][i] = i * j*k;
            }
        }
    }

    cg_field_write(index_file, index_base, index_zone, index_flow_vertex, RealDouble, "Density", &soldata_vertex, &index_field_vertex);
    if (CG_OK != result) {
        cout << "WriteField Error:" << cg_get_error() << endl;
    }



    string solname_center = "CellCenterFlowSolution";
    int index_flow_center;
    result = cg_sol_write(index_file, index_base, index_zone, solname_center.c_str(), CellCenter, &index_flow_center);
    if (CG_OK != result) {
        cout << "WriteSol Error:" << cg_get_error() << endl;
    }

    int index_field_center;
    float soldata_center[1][1][1] = { {{0.5}} };

    cg_field_write(index_file, index_base, index_zone, index_flow_center, RealDouble, "Pressure", &soldata_center, &index_field_center);
    if (CG_OK != result) {
        cout << "WriteField Error:" << cg_get_error() << endl;
    }



    int index_family;
    result = cg_family_write(index_file, index_base, "Family Node", &index_family);
    if (CG_OK != result) {
        cout << "Family Error:" << cg_get_error() << endl;
    }

    cg_close(index_file);

    return 0;
}

 

paraview预览文件效果（节点数据）：

paraview预览文件效果（单元数据）：