VTKStreamer.tpp

/*---------------------------------------------------------------------------*\
 *
 *  bitpit
 *
 *  Copyright (C) 2015-2021 OPTIMAD engineering Srl
 *
 *  -------------------------------------------------------------------------
 *  License
 *  This file is part of bitpit.
 *
 *  bitpit is free software: you can redistribute it and/or modify it
 *  under the terms of the GNU Lesser General Public License v3 (LGPL)
 *  as published by the Free Software Foundation.
 *
 *  bitpit 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 bitpit. If not, see <http://www.gnu.org/licenses/>.
 *
\*---------------------------------------------------------------------------*/
 
namespace bitpit{


template<class T>
VTKVectorContainer<T> * VTKVectorContainer<T>::clone() const {
    return new VTKVectorContainer<T>( *this );
}

template<class T>
VTKVectorContainer<T>::VTKVectorContainer( std::vector<T> &data){
    m_ptr = &data;
}

template<class T>
void VTKBaseStreamer::flushValue( std::fstream &str, VTKFormat format, const T &value) const {
 
    if( format==VTKFormat::ASCII){
        genericIO::flushASCII( str, value) ;
 
    } else if( format==VTKFormat::APPENDED) {
        genericIO::flushBINARY( str, value) ;
 
    }
 
}

template<class T>
void VTKBaseStreamer::flushValue( std::fstream &str, VTKFormat format, const T *values, int nValues) const {
 
    if( format==VTKFormat::ASCII){
        genericIO::flushASCII( str, values, nValues) ;
 
    } else if( format==VTKFormat::APPENDED) {
        genericIO::flushBINARY( str, values, nValues) ;
 
    } else {
        throw std::runtime_error("VTK format not supported.") ;
 
    }
 
}

template<class T>
void VTKVectorContainer<T>::flushData( std::fstream &str, VTKFormat format){
 
    if( format==VTKFormat::ASCII){
        genericIO::flushASCII( str, 8, *m_ptr) ;
 
    } else if( format==VTKFormat::APPENDED) {
        genericIO::flushBINARY( str, *m_ptr) ;
 
    } else {
        throw std::runtime_error("VTK format not supported.") ;
 
    }
 
}

template<class T>
void VTKVectorContainer<T>::absorbData( std::fstream &str, VTKFormat format, uint64_t entries, uint8_t components){
 
    resize( std::is_fundamental<T>{}, entries, components) ;
 
    if( format==VTKFormat::ASCII){
        genericIO::absorbASCII( str, *m_ptr) ;
 
    } else if( format==VTKFormat::APPENDED) {
        genericIO::absorbBINARY( str, *m_ptr) ;
 
    }
 
}

template<class T>
void VTKVectorContainer<T>::resize( std::true_type, uint64_t entries, uint8_t components){
 
    BITPIT_UNUSED(components) ;
    m_ptr->resize(entries) ;
 
}

template<class T >
void VTKVectorContainer<T>::resize( std::false_type, uint64_t entries, uint8_t components){
 
    uint64_t elements = entries /components ;
    m_ptr->resize(elements) ;
 
    for( auto & element : (*m_ptr) ){
        vtk::allocate( element, components) ;
    }
 
}

template<class T>
void VTKNativeStreamer::addData( const std::string &name, std::vector<T> &data ){
 
    auto fieldItr = m_field.find(name) ;
 
    if( fieldItr!=m_field.end() ){
        m_field.erase(fieldItr) ;
    }
    
    std::unique_ptr<VTKBaseContainer> temp = std::unique_ptr<VTKBaseContainer>( new VTKVectorContainer<T>(data) ) ;
    m_field.emplace( name, std::move(temp) ) ;
 
}
 
}