geometry_structure.hpp

Go to the documentation of this file.

#pragma once

#include "./mpi_structure.hpp"

#ifdef HAVE_METIS
  #include "metis.h"
#endif
#ifdef HAVE_PARMETIS
extern "C" {
#include "parmetis.h"
}
#endif
#ifdef HAVE_CGNS
  #include "cgnslib.h"
#endif
#include <string>
#include <fstream>
#include <sstream>
#include <cmath>
#include <algorithm>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>

#include "primal_grid_structure.hpp"
#include "dual_grid_structure.hpp"
#include "config_structure.hpp"

using namespace std;

class CGeometry {
protected:
    unsigned long nPoint,   
    nPointDomain,                       
    nPointGhost,                    
    nPointNode,                 
  Global_nPoint,    
    Global_nPointDomain,    
    nElem,                  
  Global_nElem, 
  Global_nElemDomain,  
    nEdge,                  
    nFace,                  
  nelem_edge,             
  Global_nelem_edge,      
  nelem_triangle,       
  Global_nelem_triangle,       
  nelem_quad,           
  Global_nelem_quad,           
  nelem_tetra,          
  Global_nelem_tetra,          
  nelem_hexa,           
  Global_nelem_hexa,           
  nelem_prism,          
  Global_nelem_prism,          
  nelem_pyramid,        
  Global_nelem_pyramid,        
  nelem_edge_bound,           
  Global_nelem_edge_bound,           
  nelem_triangle_bound,          
  Global_nelem_triangle_bound,          
  nelem_quad_bound,        
  Global_nelem_quad_bound;        
    unsigned short nDim,    
    nZone,                              
    nMarker;                
  unsigned long Max_GlobalPoint;  
public:
    unsigned long *nElem_Bound;         
    string *Tag_to_Marker;  
    CPrimalGrid** elem; 
    CPrimalGrid** face;         
    CPrimalGrid*** bound;   
    CPoint** node;          
    CEdge** edge;           
    CVertex*** vertex;      
    unsigned long *nVertex; 
    unsigned short nCommLevel;      
    vector<unsigned long> PeriodicPoint[MAX_NUMBER_PERIODIC][2];            
    vector<unsigned long> PeriodicElem[MAX_NUMBER_PERIODIC];                
  short *Marker_All_SendRecv;
  
    /*--- Create vectors and distribute the values among the different planes queues ---*/
    vector<vector<su2double> > Xcoord_plane; 
    vector<vector<su2double> > Ycoord_plane; 
    vector<vector<su2double> > Zcoord_plane;    
    vector<vector<su2double> > FaceArea_plane; 
    vector<vector<unsigned long> > Plane_points; 
    vector<su2double> XCoordList;   
    CPrimalGrid*** newBound;            
    unsigned long *nNewElem_Bound;          
  /*--- Partitioning-specific variables ---*/
  map<unsigned long,unsigned long> Global_to_Local_Elem;
  unsigned long xadj_size;
  unsigned long adjacency_size;
  unsigned long *starting_node;
  unsigned long *ending_node;
  unsigned long *npoint_procs;
#ifdef HAVE_MPI
#ifdef HAVE_PARMETIS
  idx_t * adjacency;
  idx_t * xadj;
#endif
#endif
  
    CGeometry(void);

    virtual ~CGeometry(void);

    unsigned short GetnDim(void);

    unsigned short GetnZone(void);

    unsigned long GetnPoint(void);

    unsigned long GetnPointDomain(void);

    unsigned long GetnLine(void);
  
    unsigned long GetnElem(void);
  
    unsigned long GetnEdge(void);

    unsigned short GetnMarker(void);

    unsigned long GetnVertex(unsigned short val_marker);

    long FindEdge(unsigned long first_point, unsigned long second_point);

    bool CheckEdge(unsigned long first_point, unsigned long second_point);
    
    su2double Point2Plane_Distance(su2double *Coord, su2double *iCoord, su2double *jCoord, su2double *kCoord);

    void TestGeometry(void);

    void SetnMarker(unsigned short val_nmarker);

    void SetnDim(unsigned short val_nDim);

    string GetMarker_Tag(unsigned short val_marker);

    void SetMarker_Tag(unsigned short val_marker, string val_index);

