parseDomainFile.cu

Go to the documentation of this file.

#include <fstream>

#include "io.h"
#include "yaml-cpp/yaml.h"


namespace io
{

void parseDomain(const YAML::Node &node, domain &D)
{
    std::string dir;
    real start, end, stretchRatio, h;
    int  numCells;
    
    dir = node["direction"].as<std::string>();
    start = node["start"].as<real>();
    
    if (dir=="x")
        D.nx = 0;
    else if(dir=="y")
        D.ny = 0;
    
    const YAML::Node &subDomains = node["subDomains"];
    // first pass
    for (unsigned int i=0; i<subDomains.size(); i++)
    {
        numCells = subDomains[i]["cells"].as<int>();
        if (dir=="x")
            D.nx += numCells;
        else if(dir=="y")
            D.ny += numCells;
    }
    
    // allocate memory
    int beg = 0;
    if(dir=="x")
    {
        D.x.resize(D.nx+1);
        D.dx.resize(D.nx);
        D.xD.resize(D.nx+1);
        D.dxD.resize(D.nx);
        D.x[beg] = start;
    }
    if(dir=="y")
    {
        D.y.resize(D.ny+1);
        D.dy.resize(D.ny);  
        D.yD.resize(D.ny+1);
        D.dyD.resize(D.ny);
        D.y[beg] = start;
    }
    
    // second pass
    for (unsigned int i=0; i<subDomains.size(); i++)
    {
        end = subDomains[i]["end"].as<real>();
        numCells = subDomains[i]["cells"].as<int>();
        stretchRatio = subDomains[i]["stretchRatio"].as<real>();
        
        if(fabs(stretchRatio-1.0) < 1.0e-6)
        {
            h = (end - start)/numCells;
            for(int j=beg; j<beg+numCells; j++)
            {
                if(dir=="x")
                {
                    D.dx[j]  = h;
                    D.x[j+1] = D.x[j] + D.dx[j];
                }
                else if(dir=="y")
                {
                    D.dy[j]  = h;
                    D.y[j+1] = D.y[j] + D.dy[j];
                } 
            }
        }
        else
        {
            h = (end - start)*(stretchRatio-1)/(pow(stretchRatio, numCells)-1);
            for(int j=beg; j<beg+numCells; j++)
            {
                if(dir=="x")
                {
                    D.dx[j]  = h*pow(stretchRatio, j-beg);
                    D.x[j+1] = D.x[j] + D.dx[j];
                }
                else if(dir=="y")
                {
                    D.dy[j]  = h*pow(stretchRatio, j-beg);
                    D.y[j+1] = D.y[j] + D.dy[j];
                }
            }
        }
        beg += numCells;
        start = end;
    }
    
    if(dir=="x")
    {
        D.xD  = D.x;
        D.dxD = D.dx;
    }
    else if(dir=="y")
    {
        D.yD  = D.y;
        D.dyD = D.dy;
    }
} // parseDomain


void parseDomainFile(std::string &domFile, domain &D)
{
    printf("Parsing YAML file with grid info ...\n");
    YAML::Node nodes = YAML::LoadFile(domFile);
    for (unsigned int i=0; i<nodes.size(); i++)
        parseDomain(nodes[i], D);
        
    D.xu.resize(D.nx-1);
    D.yu.resize(D.ny);
    D.xv.resize(D.nx);
    D.yv.resize(D.ny-1);
    
    int i, j;
    for(i=0; i<D.nx-1; i++)
    {
        D.xu[i] = D.x[i+1];
        D.xv[i] = (D.x[i]+D.x[i+1])/2.0;
    }
    D.xv[i] = (D.x[i]+D.x[i+1])/2.0;
    
    for(j=0; j<D.ny-1; j++)
    {
        D.yu[j] = (D.y[j]+D.y[j+1])/2.0;
        D.yv[j] = D.y[j+1];
    }
    D.yu[j] = (D.y[j]+D.y[j+1])/2.0;
} // parseDomainFile

} // End of namespace io