began implementing add_hist method

Author: jvleta

src/pyplot_module.f90

@@ -53,7 +53,7 @@ module pyplot_module
         procedure, public :: add_3d_plot   !! add a 3d plot to pyplot instance
         procedure, public :: add_contour   !! add a contour plot to pyplot instance
         procedure, public :: add_bar       !! add a barplot to pyplot instance
-
+        procedure, public :: add_hist      !! add a histogram plot to pyplot instance
         procedure, public :: savefig       !! save plots of pyplot instance
         procedure, public :: destroy       !! destroy pyplot instance
 
@@ -301,6 +301,61 @@ subroutine add_plot(me, x, y, label, linestyle, markersize, linewidth, xlim, yli
     end subroutine add_plot
 !*****************************************************************************************
 
+!*****************************************************************************************
+!> author: Jacob Williams
+!
+! Add a histogram plot.
+
+    subroutine add_hist(me, x, label, xlim, ylim, xscale, yscale)
+
+    class(pyplot),          intent (inout)        :: me           !! pyplot handler
+    real(wp), dimension(:), intent (in)           :: x            !! x values
+    character(len=*),       intent (in)           :: label        !! plot label
+    real(wp),dimension(2),  intent (in), optional :: xlim         !! x-axis range
+    real(wp),dimension(2),  intent (in), optional :: ylim         !! y-axis range
+    character(len=*),       intent (in), optional :: xscale       !! example: 'linear' (default), 'log'
+    character(len=*),       intent (in), optional :: yscale       !! example: 'linear' (default), 'log'
+
+    character(len=:), allocatable :: xstr         !! x values stringified
+    character(len=:), allocatable :: xlimstr      !! xlim values stringified
+    character(len=:), allocatable :: ylimstr      !! ylim values stringified
+    character(len=*), parameter   :: xname = 'x'  !! x variable name for script
+
+    if (allocated(me%str)) then
+
+        !axis limits (optional):
+        if (present(xlim)) call vec_to_string(xlim, me%real_fmt, xlimstr, me%use_numpy)
+        if (present(ylim)) call vec_to_string(ylim, me%real_fmt, ylimstr, me%use_numpy)
+
+        !convert the arrays to strings:
+        call vec_to_string(x, me%real_fmt, xstr, me%use_numpy)
+
+        !write the arrays:
+        call me%add_str(trim(xname)//' = '//xstr)
+        call me%add_str('')
+
+        !write the plot statement:
+        call me%add_str('ax.hist('//&
+                        trim(xname)//','//&
+                        'label="'//trim(label)//'")')
+
+        !axis limits:
+        if (allocated(xlimstr)) call me%add_str('ax.set_xlim('//xlimstr//')')
+        if (allocated(ylimstr)) call me%add_str('ax.set_ylim('//ylimstr//')')
+
+        !axis scales:
+        if (present(xscale)) call me%add_str('ax.set_xscale("'//xscale//'")')
+        if (present(yscale)) call me%add_str('ax.set_yscale("'//yscale//'")')
+
+        call me%add_str('')
+
+    else
+        error stop 'Error in add_plot: pyplot class not properly initialized.'
+    end if
+
+    end subroutine add_hist
+!*****************************************************************************************
+
 !*****************************************************************************************
 !> author: Jacob Williams
 !

src/tests/test.f90

@@ -68,5 +68,21 @@ program test
     call plt%add_contour(x, y, z, label='contour', linestyle='-', linewidth=2, filled=.true., cmap='bone')
     call plt%savefig('contour.png',pyfile='contour.py')
 
+    !histogram chart:
+    do i=1,n 
+      x(i) = 1.0
+    end do
+    
+    write(*,*) x
+    call plt%initialize(grid=.true.,xlabel='x',&
+                        title='hist test',legend=.true.,figsize=[20,10],&
+                        font_size = 20,&
+                        axes_labelsize = 20,&
+                       xtick_labelsize = 20,&
+                        ytick_labelsize = 20,&
+                        legend_fontsize = 20 )
+    call plt%add_hist(x=x, label="x")
+    call plt%savefig('histtest.png', pyfile='histtest.py')
+
     end program test
 !*****************************************************************************************