    void SetnElem_Bound(unsigned short val_marker, unsigned long val_nelem_bound);

    void SetnPoint(unsigned long val_npoint);

    void SetnPointDomain(unsigned long val_npoint);

    void SetnElem(unsigned long val_nelem);

    unsigned long GetnElem_Bound(unsigned short val_marker);

    unsigned long GetMax_GlobalPoint(void);

    virtual bool FindFace(unsigned long first_elem, unsigned long second_elem, unsigned short &face_first_elem, 
            unsigned short &face_second_elem);

    virtual void ComputeWall_Distance(CConfig *config);

    virtual void SetPositive_ZArea(CConfig *config);
  
    virtual void SetPoint_Connectivity(void);
  
    virtual void SetRCM_Ordering(CConfig *config);
  
    virtual void SetElement_Connectivity(void);

    void SetEdges(void);

    void SetFaces(void);

    virtual void SetBoundVolume(void);

    virtual void SetVertex(CConfig *config);

    virtual void SetVertex(void);

    virtual void SetCoord_CG(void);

    virtual void SetControlVolume(CConfig *config, unsigned short action);

  virtual void VisualizeControlVolume(CConfig *config, unsigned short action);
  
    virtual void MatchNearField(CConfig *config);
  
    virtual void MatchActuator_Disk(CConfig *config);

    virtual void MatchInterface(CConfig *config);

    virtual void MatchZone(CConfig *config, CGeometry *geometry_donor, CConfig *config_donor, 
            unsigned short val_iZone, unsigned short val_nZone);

    virtual void SetBoundControlVolume(CConfig *config, unsigned short action);
  
    virtual void SetTecPlot(char config_filename[MAX_STRING_SIZE], bool new_file);

    virtual void SetBoundTecPlot(char mesh_filename[MAX_STRING_SIZE], bool new_file, CConfig *config);

    virtual void SetBoundSTL(char mesh_filename[MAX_STRING_SIZE], bool new_file, CConfig *config);

  
    virtual void Check_IntElem_Orientation(CConfig *config);
  
    virtual void Check_BoundElem_Orientation(CConfig *config);

    virtual void SetColorGrid(CConfig *config);
  
  virtual void SetColorGrid_Parallel(CConfig *config);
  
  virtual void DivideConnectivity(CConfig *config, unsigned short Elem_Type);

    virtual void SetPeriodicBoundary(CConfig *config);

    virtual void SetSendReceive(CConfig *config);
  
    virtual void SetBoundaries(CConfig *config);
  
    virtual void SetCoord(CGeometry *geometry);

    virtual void SetCoord_Smoothing(unsigned short val_nSmooth, su2double val_smooth_coeff, CConfig *config);

    virtual void SetPoint_Connectivity(CGeometry *geometry);

    virtual void SetVertex(CGeometry *geometry, CConfig *config);

    virtual void SetControlVolume(CConfig *config, CGeometry *geometry, unsigned short action);

    virtual void SetBoundControlVolume(CConfig *config, CGeometry *geometry, unsigned short action);

    virtual void SetMeshFile(CConfig *config, string val_mesh_out_filename);
  
    virtual void SetMeshFile(CGeometry *geometry, CConfig *config, string val_mesh_out_filename);

    virtual void SetBoundSensitivity(CConfig *config);

    virtual void SetPeriodicBoundary(CGeometry *geometry, CConfig *config);

    virtual void SetRotationalVelocity(CConfig *config, unsigned short val_iZone);

    virtual void SetTranslationalVelocity(CConfig *config);
    
    virtual void SetGridVelocity(CConfig *config, unsigned long iter);

  virtual void Set_MPI_Coord(CConfig *config);
  
  virtual void Set_MPI_GridVel(CConfig *config);
  
  virtual void Set_MPI_OldCoord(CConfig *config);

    virtual void SetRestricted_GridVelocity(CGeometry *fine_mesh, CConfig *config);

  void ComputeSurf_Curvature(CConfig *config);
  
    void ComputeAirfoil_Section(su2double *Plane_P0, su2double *Plane_Normal,
                                      su2double MinXCoord, su2double MaxXCoord, su2double *FlowVariable,
                                      vector<su2double> &Xcoord_Airfoil, vector<su2double> &Ycoord_Airfoil,
                                      vector<su2double> &Zcoord_Airfoil, vector<su2double> &Variable_Airfoil,
                                      bool original_surface, CConfig *config);
  
