meshavg.C

Go to the documentation of this file.

// The libMesh Finite Element Library.
// Copyright (C) 2002-2019 Benjamin S. Kirk, John W. Peterson, Roy H. Stogner
 
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
 
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
 
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
 
// Open the solution files named on standard input, take the average
// of their fields' values, and output to a new solution file.
#include "libmesh/libmesh.h"
#include "libmesh/equation_systems.h"
#include "libmesh/mesh.h"
#include "libmesh/namebased_io.h"
#include "libmesh/numeric_vector.h"
#include "libmesh/int_range.h"
 
using namespace libMesh;
 
unsigned char dim = 2; // This gets overridden by most mesh formats
 
int main(int argc, char ** argv)
{
  LibMeshInit init(argc, argv);
 
  Mesh mesh1(init.comm(), dim);
  EquationSystems es1(mesh1);
 
  libMesh::out << "Usage: " << argv[0]
               << " outputsolution mesh firstsolution [secondsolution ...]" << std::endl;
 
  mesh1.read(argv[2]);
  libMesh::out << "Loaded mesh " << argv[2] << std::endl;
 
  mesh1.print_info();
 
  es1.read(argv[3],
           EquationSystems::READ_HEADER |
           EquationSystems::READ_DATA |
           EquationSystems::READ_BASIC_ONLY);
  libMesh::out << "Loaded first solution " << argv[3] << std::endl;
 
  es1.print_info();
 
  std::vector<std::string> sysnames;
  std::vector<std::unique_ptr<NumericVector<libMesh::Number>>> summed_solutions;
 
  for (unsigned int s = 0; s != es1.n_systems(); ++s)
    {
      sysnames.push_back(es1.get_system(s).name());
      summed_solutions.emplace_back(es1.get_system(s).solution->clone());
    }
 
  for (int i=4; i < argc; ++i)
    {
      Mesh mesh2(init.comm(), dim);
      EquationSystems es2(mesh2);
      mesh2.read(argv[2]);
      es2.read(argv[i],
               EquationSystems::READ_HEADER |
               EquationSystems::READ_DATA |
               EquationSystems::READ_BASIC_ONLY);
      libMesh::out << "Loaded next solution " << argv[i] << std::endl;
 
      for (auto s : index_range(sysnames))
        {
          if (!es2.has_system(sysnames[s]))
            libmesh_error_msg("EquationSystems object does not have " << sysnames[s]);
 
          (*summed_solutions[s]) += *es2.get_system(s).solution;
        }
    }
 
  int n_solutions = argc - 3;
 
  for (auto s : index_range(sysnames))
    {
      (*summed_solutions[s]) /= n_solutions;
      es1.get_system(s).solution->swap(*summed_solutions[s]);
      es1.get_system(s).solution->close();
    }
 
  NameBasedIO(mesh1).write_equation_systems (argv[1], es1);
 
  return 0;
}