#! /usr/bin/env python
#
from numpy import bitwise_xor
from sobol_lib import *
import datetime

def sobol_test01 ( ):

#*****************************************************************************80
#
## SOBOL_TEST01 tests BITWISE_XOR.
#
#  Licensing:
#
#    This code is distributed under the MIT license.
#
#  Modified:
#
#    22 February 2011
#
#  Author:
#
#    Original MATLAB version by John Burkardt.
#    Python version by Corrado Chisari
#
  print ( '' )
  print ( 'SOBOL_TEST01'  )
  print ( '  BITWISE_XOR returns the bitwise exclusive OR of two integers.' )
  print ( '' )
  print ('     I     J     BITXOR(I,J)' )
  print ( ''  )

  seed = 123456789

  for test in range ( 0, 10 ):

    i, seed = i4_uniform ( 0, 100, seed )
    j, seed = i4_uniform ( 0, 100, seed )
    k = bitwise_xor ( i, j )

    print ( '  %6d  %6d  %6d' % ( i, j, k ) )

  return

def sobol_test02 ( ):

#*****************************************************************************80
#
## SOBOL_TEST02 tests I4_BIT_HI1.
#
#  Licensing:
#
#    This code is distributed under the MIT license.
#
#  Modified:
#
#    22 February 2011
#
#  Author:
#
#    Original MATLAB version by John Burkardt.
#  PYTHON version by Corrado Chisari
#
    print ( '' )
    print ( 'SOBOL_TEST02' )
    print ( '  I4_BIT_HI1 returns the location of the high 1 bit.' )
    print ( '' )
    print (  '     I     I4_BIT_HI1(I)' )
    print ( '' )

    seed = 123456789

    for test in range( 0, 10):

      [ i, seed ] = i4_uniform ( 0, 100, seed )

      j = i4_bit_hi1 ( i )

      print ( '%6d %6d' % ( i, j ) )

    return

def sobol_test03 ( ):

#*****************************************************************************80
#
## SOBOL_TEST03 tests I4_BIT_LO0.
#
#  Licensing:
#
#    This code is distributed under the MIT license.
#
#  Modified:
#
#    22 February 2011
#
#  Author:
#
#    Original MATLAB version by John Burkardt.
#    PYTHON version by Corrado Chisari
#
    print ( '' )
    print ( 'SOBOL_TEST03' )
    print ( '  I4_BIT_LO0 returns the location of the low 0 bit.' )
    print ( '' )
    print ( '     I     I4_BIT_LO0(I)' )
    print ( '' )

    seed = 123456789

    for test in range ( 0, 10 ):

      [ i, seed ] = i4_uniform ( 0, 100, seed )

      j = i4_bit_lo0 ( i )

      print ( '%6d %6d'%( i, j ) )

    return

def sobol_test04 ( ):

#*****************************************************************************80
#
## SOBOL_TEST04 tests I4_SOBOL.
#
#  Licensing:
#
#    This code is distributed under the MIT license.
#
#  Modified:
#
#    22 February 2011
#
#  Author:
#
#    Original MATLAB version by John Burkardt.
#    PYTHON version by Corrado Chisari
#
    print ( '\nSOBOL_TEST04' )
    print ( '  I4_SOBOL returns the next element' )
    print ( '  of a Sobol sequence.' )
    print ( '\n  In this test, we call I4_SOBOL repeatedly.\n' )

    dim_max = 4

    for dim_num in range( 2, dim_max+1):

      seed = 0
      qs = prime_ge ( dim_num )

      print ( '\n  Using dimension DIM_NUM =   %d'%dim_num )
      print ( '\n  Seed  Seed   I4_SOBOL' )
      print ( '  In    Out\n' )

      for i in range( 0, 111):
        [ r, seed_out ] = i4_sobol ( dim_num, seed )
        if ( i <= 11 or 95 <= i ):
          out='%6d %6d  '%(seed, seed_out )
          for j in range (0, dim_num):
            out+='%10f  '%r[j]
          print ( out )
        elif ( i == 12 ):
          print ( '......................' )
        seed = seed_out

    return

