Migration.h

浏览该文件的文档。

#ifndef __Migration_h__
#define __Migration_h__

#include <map>
#include "BinaryBuffer.h"
#include "TemplateElement.h"

namespace Migration {

  typedef int data_id_t; 
  typedef std::string data_name_t; 
  typedef BinaryBuffer<> data_buffer_t; 

  typedef std::map<data_id_t, data_buffer_t> buffer_t; 

  data_id_t name_to_id(const data_name_t& dn); 
  data_id_t register_data_name(const data_name_t& dn, bool); 
  void initialize(); 
  bool is_valid(const data_id_t&); 

  namespace details {
    template <class MESH> BinaryBuffer<>& 
      get_buffer(MESH& mesh,
                 const data_id_t& data_id,
                 u_int dimension,
                 u_int geo_idx,
                 bool is_save) {
      buffer_t& buffer = mesh.h_geometry(dimension, geo_idx)->buffer;
      if (is_save) {
        return buffer[data_id];
      } else {
        buffer_t::iterator it = buffer.find(data_id);
        if (it == buffer.end()) {
          return *((BinaryBuffer<> *)NULL);
        }
        return it->second;
      }
    }

  }

  template <class MESH, class STREAM> void
    get_export_stream(MESH& mesh,
                      const data_id_t& data_id,
                      u_int dimension,
                      u_int geo_idx,
                      STREAM& os) {
    os.set_buffer(details::get_buffer(mesh, data_id, dimension, geo_idx, true));
  }
  
  template <class MESH, class STREAM> void
    get_import_stream(MESH& mesh,
                      const data_id_t& data_id,
                      u_int dimension,
                      u_int geo_idx,
                      STREAM& is) {
    is.set_buffer(details::get_buffer(mesh, data_id, dimension, geo_idx, false));
  }

  template <class MESH, class SP, class STREAM> void
    get_dof_export_stream(MESH& mesh,
                          SP& sp,
                          const data_id_t& data_id,
                          int dof,
                          STREAM& os) {
    const DOFIndex& di = sp.dofIndex(dof);
    get_export_stream(mesh, data_id, di.dimension, di.geometry_index, os);
  }

  template <class MESH, class SP, class STREAM> void
    get_dof_import_stream(MESH& mesh,
                          SP& sp,
                          const data_id_t& data_id,
                          int dof,
                          STREAM& is) {
    const DOFIndex& di = sp.dofIndex(dof);
    get_import_stream(mesh, data_id, di.dimension, di.geometry_index, is);
  }

} // end of namespace Migration

#endif // __Migration_h__