  virtual su2double Compute_MaxThickness(su2double *Plane_P0, su2double *Plane_Normal, unsigned short iSection, CConfig *config, vector<su2double> &Xcoord_Airfoil, vector<su2double> &Ycoord_Airfoil, vector<su2double> &Zcoord_Airfoil);
 
  virtual su2double Compute_Twist(su2double *Plane_P0, su2double *Plane_Normal, unsigned short iSection, vector<su2double> &Xcoord_Airfoil, vector<su2double> &Ycoord_Airfoil, vector<su2double> &Zcoord_Airfoil);

  virtual su2double Compute_Chord(su2double *Plane_P0, su2double *Plane_Normal, unsigned short iSection, vector<su2double> &Xcoord_Airfoil, vector<su2double> &Ycoord_Airfoil, vector<su2double> &Zcoord_Airfoil);

    virtual su2double Compute_Thickness(su2double *Plane_P0, su2double *Plane_Normal, unsigned short iSection, su2double Location, CConfig *config, vector<su2double> &Xcoord_Airfoil, vector<su2double> &Ycoord_Airfoil, vector<su2double> &Zcoord_Airfoil);
    
    virtual su2double Compute_Area(su2double *Plane_P0, su2double *Plane_Normal, unsigned short iSection, CConfig *config, vector<su2double> &Xcoord_Airfoil, vector<su2double> &Ycoord_Airfoil, vector<su2double> &Zcoord_Airfoil);
  
  virtual void Compute_LeadingTrailing(su2double *LeadingEdge, su2double *TrailingEdge, su2double *Plane_P0, su2double *Plane_Normal, unsigned short iSection, vector<su2double>
                                           &Xcoord_Airfoil, vector<su2double> &Ycoord_Airfoil, vector<su2double> &Zcoord_Airfoil);

  virtual su2double Compute_Dihedral(su2double *LeadingEdge_im1, su2double *TrailingEdge_im1,
                                     su2double *LeadingEdge_i, su2double *TrailingEdge_i);

  virtual su2double Compute_Curvature(su2double *LeadingEdge_im1, su2double *TrailingEdge_im1,
                                      su2double *LeadingEdge_i, su2double *TrailingEdge_i,
                                      su2double *LeadingEdge_ip1, su2double *TrailingEdge_ip1);

  virtual void Compute_Wing(CConfig *config, bool original_surface,
                            su2double &Wing_Volume, su2double &Wing_MinMaxThickness, su2double &Wing_MaxChord, su2double &Wing_MinToC,
                            su2double &Wing_MaxTwist, su2double &Wing_MaxCurvature, su2double &Wing_MaxDihedral);
  
    virtual void FindNormal_Neighbor(CConfig *config);

    virtual long GetGlobal_to_Local_Point(long val_ipoint);

    virtual unsigned short GetGlobal_to_Local_Marker(unsigned short val_imarker);

    virtual unsigned long GetGlobal_nPoint();
  
    virtual unsigned long GetGlobal_nPointDomain();
  
    virtual unsigned long GetGlobal_nElem();

  virtual unsigned long GetGlobal_nElemDomain();
  
    virtual unsigned long GetGlobal_nElemLine();
  
    virtual unsigned long GetGlobal_nElemTria();
  
    virtual unsigned long GetGlobal_nElemQuad();
  
    virtual unsigned long GetGlobal_nElemTetr();
  
    virtual unsigned long GetGlobal_nElemHexa();
  
    virtual unsigned long GetGlobal_nElemPris();
  
    virtual unsigned long GetGlobal_nElemPyra();
  
    virtual unsigned long GetnElemLine();
  
    virtual unsigned long GetnElemTria();
  
    virtual unsigned long GetnElemQuad();
  
    virtual unsigned long GetnElemTetr();
  
    virtual unsigned long GetnElemHexa();
  
    virtual unsigned long GetnElemPris();
  
    virtual unsigned long GetnElemPyra();

    virtual void SetGeometryPlanes(CConfig *config);
  
    virtual vector<su2double> GetGeometryPlanes();

    virtual vector<vector<su2double> > GetXCoord();

    virtual vector<vector<su2double> > GetYCoord();