def sobol_test05 ( ):

#*****************************************************************************80
#
## SOBOL_TEST05 tests I4_SOBOL.
#
#  Licensing:
#
#    This code is distributed under the MIT license.
#
#  Modified:
#
#    22 February 2011
#
#  Author:
#
#    Original MATLAB version by John Burkardt.
#    Python version by Corrado Chisari
#
  print ( '' )
  print ( 'SOBOL_TEST05' )
  print ( '  I4_SOBOL computes the next element of a Sobol sequence.' )
  print ( '' )
  print ( '  In this test, we demonstrate how the SEED can be' )
  print ( '  manipulated to skip ahead in the sequence, or' )
  print ( '  to come back to any part of the sequence.' )
  print ( '' )

  dim_num = 3

  print ( '' )
  print ( '  Using dimension DIM_NUM =   %d\n'%dim_num )

  seed = 0

  print ( '' )
  print ( '  Seed  Seed   I4_SOBOL' )
  print ( '  In    Out' )
  print ( '' )

  for i in range( 0 , 10+1):
    [ r, seed_out ] = i4_sobol ( dim_num, seed )
    out= '%6d %6d  '%( seed, seed_out )
    for j in range( 1 , dim_num+1):
      out+= '%10f  '% r[j-1]
    print ( out )
    seed = seed_out

  print ( '' )
  print ( '  Jump ahead by increasing SEED:' )
  print ( '' )

  seed = 100

  print ( '' )
  print ( '  Seed  Seed   I4_SOBOL' )
  print ( '  In    Out' )
  print ( '' )

  for i in range( 1 , 6):
    [ r, seed_out ] = i4_sobol ( dim_num, seed )
    out='%6d %6d  '%( seed, seed_out)
    for j in range( 1 , dim_num+1):
      out+= '%10f  '% r[j-1]
    print ( out )
    seed = seed_out
  print ( '' )
  print ( '  Jump back by decreasing SEED:' )
  print ( '' )

  seed = 3

  print ( '' )
  print ( '  Seed  Seed   I4_SOBOL' )
  print ( '  In    Out' )
  print ( '' )

  for i in range( 0 , 11):
    [ r, seed_out ] = i4_sobol ( dim_num, seed )
    out='%6d %6d  '%( seed, seed_out)
    for j in range( 1 , dim_num+1):
      out+= '%10f  '% r[j-1]
    print ( out )
    seed = seed_out

  print ( '' )
  print ( '  Jump back by decreasing SEED:' )
  print ( '' )

  seed = 98

  print ( '' )
  print ( '  Seed  Seed   I4_SOBOL' )
  print ( '  In    Out' )
  print ( '' )

  for i in range( 1 , 6):
    [ r, seed_out ] = i4_sobol ( dim_num, seed )
    out= '%6d %6d  '%( seed, seed_out )
    for j in range( 1 , dim_num+1):
      out+= '%10f  '%r[j-1]
    print ( out )
    seed = seed_out

  return

def main ( argv = None ):

#*****************************************************************************80
#
## SOBOL_TEST tests the SOBOL library.
#
#  Licensing:
#
#    This code is distributed under the MIT license.
#
#  Modified:
#
#    25 October 2016
#
#  Author:
#
#    Original MATLAB version by John Burkardt.
#    Python version by Corrado Chisari
#
  import platform

  d=datetime.datetime.today()
  print ( d.strftime("%d-%b-%Y %H:%M:%S") )

  print ( '' )
  print ( 'SOBOL_TEST' )
  print ( '  Test the SOBOL routines.' )

  sobol_test01 ( )
  sobol_test02 ( )
  sobol_test03 ( )
  sobol_test04 ( )
  sobol_test05 ( )
#
#  Terminate.
#
  print ( '' )
  print ( 'SOBOL_TEST' )
  print ( '  Normal end of execution.' )

  d = datetime.datetime.today ( )
  print ( d.strftime("%d-%b-%Y %H:%M:%S") )

if __name__ == "__main__":
  main ( )