2010-03-11 15:47:40 +00:00
|
|
|
/*
|
|
|
|
* Copyright (C) 2010 Thorsten Liebig (Thorsten.Liebig@gmx.de)
|
|
|
|
*
|
|
|
|
* This program 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.
|
|
|
|
*
|
|
|
|
* 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.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
|
2010-04-12 07:38:24 +00:00
|
|
|
#include <iomanip>
|
2010-04-28 09:15:59 +00:00
|
|
|
#include <H5Cpp.h>
|
2010-07-20 09:42:47 +00:00
|
|
|
#include "tools/global.h"
|
|
|
|
#include "processfields.h"
|
2010-12-06 09:44:25 +00:00
|
|
|
#include "FDTD/engine_interface_fdtd.h"
|
2010-04-05 18:22:03 +00:00
|
|
|
|
2010-12-07 15:47:23 +00:00
|
|
|
ProcessFields::ProcessFields(Engine_Interface_Base* eng_if) : Processing(eng_if)
|
2010-03-02 13:54:50 +00:00
|
|
|
{
|
2010-04-05 18:22:03 +00:00
|
|
|
m_DumpType = E_FIELD_DUMP;
|
|
|
|
// vtk-file is default
|
|
|
|
m_fileType = VTK_FILETYPE;
|
2010-04-07 10:57:45 +00:00
|
|
|
SetSubSampling(1);
|
2010-04-12 07:38:24 +00:00
|
|
|
SetPrecision(6);
|
2010-12-19 19:41:08 +00:00
|
|
|
m_dualTime = false;
|
2010-03-02 13:54:50 +00:00
|
|
|
|
2010-12-06 12:04:37 +00:00
|
|
|
for (int n=0; n<3; ++n)
|
2010-03-02 13:54:50 +00:00
|
|
|
{
|
2010-03-15 15:59:37 +00:00
|
|
|
numLines[n]=0;
|
2010-03-09 20:35:57 +00:00
|
|
|
discLines[n]=NULL;
|
2010-03-02 13:54:50 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
ProcessFields::~ProcessFields()
|
|
|
|
{
|
2010-12-06 12:04:37 +00:00
|
|
|
for (int n=0; n<3; ++n)
|
2010-03-02 13:54:50 +00:00
|
|
|
{
|
2010-03-09 20:35:57 +00:00
|
|
|
delete[] discLines[n];
|
|
|
|
discLines[n]=NULL;
|
2010-03-02 13:54:50 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2010-04-05 18:22:03 +00:00
|
|
|
string ProcessFields::GetFieldNameByType(DumpType type)
|
|
|
|
{
|
|
|
|
switch (type)
|
|
|
|
{
|
2010-12-06 12:04:37 +00:00
|
|
|
case E_FIELD_DUMP:
|
|
|
|
return "E-Field";
|
|
|
|
case H_FIELD_DUMP:
|
|
|
|
return "H-Field";
|
2010-04-05 18:22:03 +00:00
|
|
|
}
|
|
|
|
return "unknown field";
|
|
|
|
}
|
|
|
|
|
2010-03-02 13:54:50 +00:00
|
|
|
void ProcessFields::DefineStartStopCoord(double* dstart, double* dstop)
|
|
|
|
{
|
2010-04-07 10:57:45 +00:00
|
|
|
vector<double> lines;
|
2010-07-20 14:30:41 +00:00
|
|
|
|
|
|
|
// determine mesh type
|
|
|
|
bool dualMesh = false;
|
|
|
|
if (m_DumpType == H_FIELD_DUMP)
|
|
|
|
dualMesh = true;
|
|
|
|
|
2010-12-02 12:51:34 +00:00
|
|
|
Engine_Interface_Base::InterpolationType m_DumpMode = m_Eng_Interface->GetInterpolationType();
|
|
|
|
if (m_DumpMode==Engine_Interface_Base::NO_INTERPOLATION)
|
2010-03-02 13:54:50 +00:00
|
|
|
{
|
2010-07-20 14:30:41 +00:00
|
|
|
if (!Op->SnapToMesh(dstart,start,dualMesh))
|
|
|
|
cerr << "ProcessFields::DefineStartStopCoord: Warning: Snapping problem, check start value!!" << endl;
|
|
|
|
if (!Op->SnapToMesh(dstop,stop,dualMesh))
|
|
|
|
cerr << "ProcessFields::DefineStartStopCoord: Warning: Snapping problem, check stop value!!" << endl;
|
|
|
|
|
2010-12-06 12:04:37 +00:00
|
|
|
for (int n=0; n<3; ++n)
|
2010-03-02 13:54:50 +00:00
|
|
|
{
|
2010-07-20 14:30:41 +00:00
|
|
|
// normalize order of start and stop
|
2010-03-09 20:35:57 +00:00
|
|
|
if (start[n]>stop[n])
|
|
|
|
{
|
|
|
|
unsigned int help = start[n];
|
|
|
|
start[n]=stop[n];
|
|
|
|
stop[n]=help;
|
|
|
|
}
|
2010-07-20 14:30:41 +00:00
|
|
|
|
|
|
|
// construct new discLines
|
2010-04-07 10:57:45 +00:00
|
|
|
lines.clear();
|
2010-12-06 12:04:37 +00:00
|
|
|
for (unsigned int i=start[n]; i<=stop[n]; i+=subSample[n])
|
2010-04-07 10:57:45 +00:00
|
|
|
{
|
2010-07-20 14:30:41 +00:00
|
|
|
lines.push_back(Op->GetDiscLine(n,i,dualMesh));
|
2010-04-07 10:57:45 +00:00
|
|
|
}
|
|
|
|
numLines[n] = lines.size();
|
2010-03-09 20:35:57 +00:00
|
|
|
delete[] discLines[n];
|
|
|
|
discLines[n] = new double[numLines[n]];
|
2010-12-06 12:04:37 +00:00
|
|
|
for (unsigned int i=0; i<numLines[n]; ++i)
|
2010-04-07 10:57:45 +00:00
|
|
|
discLines[n][i] = lines.at(i);
|
2010-03-02 13:54:50 +00:00
|
|
|
}
|
2010-03-09 20:35:57 +00:00
|
|
|
}
|
2010-12-02 12:51:34 +00:00
|
|
|
else if (m_DumpMode==Engine_Interface_Base::NODE_INTERPOLATE)
|
2010-05-11 18:38:58 +00:00
|
|
|
{
|
|
|
|
if (Op->SnapToMesh(dstart,start)==false) cerr << "ProcessFields::DefineStartStopCoord: Warning: Snapping problem, check start value!!" << endl;
|
|
|
|
if (Op->SnapToMesh(dstop,stop)==false) cerr << "ProcessFields::DefineStartStopCoord: Warning: Snapping problem, check stop value!!" << endl;
|
|
|
|
|
|
|
|
//create mesh
|
2010-12-06 12:04:37 +00:00
|
|
|
for (int n=0; n<3; ++n)
|
2010-05-11 18:38:58 +00:00
|
|
|
{
|
|
|
|
if (start[n]>stop[n])
|
|
|
|
{
|
|
|
|
unsigned int help = start[n];
|
|
|
|
start[n]=stop[n];
|
|
|
|
stop[n]=help;
|
|
|
|
}
|
2010-12-02 12:51:34 +00:00
|
|
|
// if (stop[n]==Op->GetNumberOfLines(n)-1)
|
|
|
|
// --stop[n];
|
2010-05-11 18:38:58 +00:00
|
|
|
// cerr << "start " << start[n] << "stop " << stop[n];
|
|
|
|
lines.clear();
|
2010-12-06 12:04:37 +00:00
|
|
|
for (unsigned int i=start[n]; i<=stop[n]; i+=subSample[n])
|
2010-05-11 18:38:58 +00:00
|
|
|
{
|
|
|
|
lines.push_back(Op->GetDiscLine(n,i));//0.5*(Op->discLines[n][i+1] + Op->discLines[n][i]));
|
|
|
|
}
|
|
|
|
numLines[n] = lines.size();
|
|
|
|
delete[] discLines[n];
|
|
|
|
discLines[n] = new double[numLines[n]];
|
2010-12-06 12:04:37 +00:00
|
|
|
for (unsigned int i=0; i<numLines[n]; ++i)
|
2010-05-11 18:38:58 +00:00
|
|
|
discLines[n][i] = lines.at(i);
|
|
|
|
}
|
|
|
|
}
|
2010-12-02 12:51:34 +00:00
|
|
|
else if (m_DumpMode==Engine_Interface_Base::CELL_INTERPOLATE)
|
2010-03-09 20:35:57 +00:00
|
|
|
{
|
|
|
|
if (Op->SnapToMesh(dstart,start,true)==false) cerr << "ProcessFields::DefineStartStopCoord: Warning: Snapping problem, check start value!!" << endl;
|
|
|
|
if (Op->SnapToMesh(dstop,stop,true)==false) cerr << "ProcessFields::DefineStartStopCoord: Warning: Snapping problem, check stop value!!" << endl;
|
|
|
|
|
|
|
|
//create dual mesh
|
2010-12-06 12:04:37 +00:00
|
|
|
for (int n=0; n<3; ++n)
|
2010-03-02 13:54:50 +00:00
|
|
|
{
|
2010-12-06 12:04:37 +00:00
|
|
|
// cerr << "start " << start[n] << "stop " << stop[n];
|
2010-03-09 20:35:57 +00:00
|
|
|
if (start[n]>stop[n])
|
|
|
|
{
|
|
|
|
unsigned int help = start[n];
|
|
|
|
start[n]=stop[n];
|
|
|
|
stop[n]=help;
|
|
|
|
}
|
|
|
|
++stop[n];
|
2010-04-07 10:57:45 +00:00
|
|
|
lines.clear();
|
2010-12-06 12:04:37 +00:00
|
|
|
for (unsigned int i=start[n]; i<stop[n]; i+=subSample[n])
|
2010-04-07 10:57:45 +00:00
|
|
|
{
|
2010-04-13 16:40:43 +00:00
|
|
|
lines.push_back(Op->GetDiscLine(n,i,true));//0.5*(Op->discLines[n][i+1] + Op->discLines[n][i]));
|
2010-05-11 18:38:58 +00:00
|
|
|
}
|
2010-12-02 12:51:34 +00:00
|
|
|
numLines[n] = lines.size();
|
|
|
|
delete[] discLines[n];
|
|
|
|
discLines[n] = new double[numLines[n]];
|
2010-12-06 12:04:37 +00:00
|
|
|
for (unsigned int i=0; i<numLines[n]; ++i)
|
2010-12-02 12:51:34 +00:00
|
|
|
discLines[n][i] = lines.at(i);
|
2010-03-02 13:54:50 +00:00
|
|
|
}
|
|
|
|
}
|
2010-07-20 09:42:47 +00:00
|
|
|
|
2010-12-06 12:04:37 +00:00
|
|
|
if (g_settings.showProbeDiscretization())
|
|
|
|
{
|
2010-07-20 09:42:47 +00:00
|
|
|
// FIXME the information E-Field / H-Field and therefore which mesh to use is missing
|
|
|
|
bool dualMesh = false;
|
|
|
|
cerr << m_filename << ": snapped coords: (" << Op->GetDiscLine( 0, start[0], dualMesh ) << ","
|
2010-12-06 12:04:37 +00:00
|
|
|
<< Op->GetDiscLine( 1, start[1], dualMesh ) << "," << Op->GetDiscLine( 2, start[2], dualMesh ) << ") -> ("
|
|
|
|
<< Op->GetDiscLine( 0, stop[0], dualMesh ) << ","<< Op->GetDiscLine( 1, stop[1], dualMesh ) << ","
|
|
|
|
<< Op->GetDiscLine( 2, stop[2], dualMesh ) << ")";
|
2010-07-20 09:42:47 +00:00
|
|
|
cerr << " [" << start[0] << "," << start[1] << "," << start[2] << "] -> ["
|
2010-12-06 12:04:37 +00:00
|
|
|
<< stop[0] << "," << stop[1] << "," << stop[2] << "]" << endl;
|
2010-07-20 09:42:47 +00:00
|
|
|
}
|
|
|
|
|
2010-03-02 13:54:50 +00:00
|
|
|
}
|
|
|
|
|
2010-04-21 12:28:16 +00:00
|
|
|
double ProcessFields::CalcTotalEnergy() const
|
2010-03-15 15:59:37 +00:00
|
|
|
{
|
2010-12-02 13:29:46 +00:00
|
|
|
double energy=0.0;
|
2010-04-21 12:28:16 +00:00
|
|
|
|
2010-12-02 13:29:46 +00:00
|
|
|
Engine_Interface_FDTD* EI_FDTD = dynamic_cast<Engine_Interface_FDTD*>(m_Eng_Interface);
|
|
|
|
|
|
|
|
if (EI_FDTD)
|
2010-03-15 15:59:37 +00:00
|
|
|
{
|
2010-12-02 13:29:46 +00:00
|
|
|
const Engine* Eng = EI_FDTD->GetFDTDEngine();
|
|
|
|
|
|
|
|
unsigned int pos[3];
|
2010-12-06 12:04:37 +00:00
|
|
|
for (pos[0]=0; pos[0]<Op->GetNumberOfLines(0); ++pos[0])
|
2010-03-15 15:59:37 +00:00
|
|
|
{
|
2010-12-06 12:04:37 +00:00
|
|
|
for (pos[1]=0; pos[1]<Op->GetNumberOfLines(1); ++pos[1])
|
2010-03-15 15:59:37 +00:00
|
|
|
{
|
2010-12-06 12:04:37 +00:00
|
|
|
for (pos[2]=0; pos[2]<Op->GetNumberOfLines(2); ++pos[2])
|
2010-12-02 13:29:46 +00:00
|
|
|
{
|
|
|
|
energy+=fabs(Eng->GetVolt(0,pos[0],pos[1],pos[2]) * Eng->GetCurr(1,pos[0],pos[1],pos[2]));
|
|
|
|
energy+=fabs(Eng->GetVolt(0,pos[0],pos[1],pos[2]) * Eng->GetCurr(2,pos[0],pos[1],pos[2]));
|
|
|
|
energy+=fabs(Eng->GetVolt(1,pos[0],pos[1],pos[2]) * Eng->GetCurr(0,pos[0],pos[1],pos[2]));
|
|
|
|
energy+=fabs(Eng->GetVolt(1,pos[0],pos[1],pos[2]) * Eng->GetCurr(2,pos[0],pos[1],pos[2]));
|
|
|
|
energy+=fabs(Eng->GetVolt(2,pos[0],pos[1],pos[2]) * Eng->GetCurr(0,pos[0],pos[1],pos[2]));
|
|
|
|
energy+=fabs(Eng->GetVolt(2,pos[0],pos[1],pos[2]) * Eng->GetCurr(1,pos[0],pos[1],pos[2]));
|
|
|
|
}
|
2010-03-15 15:59:37 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return energy*0.5;
|
|
|
|
}
|
|
|
|
|
2010-12-02 13:29:46 +00:00
|
|
|
void ProcessFields::SetSubSampling(unsigned int subSampleRate, int dir)
|
2010-04-07 10:57:45 +00:00
|
|
|
{
|
|
|
|
if (dir>2) return;
|
|
|
|
if (dir<0)
|
|
|
|
{
|
|
|
|
subSample[0]=subSampleRate;
|
|
|
|
subSample[1]=subSampleRate;
|
|
|
|
subSample[2]=subSampleRate;
|
|
|
|
}
|
|
|
|
else subSample[dir]=subSampleRate;
|
|
|
|
}
|
2010-03-09 20:35:57 +00:00
|
|
|
|
2010-07-15 10:58:48 +00:00
|
|
|
void ProcessFields::WriteVTKHeader(ofstream &file, double const* const* discLines, unsigned int const* numLines, unsigned int precision, string header_info, MeshType meshT, double discLines_scaling)
|
2010-06-02 12:18:25 +00:00
|
|
|
{
|
|
|
|
if (meshT==CARTESIAN_MESH)
|
2010-07-15 10:58:48 +00:00
|
|
|
WriteVTKCartesianGridHeader(file, discLines, numLines, precision, header_info, discLines_scaling);
|
2010-06-02 12:18:25 +00:00
|
|
|
else if (meshT==CYLINDRICAL_MESH)
|
2010-07-15 10:58:48 +00:00
|
|
|
WriteVTKCylindricalGridHeader(file, discLines, numLines, precision, header_info, discLines_scaling);
|
2010-06-02 12:18:25 +00:00
|
|
|
else
|
|
|
|
cerr << "ProcessFields::WriteVTKHeader: Warning: unknown mesh type, skipping header -> file will be invalid..." << endl;
|
|
|
|
}
|
|
|
|
|
2010-07-15 10:58:48 +00:00
|
|
|
void ProcessFields::WriteVTKCartesianGridHeader(ofstream &file, double const* const* discLines, unsigned int const* numLines, unsigned int precision, string header_info, double discLines_scaling)
|
2010-03-02 13:54:50 +00:00
|
|
|
{
|
|
|
|
file << "# vtk DataFile Version 2.0" << endl;
|
2010-05-11 19:03:16 +00:00
|
|
|
file << "Rectilinear Grid openEMS_ProcessFields";
|
|
|
|
if (!header_info.empty())
|
|
|
|
file << " " << header_info;
|
|
|
|
file << endl;
|
2010-03-02 13:54:50 +00:00
|
|
|
file << "ASCII" << endl;
|
|
|
|
file << "DATASET RECTILINEAR_GRID " << endl;
|
|
|
|
file << "DIMENSIONS " << numLines[0] << " " << numLines[1] << " " << numLines[2] << endl;
|
2010-10-04 07:30:24 +00:00
|
|
|
file << "X_COORDINATES " << numLines[0] << " " << __VTK_DATA_TYPE__ << endl;
|
2010-12-06 12:04:37 +00:00
|
|
|
for (unsigned int i=0; i<numLines[0]; ++i)
|
2010-07-15 10:58:48 +00:00
|
|
|
file << setprecision(precision) << discLines[0][i] * discLines_scaling << " ";
|
2010-03-02 13:54:50 +00:00
|
|
|
file << endl;
|
2010-10-04 07:30:24 +00:00
|
|
|
file << "Y_COORDINATES " << numLines[1] << " " << __VTK_DATA_TYPE__ << endl;
|
2010-12-06 12:04:37 +00:00
|
|
|
for (unsigned int i=0; i<numLines[1]; ++i)
|
2010-07-15 10:58:48 +00:00
|
|
|
file << setprecision(precision) << discLines[1][i] * discLines_scaling << " ";
|
2010-03-02 13:54:50 +00:00
|
|
|
file << endl;
|
2010-10-04 07:30:24 +00:00
|
|
|
file << "Z_COORDINATES " << numLines[2] << " " << __VTK_DATA_TYPE__ << endl;
|
2010-12-06 12:04:37 +00:00
|
|
|
for (unsigned int i=0; i<numLines[2]; ++i)
|
2010-07-15 10:58:48 +00:00
|
|
|
file << setprecision(precision) << discLines[2][i] * discLines_scaling << " ";
|
2010-03-02 13:54:50 +00:00
|
|
|
file << endl << endl;
|
|
|
|
file << "POINT_DATA " << numLines[0]*numLines[1]*numLines[2] << endl;
|
2010-03-12 19:39:04 +00:00
|
|
|
}
|
|
|
|
|
2010-07-15 10:58:48 +00:00
|
|
|
void ProcessFields::WriteVTKCylindricalGridHeader(ofstream &file, double const* const* discLines, unsigned int const* numLines, unsigned int precision, string header_info, double discLines_scaling)
|
2010-06-02 12:18:25 +00:00
|
|
|
{
|
|
|
|
file << "# vtk DataFile Version 3.0" << endl;
|
|
|
|
file << "Structured Grid from openEMS_ProcessFields";
|
|
|
|
if (!header_info.empty())
|
|
|
|
file << " " << header_info;
|
|
|
|
file << endl;
|
|
|
|
file << "ASCII" << endl;
|
|
|
|
file << "DATASET STRUCTURED_GRID " << endl;
|
|
|
|
file << "DIMENSIONS " << numLines[0] << " " << numLines[1] << " " << numLines[2] << endl;
|
2010-10-04 07:30:24 +00:00
|
|
|
file << "POINTS " << numLines[0]*numLines[1]*numLines[2] << " " << __VTK_DATA_TYPE__ << endl;
|
2010-12-06 12:04:37 +00:00
|
|
|
for (unsigned int k=0; k<numLines[2]; ++k)
|
|
|
|
for (unsigned int j=0; j<numLines[1]; ++j)
|
|
|
|
for (unsigned int i=0; i<numLines[0]; ++i)
|
2010-06-02 12:18:25 +00:00
|
|
|
{
|
2010-07-15 10:58:48 +00:00
|
|
|
file << setprecision(precision) << discLines[0][i] * cos(discLines[1][j]) * discLines_scaling << " "
|
2010-12-06 12:04:37 +00:00
|
|
|
<< discLines[0][i] * sin(discLines[1][j]) * discLines_scaling << " "
|
|
|
|
<< discLines[2][k] * discLines_scaling << endl;
|
2010-06-02 12:18:25 +00:00
|
|
|
}
|
|
|
|
file << endl;
|
|
|
|
file << endl << endl;
|
|
|
|
file << "POINT_DATA " << numLines[0]*numLines[1]*numLines[2] << endl;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2010-09-17 12:48:32 +00:00
|
|
|
void ProcessFields::WriteVTKVectorArray(ofstream &file, string name, FDTD_FLOAT const* const* const* const* array, double const* const* discLines, unsigned int const* numLines, unsigned int precision, MeshType meshT)
|
2010-03-12 19:39:04 +00:00
|
|
|
{
|
2010-10-04 07:30:24 +00:00
|
|
|
file << "VECTORS " << name << " " << __VTK_DATA_TYPE__ << endl;
|
2010-03-02 13:54:50 +00:00
|
|
|
|
2010-09-17 12:48:32 +00:00
|
|
|
if (g_settings.NativeFieldDumps())
|
|
|
|
meshT = CARTESIAN_MESH; //dump field components as they are...
|
|
|
|
|
2010-03-02 13:54:50 +00:00
|
|
|
unsigned int pos[3];
|
2010-12-06 12:04:37 +00:00
|
|
|
for (pos[2]=0; pos[2]<numLines[2]; ++pos[2])
|
2010-03-02 13:54:50 +00:00
|
|
|
{
|
2010-12-06 12:04:37 +00:00
|
|
|
for (pos[1]=0; pos[1]<numLines[1]; ++pos[1])
|
2010-03-02 13:54:50 +00:00
|
|
|
{
|
2010-09-17 12:48:32 +00:00
|
|
|
double cos_a = cos(discLines[1][pos[1]]); //needed only for CYLINDRICAL_MESH
|
|
|
|
double sin_a = sin(discLines[1][pos[1]]); //needed only for CYLINDRICAL_MESH
|
2010-12-06 12:04:37 +00:00
|
|
|
for (pos[0]=0; pos[0]<numLines[0]; ++pos[0])
|
2010-03-02 13:54:50 +00:00
|
|
|
{
|
2010-09-17 12:48:32 +00:00
|
|
|
switch (meshT)
|
|
|
|
{
|
|
|
|
case CARTESIAN_MESH:
|
|
|
|
UNUSED(discLines); //disclines not needed for the original cartesian mesh
|
|
|
|
//in x
|
|
|
|
file << setprecision(precision) << array[0][pos[0]][pos[1]][pos[2]] << " ";
|
|
|
|
//in y
|
|
|
|
file << setprecision(precision) << array[1][pos[0]][pos[1]][pos[2]] << " ";
|
|
|
|
//in z
|
|
|
|
file << setprecision(precision) << array[2][pos[0]][pos[1]][pos[2]] << endl;
|
|
|
|
break;
|
|
|
|
case CYLINDRICAL_MESH:
|
|
|
|
//in x : F_x = F_r * cos(a) - F_a * sin(a);
|
|
|
|
file << setprecision(precision) << array[0][pos[0]][pos[1]][pos[2]] * cos_a - array[1][pos[0]][pos[1]][pos[2]] * sin_a << " ";
|
|
|
|
//in y : F_y = F_r * sin(a) + F_a * cos(a);
|
|
|
|
file << setprecision(precision) << array[0][pos[0]][pos[1]][pos[2]] * sin_a + array[1][pos[0]][pos[1]][pos[2]] * cos_a << " ";
|
|
|
|
//in z
|
|
|
|
file << setprecision(precision) << array[2][pos[0]][pos[1]][pos[2]] << endl;
|
|
|
|
break;
|
|
|
|
}
|
2010-03-02 13:54:50 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2010-03-12 19:39:04 +00:00
|
|
|
|
|
|
|
|
2010-07-15 10:58:48 +00:00
|
|
|
bool ProcessFields::DumpVectorArray2VTK(ofstream &file, string name, FDTD_FLOAT const* const* const* const* array, double const* const* discLines, unsigned int const* numLines, unsigned int precision, string header_info, MeshType meshT, double discLines_scaling)
|
2010-03-12 19:39:04 +00:00
|
|
|
{
|
2010-07-15 10:58:48 +00:00
|
|
|
WriteVTKHeader(file, discLines, numLines, precision, header_info, meshT, discLines_scaling);
|
2010-09-17 12:48:32 +00:00
|
|
|
WriteVTKVectorArray(file, name, array, discLines, numLines, precision, meshT);
|
2010-03-29 08:12:38 +00:00
|
|
|
return true;
|
2010-03-12 19:39:04 +00:00
|
|
|
}
|
|
|
|
|
2010-07-15 10:58:48 +00:00
|
|
|
bool ProcessFields::DumpMultiVectorArray2VTK(ofstream &file, string names[], FDTD_FLOAT const* const* const* const* const* array, unsigned int numFields, double const* const* discLines, unsigned int const* numLines, unsigned int precision, string header_info, MeshType meshT, double discLines_scaling)
|
2010-03-12 19:39:04 +00:00
|
|
|
{
|
2010-07-15 10:58:48 +00:00
|
|
|
WriteVTKHeader(file, discLines, numLines, precision, header_info, meshT, discLines_scaling);
|
2010-12-06 12:04:37 +00:00
|
|
|
for (unsigned int n=0; n<numFields; ++n)
|
2010-03-12 19:39:04 +00:00
|
|
|
{
|
2010-09-17 12:48:32 +00:00
|
|
|
WriteVTKVectorArray(file, names[n], array[n], discLines, numLines, precision, meshT);
|
2010-03-12 19:39:04 +00:00
|
|
|
file << endl;
|
|
|
|
}
|
2010-03-26 18:51:09 +00:00
|
|
|
return true;
|
2010-03-12 19:39:04 +00:00
|
|
|
}
|
|
|
|
|
2010-04-21 12:28:16 +00:00
|
|
|
void ProcessFields::WriteVTKScalarArray(ofstream &file, string name, FDTD_FLOAT const* const* const* array, unsigned int const* numLines, unsigned int precision)
|
2010-03-12 19:39:04 +00:00
|
|
|
{
|
2010-10-04 07:30:24 +00:00
|
|
|
file << "SCALARS " << name << " " << __VTK_DATA_TYPE__ << 1 << endl;
|
2010-03-12 19:39:04 +00:00
|
|
|
file << "LOOKUP_TABLE default" << endl;
|
|
|
|
unsigned int pos[3];
|
|
|
|
int count=0;
|
2010-12-06 12:04:37 +00:00
|
|
|
for (pos[2]=0; pos[2]<numLines[2]; ++pos[2])
|
2010-03-12 19:39:04 +00:00
|
|
|
{
|
2010-12-06 12:04:37 +00:00
|
|
|
for (pos[1]=0; pos[1]<numLines[1]; ++pos[1])
|
2010-03-12 19:39:04 +00:00
|
|
|
{
|
2010-12-06 12:04:37 +00:00
|
|
|
for (pos[0]=0; pos[0]<numLines[0]; ++pos[0])
|
2010-03-12 19:39:04 +00:00
|
|
|
{
|
2010-04-13 16:40:43 +00:00
|
|
|
file << setprecision(precision) << array[pos[0]][pos[1]][pos[2]] << " ";
|
2010-03-12 19:39:04 +00:00
|
|
|
++count;
|
|
|
|
if (count%10==0)
|
|
|
|
file << endl;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2010-07-15 10:58:48 +00:00
|
|
|
bool ProcessFields::DumpScalarArray2VTK(ofstream &file, string name, FDTD_FLOAT const* const* const* array, double const* const* discLines, unsigned int const* numLines, unsigned int precision, string header_info, MeshType meshT, double discLines_scaling)
|
2010-03-12 19:39:04 +00:00
|
|
|
{
|
2010-07-15 10:58:48 +00:00
|
|
|
WriteVTKHeader(file, discLines, numLines, precision, header_info, meshT, discLines_scaling);
|
2010-04-12 07:38:24 +00:00
|
|
|
WriteVTKScalarArray(file, name, array, numLines, precision);
|
2010-03-26 18:51:09 +00:00
|
|
|
return true;
|
2010-03-12 19:39:04 +00:00
|
|
|
}
|
|
|
|
|
2010-07-15 10:58:48 +00:00
|
|
|
bool ProcessFields::DumpMultiScalarArray2VTK(ofstream &file, string names[], FDTD_FLOAT const* const* const* const* array, unsigned int numFields, double const* const* discLines, unsigned int const* numLines, unsigned int precision, string header_info, MeshType meshT, double discLines_scaling)
|
2010-03-12 19:39:04 +00:00
|
|
|
{
|
2010-07-15 10:58:48 +00:00
|
|
|
WriteVTKHeader(file, discLines, numLines, precision, header_info, meshT, discLines_scaling);
|
2010-12-06 12:04:37 +00:00
|
|
|
for (unsigned int n=0; n<numFields; ++n)
|
2010-03-12 19:39:04 +00:00
|
|
|
{
|
2010-04-12 07:38:24 +00:00
|
|
|
WriteVTKScalarArray(file, names[n], array[n], numLines, precision);
|
2010-03-12 19:39:04 +00:00
|
|
|
file << endl;
|
|
|
|
}
|
2010-03-26 18:51:09 +00:00
|
|
|
return true;
|
2010-03-12 19:39:04 +00:00
|
|
|
}
|
|
|
|
|
2010-12-27 12:54:09 +00:00
|
|
|
|
|
|
|
bool ProcessFields::WriteMesh2HDF5(string filename, string groupName, unsigned int const* numLines, double const* const* discLines, MeshType meshT, double discLines_scaling)
|
|
|
|
{
|
|
|
|
H5::H5File file( filename, H5F_ACC_RDWR );
|
|
|
|
|
|
|
|
H5::Group hdf_group( file.openGroup( groupName ));
|
|
|
|
|
|
|
|
string names[] = {"x","y","z"};
|
|
|
|
if (meshT==CYLINDRICAL_MESH)
|
|
|
|
{
|
|
|
|
names[0]="rho";
|
|
|
|
names[1]="alpha";
|
|
|
|
}
|
|
|
|
|
|
|
|
H5::Group* group = new H5::Group( hdf_group.createGroup( "/Mesh" ));
|
|
|
|
for (int n=0; n<3; ++n)
|
|
|
|
{
|
|
|
|
hsize_t dimsf[1]; // dataset dimensions
|
|
|
|
dimsf[0] = numLines[n];
|
|
|
|
H5::DataSpace dataspace( 1, dimsf );
|
|
|
|
H5::FloatType datatype( H5::PredType::NATIVE_FLOAT );
|
|
|
|
H5::DataSet dataset = group->createDataSet( names[n].c_str(), datatype, dataspace );
|
|
|
|
//convert to float...
|
|
|
|
float* array = new float[numLines[n]];
|
|
|
|
for (unsigned int i=0; i<numLines[n]; ++i)
|
|
|
|
{
|
|
|
|
#ifdef OUTPUT_IN_DRAWINGUNITS
|
|
|
|
array[i] = Lines[n][i];
|
|
|
|
#else
|
|
|
|
if ((meshT==CYLINDRICAL_MESH) && (n==1)) //check for alpha-direction
|
|
|
|
array[i] = discLines[n][i];
|
|
|
|
else
|
|
|
|
array[i] = discLines[n][i] * discLines_scaling;
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
//write to dataset
|
|
|
|
dataset.write( array, H5::PredType::NATIVE_FLOAT );
|
|
|
|
}
|
|
|
|
delete group;
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool ProcessFields::DumpVectorArray2HDF5(string filename, string groupName, string name, FDTD_FLOAT const* const* const* const* array, unsigned int const* numLines, float time)
|
2010-04-05 18:22:03 +00:00
|
|
|
{
|
|
|
|
const H5std_string FILE_NAME(filename);
|
|
|
|
const H5std_string DATASET_NAME( name );
|
|
|
|
|
|
|
|
H5::H5File file( FILE_NAME, H5F_ACC_RDWR );
|
|
|
|
|
2010-12-27 12:54:09 +00:00
|
|
|
H5::Group group( file.openGroup( groupName ));
|
2010-04-05 18:22:03 +00:00
|
|
|
|
|
|
|
hsize_t dimsf[4]; // dataset dimensions
|
|
|
|
|
|
|
|
dimsf[0] = 3;
|
|
|
|
dimsf[1] = numLines[2];
|
|
|
|
dimsf[2] = numLines[1];
|
|
|
|
dimsf[3] = numLines[0];
|
|
|
|
|
|
|
|
H5::DataSpace dataspace( 4, dimsf );
|
|
|
|
|
|
|
|
H5::FloatType datatype( H5::PredType::NATIVE_FLOAT );
|
|
|
|
// datatype.setOrder( H5T_ORDER_LE );
|
|
|
|
H5::DataSet dataset = group.createDataSet( DATASET_NAME, datatype, dataspace );
|
|
|
|
|
2010-04-29 17:26:45 +00:00
|
|
|
hsize_t t_dimsf[] = {1};
|
|
|
|
H5::DataSpace t_dataspace( 1, t_dimsf );
|
|
|
|
H5::Attribute attr = dataset.createAttribute("time",H5::PredType::NATIVE_FLOAT,t_dataspace);
|
|
|
|
attr.write( H5::PredType::NATIVE_FLOAT , &time);
|
|
|
|
|
2010-04-05 18:22:03 +00:00
|
|
|
// I have not the slightest idea why this array-copy action is necessary... but it's the only way hdf5 does what it is supposed to do anyway!!
|
|
|
|
// at least it is save in case FDTD_FLOAT was defined as double...
|
|
|
|
// why does hdf5 write the dimensions backwards??? or matlab???
|
|
|
|
float hdf5array[3][numLines[2]][numLines[1]][numLines[0]];
|
2010-12-06 12:04:37 +00:00
|
|
|
for (int n=0; n<3; ++n)
|
2010-04-05 18:22:03 +00:00
|
|
|
{
|
2010-12-06 12:04:37 +00:00
|
|
|
for (unsigned int i=0; i<numLines[0]; ++i)
|
2010-04-05 18:22:03 +00:00
|
|
|
{
|
2010-12-06 12:04:37 +00:00
|
|
|
for (unsigned int j=0; j<numLines[1]; ++j)
|
2010-04-05 18:22:03 +00:00
|
|
|
{
|
2010-12-06 12:04:37 +00:00
|
|
|
for (unsigned int k=0; k<numLines[2]; ++k)
|
2010-04-05 18:22:03 +00:00
|
|
|
{
|
|
|
|
hdf5array[n][k][j][i] = array[n][i][j][k];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
dataset.write( hdf5array, H5::PredType::NATIVE_FLOAT );
|
|
|
|
return true;
|
|
|
|
}
|
2010-12-17 14:14:34 +00:00
|
|
|
|
2010-12-27 12:54:09 +00:00
|
|
|
bool ProcessFields::DumpVectorArray2HDF5(string filename, string groupName, string name, std::complex<float> const* const* const* const* array, unsigned int const* numLines, float weight, float frequency)
|
2010-12-19 19:41:08 +00:00
|
|
|
{
|
|
|
|
const H5std_string FILE_NAME(filename);
|
|
|
|
const H5std_string DATASET_NAME_RE( name + "_real");
|
|
|
|
const H5std_string DATASET_NAME_IM( name + "_imag");
|
|
|
|
|
|
|
|
H5::H5File file( FILE_NAME, H5F_ACC_RDWR );
|
|
|
|
|
2010-12-27 12:54:09 +00:00
|
|
|
H5::Group group( file.openGroup( groupName ));
|
2010-12-19 19:41:08 +00:00
|
|
|
|
|
|
|
hsize_t t_dimsf[] = {1};
|
|
|
|
H5::DataSpace t_dataspace( 1, t_dimsf );
|
|
|
|
|
|
|
|
hsize_t dimsf[4]; // dataset dimensions
|
|
|
|
dimsf[0] = 3;
|
|
|
|
dimsf[1] = numLines[2];
|
|
|
|
dimsf[2] = numLines[1];
|
|
|
|
dimsf[3] = numLines[0];
|
|
|
|
|
|
|
|
H5::DataSpace dataspace( 4, dimsf );
|
|
|
|
H5::FloatType datatype( H5::PredType::NATIVE_FLOAT );
|
|
|
|
|
|
|
|
//create and write real part
|
|
|
|
H5::DataSet dataset = group.createDataSet( DATASET_NAME_RE, datatype, dataspace );
|
|
|
|
H5::Attribute attr = dataset.createAttribute("frequency",H5::PredType::NATIVE_FLOAT,t_dataspace);
|
|
|
|
attr.write( H5::PredType::NATIVE_FLOAT , &frequency);
|
|
|
|
// I have not the slightest idea why this array-copy action is necessary... but it's the only way hdf5 does what it is supposed to do anyway!!
|
|
|
|
// at least it is save in case FDTD_FLOAT was defined as double...
|
|
|
|
// why does hdf5 write the dimensions backwards??? or matlab???
|
|
|
|
float hdf5array[3][numLines[2]][numLines[1]][numLines[0]];
|
|
|
|
for (int n=0; n<3; ++n)
|
|
|
|
{
|
|
|
|
for (unsigned int i=0; i<numLines[0]; ++i)
|
|
|
|
{
|
|
|
|
for (unsigned int j=0; j<numLines[1]; ++j)
|
|
|
|
{
|
|
|
|
for (unsigned int k=0; k<numLines[2]; ++k)
|
|
|
|
{
|
|
|
|
hdf5array[n][k][j][i] = array[n][i][j][k].real() * weight;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
dataset.write( hdf5array, H5::PredType::NATIVE_FLOAT );
|
|
|
|
|
|
|
|
//create and write imaginary part
|
|
|
|
dataset = group.createDataSet( DATASET_NAME_IM, datatype, dataspace );
|
|
|
|
attr = dataset.createAttribute("frequency",H5::PredType::NATIVE_FLOAT,t_dataspace);
|
|
|
|
attr.write( H5::PredType::NATIVE_FLOAT , &frequency);
|
|
|
|
// I have not the slightest idea why this array-copy action is necessary... but it's the only way hdf5 does what it is supposed to do anyway!!
|
|
|
|
// at least it is save in case FDTD_FLOAT was defined as double...
|
|
|
|
// why does hdf5 write the dimensions backwards??? or matlab???
|
|
|
|
for (int n=0; n<3; ++n)
|
|
|
|
{
|
|
|
|
for (unsigned int i=0; i<numLines[0]; ++i)
|
|
|
|
{
|
|
|
|
for (unsigned int j=0; j<numLines[1]; ++j)
|
|
|
|
{
|
|
|
|
for (unsigned int k=0; k<numLines[2]; ++k)
|
|
|
|
{
|
|
|
|
hdf5array[n][k][j][i] = array[n][i][j][k].imag() * weight;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
dataset.write( hdf5array, H5::PredType::NATIVE_FLOAT );
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2010-12-17 14:14:34 +00:00
|
|
|
FDTD_FLOAT**** ProcessFields::CalcField()
|
|
|
|
{
|
|
|
|
unsigned int pos[3];
|
|
|
|
unsigned int OpPos[3];
|
|
|
|
double out[3];
|
|
|
|
//create array
|
|
|
|
FDTD_FLOAT**** field = Create_N_3DArray<FDTD_FLOAT>(numLines);
|
|
|
|
if (m_DumpType==E_FIELD_DUMP)
|
|
|
|
{
|
|
|
|
for (pos[0]=0; pos[0]<numLines[0]; ++pos[0])
|
|
|
|
{
|
|
|
|
OpPos[0]=start[0]+pos[0]*subSample[0];
|
|
|
|
for (pos[1]=0; pos[1]<numLines[1]; ++pos[1])
|
|
|
|
{
|
|
|
|
OpPos[1]=start[1]+pos[1]*subSample[1];
|
|
|
|
for (pos[2]=0; pos[2]<numLines[2]; ++pos[2])
|
|
|
|
{
|
|
|
|
OpPos[2]=start[2]+pos[2]*subSample[2];
|
|
|
|
m_Eng_Interface->GetEField(OpPos,out);
|
|
|
|
field[0][pos[0]][pos[1]][pos[2]] = out[0];
|
|
|
|
field[1][pos[0]][pos[1]][pos[2]] = out[1];
|
|
|
|
field[2][pos[0]][pos[1]][pos[2]] = out[2];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2010-12-19 19:41:08 +00:00
|
|
|
return field;
|
2010-12-17 14:14:34 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if (m_DumpType==H_FIELD_DUMP)
|
|
|
|
{
|
|
|
|
for (pos[0]=0; pos[0]<numLines[0]; ++pos[0])
|
|
|
|
{
|
|
|
|
OpPos[0]=start[0]+pos[0]*subSample[0];
|
|
|
|
for (pos[1]=0; pos[1]<numLines[1]; ++pos[1])
|
|
|
|
{
|
|
|
|
OpPos[1]=start[1]+pos[1]*subSample[1];
|
|
|
|
for (pos[2]=0; pos[2]<numLines[2]; ++pos[2])
|
|
|
|
{
|
|
|
|
OpPos[2]=start[2]+pos[2]*subSample[2];
|
|
|
|
m_Eng_Interface->GetHField(OpPos,out);
|
|
|
|
field[0][pos[0]][pos[1]][pos[2]] = out[0];
|
|
|
|
field[1][pos[0]][pos[1]][pos[2]] = out[1];
|
|
|
|
field[2][pos[0]][pos[1]][pos[2]] = out[2];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2010-12-19 19:41:08 +00:00
|
|
|
return field;
|
2010-12-17 14:14:34 +00:00
|
|
|
}
|
2010-12-19 19:41:08 +00:00
|
|
|
|
|
|
|
cerr << "ProcessFields::CalcField(): Error, unknown dump type..." << endl;
|
2010-12-17 14:14:34 +00:00
|
|
|
return field;
|
|
|
|
}
|
|
|
|
|