Model.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Stanford University, The Regents of the
// University of California, and others. SPDX-License-Identifier: BSD-3-Clause
/**
 * @file Model.h
 * @brief model::Model source file
 */
 
#ifndef SVZERODSOLVER_MODEL_MODEL_HPP_
#define SVZERODSOLVER_MODEL_MODEL_HPP_
 
#include <algorithm>
#include <list>
#include <map>
#include <memory>
#include <string>
#include <vector>
 
#include "Block.h"
#include "BlockFactory.h"
#include "BloodVessel.h"
#include "BloodVesselCRL.h"
#include "BloodVesselJunction.h"
#include "BloodVesselRC.h"
#include "ChamberElastanceInductor.h"
#include "ChamberElastanceInductorExponential.h"
#include "ChamberSphere.h"
#include "ClosedLoopCoronaryLeftBC.h"
#include "ClosedLoopCoronaryRightBC.h"
#include "ClosedLoopHeartPulmonary.h"
#include "ClosedLoopRCRBC.h"
#include "DOFHandler.h"
#include "FlowReferenceBC.h"
#include "ImpedanceBC.h"
#include "Junction.h"
#include "LinearElastanceChamber.h"
#include "Node.h"
#include "OpenLoopCoronaryBC.h"
#include "OpenLoopCoronaryVarResBC.h"
#include "Parameter.h"
#include "PiecewiseValve.h"
#include "PressureReferenceBC.h"
#include "ResistanceBC.h"
#include "ResistiveJunction.h"
#include "State.h"
#include "ValveTanh.h"
#include "WindkesselBC.h"
#include "debug.h"
 
/**
 * @brief Model of 0D elements
 *
 * This class represents a full 0D model. It contains attributes and
 * methods to store and modify 0D elements.
 *
 */
class Model {
 public:
  /// Factory that holds all implemented blocks
  std::map<std::string_view, BlockFactoryFunc> block_factory_map;
 
  /**
   * @brief Construct a new Model object
   *
   */
  Model();
 
  /**
   * @brief Destroy the Model object
   *
   */
  ~Model();
 
  DOFHandler dofhandler;  ///< Degree-of-freedom handler of the model
 
  double cardiac_cycle_period = -1.0;  ///< Cardiac cycle period
  double time = 0.0;                   ///< Current time
 
  /**
   * @brief Create a new block
   *
   * @param block_name The block name (defined in block_factory_map)
   * @return int Global ID of the block
   */
  Block* create_block(const std::string& block_name);
 
  /**
   * @brief Add a block to the model (without parameters)
   *
   * @param block The block to add
   * @param name The name of the block
   * @param block_param_ids Global IDs of the parameters of the block
   * @param internal Toggle whether block is internal
   * @return int Global ID of the block
   */
  int add_block(Block* block, const std::string_view& name,
                const std::vector<int>& block_param_ids, bool internal = false);
 
  /**
   * @brief Add a block to the model (with parameters)
   *
   * @param block_name Type of the block
   * @param block_param_ids Global IDs of the parameters of the block
   * @param name The name of the block
   * @param internal Toggle whether block is internal
   * @return int Global ID of the block
   */
  int add_block(const std::string& block_name,
                const std::vector<int>& block_param_ids,
                const std::string_view& name, bool internal = false);
 
  /**
   * @brief Check if a block with given name exists
   *
   * @param name Name of the Block
   * @return bool whether block exists
   */
  bool has_block(const std::string& name) const;
 
  /**
   * @brief Get a block by its name
   *
   * @param name Name of the Block
   * @return Block* The block
   */
  Block* get_block(const std::string_view& name) const;
 
  /**
   * @brief Get a block by its global ID
   *
   * @param block_id Global ID of the Block
   * @return Block* The block
   */
  Block* get_block(int block_id) const;
 
  /**
   * @brief Get a block type by its name
   *
   * @param name The name of the block
   * @return BlockType The block type
   */
  BlockType get_block_type(const std::string_view& name) const;
 
  /**
   * @brief Get the name of a block by it's ID
   *
   * @param block_id Global ID of the block
   * @return std::string Name of the block
   */
  std::string get_block_name(int block_id) const;
 
  /**
   * @brief Add a node to the model
   *
   * @param inlet_eles Inlet blocks of the node
   * @param outlet_eles Outlet blocks of the node
   * @param name Name of node
   * @return int Global ID of the node
   */
  int add_node(const std::vector<Block*>& inlet_eles,
               const std::vector<Block*>& outlet_eles,
               const std::string_view& name);
 
  /**
   * @brief Get the name of a node by it's ID
   *
   * @param node_id Global ID of the node
   * @return std::string Name of the node
   */
  std::string get_node_name(int node_id) const;
 
  /**
   * @brief Add a constant model parameter
   *
   * @param value Value of the parameter
   * @return int Global ID of the parameter
   */
  int add_parameter(double value);
 
  /**
   * @brief Add a time-dependent model parameter
   *
   * @param times Times corresponding to the parameter values
   * @param values Values of the parameter
   * @param periodic Toggle whether parameter is periodic
   * @return int Global ID of the parameter
   */
  int add_parameter(const std::vector<double>& times,
                    const std::vector<double>& values, bool periodic = true);
 
  /**
   * @brief Get a parameter by its global ID
   *
   * @param param_id Global ID of the parameter
   * @return Parameter* The parameter
   */
  Parameter* get_parameter(int param_id);
 
  /**
   * @brief Get the current value of a parameter
   *
   * @param param_id Global ID of the parameter
   * @return T Current value of the parameter
   */
  double get_parameter_value(int param_id) const;
 
