unknown_manager.cc

Go to the documentation of this file.

#include "unknown_manager.h"
 
#include "chi_runtime.h"
#include "chi_log.h"
 
//###################################################################
/**Adds an unknown to the manager. This method will figure out
 * where the last unknown ends and where to begin the next one.*/
unsigned int chi_math::UnknownManager::
  AddUnknown(UnknownType unk_type,
             unsigned int dimension)
{
  auto& log = chi::ChiLog::GetInstance();
 
  unsigned int last_unknown_end = -1;
  if (not unknowns_.empty())
    last_unknown_end = unknowns_.back().GetMapEnd();
 
  unsigned int new_unknown_index = unknowns_.size();
 
  if (unk_type == UnknownType::SCALAR)
  {
    unknowns_.emplace_back(UnknownType::SCALAR, 1, last_unknown_end + 1);
    unknowns_.back().text_name_ = "Unknown_" + std::to_string(unknowns_.size() - 1);
  }
  else if (unk_type == UnknownType::VECTOR_2)
  {
    unknowns_.emplace_back(UnknownType::VECTOR_2, 2, last_unknown_end + 1);
    unknowns_.back().text_name_ = "Unknown_" + std::to_string(unknowns_.size() - 1);
  }
  else if (unk_type == UnknownType::VECTOR_3)
  {
    unknowns_.emplace_back(UnknownType::VECTOR_3, 3, last_unknown_end + 1);
    unknowns_.back().text_name_ = "Unknown_" + std::to_string(unknowns_.size() - 1);
  }
  else if (unk_type == UnknownType::VECTOR_N)
  {
    if (dimension == 0)
    {
      log.LogAllError()
        << "UnknownManager: When adding unknown of type VECTOR_N, "
        << "the dimension must not be 0.";
      Chi::Exit(EXIT_FAILURE);
    }
 
    unknowns_.emplace_back(UnknownType::VECTOR_N, dimension, last_unknown_end + 1);
    unknowns_.back().text_name_ = "Unknown_" + std::to_string(unknowns_.size() - 1);
  }
  else if (unk_type == UnknownType::TENSOR)
  {
    if (dimension == 0 or dimension == 1)
    {
      log.LogAllError()
        << "UnknownManager: When adding unknown of type TENSOR, "
        << "the dimension must not be 0 or 1.";
      Chi::Exit(EXIT_FAILURE);
    }
 
    throw std::invalid_argument("UnknownManager: TENSOR unknowns are not "
                                "supported yet.");
  }
  else
  {
    throw std::logic_error("UnknownManager: Invalid call to AddUnknown(). "
                           "Unknown type is probably not supported yet.");
  }
 
  return new_unknown_index;
}
 
//###################################################################
/**Maps the unknown's component within the storage of a node.*/
unsigned int chi_math::UnknownManager::
  MapUnknown(unsigned int unknown_id, unsigned int component) const
{
  auto& log = chi::ChiLog::GetInstance();
 
  if (unknown_id < 0 or unknown_id >= unknowns_.size())
  {
    log.LogAllError()
      << "UnknownManager failed call to MapUnknown " << unknown_id;
    Chi::Exit(EXIT_FAILURE);
  }
  return unknowns_[unknown_id].GetMap(component);
}
 
//###################################################################
/**Determines the total number of components over all unknowns.*/
unsigned int chi_math::UnknownManager::GetTotalUnknownStructureSize() const
{
  if (unknowns_.empty())
    return 0;
 
  return unknowns_.back().GetMapEnd() + 1;
}
 
//###################################################################
/**Sets the number of off block connections for the given unknown.
 * All the components will be set to the same amount.*/
void chi_math::UnknownManager::
  SetUnknownNumOffBlockConnections(unsigned int unknown_id,
                                   int num_conn)
{
  auto& log = chi::ChiLog::GetInstance();
 
  if (unknown_id < 0 or unknown_id >= unknowns_.size())
  {
    log.LogAllError()
      << "UnknownManager failed call to SetUnknownNumOffBlockConnections,"
         " illegal index. " << unknown_id;
    Chi::Exit(EXIT_FAILURE);
  }
 
  for (auto& val : unknowns_[unknown_id].num_off_block_connections_)
    val = num_conn;
}
//###################################################################
/**Sets the number of off block connections for the given unknown-
 * component pair.*/
void chi_math::UnknownManager::
  SetUnknownComponentNumOffBlockConnections(unsigned int unknown_id,
                                            unsigned int component,
                                            int num_conn)
{
  auto& log = chi::ChiLog::GetInstance();
 
  if (unknown_id < 0 or unknown_id >= unknowns_.size())
  {
    log.LogAllError()
      << "UnknownManager failed call to SetUnknownComponentTextName,"
         " illegal unknown index. " << unknown_id;
    Chi::Exit(EXIT_FAILURE);
  }
 
  if (component < 0 or component >= unknowns_[unknown_id].num_components_)
  {
    log.LogAllError()
      << "UnknownManager failed call to SetUnknownComponentTextName,"
         " illegal component index. " << component;
    Chi::Exit(EXIT_FAILURE);
  }
 
  unknowns_[unknown_id].num_off_block_connections_[component] = num_conn;
 
}
 
//###################################################################
/**Sets a text name for the indicated unknown.*/
void chi_math::UnknownManager::
  SetUnknownTextName(unsigned int unknown_id,
                     const std::string& in_text_name)
{
  auto& log = chi::ChiLog::GetInstance();
 
  if (unknown_id < 0 or unknown_id >= unknowns_.size())
  {
    log.LogAllError()
      << "UnknownManager failed call to SetUnknownTextName,"
         " illegal index. " << unknown_id;
    Chi::Exit(EXIT_FAILURE);
  }
 
  unknowns_[unknown_id].text_name_ = in_text_name;
}
 
//###################################################################
/**Sets the text name to be associated with each component of the
 * unknown.*/
void chi_math::UnknownManager::
  SetUnknownComponentTextName(unsigned int unknown_id,
                              unsigned int component,
                              const std::string& in_text_name)
{
  auto& log = chi::ChiLog::GetInstance();
 
  if (unknown_id < 0 or unknown_id >= unknowns_.size())
  {
    log.LogAllError()
      << "UnknownManager failed call to SetUnknownComponentTextName,"
         " illegal unknown index. " << unknown_id;
    Chi::Exit(EXIT_FAILURE);
  }
 
  if (component < 0 or component >= unknowns_[unknown_id].num_components_)
  {
    log.LogAllError()
      << "UnknownManager failed call to SetUnknownComponentTextName,"
         " illegal component index. " << component;
    Chi::Exit(EXIT_FAILURE);
  }
 
  unknowns_[unknown_id].component_text_names_[component] = in_text_name;
 
}