mirror of
https://github.com/OPM/ResInsight.git
synced 2025-01-21 14:03:18 -06:00
138 lines
4.9 KiB
C++
138 lines
4.9 KiB
C++
/////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Copyright (C) 2018- Equinor ASA
|
|
//
|
|
// ResInsight 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 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// ResInsight 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 at <http://www.gnu.org/licenses/gpl.html>
|
|
// for more details.
|
|
//
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "VdeFileExporter.h"
|
|
#include "VdeArrayDataPacket.h"
|
|
#include "VdeCachingHashedIdFactory.h"
|
|
#include "VdePacketDirectory.h"
|
|
|
|
#include "cvfTrace.h"
|
|
|
|
#include <QDir>
|
|
#include <QFile>
|
|
|
|
//==================================================================================================
|
|
//
|
|
//
|
|
//
|
|
//==================================================================================================
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
VdeFileExporter::VdeFileExporter( QString absOutputFolder )
|
|
: m_absOutputFolder( absOutputFolder )
|
|
{
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
bool VdeFileExporter::exportToFile( const QString& modelMetaJsonStr,
|
|
const VdePacketDirectory& packetDirectory,
|
|
const std::vector<int>& packetIdsToExport )
|
|
{
|
|
cvf::Trace::show( "Exporting to folder: %s", m_absOutputFolder.toLatin1().constData() );
|
|
|
|
QString jsonFileName = m_absOutputFolder + "/modelMeta.json";
|
|
if ( !writeModelMetaJsonFile( modelMetaJsonStr, jsonFileName ) )
|
|
{
|
|
cvf::Trace::show( "Error writing: %s", jsonFileName.toLatin1().constData() );
|
|
return false;
|
|
}
|
|
|
|
for ( const int packetArrayId : packetIdsToExport )
|
|
{
|
|
const VdeArrayDataPacket* dataPacket = packetDirectory.lookupPacket( packetArrayId );
|
|
if ( !dataPacket )
|
|
{
|
|
cvf::Trace::show( "Error during export, no data for arrayId %d", packetArrayId );
|
|
return false;
|
|
}
|
|
|
|
CVF_ASSERT( packetArrayId == dataPacket->arrayId() );
|
|
if ( !writeDataPacketToFile( dataPacket->arrayId(), *dataPacket ) )
|
|
{
|
|
cvf::Trace::show( "Error writing packet data to file, arrayId %d", packetArrayId );
|
|
return false;
|
|
}
|
|
}
|
|
|
|
cvf::Trace::show( "Data exported (%d packets) to folder: %s", packetIdsToExport.size(), m_absOutputFolder.toLatin1().constData() );
|
|
|
|
return true;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
bool VdeFileExporter::exportViewContents( const RimGridView& view )
|
|
{
|
|
VdeCachingHashedIdFactory localIdFactory;
|
|
VdeVizDataExtractor extractor( view, &localIdFactory );
|
|
|
|
QString modelMetaJsonStr;
|
|
std::vector<int> allReferencedArrayIds;
|
|
VdePacketDirectory packetDirectory;
|
|
extractor.extractViewContents( &modelMetaJsonStr, &allReferencedArrayIds, &packetDirectory );
|
|
|
|
return exportToFile( modelMetaJsonStr, packetDirectory, allReferencedArrayIds );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
bool VdeFileExporter::writeDataPacketToFile( int arrayId, const VdeArrayDataPacket& packet ) const
|
|
{
|
|
const QString fileName = m_absOutputFolder + QString( "/arrayData_%1.bin" ).arg( arrayId );
|
|
|
|
QFile file( fileName );
|
|
if ( !file.open( QIODevice::WriteOnly ) )
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if ( file.write( packet.fullPacketRawPtr(), packet.fullPacketSize() ) == -1 )
|
|
{
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
bool VdeFileExporter::writeModelMetaJsonFile( const QString& modelMetaJsonStr, QString fileName )
|
|
{
|
|
const QByteArray jsonByteArr = modelMetaJsonStr.toLatin1();
|
|
|
|
QFile file( fileName );
|
|
if ( !file.open( QIODevice::WriteOnly | QIODevice::Text ) )
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if ( file.write( jsonByteArr ) == -1 )
|
|
{
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|