src/FrontTrack/FrontTrack.cxx

   1 // File      : FrontTrack.cxx
   2 // Created   : Tue Apr 25 17:20:28 2017
   3 // Author    : Edward AGAPOV (eap)
   4
   5 #include "FrontTrack.hxx"
   6 #include "FrontTrack_NodeGroups.hxx"
   7 #include "FrontTrack_Utils.hxx"
   8
   9 #include <MCAuto.hxx>
  10 #include <MEDCouplingMemArray.hxx>
  11 #include <MEDFileMesh.hxx>
  12
  13 #include <XAO_Xao.hxx>
  14 #include <XAO_BrepGeometry.hxx>
  15
  16 #include <stdexcept>
  17
  18 #include <OSD_Parallel.hxx>
  19
  20 /*!
  21  * \brief Relocate nodes to lie on geometry
  22  *  \param [in] theInputMedFile - a MED file holding a mesh including nodes that will be
  23  *         moved onto the geometry
  24  *  \param [in] theOutputMedFile - a MED file to create, that will hold a modified mesh
  25  *  \param [in] theNodeFiles - an array of names of files describing groups of nodes that
  26  *         will be moved onto the geometry
  27  *  \param [in] theXaoFileName - a path to a file in XAO format holding the geometry and
  28  *         the geometrical groups.
  29  *  \param [in] theIsParallel - if \c true, all processors are used to treat boundary shapes
  30  *          in parallel.
  31  */
  32 void FrontTrack::track( const std::string&                 theInputMedFile,
  33                         const std::string&                 theOutputMedFile,
  34                         const std::vector< std::string > & theNodeFiles,
  35                         const std::string&                 theXaoFileName,
  36                         bool                               theIsParallel )
  37 {
  38   // check arguments
  39
  40   if ( theNodeFiles.empty() )
  41     return;
  42
  43   if ( !FT_Utils::fileExists( theInputMedFile ))
  44     throw std::invalid_argument( "Input MED file does not exist: " + theInputMedFile );
  45
  46   if ( !FT_Utils::canWrite( theOutputMedFile ))
  47     throw std::invalid_argument( "Can't create the output MED file: " + theOutputMedFile );
  48
  49   for ( size_t i = 0; i < theNodeFiles.size(); ++i )
  50     if ( !FT_Utils::fileExists( theNodeFiles[i] ))
  51       throw std::invalid_argument( "Input node file does not exist: " + theNodeFiles[i] );
  52
  53   if ( !FT_Utils::fileExists( theXaoFileName ))
  54     throw std::invalid_argument( "Input XAO file does not exist: " + theXaoFileName );
  55
  56
  57   // read a mesh
  58
  59   MEDCoupling::MCAuto< MEDCoupling::MEDFileUMesh >
  60     mfMesh( MEDCoupling::MEDFileUMesh::New( theInputMedFile ));
  61   if ( mfMesh.isNull() )
  62     throw std::invalid_argument( "Failed to read the input MED file: " + theInputMedFile );
  63
  64   MEDCoupling::DataArrayDouble * nodeCoords = mfMesh->getCoords();
  65   if ( !nodeCoords || nodeCoords->empty() )
  66     throw std::invalid_argument( "No nodes in the input mesh" );
  67
  68
  69   // read a geometry
  70
  71   XAO::Xao xao;
  72   if ( !xao.importXAO( theXaoFileName ) || !xao.getGeometry() )
  73     throw std::invalid_argument( "Failed to read the XAO input file: " + theXaoFileName );
  74
  75   XAO::BrepGeometry* xaoGeom = dynamic_cast<XAO::BrepGeometry*>( xao.getGeometry() );
  76   if ( !xaoGeom || xaoGeom->getTopoDS_Shape().IsNull() )
  77     throw std::invalid_argument( "Failed to get a BREP shape from the XAO input file" );
  78
  79
  80   // read groups of nodes and associate them with boundary shapes using names (no projection so far)
  81
  82   FT_NodeGroups nodeGroups;
  83   nodeGroups.read( theNodeFiles, &xao, nodeCoords );
  84
  85
  86   // project nodes to the boundary shapes and change their coordinates
  87
  88   OSD_Parallel::For( 0, nodeGroups.nbOfGroups(), nodeGroups, !theIsParallel );
  89
  90   // save the modified mesh
  91
  92   const int erase = 2;
  93   mfMesh->write( theOutputMedFile, /*mode=*/erase );
  94
  95   if ( !nodeGroups.isOK() )
  96     throw std::runtime_error("Unable to project some nodes");
  97 }