    virtual vector<vector<su2double> > GetZCoord();

    virtual vector<vector<unsigned long> > GetPlanarPoints();

    void SetSpline(vector<su2double> &x, vector<su2double> &y, unsigned long n, su2double yp1, su2double ypn, vector<su2double> &y2);
    
    su2double GetSpline(vector<su2double> &xa, vector<su2double> &ya, vector<su2double> &y2a, unsigned long n, su2double x);
      
  bool SegmentIntersectsPlane(su2double *Segment_P0, su2double *Segment_P1, su2double Variable_P0, su2double Variable_P1,
                              su2double *Plane_P0, su2double *Plane_Normal, su2double *Intersection, su2double &Variable_Interp);
  
  bool RayIntersectsTriangle(su2double orig[3], su2double dir[3],
                             su2double vert0[3], su2double vert1[3], su2double vert2[3],
                             su2double *intersect);
  
  bool SegmentIntersectsTriangle(su2double point0[3], su2double point1[3],
                                 su2double vert0[3], su2double vert1[3], su2double vert2[3]);

  bool SegmentIntersectsLine(su2double point0[2], su2double point1[2], su2double vert0[2], su2double vert1[2]);

  void RegisterCoordinates(CConfig *config);

  void UpdateGeometry(CGeometry **geometry_container, CConfig *config);

  virtual void SetSensitivity(CConfig *config);

  virtual su2double GetSensitivity(unsigned long iPoint, unsigned short iDim);

  virtual void SetSensitivity(unsigned long iPoint, unsigned short iDim, su2double val);
  
  virtual void Check_Periodicity(CConfig *config);
  
};

class CPhysicalGeometry : public CGeometry {

  map<long,long> Global_to_Local_Point; 
  long *Local_to_Global_Point;              
  unsigned short *Local_to_Global_Marker;   
  unsigned short *Global_to_Local_Marker;   
  unsigned long *adj_counter; 
  unsigned long **adjacent_elem; 
  su2double* Sensitivity; 
public:
  
    CPhysicalGeometry(void);

    CPhysicalGeometry(CConfig *config, unsigned short val_iZone, unsigned short val_nZone);
  
  CPhysicalGeometry(CGeometry *geometry, CConfig *config);
  
    ~CPhysicalGeometry(void);
  
    void SetSendReceive(CConfig *config);
  
    void SetBoundaries(CConfig *config);
  
    long GetGlobal_to_Local_Point(long val_ipoint);
  
    unsigned short GetGlobal_to_Local_Marker(unsigned short val_imarker);
  
  void Read_SU2_Format_Parallel(CConfig *config, string val_mesh_filename, unsigned short val_iZone, unsigned short val_nZone);
    

  void Read_CGNS_Format_Parallel(CConfig *config, string val_mesh_filename, unsigned short val_iZone, unsigned short val_nZone);

    bool FindFace(unsigned long first_elem, unsigned long second_elem, unsigned short &face_first_elem, 
            unsigned short &face_second_elem);

    void ComputeWall_Distance(CConfig *config);

    void SetPositive_ZArea(CConfig *config);
  
    void SetPoint_Connectivity(void);
  
    void SetRCM_Ordering(CConfig *config);
  
    void SetPoint_Connectivity(CGeometry *geometry);

    void SetElement_Connectivity(void);

    void SetBoundVolume(void);

    void SetVertex(CConfig *config);

    void SetCoord_CG(void);

    void SetControlVolume(CConfig *config, unsigned short action);

  void VisualizeControlVolume(CConfig *config, unsigned short action);
  
    void MatchNearField(CConfig *config);
  
    void MatchActuator_Disk(CConfig *config);

    void MatchInterface(CConfig *config);

    void MatchZone(CConfig *config, CGeometry *geometry_donor, CConfig *config_donor, 
            unsigned short val_iZone, unsigned short val_nZone);

    void SetBoundControlVolume(CConfig *config, unsigned short action);

    void SetTecPlot(char config_filename[MAX_STRING_SIZE], bool new_file);

    void SetBoundTecPlot(char mesh_filename[MAX_STRING_SIZE], bool new_file, CConfig *config);

    void SetBoundSTL(char mesh_filename[MAX_STRING_SIZE], bool new_file, CConfig *config) ;

