msh_basereader.cc

Go to the documentation of this file.

/*!
 *
 * Copyright (C) 2015 Technical University of Liberec.  All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License version 3 as published by the
 * Free Software Foundation. (http://www.gnu.org/licenses/gpl-3.0.en.html)
 *
 * This program 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.
 *
 *
 * @file    msh_basereader.cc
 * @brief
 * @author  dalibor
 */


#include "io/msh_basereader.hh"
#include "io/msh_gmshreader.h"
#include "io/msh_vtkreader.hh"
#include "io/msh_pvdreader.hh"
#include "system/sys_profiler.hh"


BaseMeshReader::BaseMeshReader(const FilePath &file_name)
: element_data_values_(std::make_shared<ElementDataFieldMap>()),
  tok_(file_name)
{}

BaseMeshReader::BaseMeshReader(const FilePath &file_name, std::shared_ptr<ElementDataFieldMap> element_data_values)
: element_data_values_(element_data_values),
  tok_(file_name)
{}

std::shared_ptr< BaseMeshReader > BaseMeshReader::reader_factory(const FilePath &file_name) {
    std::shared_ptr<BaseMeshReader> reader_ptr;
    if ( file_name.extension() == ".msh" ) {
        reader_ptr = std::make_shared<GmshMeshReader>(file_name);
    } else if ( file_name.extension() == ".vtu" ) {
        reader_ptr = std::make_shared<VtkMeshReader>(file_name);
    } else if ( file_name.extension() == ".pvd" ) {
        reader_ptr = std::make_shared<PvdMeshReader>(file_name);
    } else {
        THROW(ExcWrongExtension() << EI_FileExtension(file_name.extension()) << EI_MeshFile((string)file_name) );
    }
    return reader_ptr;
}

Mesh * BaseMeshReader::mesh_factory(const Input::Record &input_mesh_rec) {
    START_TIMER("BaseMeshReader - mesh factory");

    Input::Array region_list;
    Mesh * mesh = new Mesh( input_mesh_rec );

    try {
        std::shared_ptr< BaseMeshReader > reader = BaseMeshReader::reader_factory(input_mesh_rec.val<FilePath>("mesh_file"));
        reader->read_physical_names(mesh);
        if (input_mesh_rec.opt_val("regions", region_list)) {
            mesh->read_regions_from_input(region_list);
        }
        reader->read_raw_mesh(mesh);
    } INPUT_CATCH(FilePath::ExcFileOpen, FilePath::EI_Address_String, input_mesh_rec)
    mesh->setup_topology();
    mesh->check_and_finish();
    return mesh;

}

void BaseMeshReader::read_raw_mesh(Mesh * mesh) {
    ASSERT(mesh).error("Argument mesh is NULL.\n");
    tok_.set_position( Tokenizer::Position() );
    read_nodes(mesh);
    read_elements(mesh);
}


std::vector<int> const & BaseMeshReader::get_element_vector(bool boundary_domain) {
    if (boundary_domain) return boundary_elements_id_;
    else return bulk_elements_id_;
}


template<typename T>
typename ElementDataCache<T>::ComponentDataPtr BaseMeshReader::get_element_data( unsigned int n_entities, unsigned int n_components,
        bool boundary_domain, unsigned int component_idx) {
    ASSERT(has_compatible_mesh_)
            .error("Vector of mapping VTK to GMSH element is not initialized. Did you call check_compatible_mesh?");
    ASSERT(actual_header_.field_name != "").error("Unset MeshDataHeader. Did you call find_header?\n");

    std::string field_name = actual_header_.field_name;

    ElementDataFieldMap::iterator it=element_data_values_->find(field_name);
    if (it == element_data_values_->end()) {
        (*element_data_values_)[field_name] = std::make_shared< ElementDataCache<T> >();
        it=element_data_values_->find(field_name);
    }

    if ( !it->second->is_actual(actual_header_.time, field_name) ) {
        unsigned int size_of_cache; // count of vectors stored in cache

        // check that the header is valid, try to correct
        if (actual_header_.n_entities != n_entities) {
            WarningOut().fmt("In file '{}', '{}' section for field '{}', time: {}.\nWrong number of entities: {}, using {} instead.\n",
                    tok_.f_name(), data_section_name_, field_name, actual_header_.time, actual_header_.n_entities, n_entities);
            // actual_header.n_entities=n_entities;
        }

        if (n_components == 1) {
            // read for MultiField to 'n_comp' vectors
            // or for Field if ElementData contains only one value
            size_of_cache = actual_header_.n_components;
        }
        else {
            // read for Field if more values is stored to one vector
            size_of_cache = 1;
            if (actual_header_.n_components != n_components) {
                WarningOut().fmt("In file '{}', '{}' section for field '{}', time: {}.\nWrong number of components: {}, using {} instead.\n",
                        tok_.f_name(), data_section_name_, field_name, actual_header_.time, actual_header_.n_components, n_components);
                actual_header_.n_components=n_components;
            }
        }

        (*element_data_values_)[field_name]
                    = std::make_shared< ElementDataCache<T> >(field_name, actual_header_.time, size_of_cache, n_components*n_entities);
        this->read_element_data(*(it->second), actual_header_, n_components, boundary_domain );
    }

    actual_header_.reset();

    if (component_idx == std::numeric_limits<unsigned int>::max()) component_idx = 0;
    ElementDataCache<T> &current_cache = dynamic_cast<ElementDataCache<T> &>(*(it->second));
    return current_cache.get_component_data(component_idx);
}


// explicit instantiation of template methods
#define MESH_READER_GET_ELEMENT_DATA(TYPE) \
template typename ElementDataCache<TYPE>::ComponentDataPtr BaseMeshReader::get_element_data<TYPE>(unsigned int n_entities, \
        unsigned int n_components, bool boundary_domain, unsigned int component_idx);

MESH_READER_GET_ELEMENT_DATA(int);
MESH_READER_GET_ELEMENT_DATA(unsigned int);
MESH_READER_GET_ELEMENT_DATA(double);