netgen/libsrc/interface/writeuser.hpp

#ifndef WRITEUSER
#define WRITEUSER

/**************************************************************************/
/* File:    writeuser.hh                                                  */
/* Authors: many                                                          */
/* Date:    10. Dec. 97                                                   */
/**************************************************************************/

#include <filesystem>
#include <functional>
#include <optional>

#include <meshing.hpp>

namespace netgen {

using namespace std::filesystem;

typedef std::function<void (const Mesh & mesh, const filesystem::path & filename)> FWrite;
typedef std::function<void (Mesh & mesh, const filesystem::path & filename)> FRead;
typedef std::function<bool (const filesystem::path & filename)> FTest;

struct UserFormatRegister {
  struct UserFormatEntry {
    string format;
    Array<string> extensions;
    optional<FRead> read;
    optional<FWrite> write;
  };
  DLL_HEADER static Array<UserFormatEntry> entries;
  DLL_HEADER static std::map<string, int> format_to_entry_index;

  static void Register(UserFormatEntry && entry) {
    format_to_entry_index[entry.format] = entries.Size();
    entries.Append( std::move(entry) );
  }

  static const bool HaveFormat(string format) {
    return format_to_entry_index.count(format) > 0;
  }
  static const UserFormatEntry & Get(string format) {
    return entries[format_to_entry_index[format]];
  }

  template<typename TFunc>
  static void IterateFormats(TFunc func, bool need_read=false, bool need_write=false) {
    Array<string> import_formats;
    for(const auto & e: entries)
    if((!need_read || e.read) && (!need_write || e.write))
      import_formats.Append(e.format);
    QuickSort(import_formats);
    for(auto format : import_formats)
      func(entries[format_to_entry_index[format]]);
  }

};

struct RegisterUserFormat {
  RegisterUserFormat(string format, Array<string> extensions, optional<FRead> read, optional<FWrite> write, FTest ftest = [](const filesystem::path & ){return true;})
  {
    UserFormatRegister::Register({format, std::move(extensions), std::move(read), std::move(write)});
  }
};

DLL_HEADER void ReadFile(Mesh & mesh, const filesystem::path & filename);
DLL_HEADER void ReadUserFormat(Mesh & mesh, const filesystem::path & filename, const string & format = "");


extern bool DLL_HEADER WriteUserFormat (const string & format,
                                        const Mesh & mesh,
                                        const filesystem::path & filename);

extern void DLL_HEADER RegisterUserFormats (NgArray<const char*> & names,
                                 NgArray<const char*> & extensions);

}

#endif