reader_cache.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    reader_cache.cc
 * @brief   
 */
 
#include "io/reader_cache.hh"
#include "io/msh_basereader.hh"
#include "io/msh_gmshreader.h"
#include "io/msh_vtkreader.hh"
#include "io/msh_pvdreader.hh"
#include "mesh/mesh.h"
#include "mesh/bc_mesh.hh"
#include "input/accessors.hh"
 
 
/***********************************************************************************************
 * Implementation of ReaderCache
 */
 
ReaderCache * ReaderCache::instance() {
    static ReaderCache *instance = new ReaderCache;
    return instance;
}
 
std::shared_ptr<BaseMeshReader> ReaderCache::get_reader(const FilePath &file_path) {
    return ReaderCache::get_reader_data(file_path)->second.reader_;
}
 
std::shared_ptr<Mesh> ReaderCache::get_mesh(const FilePath &file_path) {
    auto it = ReaderCache::get_reader_data(file_path);
    // Create and fill mesh if doesn't exist
    if ( (*it).second.mesh_ == nullptr ) {
        (*it).second.mesh_ = std::make_shared<Mesh>( Input::Record() );
        (*it).second.reader_->read_physical_names( (*it).second.mesh_.get() );
        (*it).second.reader_->read_raw_mesh( (*it).second.mesh_.get() );
        //(*it).second.reader_->check_compatible_mesh( *((*it).second.mesh_) );
 
    }
    return (*it).second.mesh_;
}
 
ReaderCache::ReaderTable::iterator ReaderCache::get_reader_data(const FilePath &file_path) {
    ReaderTable::iterator it = ReaderCache::instance()->reader_table_.find( string(file_path) );
    if (it == ReaderCache::instance()->reader_table_.end()) {
        ReaderData reader_data;
        if ( file_path.extension() == ".msh" || file_path.extension() == ".msh2") {
            reader_data.reader_ = std::make_shared<GmshMeshReader>(file_path);
        } else if ( file_path.extension() == ".vtu" ) {
            reader_data.reader_ = std::make_shared<VtkMeshReader>(file_path);
        } else if ( file_path.extension() == ".pvd" ) {
            reader_data.reader_ = std::make_shared<PvdMeshReader>(file_path);
        } else {
            THROW(BaseMeshReader::ExcWrongExtension()
                << BaseMeshReader::EI_FileExtension(file_path.extension()) << BaseMeshReader::EI_MeshFile((string)file_path) );
        }
        ReaderCache::instance()->reader_table_.insert( std::pair<string, ReaderData>(string(file_path), reader_data) );
        it = ReaderCache::instance()->reader_table_.find( string(file_path) );
    }
    return it;
}
 
void ReaderCache::get_element_ids(const FilePath &file_path, const Mesh &mesh) {
    auto reader_ptr = ReaderCache::get_reader(file_path);
    reader_ptr->set_element_ids(mesh);
}
 
 
void _set_identic_mesh_map(std::vector<int> &map, std::vector<int> ids, const MeshBase * mesh)
{
 
    // assume both bulk_elements_id and boundary_elements_id are sorted
    uint i = 0;
    for(int id : ids) {
        if ((uint)id == Mesh::undef_idx) return; // Boundary IDs may be undef.
        map[mesh->elem_index(id)] = i;
        i++;
    }
    ASSERT(i == map.size());
}
 
 
std::shared_ptr<EquivalentMeshMap> ReaderCache::identic_mesh_map(const FilePath &file_path,
                                                                            Mesh *computational_mesh) {
    //ASSERT_PERMANENT(false).error("Not implemented yet." );
    auto it = ReaderCache::get_reader_data(file_path);
    auto &reader_data = (*it).second;
    if ( reader_data.target_mesh_element_map_ == nullptr ) {
        // Create map for the identic mesh taking the computational mesh permutation into account.
        // Assume that element IDs in the source and computational mesh match.
        // map index of the sorted IDs to the index of the element in permuted computational mesh.
        // make for both bulk and boundary mesh.
 
        reader_data.reader_->set_element_ids(*computational_mesh);
        std::shared_ptr<EquivalentMeshMap> map_ptr =
            std::make_shared<EquivalentMeshMap>(computational_mesh->n_elements(),
                    computational_mesh->bc_mesh()->n_elements(), (LongIdx)undef_idx);
        _set_identic_mesh_map(map_ptr->bulk, reader_data.reader_->get_element_ids(false), computational_mesh);
        _set_identic_mesh_map(map_ptr->boundary, reader_data.reader_->get_element_ids(true), computational_mesh->bc_mesh());
 
        reader_data.target_mesh_element_map_ = map_ptr;
    }
    return reader_data.target_mesh_element_map_;
 
}
 
std::shared_ptr<EquivalentMeshMap> ReaderCache::eqivalent_mesh_map(const FilePath &file_path,
                                                                            Mesh *computational_mesh) {
    auto it = ReaderCache::get_reader_data(file_path);
    auto source_mesh = ReaderCache::get_mesh(file_path);
    auto &reader_data = (*it).second;
    //ASSERT_PTR( reader_data.mesh_ ).error("Mesh is not created. Did you call 'ReaderCache::get_mesh(file_path)'?\n");
    if ( reader_data.target_mesh_element_map_ == nullptr ) {
        reader_data.reader_->set_element_ids(*source_mesh);
        reader_data.target_mesh_element_map_ = computational_mesh->check_compatible_mesh( *((*it).second.mesh_.get()) );
    }
    return reader_data.target_mesh_element_map_;
}