file_1D_in.c

浏览该文件的文档.

 
#include <math.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
 
#include "../include/var_struc.h"
#include "../include/file_io.h"
 
 
#define N_MAX_1D 1000
 
#define STR_FLU_INI(sfv, err_exit)                  \
    do {                                \
    strcpy(add, add_in);                        \
    strcat(add, #sfv ".txt");                   \
    if((fp = fopen(add, "r")) == NULL)              \
        {                               \
        strcpy(add, add_in);                    \
        strcat(add, #sfv ".dat");               \
        }                               \
    if((fp = fopen(add, "r")) == NULL)              \
        {                               \
        printf("Cannot open initial data file: %s!\n", #sfv);   \
        if(err_exit)                        \
            exit(1);                        \
        r = false;                      \
        }                               \
    if(r)                               \
        {                               \
        num_cell = flu_var_count(fp, add);          \
        if (num_cell < 1)                   \
            {                           \
            printf("Error in counting fluid variables in initial data file: %s!\n", #sfv); \
            fclose(fp);                 \
            exit(2);                    \
            }                           \
        if(isinf(config[3]))                    \
            config[3] = (double)num_cell;           \
        else if(num_cell != (int)config[3])         \
            {                           \
            printf("Input unequal! num_%s=%d, num_cell=%d.\n", #sfv, num_cell, (int)config[3]); \
            exit(2);                    \
            }                           \
        FV0.sfv = (double*)malloc(num_cell * sizeof(double));   \
        if(FV0.sfv == NULL)                 \
            {                           \
            printf("NOT enough memory! %s\n", #sfv);    \
            exit(5);                    \
            }                           \
        if(flu_var_read(fp, FV0.sfv, num_cell))         \
            {                           \
            fclose(fp);                 \
            exit(2);                    \
            }                           \
        fclose(fp);                     \
        }                               \
    else                                \
        {                               \
        FV0.sfv = (double*)malloc(num_cell * sizeof(double));   \
        if(FV0.sfv == NULL)                 \
            {                           \
            printf("NOT enough memory! %s\n", #sfv);    \
            exit(5);                    \
            }                           \
        }                               \
    } while(0)
 
struct flu_var initialize_1D(const char * name, int * N, int * N_plot, double * time_plot[])
{
    struct flu_var FV0 = {NULL}; // Structure of initial data array pointer.
 
    char add_in[FILENAME_MAX+40]; 
    // Get the address of the initial data folder of the test example.
    example_io(name, add_in, 1);
    
    /* 
     * Read the configuration data.
     * The detail could be seen in the definition of array config
     * referring to file 'doc/config.csv'.
     */
    configurate(add_in);
    printf("  delta_x\t= %g\n", config[10]);
    printf("  bondary\t= %d\n", (int)config[17]);
 
    (*N) = time_plot_read(add_in, N_MAX_1D, N_plot, time_plot);
 
    char add[FILENAME_MAX+40]; // The address of the velocity/pressure/density file to read in.
    FILE * fp;      // The pointer to the above data files.
    int num_cell = (int)config[3]; // The number of the numbers in the above data files.
    _Bool r = true; // r: Whether to read data file successfully.
 
    // Open the initial data files and initializes the reading of data.
    STR_FLU_INI(RHO,  1);
    STR_FLU_INI(U,    1);
    STR_FLU_INI(P,    1);
#ifdef MULTIFLUID_BASICS
#ifdef MULTIPHASE_BASICS
    STR_FLU_INI(Z_a,  1);
    STR_FLU_INI(RHO_b,1);
    STR_FLU_INI(U_b,  1);
    STR_FLU_INI(V_b,  1);
    STR_FLU_INI(P_b,  1);
#else
    STR_FLU_INI(PHI,  1);
    STR_FLU_INI(Z_a,  0);
    if(!r)
    {
        for(int i = 0; i < num_cell; i++)
        FV0.Z_a[i] = FV0.PHI[i];
        printf("\t Initial volume fraction 'Z_a' is initialized by mass fraction 'PHI'.\n");
        r = true;
    }
    STR_FLU_INI(gamma,0);
    if(!r)
    {
        for(int i = 0; i < num_cell; i++)
        FV0.gamma[i] = 1.0 + 1.0 / (FV0.Z_a[i]/(config[6]-1.0) + (1.0-FV0.Z_a[i])/(config[106]-1.0));
        printf("\t Initial specific heat rate 'gamma' is initialized by volume fraction 'Z_a'.\n");
        r = true;
    }
#endif
#endif
 
    printf("'%s' data initialized, grid cell number = %d.\n", add_in, num_cell);
    return FV0;
}