  /**
   * @brief Update the current value of a parameter in the `parameter_values`
   * vector. Note that this is different from updating the value within each
   * parameter object, which is done in Parameter::update()
   *
   * @param param_id Global ID of the parameter
   * @param param_value The new parameter value
   */
  void update_parameter_value(int param_id, double param_value);
 
  /**
   * @brief Finalize the model after all blocks, nodes and parameters have been
   * added
   *
   */
  void finalize();
 
  /**
   * @brief Update the constant contributions of all elements in a sparse system
   *
   * @param system System to update contributions at
   */
  void update_constant(SparseSystem& system);
 
  /**
   * @brief Update the time-dependent contributions of all elements in a sparse
   * system
   *
   * @param system System to update contributions at
   * @param time Current time
   */
  void update_time(SparseSystem& system, double time);
 
  /**
   * @brief Update the solution-dependent contributions of all elements in a
   * sparse system
   *
   * @param system System to update contributions at
   * @param y Current solution
   * @param dy Current derivate of the solution
   */
  void update_solution(SparseSystem& system,
                       Eigen::Matrix<double, Eigen::Dynamic, 1>& y,
                       Eigen::Matrix<double, Eigen::Dynamic, 1>& dy);
 
  /**
   * @brief Modify the solution after solving it
   *
   * @param y Current solution
   */
  void post_solve(Eigen::Matrix<double, Eigen::Dynamic, 1>& y);
 
  /**
   * @brief Accept one converged time-step for all blocks.
   *
   * @param y Accepted state vector
   */
  void accept_timestep(const Eigen::Matrix<double, Eigen::Dynamic, 1>& y);
 
  /**
   * @brief Get persistent state payloads for all blocks (including internal).
   *
   * @return Per-block persistent states
   */
  std::vector<std::vector<double>> get_persistent_states() const;
 
  /**
   * @brief Restore persistent states for all blocks.
   *
   * @param states Per-block persistent states
   */
  void set_persistent_states(const std::vector<std::vector<double>>& states);
 
  /**
   * @brief Clear persistent state for all blocks.
   */
  void clear_persistent_states();
 
  /**
   * @brief Set model time step size used by blocks with dt-dependent logic.
   *
   * @param time_step_size Model time step size
   */
  void set_time_step_size(double time_step_size);
 
  /**
   * @brief Get model time step size used by blocks with dt-dependent logic.
   *
   * @return double Model time step size
   */
  double get_time_step_size() const;
 
  /**
   * @brief Convert the blocks to a steady behavior
   *
   */
  void to_steady();
 
  /**
   * @brief Convert the blocks to an unsteady behavior
   *
   */
  void to_unsteady();
 
  /**
   * @brief Get number of triplets all elements
   *
   * Get the number of triplets the elements contribute to the global system
   * (relevant for sparse memory reservation)
   *
   * @return Number of triplets that are used in each system matrix
   */
  // std::map<std::string, int> get_num_triplets();
  TripletsContributions get_num_triplets() const;
 
  /**
   * @brief Get the number of blocks in the model
   *
   * @param internal Toggle whether to return internal/hidden blocks
   *
   * @return int Number of blocks
   */
  int get_num_blocks(bool internal = false) const;
 
  /**
   * @brief Specify if model has at least one Windkessel boundary condition
   *
   * @param has_windkessel Toggle if model has at least one Windkessel boundary
   * condition
   */
  void update_has_windkessel_bc(bool has_windkessel);
 
  /**
   * @brief Update model with largest time constant among all Windkessel
   * boundary conditions present in model
   *
   * @param time_constant Largest Windkessel time constant
   */
  void update_largest_windkessel_time_constant(double time_constant);
 
  /**
   * @brief Check if model has at least one Windkessel boundary condition
   *
   * @return bool True if model has at least one Windkessel boundary condition
   */
  bool get_has_windkessel_bc();
 
  /**
   * @brief Get largest Windkessel time constant in model
   *
   * @return double Largest Windkessel time constant of model
   */
  double get_largest_windkessel_time_constant();
  /**
   * @brief Setup model parameters that depend on the initial state
   *
   * @param initial_state The initial state vector
   */
  void setup_initial_state_dependent_parameters(State initial_state);
 
 private:
  int block_count = 0;
  int node_count = 0;
  int parameter_count = 0;
  std::map<int, double> param_value_cache;
 
  std::vector<std::shared_ptr<Block>> blocks;  ///< Blocks of the model
  std::vector<BlockType> block_types;          ///< Types of the blocks
  std::vector<std::string> block_names;        ///< Names of the blocks
  std::map<std::string, int>
      block_index_map;  ///< Map between block name and index
 
  std::vector<std::shared_ptr<Block>>
      hidden_blocks;  ///< Hidden blocks of the model
 
  std::vector<std::shared_ptr<Node>> nodes;  ///< Nodes of the model
  std::vector<std::string> node_names;       ///< Names of the nodes
 
  std::vector<Parameter>
      parameters;  ///< Parameters of the model. This vector stores the
                   ///< parameter objects and is primarily used to update
                   ///< `parameter_values` at each time-step for time-dependent
                   ///< parameters and also for steady initial conditions.
  std::vector<double>
      parameter_values;  ///< Current values of the parameters. This is passed
                         ///< to blocks to set up the linear system in
                         ///< `update_constant`, `update_time` and
                         ///< `update_solution`.
 
  bool has_windkessel_bc = false;
  double largest_windkessel_time_constant = 0.0;
  double time_step_size = -1.0;
};
 
#endif  // SVZERODSOLVER_MODEL_MODEL_HPP_