zsvd Subroutine

subroutine zsvd(A, s, U, Vtransp)
  ! complex m x m matrix A
  ! U is m x min(m, n)
  ! Vtransp is n x n
  ! sigma is m x n with with sigma_ii = s_i
  ! note that this routine returns V^H, not V!
  complex(8), intent(in)  :: A(:,:)
  real(8), intent(out)    :: s(:)
  complex(8), intent(out) :: U(:,:), Vtransp(:,:)
  integer                 :: info, lwork, m, n, ldu, lrwork
  real(8), allocatable    :: rwork(:)
  complex(8), allocatable :: work(:), At(:,:)
  ! TODO: check shapes here and in other routines?
  m = size(A(:,1))  ! = lda
  n = size(A(1,:))
  ldu = m
  lrwork = 5*min(m,n)
  allocate(rwork(lrwork), At(m,n))
  At(:,:) = A(:,:)  ! use a temporary as zgesvd destroys its input
  call assert_shape(U, [m, m], "svd", "U")
  call assert_shape(Vtransp, [n, n], "svd", "Vtransp")
  ! query optimal lwork and allocate workspace:
  allocate(work(1))
  call zgesvd('A', 'A', m, n, At, m, s, U, ldu, Vtransp, n, work, -1,&
       rwork, info)
  lwork = int(real(work(1)))
  deallocate(work)
  allocate(work(lwork))
  call zgesvd('A', 'A', m, n, At, m, s, U, ldu, Vtransp, n, work, &
       lwork, rwork, info)
  if(info /= 0) then
     print *, "zgesvd returned info = ", info
     if(info < 0) then
        print *, "the ", -info, "-th argument had an illegal value"
     else
        print *, "ZBDSQR did not converge, there are ", info
        print *, "superdiagonals of an intermediate bidiagonal form B"
        print *, "did not converge to zero. See the description of WORK"
        print *, "in DGESVD's man page for details."
     endif
     stop 'zsvd error: zgesvd'
  endif
end subroutine zsvd