inputGen.C

Go to the documentation of this file.

/*******************************************************************************
* Promesh                                                                      *
* Copyright (C) 2022, IllinoisRocstar LLC. All rights reserved.                *
*                                                                              *
* Promesh is the property of IllinoisRocstar LLC.                              *
*                                                                              *
* IllinoisRocstar LLC                                                          *
* Champaign, IL                                                                *
* www.illinoisrocstar.com                                                      *
* promesh@illinoisrocstar.com                                                  *
*******************************************************************************/
/*******************************************************************************
* This file is part of Promesh                                                 *
*                                                                              *
* This version of Promesh 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 3 of the License,  *
* or (at your option) any later version.                                       *
*                                                                              *
* Promesh 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 program. If not, see <https://www.gnu.org/licenses/>.        *
*                                                                              *
*******************************************************************************/
// Nemosys headers
#include "InputGeneration/inputGen.H"

// other headers
#include <iostream>
#include <fstream>

#include "AuxiliaryFunctions.H"


void inputGen::setNameType(const std::string &fname,
                           inpFileType ftyp,
                           const std::string &key)
{
  if (!key.empty())
    _key = key;
  _fn[_key] = fname;
  _tpe[_key] = ftyp;
}


void inputGen::setOrder(const std::vector<std::string> &__ord,
                        const std::string &key)
{
  if (!key.empty())
    _key = key;
  std::vector<std::string> ord = __ord;
  for (auto &&it : ord)
    nemAux::toLower(it);
  _ord[_key].insert(_ord[_key].end(), ord.begin(), ord.end());
}


std::vector<std::string> inputGen::getOrder(const std::string &key)
{
  if (!key.empty())
    _key = key;
  return _ord[_key];
}


void inputGen::pushOrder(const std::string &__ord, const std::string &key)
{
    if (!key.empty())
        _key = key;
    std::string ord = __ord;
    nemAux::toLower(ord);
    _ord[_key].push_back(ord);
}


void inputGen::setMsh(meshBase *mb, const std::string &key)
{
  if (!key.empty())
    _key = key;
  _mb[_key].push_back(mb);
}


void inputGen::setCmntStr(const std::string &cmstr, const std::string &key)
{
  if (!key.empty())
    _key = key;
  _cmnt[_key] = cmstr;
}


std::string inputGen::getCmntStr(const std::string &key)
{
  if (!key.empty())
    _key = key;
  return _cmnt[_key];
}

void inputGen::write(const std::string &key) const
{
  bool onlyKey = false;
  if (!key.empty())
    onlyKey = true;
  for (const auto &it : _inp)
  {
    if (onlyKey && it.first != key)
      continue;

    std::string fname = _fn.at(it.first);
    std::ofstream ofile;
    ofile.open(fname);
    if (!ofile.good())
    {
      std::cerr << "Error opening file " << fname << std::endl;
      throw;
    }
    ofile << (it.second)->str();
    ofile.close();
  }
}