    void Check_IntElem_Orientation(CConfig *config);
  
    void Check_BoundElem_Orientation(CConfig *config);

    void SetColorGrid(CConfig *config);
  
  void SetColorGrid_Parallel(CConfig *config);
  
    void SetRotationalVelocity(CConfig *config, unsigned short val_iZone);
    
    void SetTranslationalVelocity(CConfig *config);

    void SetGridVelocity(CConfig *config, unsigned long iter);
  
  void Set_MPI_Coord(CConfig *config);
  
  void Set_MPI_GridVel(CConfig *config);
  
  void Set_MPI_OldCoord(CConfig *config);

    void SetPeriodicBoundary(CConfig *config);
  
    void SetCoord_Smoothing(unsigned short val_nSmooth, su2double val_smooth_coeff, CConfig *config);

    void SetMeshFile(CConfig *config, string val_mesh_out_filename);

    void GetQualityStatistics(su2double *statistics);

    void ComputeSurf_Curvature(CConfig *config);

    void FindNormal_Neighbor(CConfig *config);

    unsigned long GetGlobal_nPoint();
  
    unsigned long GetGlobal_nPointDomain();
  
  unsigned long GetGlobal_nElem();
  
  unsigned long GetGlobal_nElemDomain();

    unsigned long GetGlobal_nElemLine();
  
    unsigned long GetGlobal_nElemTria();
  
    unsigned long GetGlobal_nElemQuad();
  
    unsigned long GetGlobal_nElemTetr();
  
    unsigned long GetGlobal_nElemHexa();
  
    unsigned long GetGlobal_nElemPris();
  
    unsigned long GetGlobal_nElemPyra();
  
    unsigned long GetnElemLine();
  
    unsigned long GetnElemTria();
  
    unsigned long GetnElemQuad();
  
    unsigned long GetnElemTetr();
  
    unsigned long GetnElemHexa();
  
    unsigned long GetnElemPris();
  
    unsigned long GetnElemPyra();

    void SetGeometryPlanes(CConfig *config);
  
    vector<su2double> GetGeometryPlanes();

    vector<vector<su2double> > GetXCoord();

    vector<vector<su2double> > GetYCoord();

    vector<vector<su2double> > GetZCoord();

    vector<vector<unsigned long> > GetPlanarPoints();
  
  void SetBoundSensitivity(CConfig *config);
  
  su2double Compute_MaxThickness(su2double *Plane_P0, su2double *Plane_Normal, unsigned short iSection, CConfig *config, vector<su2double> &Xcoord_Airfoil, vector<su2double> &Ycoord_Airfoil, vector<su2double> &Zcoord_Airfoil);
  
  su2double Compute_Twist(su2double *Plane_P0, su2double *Plane_Normal, unsigned short iSection, vector<su2double> &Xcoord_Airfoil, vector<su2double> &Ycoord_Airfoil, vector<su2double> &Zcoord_Airfoil);

  void Compute_LeadingTrailing(su2double *LeadingEdge, su2double *TrailingEdge, su2double *Plane_P0, su2double *Plane_Normal, unsigned short iSection, vector<su2double> &Xcoord_Airfoil,
                               vector<su2double> &Ycoord_Airfoil, vector<su2double> &Zcoord_Airfoil);

  su2double Compute_Chord(su2double *Plane_P0, su2double *Plane_Normal, unsigned short iSection, vector<su2double> &Xcoord_Airfoil, vector<su2double> &Ycoord_Airfoil, vector<su2double> &Zcoord_Airfoil);

  su2double Compute_Thickness(su2double *Plane_P0, su2double *Plane_Normal, unsigned short iSection, su2double Location, CConfig *config, vector<su2double> &Xcoord_Airfoil, vector<su2double> &Ycoord_Airfoil, vector<su2double> &Zcoord_Airfoil);

  su2double Compute_Area(su2double *Plane_P0, su2double *Plane_Normal, unsigned short iSection, CConfig *config, vector<su2double> &Xcoord_Airfoil, vector<su2double> &Ycoord_Airfoil, vector<su2double> &Zcoord_Airfoil);

  su2double Compute_Dihedral(su2double *LeadingEdge_im1, su2double *TrailingEdge_im1,
                             su2double *LeadingEdge_i, su2double *TrailingEdge_i);

