WriteDamsel.hpp

Go to the documentation of this file.

//-------------------------------------------------------------------------
// Filename      : WriteDamsel.hpp
//
// Purpose       : ExodusII writer
//
// Special Notes : Lots of code taken from verde implementation
//
// Creator       : Corey Ernst 
//
// Date          : 8/02
//
// Owner         : Corey Ernst 
//-------------------------------------------------------------------------

#ifndef WRITEDAMSEL_HPP
#define WRITEDAMSEL_HPP

#ifndef IS_BUILDING_MB
#error "WriteDamsel.hpp isn't supposed to be included into an application"
#endif

#include <vector>
#include <string>

#include "moab/Interface.hpp"
#include "moab/Range.hpp"
#include "moab/WriterIface.hpp"
#include "RangeSeqIntersectIter.hpp"
#include "moab/FileOptions.hpp"
#include "DamselUtil.hpp"

#include "damsel.h"

namespace moab {

class WriteUtilIface;
class SequenceManager;
class Error;

class WriteDamsel : public WriterIface
{
 
public:
  static WriterIface* factory( Interface* iface );

  WriteDamsel(Interface *impl);

  virtual ~WriteDamsel();

  ErrorCode write_file(const char *file_name, 
                       const bool /* overwrite */,
                       const FileOptions& opts,
                       const EntityHandle *meshset_list,
                       const int num_sets,
                       const std::vector<std::string>& /* qa_records */,
                       const Tag* /* tag_list */ = NULL,
                       int /* num_tags */ = 0,
                       int /* requested_output_dimension */ = 3);

  enum {DAMSEL_IS_TRACKING = 0x1
  } DAMSEL_FLAGS;
  
private:

  ErrorCode init_tag_info();

  ErrorCode write_subrange(RangeSeqIntersectIter &rsi);
  
  ErrorCode write_vertices(RangeSeqIntersectIter &rsi);
  
  ErrorCode write_entities(RangeSeqIntersectIter &rsi);
  
  ErrorCode write_sets(RangeSeqIntersectIter &rsi);
  
  ErrorCode map_dense_tags(RangeSeqIntersectIter &rsi, damsel_container &ent_cont);

  ErrorCode map_sparse_tags();

  Interface *mbImpl;

  WriteUtilIface* mWriteIface;

  Error *mError;
  
  SequenceManager *sequenceManager;

  std::string fileName;

  DamselUtil dU;
};

inline ErrorCode WriteDamsel::write_subrange(RangeSeqIntersectIter &rsi) 
{
  ErrorCode rval = MB_SUCCESS;

  if (MBVERTEX == mbImpl->type_from_handle(rsi.get_start_handle())) 
    rval = write_vertices(rsi);

  else if (MBENTITYSET > mbImpl->type_from_handle(rsi.get_start_handle())) 
    rval = write_entities(rsi);

//  else
//    rval = write_sets(rsi);

  return rval;
}

} // namespace moab

#endif