grid/gridInitialData.h

Go to the documentation of this file.

/**
 *  @file      gridInitialData.h
 *  @brief     Reads initial data into the grid.
 *  @author    Mikica Kocic
 *  @copyright GNU General Public License (GPLv3).
 */

/** Reads the initial data from the given file.
 */
class GridInitialData
{
    UniformGrid* gridDriver; //!< The grid
    bool  binaryFormat;   //!< Read binary data from the input file.
    std::string fileName; //!< Input filename.
    std::vector<Int> input; //! The list of grid functions

public:

    /** Constructs the initial data reader as specified in the parameter file.
     */
    GridInitialData( Parameters& params, UniformGrid& ug )
        : gridDriver(&ug), binaryFormat(false)
    {
        input.reserve( 100 );
        input.push_back( fld::r );

        params.get( "input.file",   fileName, "stdin" );
        params.get( "input.binary", binaryFormat, true );

        slog << "Initial Data:" << std::endl << std::endl
             << "    file = " << fileName << ",  binary = " << binaryFormat
             << std::endl << std::endl;
    }

    /** The given grid functions will be read as the initial data.
     */
    void gridFunctions( const std::vector<Int>& gfs )
    {
        input.insert( input.end(), gfs.begin(), gfs.end() );
    }

    /** Loads the initial data.
     */
    bool load ()
    {
        GridPoint** grid = gridDriver->get_grid ();
        Int nLen   = gridDriver->get_nLen ();
        Int nGhost = gridDriver->get_nGhost ();

        FILE* inf = fileName == "stdin" ? stdin
                  : fopen( fileName.c_str(), binaryFormat ? "rb" : "r" );
        if( ! inf ) {
            std::cerr << "Error: Could not open " << fileName << std::endl;
            return false;
        }

        // If using MPI, offset the initial data depending on our rank
        if ( gridDriver->mpiSize () > 1 )
        {
            for( Int n = 0; n < nLen * gridDriver->mpiRank(); ++n )
            {
                if( ! IORecord( grid[0][0] ).read( inf, binaryFormat, input ) ) { // dummy read
                    return false;
                }
            }
        }

        for( Int n = nGhost; n < nGhost + nLen; ++n )
        {
            if( ! IORecord( grid[0][n] ).read( inf, binaryFormat, input ) ) {
                return false;
            }
        }

        if ( inf != stdin ) {
            fclose( inf );
        }

        return true;
    }
};