#! /usr/bin/env python
#
def subcompnz2_next ( n_lo, n_hi, k, a, more, h, t, n2, more2 ):

#*****************************************************************************80
#
## SUBCOMPNZ2_NEXT computes the next subcomposition of N into K nonzero parts.
#
#  Discussion:
#
#    A composition of the integer N into K nonzero parts is an ordered sequence
#    of K positive integers which sum to a value of N.
#
#    A subcomposition of the integer N into K nonzero parts is a composition
#    of M into K nonzero parts, where 0 < M <= N.
#
#    This routine computes all compositions of K into nonzero parts which sum
#    to values between N_LO and N_HI.
#
#    The routine SUBCOMPNZ_NEXT can be regarded as a special case where 
#    N_LO = K.
#
#  Licensing:
#
#    This code is distributed under the GNU LGPL license. 
#
#  Modified:
#
#    06 June 2015
#
#  Author:
#
#    John Burkardt
#
#  Parameters:
#
#    Input, integer N_LO, N_HI, the range of values N for which
#    compositions are desired.
#
#    Input, integer K, the number of parts in the subcomposition.
#    K must be no greater than N_HI.
#
#    Input, integer A(K), the parts of the subcomposition.
#
#    Input, logical MORE, set to FALSE by the user to start the computation.
#
#    Input, integer H, T, N2, internal parameters needed for the
#    computation.  The user may need to initialize these before the
#    very first call, but these initial values are not important.
#    The user should not alter these parameters once the computation
#    begins.
#
#    Input, logical MORE2, an internal parameter needed for the
#    computation.  The user may need to initialize this before the
#    very first call, but the initial value is not important.
#    The user should not alter this parameter once the computation
#    begins.
#
#    Output, integer A(K), the parts of the subcomposition.
#
#    Output, logical MORE, set to FALSE by the routine to terminate the computation.
#
#    Output, integer H, T, N2, updated values.
#
#    Output, logical MORE2, an updated value.
#
  from compnz_next import compnz_next

  if ( n_hi < k ):
    more = False
    for i in range ( 0, k ):
      a[i] = -1
    return a, more, h, t, n2, more2

  if ( n_hi < n_lo ):
    more = False
    for i in range ( 0, k ):
      a[i] = -1
    return a, more, h, t, n2, more2
#
#  The first computation.
#
  if ( not more ):

    more = True

    n2 = max ( k, n_lo )
    more2 = False
    h = 0
    t = 0

    a, more2, h, t = compnz_next ( n2, k, a, more2, h, t )
#
#  Do the next element at the current value of N.
#
  elif ( more2 ):

    a, more2, h, t = compnz_next ( n2, k, a, more2, h, t )

  else:

    n2 = n2 + 1

    a, more2, h, t = compnz_next ( n2, k, a, more2, h, t )
#
#  Termination occurs if MORE2 = FALSE and N2 = N_HI.
#
  if ( not more2 and n2 == n_hi ):
    more = False

  return a, more, h, t, n2, more2

def subcompnz2_next_test ( ):

#*****************************************************************************80
#
## SUBCOMPNZ2_NEXT_TEST tests SUBCOMPNZ2_NEXT.
#
#  Licensing:
#
#    This code is distributed under the GNU LGPL license.
#
#  Modified:
#
#    06 June 2015
#
#  Author:
#
#    John Burkardt
#
  import numpy as np
  import platform
  from i4vec_sum import i4vec_sum

  n_lo = 5
  n_hi = 7
  k = 3
  a = np.zeros ( k )
  more = False
  h = 0
  t = 0
  n2 = 0
  more2 = False

  print ( '' )
  print ( 'SUBCOMPNZ2_NEXT_TEST' )
  print ( '  Python version: %s' % ( platform.python_version ( ) ) )
  print ( '  SUBCOMPNZ2_NEXT generates subcompositions' )
  print ( '  using nonzero parts.' )
  print ( '' )
  print ( '  Seek all subcompositions of N' )
  print ( '  using K = %d nonzero parts' % ( k ) )
  print ( '  for %d <= N <= %d' % ( n_lo, n_hi ) )
  print ( '' )
  print ( '     #     N' )
  print ( '' )

  rank = 0

  while ( True ):

    a, more, h, t, n2, more2 = subcompnz2_next ( n_lo, n_hi, k, a, more, h, t, n2, more2 )

    rank = rank + 1
    n = i4vec_sum ( k, a )
    print ( '  %4d  %4d  ' % ( rank, n ) ),
    for i in range ( 0, k ):
      print ( '%4d' % ( a[i] ) ),
    print ( '' )

    if ( not more ):
      break
#
#  Terminate.
#
  print ( '' )
  print ( 'SUBCOMPNZ2_NEXT_TEST' )
  print ( '  Normal end of execution.' )
  return

if ( __name__ == '__main__' ):
  from timestamp import timestamp
  timestamp ( )
  subcompnz2_next_test ( )
  timestamp ( )