timer, flat source, and source region additions

Author: yardasol

include/openmc/random_ray/flat_source_domain.h

@@ -39,6 +39,7 @@ class FlatSourceDomain {
   void reset_tally_volumes();
   void random_ray_tally();
   virtual void accumulate_iteration_flux();
+  void accumulate_iteration_source();
   void output_to_vtk() const;
   void convert_external_sources();
   void count_external_source_regions();
@@ -49,6 +50,7 @@ class FlatSourceDomain {
   void flatten_xs();
   void transpose_scattering_matrix();
   void serialize_final_fluxes(vector<double>& flux);
+  void serialize_final_sources(vector<double>& source);
   void apply_meshes();
   void apply_mesh_to_cell_instances(int32_t i_cell, int32_t mesh_idx,
     int target_material_id, const vector<int32_t>& instances,
@@ -72,6 +74,37 @@ class FlatSourceDomain {
   int64_t lookup_mesh_bin(int64_t sr, Position r) const;
   int lookup_mesh_idx(int64_t sr) const;
 
+  //----------------------------------------------------------------------------
+  // Methods for kinetic simulations
+  virtual void update_single_neutron_source_td(SourceRegionHandle& srh);
+  void compute_single_neutron_source_time_derivative(SourceRegionHandle& srh);
+  void compute_single_scalar_flux_time_derivative_2(SourceRegionHandle& srh);
+  virtual void update_all_neutron_sources_td();
+
+  void compute_single_delayed_fission_source(SourceRegionHandle& srh);
+  void compute_single_precursors(SourceRegionHandle& srh);
+  void compute_all_precursors();
+
+  void serialize_final_td_fluxes(vector<double>& flux_td);
+  void serialize_final_td_sources(vector<double>& source_td);
+  void serialize_final_precursors(vector<double>& precursors);
+  void serialize_final_delayed_fission_source(
+    vector<double>& delayed_fission_source);
+
+  void flux_td_swap();
+  void precursors_swap();
+  void accumulate_iteration_quantities();
+  void normalize_final_quantities();
+  void propagate_final_quantities();
+  void store_time_step_quantities(bool increment_not_initialize = true);
+  void compute_rhs_bd_quantities();
+  void update_material_density(int i);
+
+  int64_t n_delay_elements() const
+  {
+    return source_regions_.n_source_regions() * ndgroups_;
+  }
+
   //----------------------------------------------------------------------------
   // Static Data members
   static bool volume_normalized_flux_tallies_;
@@ -149,6 +182,31 @@ class FlatSourceDomain {
   // technique.
   bool is_transport_stabilization_needed_ {false};
 
+  //---------------------------------------------------------------------------
+  // Public Data Members for kinetic simulations
+
+  // 2D arrays stored in 1D representing values for all materials x
+  // delay_groups
+  vector<double> lambda_;
+
+  // 3D arrays stored in 1D representing values for all materials x energy
+  // groups x delay groups
+  vector<double> nu_d_sigma_f_;
+  vector<double> chi_d_;
+
+  // 2D arrays stored in 1D representing values for all materials x energy
+  // groups
+  vector<double> nu_p_sigma_f_;
+  vector<double> chi_p_;
+  vector<double> inverse_vbar_;
+
+  // Time-dependent cross section arrays for use with material density
+  // timeseries
+  vector<double> sigma_t_td_;
+  vector<double> nu_sigma_f_td_;
+  vector<double> sigma_f_td_;
+  vector<double> sigma_s_td_;
+
 protected:
   //----------------------------------------------------------------------------
   // Methods
@@ -165,6 +223,7 @@ class FlatSourceDomain {
   //----------------------------------------------------------------------------
   // Private data members
   int negroups_; // Number of energy groups in simulation
+  int ndgroups_; // Number of delay groups in simulation
 
   double
     simulation_volume_; // Total physical volume of the simulation domain, as