xdr_mgf.h

Go to the documentation of this file.

// The libMesh Finite Element Library.
// Copyright (C) 2002-2014 Benjamin S. Kirk, John W. Peterson, Roy H. Stogner

// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.

// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.

// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

#ifndef LIBMESH_XDR_MGF_H
#define LIBMESH_XDR_MGF_H

// Local includes
#include "libmesh/legacy_xdr_io.h"          // for LegacyXdrIO::FileFormat
#include "libmesh/libmesh_config.h"  // for LIBMESH_HAVE_XDR

// C++ includes
#include <cstdio>  // for std::FILE
#include <string>
#include <cstring> // std::strlen, std::strcmp
#include <fstream> // for std::ifstream
#include <sstream>

// Forward Declarations

#ifdef LIBMESH_HAVE_XDR
#  ifdef LIBMESH_HAVE_RPC_RPC_H
#    include <rpc/rpc.h>
#  elif  LIBMESH_HAVE_RPC_XDR_H
#    include <rpc/xdr.h>
#  endif
#  ifndef LIBMESH_DEFAULT_SINGLE_PRECISION
#    ifdef LIBMESH_DEFAULT_TRIPLE_PRECISION
// #      define xdr_REAL xdr_quadruple
// For some reason my xdr implementation doesn't define
// xdr_quadruple... - RHS
#      define xdr_REAL xdr_double
#      define xdr_Real double
#    else
#      define xdr_REAL xdr_double
#      define xdr_Real Real
#    endif
#  else
#    define xdr_REAL xdr_float
#    define xdr_Real Real
#  endif
#else
#  define xdr_Real Real
#endif


namespace libMesh
{

class XdrMGF
{
public:
  enum XdrIO_TYPE {UNKNOWN = -1, ENCODE=0, DECODE,
                   W_ASCII , R_ASCII};


#ifdef LIBMESH_HAVE_XDR
  XdrMGF() : _num_levels(0), m_type(UNKNOWN), mp_xdr_handle(0), orig_flag(LegacyXdrIO::LIBM), mp_fp(0) {}
#else
  XdrMGF() : _num_levels(0), m_type(UNKNOWN), orig_flag(LegacyXdrIO::LIBM), mp_fp(0) {}
#endif

  void init(XdrIO_TYPE t, const char* fn, const char* type, int icnt);

  virtual ~XdrMGF();

  void fini();

  int dataBlk(int*  array, int numvar, int size);

  int dataBlk(Real* array, int numvar, int size);

  LegacyXdrIO::FileFormat get_orig_flag() const { return orig_flag; }

  void set_orig_flag(LegacyXdrIO::FileFormat in_orig_flag) { orig_flag = in_orig_flag; }


  void set_num_levels(unsigned int num_levels) { _num_levels = num_levels; }

  unsigned int get_num_levels() { return _num_levels; }

protected:

  unsigned int _num_levels;

  XdrIO_TYPE m_type;

#ifdef LIBMESH_HAVE_XDR

  XDR*  mp_xdr_handle;

#endif

  LegacyXdrIO::FileFormat orig_flag;

  std::ifstream mp_in;

  std::ofstream mp_out;

private:
  std::FILE* mp_fp;

  void tokenize_first_line(const char* p)
  {
    std::string buf_str(p);
    std::stringstream ss(buf_str);

    char token[256];
    ss >> token;
    if(std::strcmp(token,"LIBM") == 0)
      {
        ss >> token;
        _num_levels = std::atoi(token);
      }

  }
};


} // namespace libMesh


#endif // LIBMESH_XDR_MGF_H