readme and test updates

Author: jacobwilliams

LICENSE.md

@@ -1,7 +1,7 @@
 Polyroots-Fortran: Root solvers for modern Fortran
 <https://github.com/jacobwilliams/polyroots-fortran>
 
-Copyright (c) 2022, Jacob Williams
+Copyright (c) 2022-2024, Jacob Williams
 All rights reserved.
 
 Redistribution and use in source and binary forms, with or without modification,

README.md

@@ -3,6 +3,7 @@
 
 **polyroots-fortran**: Polynomial Roots with Modern Fortran
 
+[![Language](https://img.shields.io/badge/-Fortran-734f96?logo=fortran&logoColor=white)](https://github.com/topics/fortran)
 [![GitHub release](https://img.shields.io/github/release/jacobwilliams/polyroots-fortran.svg)](https://github.com/jacobwilliams/polyroots-fortran/releases/latest)
 [![CI Status](https://github.com/jacobwilliams/polyroots-fortran/actions/workflows/CI.yml/badge.svg)](https://github.com/jacobwilliams/polyroots-fortran/actions)
 [![codecov](https://codecov.io/gh/jacobwilliams/polyroots-fortran/branch/master/graph/badge.svg)](https://codecov.io/gh/jacobwilliams/polyroots-fortran)

fpm.toml

@@ -1,6 +1,6 @@
 name = "polyroots-fortran"
 author = "Jacob Williams"
-copyright  = "Copyright (c) 2022, Jacob Williams"
+copyright  = "Copyright (c) 2022-2024, Jacob Williams"
 license = "BSD-3"
 description = "Polynomial Roots with Modern Fortran"
 homepage = "https://github.com/jacobwilliams/polyroots-fortran"

test/polyroots_test_10.f90

@@ -4,24 +4,40 @@
 
     program polyroots_test_10
 
-    use polyroots_module, only: dpolz, wp => polyroots_module_rk
+    use polyroots_module, only: polyroots, dpolz, wp => polyroots_module_rk
     use pyplot_module,    only: pyplot
 
     implicit none
 
-    integer,parameter :: degree = 10 !! polynomial degree
-    integer,dimension(*),parameter :: icoeffs = [-1,1] !! set of coefficients
-    integer,dimension (degree+1) :: a !! coefficients of polynomial
-    real(wp),dimension(degree) :: zr, zi !! roots
+    integer :: degree = 5 !! polynomial degree
+    integer,dimension(2) :: icoeffs = [-1,1] !! set of coefficients
     integer :: ierr !! error code from [[dpolz]]
     type(pyplot) :: plt !! for making the plot
-
-    call plt%initialize(grid=.true.,xlabel='$\Re(z)$',ylabel='$\Im(z)$',&
-                        title='Degree 10 Polynomial Roots',usetex=.true.,&
-                        figsize=[20,10])
-
-    call generate(1)
-    call plt%savefig('roots.png')
+    integer :: i ,j
+    character(len=5) :: istr
+    integer,dimension (:),allocatable :: a !! coefficients of polynomial
+    real(wp),dimension(:),allocatable :: zr, zi !! roots
+
+    do i = 2, 10, 2
+
+        write(istr,'(I5)') i; istr = adjustl(istr)
+        degree = i
+        ! resize
+        a  = [(0, j = 1, degree+1)]
+        zr = [(0, j = 1, degree)]
+        zi = [(0, j = 1, degree)]
+
+        !icoeffs = icoeffs + 1
+        write(*,*) 'degree = ', degree
+        call plt%initialize(grid=.true.,xlabel='$\Re(z)$',ylabel='$\Im(z)$',&
+                            title='Degree '//trim(istr)//' Polynomial Roots',usetex=.true.,&
+                            font_size=25, axes_labelsize=25, &
+                            xtick_labelsize=25, ytick_labelsize=25, &
+                            figsize=[20,10])
+        call generate(1)
+        call plt%savefig('roots_'//trim(istr)//'.png')
+
+    end do
 
     contains
 
@@ -30,10 +46,11 @@ recursive subroutine generate (i)
         integer :: ix
         if (i > degree+1) then
             !write (*, '(*(I2,","))') a
-            call dpolz(degree,real(a,wp),zr,zi,ierr); if (ierr/=0) error stop ierr
-            call plt%add_plot(zr,zi,label='',linestyle='bo',markersize=1)
+            call polyroots(degree,real(a,wp),zr,zi,ierr)  !polyroots !! dpolz
+            if (ierr/=0) return !error stop ierr
+            call plt%add_plot(zr,zi,label='',linestyle='bo',markersize=2)
         else
-            do ix = 1,size(icoeffs) 
+            do ix = 1,size(icoeffs)
                 a(i) = icoeffs(ix)
                 call generate(i+1)
             end do