diff --git a/openEMS.pro b/openEMS.pro index 0d0e6b3..8c085ba 100644 --- a/openEMS.pro +++ b/openEMS.pro @@ -42,19 +42,25 @@ win32 { LIBS += ../tinyxml/libtinyxml.so LIBS += -lboost_thread LIBS += -lhdf5 -lhdf5_cpp + ### vtk ### + INCLUDEPATH += /usr/include/vtk-5.2 \ + /usr/include/vtk-5.4 \ + /usr/include/vtk-5.6 + LIBS += -lvtkCommon \ + -lvtkIO \ + -lvtksys } - QMAKE_LFLAGS += \'-Wl,-rpath,\$$ORIGIN/../CSXCAD\' QMAKE_LFLAGS += \'-Wl,-rpath,\$$ORIGIN/../fparser\' QMAKE_LFLAGS += \'-Wl,-rpath,\$$ORIGIN/../tinyxml\' +#### SOURCES ################################################################ SOURCES += main.cpp \ - tools/ErrorMsg.cpp \ - tools/AdrOp.cpp \ - FDTD/engine.cpp \ + openems.cpp + +# FDTD +SOURCES += FDTD/engine.cpp \ FDTD/operator.cpp \ - tools/array_ops.cpp \ - openems.cpp \ FDTD/engine_multithread.cpp \ FDTD/operator_cylinder.cpp \ FDTD/engine_sse.cpp \ @@ -62,8 +68,6 @@ SOURCES += main.cpp \ FDTD/operator_sse_compressed.cpp \ FDTD/engine_sse_compressed.cpp \ FDTD/operator_multithread.cpp \ - tools/global.cpp \ - tools/useful.cpp \ FDTD/excitation.cpp \ FDTD/operator_cylindermultigrid.cpp \ FDTD/engine_cylindermultigrid.cpp \ @@ -99,16 +103,24 @@ SOURCES += Common/operator_base.cpp \ Common/processfields_td.cpp \ Common/processcurrent.cpp \ Common/processfields_fd.cpp \ - Common/processfieldprobe.cpp \ - Common/processfields_sar.cpp + Common/processfieldprobe.cpp \ + Common/processfields_sar.cpp -HEADERS += tools/ErrorMsg.h \ - tools/AdrOp.h \ - tools/constants.h \ - FDTD/engine.h \ +# tools + SOURCES += tools/global.cpp \ + tools/useful.cpp \ + tools/array_ops.cpp \ + tools/ErrorMsg.cpp \ + tools/AdrOp.cpp \ + tools/vtk_file_io.cpp \ + tools/base_file_io.cpp + +#### HEADERS ################################################################ +HEADERS += openems.h + +# FDTD +HEADERS += FDTD/engine.h \ FDTD/operator.h \ - tools/array_ops.h \ - openems.h \ FDTD/engine_multithread.h \ FDTD/operator_cylinder.h \ FDTD/engine_sse.h \ @@ -117,11 +129,8 @@ HEADERS += tools/ErrorMsg.h \ FDTD/operator_sse_compressed.h \ FDTD/engine_sse_compressed.h \ FDTD/operator_multithread.h \ - tools/global.h \ - tools/useful.h \ FDTD/operator_cylindermultigrid.h \ FDTD/engine_cylindermultigrid.h \ - tools/aligned_allocator.h \ FDTD/engine_interface_fdtd.h # FDTD/extensions header files @@ -154,8 +163,19 @@ HEADERS += Common/operator_base.h \ Common/processcurrent.h \ Common/processmodematch.h \ Common/processfields_fd.h \ - Common/processfieldprobe.h \ - Common/processfields_sar.h + Common/processfieldprobe.h \ + Common/processfields_sar.h + +# tools +HEADERS += tools/ErrorMsg.h \ + tools/AdrOp.h \ + tools/constants.h \ + tools/array_ops.h \ + tools/global.h \ + tools/useful.h \ + tools/aligned_allocator.h \ + tools/vtk_file_io.h \ + tools/base_file_io.h QMAKE_CXXFLAGS_RELEASE = -O3 \ -g \ diff --git a/tools/base_file_io.cpp b/tools/base_file_io.cpp new file mode 100644 index 0000000..ce33a68 --- /dev/null +++ b/tools/base_file_io.cpp @@ -0,0 +1,32 @@ +/* +* Copyright (C) 2011 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 . +*/ + +#include "base_file_io.h" + +Base_File_IO::Base_File_IO(std::string filename, int meshType) +{ + SetFilename(filename); + m_MeshType = meshType; + m_Binary = true; + m_Compress = true; + m_AppendMode = false; + m_ActiveTS = false; +} + +Base_File_IO::~Base_File_IO() +{ +} diff --git a/tools/base_file_io.h b/tools/base_file_io.h new file mode 100644 index 0000000..44c1d1d --- /dev/null +++ b/tools/base_file_io.h @@ -0,0 +1,87 @@ +/* +* Copyright (C) 2011 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 . +*/ + +#ifndef BASE_FILE_IO_H +#define BASE_FILE_IO_H + +#include +#include +#include +#include +#include + +//! Abstract base class for dumping scalar or vector field data +class Base_File_IO +{ +public: + virtual ~Base_File_IO(); + + //! Set the filename + virtual void SetFilename(std::string filename) {m_filename=filename;} + //! Set the header information. May not be supported by all file types or setting. + virtual void SetHeader(std::string header) {m_header=header;} + + //! Tell write to append data. May fail if filename has changed or filetype doesn't support this. + virtual void SetAppendMode(bool val) {m_AppendMode=val;} + //! Set binary flag (if the file type supports it) + virtual void SetBinary(bool val) {m_Binary=val;} + //! Set compression flag (if the file type supports it) + virtual void SetCompress(bool val) {m_Compress=val;} + + //! Set the mesh lines for the given mesh type. + virtual void SetMeshLines(double const* const* lines, unsigned int const* count, double scaling=1) = 0 ; + + //! Add a scalar field. \sa GetNumberOfFields \sa ClearAllFields + virtual void AddScalarField(std::string fieldname, double const* const* const* field, unsigned int const* size) = 0; + //! Add a scalar field. \sa GetNumberOfFields \sa ClearAllFields + virtual void AddScalarField(std::string fieldname, float const* const* const* field, unsigned int const* size) = 0; + //! Add a vector field. \sa GetNumberOfFields \sa ClearAllFields + virtual void AddVectorField(std::string fieldname, double const* const* const* const* field, unsigned int const* size) = 0; + //! Add a vector field. \sa GetNumberOfFields \sa ClearAllFields + virtual void AddVectorField(std::string fieldname, float const* const* const* const* field, unsigned int const* size) = 0; + + //! Get the number of fields. \sa ClearAllFields + virtual int GetNumberOfFields() const = 0; + //! Clear all included fields. \sa GetNumberOfFields + virtual void ClearAllFields() = 0; + + //! Get if timestep file series is active. \sa SetTimestepActive + virtual bool GetTimestepActive() {return m_ActiveTS;} + //! Set the timestep file series flag. \sa GetTimestepActive \sa SetTimestep + virtual void SetTimestepActive(bool val) {m_ActiveTS = val;} + //! Set the current timestep, this will set the timestep flag to true. \sa SetTimestepActive + virtual void SetTimestep(unsigned int ts) {m_timestep=ts;SetTimestepActive(true);} + + virtual bool Write() = 0; + +protected: + Base_File_IO(std::string filename, int meshType=0); + std::string m_filename; + std::string m_header; + + bool m_ActiveTS; + unsigned int m_timestep; + + int m_MeshType; + + bool m_AppendMode; + bool m_Binary; + bool m_Compress; +}; + + +#endif // BASE_FILE_IO_H diff --git a/tools/vtk_file_io.cpp b/tools/vtk_file_io.cpp new file mode 100644 index 0000000..22664f2 --- /dev/null +++ b/tools/vtk_file_io.cpp @@ -0,0 +1,285 @@ +/* +* Copyright (C) 2011 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 . +*/ + +using namespace std; + +#include "vtk_file_io.h" + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#include +#include + + +VTK_File_IO::VTK_File_IO(string filename, int meshType) : Base_File_IO(filename, meshType) +{ + if (m_MeshType==0) //cartesian mesh + m_GridData = vtkRectilinearGrid::New(); + else if (m_MeshType==1) //cylindrical mesh + m_GridData = vtkStructuredGrid::New(); + else + { + cerr << "VTK_File_IO::VTK_File_IO: Error, unknown mesh type: " << m_MeshType << endl; + m_GridData=NULL; + } +} + +VTK_File_IO::~VTK_File_IO() +{ + if (m_GridData) + m_GridData->Delete(); + m_GridData = NULL; +} + +void VTK_File_IO::SetMeshLines(double const* const* lines, unsigned int const* count, double scaling) +{ + if (m_MeshType==0) //cartesian mesh + { + vtkRectilinearGrid* RectGrid = dynamic_cast(m_GridData); + if (RectGrid==NULL) + { + cerr << "VTK_File_IO::SetMeshLines: Error, grid invalid, this should not have happend! " << endl; + exit(1); + } + RectGrid->SetDimensions(count[0],count[1],count[2]); + vtkDoubleArray *Coords[3]; + for (int n=0;n<3;++n) + { + Coords[n] = vtkDoubleArray::New(); + for (unsigned int i=0; iInsertNextValue(lines[n][i]*scaling); + } + RectGrid->SetXCoordinates(Coords[0]); + RectGrid->SetYCoordinates(Coords[1]); + RectGrid->SetZCoordinates(Coords[2]); + for (int n=0;n<3;++n) + Coords[n]->Delete(); + } + else if (m_MeshType==1) //cylindrical mesh + { + vtkStructuredGrid* StructGrid = dynamic_cast(m_GridData); + if (StructGrid==NULL) + { + cerr << "VTK_File_IO::SetMeshLines: Error, grid invalid, this should not have happend! " << endl; + exit(1); + } + StructGrid->SetDimensions(count[0],count[1],count[2]); + vtkPoints *points = vtkPoints::New(); + points->SetNumberOfPoints(count[0]*count[1]*count[2]); + double r[3]; + int id=0; + for (unsigned int k=0; kSetPoint(id++,r); + } + StructGrid->SetPoints(points); + points->Delete(); + } + else + { + cerr << "VTK_File_IO::SetMeshLines: Error, unknown mesh type: " << m_MeshType << endl; + } +} + +void VTK_File_IO::AddScalarField(string fieldname, double const* const* const* field, unsigned int const* size) +{ + vtkDoubleArray* array = vtkDoubleArray::New(); + array->SetNumberOfTuples(size[0]*size[1]*size[2]); + array->SetName(fieldname.c_str()); + int id=0; + for (unsigned int k=0;kSetTuple1(id++,field[i][j][k]); + } + } + } + m_GridData->GetPointData()->AddArray(array); + array->Delete(); +} + +void VTK_File_IO::AddScalarField(string fieldname, float const* const* const* field, unsigned int const* size) +{ + vtkFloatArray* array = vtkFloatArray::New(); + array->SetNumberOfTuples(size[0]*size[1]*size[2]); + array->SetName(fieldname.c_str()); + int id=0; + for (unsigned int k=0;kSetTuple1(id++,field[i][j][k]); + } + } + } + m_GridData->GetPointData()->AddArray(array); + array->Delete(); +} + +void VTK_File_IO::AddVectorField(string fieldname, double const* const* const* const* field, unsigned int const* size) +{ + vtkDoubleArray* array = vtkDoubleArray::New(); + array->SetNumberOfComponents(3); + array->SetNumberOfTuples(size[0]*size[1]*size[2]); + array->SetName(fieldname.c_str()); + int id=0; + for (unsigned int k=0;kSetTuple3(id++,field[0][i][j][k],field[1][i][j][k],field[2][i][j][k]); + } + } + } + m_GridData->GetPointData()->AddArray(array); + array->Delete(); +} + +void VTK_File_IO::AddVectorField(string fieldname, float const* const* const* const* field, unsigned int const* size) +{ + vtkFloatArray* array = vtkFloatArray::New(); + array->SetNumberOfComponents(3); + array->SetNumberOfTuples(size[0]*size[1]*size[2]); + array->SetName(fieldname.c_str()); + int id=0; + for (unsigned int k=0;kSetTuple3(id++,field[0][i][j][k],field[1][i][j][k],field[2][i][j][k]); + } + } + } + m_GridData->GetPointData()->AddArray(array); + array->Delete(); +} + + +int VTK_File_IO::GetNumberOfFields() const +{ + return m_GridData->GetPointData()->GetNumberOfArrays(); +} + +void VTK_File_IO::ClearAllFields() +{ + while (m_GridData->GetPointData()->GetNumberOfArrays()>0) + { + const char* name = m_GridData->GetPointData()->GetArrayName(0); + m_GridData->GetPointData()->RemoveArray(name); + } +} + +bool VTK_File_IO::Write() +{ + return WriteXML(); +} + +string VTK_File_IO::GetTimestepFilename(int pad_length) const +{ + if (m_ActiveTS==false) + return m_filename; + + stringstream ss; + ss << m_filename << "_" << std::setw( pad_length ) << std::setfill( '0' ) << m_timestep; + + return ss.str(); +} + + +bool VTK_File_IO::WriteASCII() +{ + vtkDataWriter* writer = NULL; + if (m_MeshType==0) //cartesian mesh + writer = vtkRectilinearGridWriter::New(); + else if (m_MeshType==1) //cylindrical mesh + writer = vtkStructuredGridWriter::New(); + else + { + cerr << "VTK_File_IO::WriteASCII: Error, unknown mesh type: " << m_MeshType << endl; + return false; + } + + writer->SetHeader(m_header.c_str()); + writer->SetInput(m_GridData); + + string filename = GetTimestepFilename() + ".vtk"; + writer->SetFileName(filename.c_str()); + if (m_Binary) + writer->SetFileTypeToBinary(); + else + writer->SetFileTypeToASCII(); + + writer->Write(); + writer->Delete(); + return true; +} + +bool VTK_File_IO::WriteXML() +{ + vtkXMLStructuredDataWriter* writer = NULL; + if (m_MeshType==0) //cartesian mesh + writer = vtkXMLRectilinearGridWriter::New(); + else if (m_MeshType==1) //cylindrical mesh + writer = vtkXMLStructuredGridWriter::New(); + else + { + cerr << "VTK_File_IO::WriteXML: Error, unknown mesh type: " << m_MeshType << endl; + return false; + } + + writer->SetInput(m_GridData); + + string filename = GetTimestepFilename() + "." + writer->GetDefaultFileExtension(); + writer->SetFileName(filename.c_str()); + if (m_Compress) + writer->SetCompressor(vtkZLibDataCompressor::New()); + else + writer->SetCompressor(NULL); + + if (m_Binary) + writer->SetDataModeToBinary(); + else + writer->SetDataModeToAscii(); + + writer->Write(); + writer->Delete(); + return true; +} diff --git a/tools/vtk_file_io.h b/tools/vtk_file_io.h new file mode 100644 index 0000000..5f1be3c --- /dev/null +++ b/tools/vtk_file_io.h @@ -0,0 +1,52 @@ +/* +* Copyright (C) 2011 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 . +*/ + +#ifndef VTK_FILE_IO_H +#define VTK_FILE_IO_H + +#include "base_file_io.h" + +class vtkDataSet; + +class VTK_File_IO : public Base_File_IO +{ +public: + VTK_File_IO(std::string filename, int meshType=0); + virtual ~VTK_File_IO(); + + virtual void SetMeshLines(double const* const* lines, unsigned int const* count, double scaling=1); + + virtual void AddScalarField(std::string fieldname, double const* const* const* field, unsigned int const* size); + virtual void AddScalarField(std::string fieldname, float const* const* const* field, unsigned int const* size); + virtual void AddVectorField(std::string fieldname, double const* const* const* const* field, unsigned int const* size); + virtual void AddVectorField(std::string fieldname, float const* const* const* const* field, unsigned int const* size); + + virtual int GetNumberOfFields() const; + virtual void ClearAllFields(); + + virtual bool Write(); + + virtual bool WriteASCII(); + virtual bool WriteXML(); + +protected: + vtkDataSet* m_GridData; + + virtual std::string GetTimestepFilename(int pad_length=10) const; +}; + +#endif // VTK_FILE_IO_H