/home/zenker/tmp/CSTDataExport/CST_Readout.h

Go to the documentation of this file.

#pragma once

#ifndef CST_Readout_h
#define CST_Readout_h
#include "CSTResultReaderInterf.h"
#include <vector>

using namespace std;

#define eps0 8.85418782e-12
#define mue0 M_PI*4e-7

struct material
{
  float id; 
  float eps; 
  float mue; 
  float rho; 
  float rho_set; 
  string description; 
};

class CST_Readout {
private:
  CSTProjHandle m_pHandle;
  int uReturnVal;
  int m_fieldType;
  float* m_meshMatrix; 

  unsigned int m_materialProperty;
  //mesh information
  int* m_nxyz; 
  vector<double> m_xlines; 
  vector<double> m_ylines; 
  vector<double> m_zlines; 

  void GetInv(int i, int j, int k, double &inv_x, double &inv_y, double &inv_z, bool epsOrkappa = false);
  int GetSuperIndex(int i, int j, int k){return i + j * m_nxyz[0] + k * m_nxyz[0] * m_nxyz[1];}
  /* @{ Private interface part needed for binary file export */
  string m_materialFile;
  string m_potName;
  vector<material> m_materials;
  vector<float> m_elementMaterialProperty; 
  vector<unsigned char> m_elementMaterialId; 
  vector<float> m_potentials; 
  string m_fieldTypeNames[5];
  static const int headerSize = 1000; 

  void PrepareMaterialData();
  void RetrieveMaterialProperties();
  void AssignMaterialIdToElements();
  /* @} */
public:
  CST_Readout(string cst_file);
  virtual ~CST_Readout();
  static double Round(double number, int digits);
  static void PrintStatus(int event, int all_events, double percent = 0.1);
  int* Getnxyz() {
    return m_nxyz;
  }
  int GetElementMaterial(int element, int node = 3);
  double* GetElementMaterialProperty(int element, unsigned int materialProperty,
      int testnode = 3);
  void GetNodesForElement(int element, vector<int> &nodes);
  void Node2Index(int node, int &i, int &j, int &k);
  double* GetMaterialPropertyFromIndex(int i, int j, int k, int n, unsigned int materialProperty);
  void Element2Index(int element, int &i, int &j, int &k);
  void  RetrieveFieldData(string TreeFieldName);
  int GetNumberOfNodes() {
    return m_nxyz[0] * m_nxyz[1] * m_nxyz[2];
  }
  int GetNumberOfElements() {
    return (m_nxyz[0] - 1) * (m_nxyz[1] - 1) * (m_nxyz[2] - 1);
  }
  double* GetMaterialPropertyFromIndex(float* matMatrix_cst, int i, int j, int k, int n,
      unsigned int materialProperty);
  void SetPotentialName(string pot){m_potName = pot;}
  string GetPotentialName(){return m_potName;}
  void UsePropertyRho(){m_materialProperty = 3;}
  void WriteElementComp(string path, string prefix = "");
  void WritePosLines(string path, string prefix = "");
  void WritePotentials(string TreeFieldName, string path, string prefix = "");
  void ExtractData(string fileName, bool isWeightingField, bool writeMaterialProp = false);
  void SetMaterialFile(string file){m_materialFile = file;}
};

#endif