#! /usr/bin/env python3 # def newton_coef_1d ( nd, xd, yd ): #*****************************************************************************80 # ## newton_coef_1d() computes coefficients of a Newton 1D interpolant. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 10 July 2015 # # Author: # # John Burkardt # # Reference: # # Carl deBoor, # A Practical Guide to Splines, # Springer, 2001, # ISBN: 0387953663, # LC: QA1.A647.v27. # # Input: # # integer ND, the number of data values. # # real XD(ND), the X values at which data was taken. # # real YD(ND), the corresponding Y values. # # Output: # # real CD(ND), the divided difference # coefficients corresponding to the input (XTAB,YTAB). # import numpy as np cd = np.zeros ( nd, dtype = np.float64 ) for i in range ( 0, nd ): cd[i] = yd[i] for i in range ( 1, nd ): for j in range ( nd - 1, i - 1, -1 ): cd[j] = ( cd[j] - cd[j-1] ) / ( xd[j] - xd[j-i] ) return cd def newton_coef_1d_test ( ): #*****************************************************************************80 # ## newton_coef_1d_test() tests newton_coef_1d(). # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 10 July 2015 # # Author: # # John Burkardt # import numpy as np nd = 5 xd = np.array ( [ 0.0, 1.0, 2.0, 3.0, 4.0 ] ) yd = np.array ( [ 24.0, 0.0, 0.0, 0.0, 0.0 ] ) print ( '' ) print ( 'newton_coef_1d_test' ) print ( ' newton_coef_1d sets the coefficients for a 1D Newton interpolation.' ) r8vec2_print ( nd, xd, yd, ' Interpolation data:' ) cd = newton_coef_1d ( nd, xd, yd ) r8vec_print ( nd, cd, ' Newton interpolant coefficients:' ) # # Terminate. # print ( '' ) print ( 'newton_coef_1d_test:' ) print ( ' Normal end of execution.' ) return def newton_interp_1d_test ( ): #*****************************************************************************80 # ## newton_interp_1d_test() tests newton_interp_1d(). # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 12 July 2015 # # Author: # # John Burkardt # print ( '' ) print ( 'newton_interp_1d_test():' ) print ( ' Python version:' ) print ( ' Test newton_interp_1d().' ) # # Utility functions. # p00_data_test ( ) p00_data_num_test ( ) p00_dim_num_test ( ) p00_prob_num_test ( ) # # Library functions. # newton_coef_1d_test ( ) newton_value_1d_test ( ) prob_num = p00_prob_num ( ); for prob in range ( 1, prob_num + 1 ): newton_interp_1d_test01 ( prob ) # # Terminate. # print ( '' ) print ( 'newton_interp_1d_test:' ) print ( ' Normal end of execution.' ) return def newton_interp_1d_test01 ( prob ): #*****************************************************************************80 # ## newton_interp_1d_test01() tests newton_value_1d(). # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 01 July 2015 # # Author: # # John Burkardt # # Input: # # integer PROB, the problem index. # # integer ND, the number of data points to use. # import matplotlib.pyplot as plt import numpy as np print ( '' ) print ( 'newton_interp_1d_test01:' ) print ( ' Interpolate data from TEST_interp problem #%d.' % ( prob ) ) nd = p00_data_num ( prob ) print ( ' Number of data points = %d' % ( nd ) ) xy = p00_data ( prob, 2, nd ) xd = np.zeros ( nd ) yd = np.zeros ( nd ) for i in range ( 0, nd ): xd[i] = xy[0,i] yd[i] = xy[1,i] if ( nd < 10 ): r8vec2_print ( nd, xd, yd, ' Data array:' ) # # Get interpolant coefficients. # cd = newton_coef_1d ( nd, xd, yd ) # # #1: Does interpolant match function at interpolation points? # ni = nd xi = xd yi = newton_value_1d ( nd, xd, cd, ni, xi ) int_error = np.linalg.norm ( yi - yd ) / float ( ni ) print ( '' ) print ( ' L2 interpolation error averaged per interpolant node = %g' % ( int_error ) ) # # #2: Compare estimated curve length to piecewise linear (minimal) curve length. # Assume data is sorted, and normalize X and Y dimensions by (XMAX-XMIN) and # (YMAX-YMIN). # xmin = np.min ( xd ) xmax = np.max ( xd ) ymin = np.min ( yd ) ymax = np.max ( yd ) ni = 501 xi = np.linspace ( xmin, xmax, ni ) yi = newton_value_1d ( nd, xd, cd, ni, xi ) ld = 0.0 for i in range ( 0, nd - 1 ): ld = ld + np.sqrt \ ( \ ( ( xd[i+1] - xd[i] ) / ( xmax - xmin ) ) ** 2 \ + ( ( yd[i+1] - yd[i] ) / ( ymax - ymin ) ) ** 2 \ ) li = 0.0 for i in range ( 0, ni - 1 ): li = li + np.sqrt \ ( \ ( ( xi[i+1] - xi[i] ) / ( xmax - xmin ) ) ** 2 \ + ( ( yi[i+1] - yi[i] ) / ( ymax - ymin ) ) ** 2 \ ) li = np.sqrt ( li ) print ( '' ) print ( ' Normalized length of piecewise linear interpolant = %g' % ( ld ) ) print ( ' Normalized length of Shepard interpolant = %g' % ( li ) ) # # #3: Plot the data. # plt.plot ( xd, yd, 'b-', linewidth = 3.0 ) plt.plot ( xd, yd, 'k.', markersize = 10 ) t = 'p0' + str ( prob ) + ' Piecewise Linear Interpolant' plt.title ( t ) plt.grid ( True ) plt.xlabel ( '<---X--->' ) plt.ylabel ( '<---Y--->' ) filename = 'p0' + str ( prob ) + '_data.png' plt.savefig ( filename ) plt.show ( block = False ) plt.close ( ) print ( '' ) print ( ' Created plot file "%s"' % ( filename ) ) # # #4: Plot the piecewise linear and Vandermonde interpolants. # ni = 501 xmin = np.min ( xd ) xmax = np.max ( xd ) xi = np.linspace ( xmin, xmax, ni ) yi = newton_value_1d ( nd, xd, cd, ni, xi ) plt.plot ( xd, yd, 'b-', linewidth = 3.0 ) plt.plot ( xi, yi, 'r-', linewidth = 4.0 ) plt.plot ( xd, yd, 'k.', markersize = 10 ) t = 'p0' + str ( prob ) + ' Newton Interpolant for ' + str ( nd ) + ' nodes.' plt.title ( t ) plt.grid ( True ) plt.xlabel ( '<---X--->' ) plt.ylabel ( '<---Y--->' ) filename = 'p0' + str ( prob ) + '_newton.png' plt.savefig ( filename ) plt.show ( block = False ) plt.close ( ) print ( ' Created plot file "%s".' % ( filename ) ) # # Terminate. # print ( '' ) print ( 'newton_interp_1d_test01:' ) print ( ' Normal end of execution.' ) return def newton_value_1d ( nd, xd, cd, ni, xi ): #*****************************************************************************80 # ## newton_value_1d() evaluates a Newton 1D interpolant. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 11 July 2015 # # Author: # # John Burkardt # # Reference: # # Carl deBoor, # A Practical Guide to Splines, # Springer, 2001, # ISBN: 0387953663, # LC: QA1.A647.v27. # # Input: # # integer ND, the order of the difference table. # # real XD(ND), the X values of the difference table. # # real CD(ND), the divided differences. # # integer NI, the number of interpolation points. # # real XI(NI), the interpolation points. # # Output: # # real YI(NI), the interpolation values. # import numpy as np yi = np.zeros ( ni ) for i in range ( 0, ni ): yi[i] = cd[nd-1] for j in range ( 1, nd ): for i in range ( 0, ni ): yi[i] = cd[nd-j-1] + ( xi[i] - xd[nd-j-1] ) * yi[i] return yi def newton_value_1d_test ( ): #*****************************************************************************80 # ## newton_value_1d_test() tests newton_value_1d(). # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 11 July 2015 # # Author: # # John Burkardt # import numpy as np print ( '' ) print ( 'newton_value_1d_test' ) print ( ' newton_value_1d evaluates a Newton 1d interpolant.' ) nd = 5 xd = np.array ( [ 0.0, 1.0, 2.0, 3.0, 4.0 ] ) cd = np.array ( [ 24.0, -24.0, +12.0, -4.0, 1.0 ] ) r8vec2_print ( nd, xd, cd, ' The Newton interpolant data:' ) ni = 16 x_lo = 0.0 x_hi = 5.0 xi = np.linspace ( x_lo, x_hi, ni ) yi = newton_value_1d ( nd, xd, cd, ni, xi ) r8vec2_print ( ni, xi, yi, ' Newton interpolant values:' ) # # Terminate. # print ( '' ) print ( 'newton_value_1d_test:' ) print ( ' Normal end of execution.' ) return def p00_data_num ( prob ): #*****************************************************************************80 # ## p00_data_num() returns the number of data points for any problem. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Input: # # integer PROB, the problem index. # # Output: # # integer DATA_num, the number of data points. # if ( prob == 1 ): data_num = p01_data_num ( ) elif ( prob == 2 ): data_num = p02_data_num ( ) elif ( prob == 3 ): data_num = p03_data_num ( ) elif ( prob == 4 ): data_num = p04_data_num ( ) elif ( prob == 5 ): data_num = p05_data_num ( ) elif ( prob == 6 ): data_num = p06_data_num ( ) elif ( prob == 7 ): data_num = p07_data_num ( ) elif ( prob == 8 ): data_num = p08_data_num ( ) else: print ( '' ) print ( 'p00_data_num - Fatal error!' ) print ( ' Unexpected input value of PROB.' ) raise Exception ( 'p00_data_num - Fatal error!' ) return data_num def p01_data_num ( ): #*****************************************************************************80 # ## p01_data_num() returns the number of data points for problem p01. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Output: # # integer DATA_num, the number of data points. # data_num = 18 return data_num def p02_data_num ( ): #*****************************************************************************80 # ## p02_data_num() returns the number of data points for problem p02. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Output: # # integer DATA_num, the number of data points. # data_num = 18 return data_num def p03_data_num ( ): #*****************************************************************************80 # ## p03_data_num() returns the number of data points for problem p03. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Output: # # integer DATA_num, the number of data points. # data_num = 11 return data_num def p04_data_num ( ): #*****************************************************************************80 # ## p04_data_num() returns the number of data points for problem p04. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Output: # # integer DATA_num, the number of data points. # data_num = 8 return data_num def p05_data_num ( ): #*****************************************************************************80 # ## p05_data_num() returns the number of data points for problem p05. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Output: # # integer DATA_num, the number of data points. # data_num = 9 return data_num def p06_data_num ( ): #*****************************************************************************80 # ## p06_data_num() returns the number of data points for problem p06. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Output: # # integer DATA_num, the number of data points. # data_num = 49 return data_num def p07_data_num ( ): #*****************************************************************************80 # ## p07_data_num() returns the number of data points for problem p07. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Output: # # integer DATA_num, the number of data points. # data_num = 4 return data_num def p08_data_num ( ): #*****************************************************************************80 # ## p08_data_num() returns the number of data points for problem p08. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Output: # # integer DATA_num, the number of data points. # data_num = 12 return data_num def p00_data_num_test ( ): #*****************************************************************************80 # ## p00_data_num_test() tests p00_data_num. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # print ( '' ) print ( 'p00_data_num_test' ) print ( ' p00_data_num() returns the number of data points for any problem.' ) print ( '' ) print ( ' Problem Data Num' ) print ( '' ) prob_num = p00_prob_num ( ) for prob in range ( 1, prob_num + 1 ): data_num = p00_data_num ( prob ) print ( ' %7d %9d' % ( prob, data_num ) ) # # Terminate. # print ( '' ) print ( 'p00_data_num_test:' ) print ( ' Normal end of execution.' ) return def p00_data ( prob, dim_num, data_num ): #*****************************************************************************80 # ## p00_data() returns the data for any problem. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Input: # # integer PROB, the problem index. # # integer DIM_num, the spatial dimension of the dependent # variables. # # integer DATA_num, the number of data points. # # Output: # # real P_data(DIM_num,DATA_num), the data. # if ( prob == 1 ): p_data = p01_data ( dim_num, data_num ) elif ( prob == 2 ): p_data = p02_data ( dim_num, data_num ) elif ( prob == 3 ): p_data = p03_data ( dim_num, data_num ) elif ( prob == 4 ): p_data = p04_data ( dim_num, data_num ) elif ( prob == 5 ): p_data = p05_data ( dim_num, data_num ) elif ( prob == 6 ): p_data = p06_data ( dim_num, data_num ) elif ( prob == 7 ): p_data = p07_data ( dim_num, data_num ) elif ( prob == 8 ): p_data = p08_data ( dim_num, data_num ) else: print ( '' ) print ( 'p00_data - Fatal error!' ) print ( ' Unexpected input value of PROB.' ) raise Exception ( 'p00_data - Fatal error!' ) return p_data def p01_data ( dim_num, data_num ): #*****************************************************************************80 # ## p01_data() returns the data for problem p01. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Input: # # integer DIM_num, the spatial dimension of the dependent # variables. # # integer DATA_num, the number of data points. # # Output: # # real P_data(DIM_num,DATA_num), the data. # import numpy as np p_data = np.array ( [ \ [ 0.0, 4.0 ], \ [ 1.0, 5.0 ], \ [ 2.0, 6.0 ], \ [ 4.0, 6.0 ], \ [ 5.0, 5.0 ], \ [ 6.0, 3.0 ], \ [ 7.0, 1.0 ], \ [ 8.0, 1.0 ], \ [ 9.0, 1.0 ], \ [ 10.0, 3.0 ], \ [ 11.0, 4.0 ], \ [ 12.0, 4.0 ], \ [ 13.0, 3.0 ], \ [ 14.0, 3.0 ], \ [ 15.0, 4.0 ], \ [ 16.0, 4.0 ], \ [ 17.0, 3.0 ], \ [ 18.0, 0.0 ] ] ) p_data = np.transpose ( p_data ) return p_data def p02_data ( dim_num, data_num ): #*****************************************************************************80 # ## p02_data() returns the data for problem p02. # # Discussion: # # Two pairs of identical X values have now been slightly separated. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Input: # # integer DIM_num, the spatial dimension of the dependent # variables. # # integer DATA_num, the number of data points. # # Output: # # real P_data(DIM_num,DATA_num), the data. # import numpy as np p_data = np.array ( [ \ [ 0.00, 0.00 ], \ [ 1.34, 5.00 ], \ [ 5.00, 8.66 ], \ [ 10.00, 10.00 ], \ [ 10.60, 10.40 ], \ [ 10.70, 12.00 ], \ [ 10.705, 28.60 ], \ [ 10.80, 30.20 ], \ [ 11.40, 30.60 ], \ [ 19.60, 30.60 ], \ [ 20.20, 30.20 ], \ [ 20.295, 28.60 ], \ [ 20.30, 12.00 ], \ [ 20.40, 10.40 ], \ [ 21.00, 10.00 ], \ [ 26.00, 8.66 ], \ [ 29.66, 5.00 ], \ [ 31.00, 0.00 ] ] ) p_data = np.transpose ( p_data ) return p_data def p03_data ( dim_num, data_num ): #*****************************************************************************80 # ## p03_data() returns the data for problem p03. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Input: # # integer DIM_num, the spatial dimension of the dependent # variables. # # integer DATA_num, the number of data points. # # Output: # # real P_data(DIM_num,DATA_num), the data. # import numpy as np p_data = np.array ( [ \ [ 0.0, 0.0 ], \ [ 2.0, 10.0 ], \ [ 3.0, 10.0 ], \ [ 5.0, 10.0 ], \ [ 6.0, 10.0 ], \ [ 8.0, 10.0 ], \ [ 9.0, 10.5 ], \ [ 11.0, 15.0 ], \ [ 12.0, 50.0 ], \ [ 14.0, 60.0 ], \ [ 15.0, 85.0 ] ] ) p_data = np.transpose ( p_data ) return p_data def p04_data ( dim_num, data_num ): #*****************************************************************************80 # ## p04_data() returns the data for problem p04. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Input: # # integer DIM_num, the spatial dimension of the dependent # variables. # # integer DATA_num, the number of data points. # # Output: # # real P_data(DIM_num,DATA_num), the data. # import numpy as np p_data = np.array ( [ \ [ 0.00, 0.00 ], \ [ 0.05, 0.70 ], \ [ 0.10, 1.00 ], \ [ 0.20, 1.00 ], \ [ 0.80, 0.30 ], \ [ 0.85, 0.05 ], \ [ 0.90, 0.10 ], \ [ 1.00, 1.00 ] ] ) p_data = np.transpose ( p_data ) return p_data def p05_data ( dim_num, data_num ): #*****************************************************************************80 # ## p05_data() returns the data for problem p05. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Input: # # integer DIM_num, the spatial dimension of the dependent # variables. # # integer DATA_num, the number of data points. # # Output: # # real P_data(DIM_num,DATA_num), the data. # import numpy as np p_data = np.array ( [ \ [ 0.00, 0.00 ], \ [ 0.10, 0.90 ], \ [ 0.20, 0.95 ], \ [ 0.30, 0.90 ], \ [ 0.40, 0.10 ], \ [ 0.50, 0.05 ], \ [ 0.60, 0.05 ], \ [ 0.80, 0.20 ], \ [ 1.00, 1.00 ] ] ) p_data = np.transpose ( p_data ) return p_data def p06_data ( dim_num, data_num ): #*****************************************************************************80 # ## p06_data() returns the data for problem p06. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Input: # # integer DIM_num, the spatial dimension of the dependent # variables. # # integer DATA_num, the number of data points. # # Output: # # real P_data(DIM_num,DATA_num), the data. # import numpy as np p_data = np.array ( [ \ [ 595.0, 0.644 ], \ [ 605.0, 0.622 ], \ [ 615.0, 0.638 ], \ [ 625.0, 0.649 ], \ [ 635.0, 0.652 ], \ [ 645.0, 0.639 ], \ [ 655.0, 0.646 ], \ [ 665.0, 0.657 ], \ [ 675.0, 0.652 ], \ [ 685.0, 0.655 ], \ [ 695.0, 0.644 ], \ [ 705.0, 0.663 ], \ [ 715.0, 0.663 ], \ [ 725.0, 0.668 ], \ [ 735.0, 0.676 ], \ [ 745.0, 0.676 ], \ [ 755.0, 0.686 ], \ [ 765.0, 0.679 ], \ [ 775.0, 0.678 ], \ [ 785.0, 0.683 ], \ [ 795.0, 0.694 ], \ [ 805.0, 0.699 ], \ [ 815.0, 0.710 ], \ [ 825.0, 0.730 ], \ [ 835.0, 0.763 ], \ [ 845.0, 0.812 ], \ [ 855.0, 0.907 ], \ [ 865.0, 1.044 ], \ [ 875.0, 1.336 ], \ [ 885.0, 1.881 ], \ [ 895.0, 2.169 ], \ [ 905.0, 2.075 ], \ [ 915.0, 1.598 ], \ [ 925.0, 1.211 ], \ [ 935.0, 0.916 ], \ [ 945.0, 0.746 ], \ [ 955.0, 0.672 ], \ [ 965.0, 0.627 ], \ [ 975.0, 0.615 ], \ [ 985.0, 0.607 ], \ [ 995.0, 0.606 ], \ [1005.0, 0.609 ], \ [1015.0, 0.603 ], \ [1025.0, 0.601 ], \ [1035.0, 0.603 ], \ [1045.0, 0.601 ], \ [1055.0, 0.611 ], \ [1065.0, 0.601 ], \ [1075.0, 0.608 ] ] ) p_data = np.transpose ( p_data ) return p_data def p07_data ( dim_num, data_num ): #*****************************************************************************80 # ## p07_data() returns the data for problem p07. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Input: # # integer DIM_num, the spatial dimension of the dependent # variables. # # integer DATA_num, the number of data points. # # Output: # # real P_data(DIM_num,DATA_num), the data. # import numpy as np p_data = np.array ( [ \ [ 0.0, 1.0 ], \ [ 1.0, 2.0 ], \ [ 4.0, 2.0 ], \ [ 5.0, 1.0 ] ] ) p_data = np.transpose ( p_data ) return p_data def p08_data ( dim_num, data_num ): #*****************************************************************************80 # ## p08_data() returns the data for problem p08. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # # Input: # # integer DIM_num, the spatial dimension of the dependent # variables. # # integer DATA_num, the number of data points. # # Output: # # real P_data(DIM_num,DATA_num), the data. # import numpy as np p_data = np.array ( [ \ [ -1.0, 1.00 ], \ [ -0.8, 0.64 ], \ [ -0.6, 0.36 ], \ [ -0.4, 0.16 ], \ [ -0.2, 0.04 ], \ [ 0.0, 0.00 ], \ [ 0.2, 0.04 ], \ [ 0.20001, 0.05 ], \ [ 0.4, 0.16 ], \ [ 0.6, 0.36 ], \ [ 0.8, 0.64 ], \ [ 1.0, 1.00 ] ] ) p_data = np.transpose ( p_data ) return p_data def p00_data_test ( ): #*****************************************************************************80 # ## p00_data_test() tests p00_data. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 29 June 2015 # # Author: # # John Burkardt # print ( '' ) print ( 'p00_data_test() tests p00_data' ) print ( ' p00_data() returns the actual (MxN) data for any problem.' ) prob_num = p00_prob_num ( ) for prob in range ( 1, prob_num + 1 ): print ( '' ) print ( ' Problem %d' % ( prob ) ) data_num = p00_data_num ( prob ) print ( ' DATA_num = %d' % ( data_num ) ) dim_num = p00_dim_num ( prob ) print ( ' DIM_num = %d' % ( dim_num ) ) p = p00_data ( prob, dim_num, data_num ) r8mat_transpose_print ( dim_num, data_num, p, ' Data array:' ) # # Terminate. # print ( '' ) print ( 'p00_data_test:' ) print ( ' Normal end of execution.' ) return def p00_dim_num ( prob ): #*****************************************************************************80 # ## p00_dim_num() returns the spatial dimension for any problem. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 28 June 2015 # # Author: # # John Burkardt # # Input: # # integer PROB, the problem index. # # Output: # # integer DIM_num, the spatial dimension of the # dependent variables. # if ( prob == 1 ): dim_num = p01_dim_num ( ) elif ( prob == 2 ): dim_num = p02_dim_num ( ) elif ( prob == 3 ): dim_num = p03_dim_num ( ) elif ( prob == 4 ): dim_num = p04_dim_num ( ) elif ( prob == 5 ): dim_num = p05_dim_num ( ) elif ( prob == 6 ): dim_num = p06_dim_num ( ) elif ( prob == 7 ): dim_num = p07_dim_num ( ) elif ( prob == 8 ): dim_num = p08_dim_num ( ) else: print ( '' ) print ( 'p00_dim_num - Fatal error!' ) print ( ' Unexpected input value of PROB.' ) raise Exception ( 'p00_dim_num - Fatal error!' ) return dim_num def p01_dim_num ( ): #*****************************************************************************80 # ## p01_dim_num() returns the spatial dimension for problem p01. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 28 June 2015 # # Author: # # John Burkardt # # Output: # # integer DIM_num, the spatial dimension of the # dependent variables. # dim_num = 2 return dim_num def p02_dim_num ( ): #*****************************************************************************80 # ## p02_dim_num() returns the spatial dimension for problem p02. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 28 June 2015 # # Author: # # John Burkardt # # Output: # # integer DIM_num, the spatial dimension of the # dependent variables. # dim_num = 2 return dim_num def p03_dim_num ( ): #*****************************************************************************80 # ## p03_dim_num() returns the spatial dimension for problem p03. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 28 June 2015 # # Author: # # John Burkardt # # Output: # # integer DIM_num, the spatial dimension of the # dependent variables. # dim_num = 2 return dim_num def p04_dim_num ( ): #*****************************************************************************80 # ## p04_dim_num() returns the spatial dimension for problem p04. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 28 June 2015 # # Author: # # John Burkardt # # Output: # # integer DIM_num, the spatial dimension of the # dependent variables. # dim_num = 2 return dim_num def p05_dim_num ( ): #*****************************************************************************80 # ## p05_dim_num() returns the spatial dimension for problem p05. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 28 June 2015 # # Author: # # John Burkardt # # Output: # # integer DIM_num, the spatial dimension of the # dependent variables. # dim_num = 2 return dim_num def p06_dim_num ( ): #*****************************************************************************80 # ## p06_dim_num() returns the spatial dimension for problem p06. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 28 June 2015 # # Author: # # John Burkardt # # Output: # # integer DIM_num, the spatial dimension of the # dependent variables. # dim_num = 2 return dim_num def p07_dim_num ( ): #*****************************************************************************80 # ## p07_dim_num() returns the spatial dimension for problem p07. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 28 June 2015 # # Author: # # John Burkardt # # Output: # # integer DIM_num, the spatial dimension of the # dependent variables. # dim_num = 2 return dim_num def p08_dim_num ( ): #*****************************************************************************80 # ## p08_dim_num() returns the spatial dimension for problem p08. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 28 June 2015 # # Author: # # John Burkardt # # Output: # # integer DIM_num, the spatial dimension of the # dependent variables. # dim_num = 2 return dim_num def p00_dim_num_test ( ): #*****************************************************************************80 # ## p00_dim_num_test() tests p00_dim_num. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 28 June 2015 # # Author: # # John Burkardt # print ( '' ) print ( 'p00_dim_num_test' ) print ( ' p00_dim_num() returns the spatial dimension for any problem.' ) print ( '' ) print ( ' Problem Dimension' ) print ( '' ) prob_num = p00_prob_num ( ) for prob in range ( 1, prob_num + 1 ): dim_num = p00_dim_num ( prob ) print ( ' %7d %9d' % ( prob, dim_num ) ) # # Terminate. # print ( '' ) print ( 'p00_dim_num_test:' ) print ( ' Normal end of execution.' ) return def p00_prob_num ( ): #*****************************************************************************80 # ## p00_prob_num() returns the number of problems. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 27 June 2015 # # Author: # # John Burkardt # # Output: # # integer VALUE, the number of problems. # value = 8 return value def p00_prob_num_test ( ): #*****************************************************************************80 # ## p00_prob_num_test() tests p00_prob_num. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 27 June 2015 # # Author: # # John Burkardt # print ( '' ) print ( 'p00_prob_num_test' ) print ( ' p00_prob_num() returns the number of test problems.' ) num = p00_prob_num ( ) print ( '' ) print ( ' TEST_interp includes %d test problems.' % ( num ) ) # # Terminate. # print ( '' ) print ( 'p00_prob_num_test:' ) print ( ' Normal end of execution.' ) return def r8mat_transpose_print ( m, n, a, title ): #*****************************************************************************80 # ## r8mat_transpose_print() prints an R8MAT, transposed. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 31 August 2014 # # Author: # # John Burkardt # # Input: # # integer M, the number of rows in A. # # integer N, the number of columns in A. # # real A(M,N), the matrix. # # string TITLE, a title. # r8mat_transpose_print_some ( m, n, a, 0, 0, m - 1, n - 1, title ) return def r8mat_transpose_print_some ( m, n, a, ilo, jlo, ihi, jhi, title ): #*****************************************************************************80 # ## r8mat_transpose_print_some prints a portion of an R8MAT, transposed. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 13 November 2014 # # Author: # # John Burkardt # # Input: # # integer M, N, the number of rows and columns of the matrix. # # real A(M,N), an M by N matrix to be printed. # # integer ILO, JLO, the first row and column to print. # # integer IHI, JHI, the last row and column to print. # # string TITLE, a title. # incx = 5 print ( '' ) print ( title ) if ( m <= 0 or n <= 0 ): print ( '' ) print ( ' (None)' ) return for i2lo in range ( max ( ilo, 0 ), min ( ihi, m - 1 ), incx ): i2hi = i2lo + incx - 1 i2hi = min ( i2hi, m - 1 ) i2hi = min ( i2hi, ihi ) print ( '' ) print ( ' Row: ', end = '' ) for i in range ( i2lo, i2hi + 1 ): print ( '%7d ' % ( i ), end = '' ) print ( '' ) print ( ' Col' ) j2lo = max ( jlo, 0 ) j2hi = min ( jhi, n - 1 ) for j in range ( j2lo, j2hi + 1 ): print ( '%7d :' % ( j ), end = '' ) for i in range ( i2lo, i2hi + 1 ): print ( '%12g ' % ( a[i,j] ), end = '' ) print ( '' ) return def r8vec2_print ( n, a1, a2, title ): #*****************************************************************************80 # ## r8vec2_print() prints an R8VEC2. # # Discussion: # # An R8VEC2 is a dataset consisting of N pairs of real values, stored # as two separate vectors A1 and A2. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 27 June 2015 # # Author: # # John Burkardt # # Input: # # integer N, the number of components of the vector. # # real A1(N), A2(N), the vectors to be printed. # # string TITLE, a title. # print ( '' ) print ( title ) print ( '' ) for i in range ( 0, n ): print ( ' %6d: %12g %12g' % ( i, a1[i], a2[i] ) ) return def r8vec_print ( n, a, title ): #*****************************************************************************80 # ## r8vec_print() prints an R8VEC. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 31 August 2014 # # Author: # # John Burkardt # # Input: # # integer N, the dimension of the vector. # # real A(N), the vector to be printed. # # string TITLE, a title. # print ( '' ) print ( title ) print ( '' ) for i in range ( 0, n ): print ( '%6d: %12g' % ( i, a[i] ) ) return def timestamp ( ): #*****************************************************************************80 # ## timestamp() prints the date as a timestamp. # # Licensing: # # This code is distributed under the MIT license. # # Modified: # # 06 April 2013 # # Author: # # John Burkardt # import time t = time.time ( ) print ( time.ctime ( t ) ) return None if ( __name__ == '__main__' ): timestamp ( ) newton_interp_1d_test ( ) timestamp ( )