add test

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

diff --git a/tests/mpi/blockvec_01.cc b/tests/mpi/blockvec_01.cc
new file mode 100644 (file)
index 0000000..9e24142
--- /dev/null
+++ b/tests/mpi/blockvec_01.cc
@@ -0,0 +1,123 @@
+//---------------------------------------------------------------------------
+//    $Id: ghost_01.cc 28403 2013-02-15 05:23:40Z heister $
+//    Version: $Name$ 
+//
+//    Copyright (C) 2004, 2005, 2010 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.
+//
+//---------------------------------------------------------------------------
+
+
+// Test constructor/reinit of BlockVector with IndexSets
+
+#include "../tests.h"
+#include "coarse_grid_common.h"
+#include <deal.II/lac/petsc_parallel_block_vector.h>
+#include <deal.II/lac/trilinos_parallel_block_vector.h>
+#include <deal.II/base/index_set.h>
+#include <fstream>
+#include <iostream>
+#include <iomanip>
+#include <vector>
+
+
+template <class BLOCKVEC>
+void test ()
+{
+  unsigned int myid = Utilities::MPI::this_mpi_process (MPI_COMM_WORLD);
+  unsigned int numproc = Utilities::MPI::n_mpi_processes (MPI_COMM_WORLD);
+  
+  if (myid==0)
+    deallog << "numproc=" << numproc << std::endl;
+
+  std::vector<IndexSet> local_active(2);
+
+                                  // block 0:
+  local_active[0].set_size(numproc);
+  local_active[0].add_range(myid,myid+1);
+
+  local_active[1].set_size(2*numproc);
+  local_active[1].add_range(myid*2,myid*2+2);
+
+  BLOCKVEC v(local_active, MPI_COMM_WORLD);
+
+  v(myid) = 100.0 + myid;
+
+  v.block(1)(myid*2)=myid*2.0;
+  v.block(1)(myid*2+1)=myid*2.0+1.0;
+
+  v.compress(VectorOperation::insert);
+
+                                  //Assert(!v.has_ghost_elements(), ExcInternalError());
+
+  deallog << "size: " << v.size() << std::endl;
+  deallog << "size[0]: " << v.block(0).size() << std::endl;
+  deallog << "size[1]: " << v.block(1).size() << std::endl;  
+  
+  {
+    std::ofstream file((JobIdentifier::base_name(__FILE__)+"/ncpu_" + Utilities::int_to_string(Utilities::MPI::n_mpi_processes (MPI_COMM_WORLD)) + "/dat." + Utilities::int_to_string(myid)).c_str());
+    
+    file << "**** proc " << myid << std::endl;
+    v.print(file);
+  }
+
+  MPI_Barrier(MPI_COMM_WORLD);
+
+  if (myid==0)
+    {
+      for (unsigned int i=0;i<Utilities::MPI::n_mpi_processes (MPI_COMM_WORLD);++i)
+        {
+          cat_file((JobIdentifier::base_name(__FILE__)+"/ncpu_" + Utilities::int_to_string(Utilities::MPI::n_mpi_processes (MPI_COMM_WORLD)) + "/dat." + Utilities::int_to_string(i)).c_str());
+        }
+    } 
+
+                                  // now test the constructor
+
+  IndexSet id(7);
+  if (myid==0)
+    id.add_range(0,7);
+  local_active.push_back(id);
+  
+  BLOCKVEC xy(local_active);
+  
+  deallog << "size: " << xy.size() << std::endl;
+  deallog << "size[0]: " << xy.block(0).size() << std::endl;
+  deallog << "size[1]: " << xy.block(1).size() << std::endl;  
+  deallog << "size[2]: " << xy.block(2).size() << std::endl;
+  
+                                  // done
+  if (myid==0)
+    deallog << "OK" << std::endl;
+}
+
+
+
+int main (int argc, char **argv)
+{
+  Utilities::MPI::MPI_InitFinalize mpi_initialization(argc, argv);
+  unsigned int myid = Utilities::MPI::this_mpi_process (MPI_COMM_WORLD);
+
+  deallog.push(Utilities::int_to_string(myid));
+
+  if (myid == 0)
+    {
+      std::ofstream logfile(output_file_for_mpi("blockvec_01").c_str());
+      deallog.attach(logfile);
+      deallog << std::setprecision(4);
+      deallog.depth_console(0);
+      deallog.threshold_double(1.e-10);
+
+      test<PETScWrappers::MPI::BlockVector>();
+                                      //test<TrilinosWrappers::MPI::BlockVector>();
+    }
+  else
+    {
+      test<PETScWrappers::MPI::BlockVector>();
+                                      //test<TrilinosWrappers::MPI::BlockVector>();
+    }
+  
+}