Tiding TransportBase SlaybaughLab#10

wzheng21 · Jul 26, 2017 · 4749172 · 4749172
1 parent f259461
commit 4749172
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 16 deletions.
diff --git a/src/common/bart_driver.cc b/src/common/bart_driver.cc
@@ -43,7 +43,8 @@ pcout(std::cout,
       (Utilities::MPI::this_mpi_process(mpi_communicator)
        == 0))
 {
-  initialize_aq (prm);
+  aqd_ptr = build_aq_model (prm)
+  aqd_ptr->make_aq (prm);
   n_total_ho_vars = aqd_ptr->get_n_total_ho_vars ();
   n_azi = aqd_ptr->get_sn_order ();
   n_dir = aqd_ptr->get_n_dir ();
@@ -58,20 +59,6 @@ BartDriver<dim>::~BartDriver ()
   dof_handler.clear();
 }
 
-template <int dim>
-void BartDriver<dim>::process_input ()
-{
-  // from angular quadrature data
-
-}
-
-template <int dim>
-void BartDriver<dim>::initialize_aq (ParameterHandler &prm)
-{
-  aqd_ptr = build_aq_model (prm)
-  aqd_ptr->make_aq (prm);
-}
-
 template <int dim>
 void BartDriver<dim>::report_system ()
 {

diff --git a/src/transport/transport_base.cc b/src/transport/transport_base.cc
@@ -139,7 +139,9 @@ void TransportBase<dim>::assemble_ho_system ()
 }
 
 template <int dim>
-void TransportBase<dim>::initialize_assembly_related_objects ()
+void TransportBase<dim>::initialize_assembly_related_objects
+(DoFHandler<dim> &dof_handler,
+ FE_Poly<TensorProductPolynomials<dim>,dim,dim>* fe)
 {
   // we need to input ptr to fe, dof_handler
 

diff --git a/src/transport/transport_base.h b/src/transport/transport_base.h
@@ -123,6 +123,9 @@ class TransportBase
   void scale_fiss_transfer_matrices ();
   void renormalize_sflx (std::vector<LA::MPI::Vector*> &target_sflxes);
   void initialize_aq (ParameterHandler &prm);
+  void initialize_assembly_related_objects
+  (DoFHandler<dim> &dof_handler,
+   FE_Poly<TensorProductPolynomials<dim>,dim,dim>* fe);
 
   double estimate_k (double &fiss_source,
                      double &fiss_source_prev_gen,