VTKField.cpp

/*---------------------------------------------------------------------------*\
 *
 *  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/>.
 *
\*---------------------------------------------------------------------------*/
 
#include "VTK.hpp"
 
namespace bitpit{
 
unsigned VTKField::getComponentCount(VTKFieldType fieldType) {
 
    switch (fieldType) {
 
    case VTKFieldType::SCALAR:
        return 1;
 
    case VTKFieldType::VECTOR:
        return 3;
 
    case VTKFieldType::TENSOR:
        return 9;
 
    default:
        throw std::runtime_error("Unable to identify the number of components of field type " + std::to_string(static_cast<int>(fieldType)));
 
    }
}
 
 
VTKField::VTKField(){
 
    m_name            = "undefined" ;  
    m_dataType        = VTKDataType::UNDEFINED ;
    m_location        = VTKLocation::UNDEFINED ;
    m_codification    = VTKFormat::UNDEFINED ;
    m_offset          = 0 ;
    m_fieldType       = VTKFieldType::UNDEFINED ;
    m_position        = 0 ;
    m_streamer        = nullptr ;
    m_enabled         = true ;
 
}
 
VTKField::VTKField( const std::string &name ): VTKField() {
    setName(name) ;
}
 
void VTKField::setName( const std::string &name){
    m_name= name; 
}
 
void VTKField::setDataType( VTKDataType type){
    m_dataType= type; 
}
 
void VTKField::setLocation( VTKLocation loc ){ 
    m_location= loc; 
}
 
void VTKField::setCodification( VTKFormat cod ){ 
    m_codification= cod; 
}
 
void VTKField::setFieldType( VTKFieldType type ){ 
    m_fieldType= type; 
}
 
void VTKField::setPosition( std::fstream::pos_type pos ){ 
    m_position =pos ; 
}
 
void VTKField::setOffset( uint64_t off){ 
    m_offset= off; 
}
 
void VTKField::setStreamer(VTKBaseStreamer& streamer ){ 
    m_streamer= &streamer; 
}
 
void VTKField::enable(){
    m_enabled=true;
}
 
void VTKField::disable(){
    m_enabled=false;
}
 
const std::string & VTKField::getName() const{
    return m_name; 
}
 
VTKFieldType VTKField::getFieldType() const{ 
    return m_fieldType; 
}
 
unsigned VTKField::getComponentCount() const{
 
    return getComponentCount(getFieldType());
}
 
VTKDataType VTKField::getDataType() const{ 
    return m_dataType; 
}
 
VTKLocation VTKField::getLocation() const{ 
    return m_location; 
}
 
VTKFormat VTKField::getCodification() const{ 
    return m_codification; 
}
 
uint64_t VTKField::getOffset() const{ 
    return m_offset; 
}
 
std::fstream::pos_type   VTKField::getPosition() const{ 
    return m_position; 
}
 
const VTKBaseStreamer & VTKField::getStreamer() const {
    return *m_streamer;
}
 
bool VTKField::isEnabled() const{ 
    return m_enabled; 
}
 
bool VTKField::hasAllMetaData() const{ 
 
    bool allData(true);
 
    allData &= m_name != "undefined" ;
    allData &= m_dataType != VTKDataType::UNDEFINED ;
    allData &= m_location != VTKLocation::UNDEFINED ;
    allData &= m_codification != VTKFormat::UNDEFINED ;
    allData &= m_fieldType != VTKFieldType::UNDEFINED ;
    allData &= m_streamer != nullptr ;
 
    return allData;
 
}
 
void  VTKField::write( std::fstream &str) const{ 
    if(m_streamer){
        m_streamer->flushData( str, m_name, m_codification) ;
    }
}
 
void VTKField::read( std::fstream &str, uint64_t entries, uint8_t components ) const{ 
    if(m_streamer){
        m_streamer->absorbData( str, m_name, m_codification, entries, components, m_dataType) ;
    }
}
 
}