stl_file.f90

Go to the documentation of this file.

! Copyright (c) 2022-2025, The Neko Authors
! All rights reserved.
!
! Redistribution and use in source and binary forms, with or without
! modification, are permitted provided that the following conditions
! are met:
!
!   * Redistributions of source code must retain the above copyright
!     notice, this list of conditions and the following disclaimer.
!
!   * Redistributions in binary form must reproduce the above
!     copyright notice, this list of conditions and the following
!     disclaimer in the documentation and/or other materials provided
!     with the distribution.
!
!   * Neither the name of the authors nor the names of its
!     contributors may be used to endorse or promote products derived
!     from this software without specific prior written permission.
!
! THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
! "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
! LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
! FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
! COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
! INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
! BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
! LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
! CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
! LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
! ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
! POSSIBILITY OF SUCH DAMAGE.
!
module stl_file
  use num_types, only : rp
  use generic_file, only : generic_file_t
  use tri_mesh, only : tri_mesh_t
  use logger, only : neko_log
  use point, only : point_t
  use neko_mpi_types, only : mpi_stl_header, mpi_stl_triangle
  use mpi_f08, only : mpi_mode_rdonly, mpi_info_null, &
       mpi_file_open, mpi_file_read_all, mpi_file, mpi_status, &
       mpi_file_close
  use utils, only : neko_error
  use comm, only : neko_comm
  use stl, only : stl_hdr_t, stl_triangle_t
  implicit none
  private
 
  type, public, extends(generic_file_t) :: stl_file_t
   contains
     procedure :: read => stl_file_read
     procedure :: write => stl_file_write
  end type stl_file_t
 
contains
 
  subroutine stl_file_write(this, data, t)
    class(stl_file_t), intent(inout) :: this
    class(*), target, intent(in) :: data
    real(kind=rp), intent(in), optional :: t
    call neko_log%error('Not implemented')
  end subroutine stl_file_write
 
  subroutine stl_file_read(this, data)
    class(stl_file_t) :: this
    class(*), target, intent(inout) :: data
    type(tri_mesh_t), pointer :: tri_msh => null()
    type(mpi_status) :: status
    type(mpi_file) :: fh
    type(point_t), target :: p1, p2, p3
    type(stl_hdr_t) :: stl_hdr
    type(stl_triangle_t), allocatable :: stl_tri(:)
    integer :: i, p_idx, ierr
 
    call this%check_exists()
 
    select type (data)
    type is (tri_mesh_t)
       tri_msh => data
    class default
       call neko_log%error('Invalid data')
    end select
 
    call mpi_file_open(neko_comm, trim(this%fname), &
         mpi_mode_rdonly, mpi_info_null, fh, ierr)
    call mpi_file_read_all(fh, stl_hdr, 1, mpi_stl_header, status, ierr)
 
    if (stl_hdr%hdr(1:6) .eq. 'solid') then
       call neko_log%error('Invalid STL file (ASCII)')
    end if
 
    call tri_msh%init(stl_hdr%ntri)
    allocate(stl_tri(stl_hdr%ntri))
 
    call mpi_file_read_all(fh, stl_tri, stl_hdr%ntri, &
         mpi_stl_triangle, status, ierr)
 
    p_idx = 0
    do i = 1, stl_hdr%ntri
       p_idx = p_idx + 1
       p1 = point_t(dble(stl_tri(i)%v1), p_idx)
       p_idx = p_idx + 1
       p2 = point_t(dble(stl_tri(i)%v2), p_idx)
       p_idx = p_idx + 1
       p3 = point_t(dble(stl_tri(i)%v3), p_idx)
       call tri_msh%add_element(p1, p2, p3)
    end do
 
    deallocate(stl_tri)
 
    call mpi_file_close(fh, ierr)
 
  end subroutine stl_file_read
 
end module stl_file