particles.h

Go to the documentation of this file.

/*
 Copyright (C) 2011 - 2024 by the authors of the ASPECT code.

 This file is part of ASPECT.

 ASPECT is free software; you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation; either version 2, or (at your option)
 any later version.

 ASPECT 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 General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with ASPECT; see the file LICENSE.  If not see
 <http://www.gnu.org/licenses/>.
 */

#ifndef _aspect_postprocess_particle_h
#define _aspect_postprocess_particle_h

#include <aspect/postprocess/interface.h>

#include <aspect/simulator_access.h>
#include <aspect/particle/property/interface.h>

#include <deal.II/particles/particle_handler.h>
#include <deal.II/base/data_out_base.h>

#include <thread>
#include <tuple>


namespace aspect
{
  namespace Postprocess
  {
    namespace internal
    {
      template <int dim>
      class ParticleOutput : public ::DataOutInterface<0,dim>
      {
        public:
          void build_patches(const Particles::ParticleHandler<dim> &particle_handler,
                             const aspect::Particle::Property::ParticlePropertyInformation &property_information,
                             const std::vector<std::string> &exclude_output_properties,
                             const bool only_group_3d_vectors);

        private:
          const std::vector<DataOutBase::Patch<0,dim>> &
          get_patches () const override;

          std::vector<std::string>
          get_dataset_names () const override;

          std::vector<
          std::tuple<unsigned int,
              unsigned int,
              std::string,
              DataComponentInterpretation::DataComponentInterpretation>>
              get_nonscalar_data_ranges () const override;

          std::vector<DataOutBase::Patch<0,dim>> patches;

          std::vector<std::string> dataset_names;

          std::vector<
          std::tuple<unsigned int,
              unsigned int,
              std::string,
              DataComponentInterpretation::DataComponentInterpretation>>
              vector_datasets;
      };
    }

    template <int dim>
    class Particles : public Interface<dim>, public ::aspect::SimulatorAccess<dim>
    {
      public:
        Particles();

        ~Particles() override;

        std::pair<std::string,std::string> execute (TableHandler &statistics) override;

        void save (std::map<std::string, std::string> &status_strings) const override;

        void load (const std::map<std::string, std::string> &status_strings) override;

        template <class Archive>
        void serialize (Archive &ar, const unsigned int version);

        static
        void
        declare_parameters (ParameterHandler &prm);

        void
        parse_parameters (ParameterHandler &prm) override;

      private:
        double output_interval;

        double last_output_time;

        void set_last_output_time (const double current_time);

        unsigned int output_file_number;

        std::vector<std::string> output_formats;

        std::map<std::string,std::vector<std::pair<double,std::string>>> times_and_pvtu_file_names;

        std::map<std::string,std::vector<std::vector<std::string>>> output_file_names_by_timestep;

        std::map<std::string,std::vector<XDMFEntry>>  xdmf_entries;

        unsigned int group_files;

        std::string temporary_output_location;

        bool write_in_background_thread;

        std::thread background_thread;

        std::vector<std::string> exclude_output_properties;

        static
        void writer (const std::string &filename,
                     const std::string &temporary_filename,
                     const std::string &file_contents);

        void write_description_files (const internal::ParticleOutput<dim> &data_out,
                                      const std::string &description_file_prefix,
                                      const std::string &solution_file_directory,
                                      const std::string &solution_file_prefix,
                                      const std::vector<std::string> &filenames);
    };
  }
}

#endif