Add one more testcase.

git-svn-id: https://svn.dealii.org/trunk@18340 0785d39b-7218-0410-832d-ea1e28bc413d

diff --git a/tests/bits/count_dofs_per_block_02.cc b/tests/bits/count_dofs_per_block_02.cc
new file mode 100644 (file)
index 0000000..eadcbeb
--- /dev/null
+++ b/tests/bits/count_dofs_per_block_02.cc
@@ -0,0 +1,141 @@
+//----------------------------  count_dofs_per_block_02.cc  ---------------------------
+//    $Id$
+//    Version: $Name$ 
+//
+//    Copyright (C) 2009 by the deal.II authors
+//
+//    This file is subject to QPL and may not be  distributed
+//    without copyright and license information. Please refer
+//    to the file deal.II/doc/license.html for the  text  and
+//    further information on this license.
+//
+//----------------------------  count_dofs_per_block_02.cc  ---------------------------
+
+// like the _01 testcase, but with a non-primitive element that actually
+// defines blocks of non-unit size
+
+
+#include "../tests.h"
+#include <base/logstream.h>
+#include <grid/tria.h>
+#include <grid/tria_iterator.h>
+#include <grid/grid_generator.h>
+#include <dofs/dof_accessor.h>
+#include <dofs/dof_handler.h>
+#include <dofs/dof_tools.h>
+#include <fe/fe_q.h>
+#include <fe/fe_dgq.h>
+#include <fe/fe_dgp.h>
+#include <fe/fe_nedelec.h>
+#include <fe/fe_raviart_thomas.h>
+#include <fe/fe_system.h>
+
+#include <fstream>
+#include <iomanip>
+#include <iomanip>
+#include <string>
+
+
+std::string output_file_name = "count_dofs_per_block_02/output";
+
+
+
+void print (const std::vector<unsigned int> &v)
+{  
+  deallog << v.size();
+  for (unsigned int i=0; i<v.size(); ++i)
+    deallog << ' ' << v[i];
+  deallog << std::endl;
+}
+
+  
+
+template <int dim>
+void
+check ()
+{
+                                   // create tria and dofhandler
+                                   // objects. set different boundary
+                                   // and sub-domain ids
+  Triangulation<dim> tria;
+  GridGenerator::hyper_cube(tria, 0., 1.);
+  tria.refine_global (1);
+  for (int i=0; i<2; ++i)
+    {
+      tria.begin_active()->set_refine_flag();
+      tria.execute_coarsening_and_refinement ();
+    }
+
+  FESystem<dim> fe (FE_RaviartThomas<dim>(0), 1, FE_DGQ<dim>(0), 1);
+  DoFHandler<dim> dof_handler (tria);
+  dof_handler.distribute_dofs (fe);
+
+                                  // no grouping
+  {
+    std::vector<unsigned int> dpc(dim+1);
+    DoFTools::count_dofs_per_component (dof_handler, dpc);
+    print (dpc);
+  }
+
+  {
+    std::vector<unsigned int> dpc(2);
+    DoFTools::count_dofs_per_block (dof_handler, dpc);
+    print (dpc);
+  }
+
+
+                                  // grouping into less groups than
+                                  // components
+  {
+    std::vector<unsigned int> group(dim+1, 0);
+    group[dim] = 1;
+    std::vector<unsigned int> dpc(2);
+    DoFTools::count_dofs_per_component (dof_handler, dpc, false, group);
+    Assert (dpc.size() == 2, ExcInternalError());
+    print (dpc);
+  }
+  
+  {
+    std::vector<unsigned int> group(2, 0);
+    group[1] = 1;
+    std::vector<unsigned int> dpc(2);
+    DoFTools::count_dofs_per_block (dof_handler, dpc, group);
+    Assert (dpc.size() == 2, ExcInternalError());
+    print (dpc);
+  }  
+
+                                  // grouping into more groups than
+                                  // components
+  {
+    std::vector<unsigned int> group(dim+1, 2*dim);
+    group[dim] = 0;
+    std::vector<unsigned int> dpc(2*dim+1);
+    DoFTools::count_dofs_per_component (dof_handler, dpc, false, group);
+    Assert (dpc.size() == 2*dim+1, ExcInternalError());
+    print (dpc);
+  }
+  
+  {
+    std::vector<unsigned int> group(2, 2*dim);
+    group[1] = 0;
+    std::vector<unsigned int> dpc(2*dim+1);
+    DoFTools::count_dofs_per_block (dof_handler, dpc, group);
+    Assert (dpc.size() == 2*dim+1, ExcInternalError());
+    print (dpc);
+  }  
+}
+
+
+
+int main()
+{
+  std::ofstream logfile(output_file_name.c_str());
+  logfile << std::setprecision (2);
+  deallog << std::setprecision (2);
+  deallog.attach(logfile);
+  deallog.depth_console(0);
+  deallog.threshold_double(1.e-10);
+
+  check<2> ();
+  check<3> ();
+}

diff --git a/tests/bits/count_dofs_per_block_02/cmp/generic b/tests/bits/count_dofs_per_block_02/cmp/generic
new file mode 100644 (file)
index 0000000..32daed2
--- /dev/null
+++ b/tests/bits/count_dofs_per_block_02/cmp/generic
@@ -0,0 +1,13 @@
+
+DEAL::3 29 29 10
+DEAL::2 29 10
+DEAL::2 58 10
+DEAL::2 29 10
+DEAL::5 10 0 0 0 58
+DEAL::5 10 0 0 0 29
+DEAL::4 97 97 97 22
+DEAL::2 97 22
+DEAL::2 291 22
+DEAL::2 97 22
+DEAL::7 22 0 0 0 0 0 291
+DEAL::7 22 0 0 0 0 0 97