  su2double Compute_Curvature(su2double *LeadingEdge_im1, su2double *TrailingEdge_im1,
                              su2double *LeadingEdge_i, su2double *TrailingEdge_i,
                              su2double *LeadingEdge_ip1, su2double *TrailingEdge_ip1);
  
  void Compute_Wing(CConfig *config, bool original_surface,
                    su2double &Wing_Volume, su2double &Wing_MinMaxThickness, su2double &Wing_MaxChord, su2double &Wing_MinToC,
                    su2double &Wing_MaxTwist, su2double &Wing_MaxCurvature, su2double &Wing_MaxDihedral);

  void SetSensitivity(CConfig *config);

  su2double GetSensitivity(unsigned long iPoint, unsigned short iDim);

  void SetSensitivity(unsigned long iPoint, unsigned short iDim, su2double val);
  
  void Check_Periodicity(CConfig *config);

};

class CMultiGridGeometry : public CGeometry {

public:

    CMultiGridGeometry(CGeometry ***geometry, CConfig **config_container, unsigned short iMesh, unsigned short iZone);

    ~CMultiGridGeometry(void);

    bool SetBoundAgglomeration(unsigned long CVPoint, short marker_seed, CGeometry *fine_grid, CConfig *config);

    bool GeometricalCheck(unsigned long iPoint, CGeometry *fine_grid, CConfig *config);

    void SetSuitableNeighbors(vector<unsigned long> *Suitable_Indirect_Neighbors, unsigned long iPoint, 
            unsigned long Index_CoarseCV, CGeometry *fine_grid);

    void SetVertex(CGeometry *geometry, CConfig *config);

    void SetPoint_Connectivity(CGeometry *geometry);

    void SetPoint_Connectivity(void);

    void SetControlVolume(CConfig *config, CGeometry *geometry, unsigned short action);

    void MatchNearField(CConfig *config);
  
    void MatchActuator_Disk(CConfig *config);

    void MatchInterface(CConfig *config);

    void SetBoundControlVolume(CConfig *config, CGeometry *geometry, unsigned short action);

    void SetCoord(CGeometry *geometry);

    void SetRotationalVelocity(CConfig *config, unsigned short val_iZone);
    
    void SetTranslationalVelocity(CConfig *config);

    void SetGridVelocity(CConfig *config, unsigned long iter);

    void SetRestricted_GridVelocity(CGeometry *fine_mesh, CConfig *config);

    void FindNormal_Neighbor(CConfig *config);

    void SetGeometryPlanes(CConfig *config);

    vector<su2double> GetGeometryPlanes();

    vector<vector<su2double> > GetXCoord();

    vector<vector<su2double> > GetYCoord();

    vector<vector<su2double> > GetZCoord();

    vector<vector<unsigned long> > GetPlanarPoints();

};

class CPeriodicGeometry : public CGeometry {
    CPrimalGrid*** newBoundPer;            
    unsigned long *nNewElem_BoundPer;           
public:

    CPeriodicGeometry(CGeometry *geometry, CConfig *config);

    ~CPeriodicGeometry(void);

    void SetPeriodicBoundary(CGeometry *geometry, CConfig *config);

    void SetTecPlot(char config_filename[MAX_STRING_SIZE], bool new_file);

    void SetMeshFile(CGeometry *geometry, CConfig *config, string val_mesh_out_filename);
};

class CMultiGridQueue {
    vector<vector<unsigned long> > QueueCV; 
    short *Priority;    
    bool *RightCV;  
    unsigned long nPoint; 
public:

    CMultiGridQueue(unsigned long val_npoint);

    ~CMultiGridQueue(void);

    void AddCV(unsigned long val_new_point, unsigned short val_number_neighbors);

    void RemoveCV(unsigned long val_remove_point);

    void MoveCV(unsigned long val_move_point, short val_number_neighbors);

    void IncrPriorityCV(unsigned long val_incr_point);

    void RedPriorityCV(unsigned long val_red_point);

    void VisualizeQueue(void);

    void VisualizePriority(void);

    long NextCV(void);

    bool EmptyQueue(void);

    unsigned long TotalCV(void);

    void Update(unsigned long val_update_point, CGeometry *fine_grid);

};

#include "geometry_structure.inl"