williamr@2: /*
williamr@2:  *
williamr@2:  *
williamr@2:  * Copyright (c) 1994
williamr@2:  * Hewlett-Packard Company
williamr@2:  *
williamr@2:  * Copyright (c) 1996,1997
williamr@2:  * Silicon Graphics Computer Systems, Inc.
williamr@2:  *
williamr@2:  * Copyright (c) 1997
williamr@2:  * Moscow Center for SPARC Technology
williamr@2:  *
williamr@2:  * Copyright (c) 1999 
williamr@2:  * Boris Fomitchev
williamr@2:  *
williamr@2:  * This material is provided "as is", with absolutely no warranty expressed
williamr@2:  * or implied. Any use is at your own risk.
williamr@2:  *
williamr@2:  * Permission to use or copy this software for any purpose is hereby granted 
williamr@2:  * without fee, provided the above notices are retained on all copies.
williamr@2:  * Permission to modify the code and to distribute modified code is granted,
williamr@2:  * provided the above notices are retained, and a notice that the code was
williamr@2:  * modified is included with the above copyright notice.
williamr@2:  *
williamr@2:  */
williamr@2: #ifndef _STLP_HEAP_C
williamr@2: #define _STLP_HEAP_C
williamr@2: 
williamr@2: #ifndef _STLP_INTERNAL_HEAP_H
williamr@2: # include <stl/_heap.h>
williamr@2: #endif
williamr@2: 
williamr@2: #ifndef _STLP_INTERNAL_ITERATOR_BASE_H
williamr@2: # include <stl/_iterator_base.h>
williamr@2: #endif
williamr@2: 
williamr@2: _STLP_BEGIN_NAMESPACE
williamr@2: 
williamr@2: template <class _RandomAccessIterator, class _Distance, class _Tp>
williamr@2: _STLP_INLINE_LOOP
williamr@2: void 
williamr@2: __push_heap(_RandomAccessIterator __first,
williamr@2:             _Distance __holeIndex, _Distance __topIndex, _Tp __val)
williamr@2: {
williamr@2:   _Distance __parent = (__holeIndex - 1) / 2;
williamr@2:   while (__holeIndex > __topIndex && *(__first + __parent) < __val) {
williamr@2:     *(__first + __holeIndex) = *(__first + __parent);
williamr@2:     __holeIndex = __parent;
williamr@2:     __parent = (__holeIndex - 1) / 2;
williamr@2:   }    
williamr@2:   *(__first + __holeIndex) = __val;
williamr@2: }
williamr@2: 
williamr@2: template <class _RandomAccessIterator, class _Distance, class _Tp>
williamr@2: inline void 
williamr@2: __push_heap_aux(_RandomAccessIterator __first,
williamr@2:                 _RandomAccessIterator __last, _Distance*, _Tp*)
williamr@2: {
williamr@2:   __push_heap(__first, _Distance((__last - __first) - 1), _Distance(0), 
williamr@2:               _Tp(*(__last - 1)));
williamr@2: }
williamr@2: 
williamr@2: template <class _RandomAccessIterator>
williamr@2: void 
williamr@2: push_heap(_RandomAccessIterator __first, _RandomAccessIterator __last)
williamr@2: {
williamr@2:   __push_heap_aux(__first, __last,
williamr@2:                   _STLP_DISTANCE_TYPE(__first, _RandomAccessIterator), _STLP_VALUE_TYPE(__first, _RandomAccessIterator));
williamr@2: }
williamr@2: 
williamr@2: 
williamr@2: template <class _RandomAccessIterator, class _Distance, class _Tp, 
williamr@2:           class _Compare>
williamr@2: _STLP_INLINE_LOOP
williamr@2: void
williamr@2: __push_heap(_RandomAccessIterator __first, _Distance __holeIndex,
williamr@2:             _Distance __topIndex, _Tp __val, _Compare __comp)
williamr@2: {
williamr@2:   _Distance __parent = (__holeIndex - 1) / 2;
williamr@2:   while (__holeIndex > __topIndex && __comp(*(__first + __parent), __val)) {
williamr@2:     *(__first + __holeIndex) = *(__first + __parent);
williamr@2:     __holeIndex = __parent;
williamr@2:     __parent = (__holeIndex - 1) / 2;
williamr@2:   }
williamr@2:   *(__first + __holeIndex) = __val;
williamr@2: }
williamr@2: 
williamr@2: template <class _RandomAccessIterator, class _Compare,
williamr@2:           class _Distance, class _Tp>
williamr@2: inline void 
williamr@2: __push_heap_aux(_RandomAccessIterator __first,
williamr@2:                 _RandomAccessIterator __last, _Compare __comp,
williamr@2:                 _Distance*, _Tp*) 
williamr@2: {
williamr@2:   __push_heap(__first, _Distance((__last - __first) - 1), _Distance(0), 
williamr@2:               _Tp(*(__last - 1)), __comp);
williamr@2: }
williamr@2: 
williamr@2: template <class _RandomAccessIterator, class _Compare>
williamr@2: void 
williamr@2: push_heap(_RandomAccessIterator __first, _RandomAccessIterator __last,
williamr@2:           _Compare __comp) 
williamr@2: {
williamr@2:   __push_heap_aux(__first, __last, __comp,
williamr@2:                   _STLP_DISTANCE_TYPE(__first, _RandomAccessIterator), _STLP_VALUE_TYPE(__first, _RandomAccessIterator));
williamr@2: }
williamr@2: 
williamr@2: template <class _RandomAccessIterator, class _Distance, class _Tp>
williamr@2: void 
williamr@2: __adjust_heap(_RandomAccessIterator __first, _Distance __holeIndex,
williamr@2:               _Distance __len, _Tp __val) {
williamr@2:   _Distance __topIndex = __holeIndex;
williamr@2:   _Distance __secondChild = 2 * __holeIndex + 2;
williamr@2:   while (__secondChild < __len) {
williamr@2:     if (*(__first + __secondChild) < *(__first + (__secondChild - 1)))
williamr@2:       __secondChild--;
williamr@2:     *(__first + __holeIndex) = *(__first + __secondChild);
williamr@2:     __holeIndex = __secondChild;
williamr@2:     __secondChild = 2 * (__secondChild + 1);
williamr@2:   }
williamr@2:   if (__secondChild == __len) {
williamr@2:     *(__first + __holeIndex) = *(__first + (__secondChild - 1));
williamr@2:     __holeIndex = __secondChild - 1;
williamr@2:   }
williamr@2:   __push_heap(__first, __holeIndex, __topIndex, __val);
williamr@2: }
williamr@2: 
williamr@2: 
williamr@2: template <class _RandomAccessIterator, class _Tp>
williamr@2: inline void 
williamr@2: __pop_heap_aux(_RandomAccessIterator __first, _RandomAccessIterator __last, _Tp*) {
williamr@2:   __pop_heap(__first, __last - 1, __last - 1, 
williamr@2:              _Tp(*(__last - 1)), _STLP_DISTANCE_TYPE(__first, _RandomAccessIterator));
williamr@2: }
williamr@2: 
williamr@2: template <class _RandomAccessIterator>
williamr@2: void pop_heap(_RandomAccessIterator __first, 
williamr@2: 	      _RandomAccessIterator __last) {
williamr@2:   __pop_heap_aux(__first, __last, _STLP_VALUE_TYPE(__first, _RandomAccessIterator));
williamr@2: }
williamr@2: 
williamr@2: template <class _RandomAccessIterator, class _Distance,
williamr@2:           class _Tp, class _Compare>
williamr@2: void
williamr@2: __adjust_heap(_RandomAccessIterator __first, _Distance __holeIndex,
williamr@2:               _Distance __len, _Tp __val, _Compare __comp)
williamr@2: {
williamr@2:   _Distance __topIndex = __holeIndex;
williamr@2:   _Distance __secondChild = 2 * __holeIndex + 2;
williamr@2:   while (__secondChild < __len) {
williamr@2:     if (__comp(*(__first + __secondChild), *(__first + (__secondChild - 1))))
williamr@2:       __secondChild--;
williamr@2:     *(__first + __holeIndex) = *(__first + __secondChild);
williamr@2:     __holeIndex = __secondChild;
williamr@2:     __secondChild = 2 * (__secondChild + 1);
williamr@2:   }
williamr@2:   if (__secondChild == __len) {
williamr@2:     *(__first + __holeIndex) = *(__first + (__secondChild - 1));
williamr@2:     __holeIndex = __secondChild - 1;
williamr@2:   }
williamr@2:   __push_heap(__first, __holeIndex, __topIndex, __val, __comp);
williamr@2: }
williamr@2: 
williamr@2: 
williamr@2: template <class _RandomAccessIterator, class _Tp, class _Compare>
williamr@2: inline void 
williamr@2: __pop_heap_aux(_RandomAccessIterator __first,
williamr@2:                _RandomAccessIterator __last, _Tp*, _Compare __comp)
williamr@2: {
williamr@2:   __pop_heap(__first, __last - 1, __last - 1, _Tp(*(__last - 1)), __comp,
williamr@2:              _STLP_DISTANCE_TYPE(__first, _RandomAccessIterator));
williamr@2: }
williamr@2: 
williamr@2: 
williamr@2: template <class _RandomAccessIterator, class _Compare>
williamr@2: void 
williamr@2: pop_heap(_RandomAccessIterator __first,
williamr@2:          _RandomAccessIterator __last, _Compare __comp)
williamr@2: {
williamr@2:     __pop_heap_aux(__first, __last, _STLP_VALUE_TYPE(__first, _RandomAccessIterator), __comp);
williamr@2: }
williamr@2: 
williamr@2: template <class _RandomAccessIterator, class _Tp, class _Distance>
williamr@2: _STLP_INLINE_LOOP
williamr@2: void 
williamr@2: __make_heap(_RandomAccessIterator __first,
williamr@2:             _RandomAccessIterator __last, _Tp*, _Distance*)
williamr@2: {
williamr@2:   if (__last - __first < 2) return;
williamr@2:   _Distance __len = __last - __first;
williamr@2:   _Distance __parent = (__len - 2)/2;
williamr@2:     
williamr@2:   while (true) {
williamr@2:     __adjust_heap(__first, __parent, __len, _Tp(*(__first + __parent)));
williamr@2:     if (__parent == 0) return;
williamr@2:     __parent--;
williamr@2:   }
williamr@2: }
williamr@2: 
williamr@2: template <class _RandomAccessIterator>
williamr@2: void 
williamr@2: make_heap(_RandomAccessIterator __first, _RandomAccessIterator __last)
williamr@2: {
williamr@2:   __make_heap(__first, __last,
williamr@2:               _STLP_VALUE_TYPE(__first, _RandomAccessIterator), _STLP_DISTANCE_TYPE(__first, _RandomAccessIterator));
williamr@2: }
williamr@2: 
williamr@2: template <class _RandomAccessIterator, class _Compare,
williamr@2:           class _Tp, class _Distance>
williamr@2: _STLP_INLINE_LOOP
williamr@2: void
williamr@2: __make_heap(_RandomAccessIterator __first, _RandomAccessIterator __last,
williamr@2:             _Compare __comp, _Tp*, _Distance*)
williamr@2: {
williamr@2:   if (__last - __first < 2) return;
williamr@2:   _Distance __len = __last - __first;
williamr@2:   _Distance __parent = (__len - 2)/2;
williamr@2:     
williamr@2:   while (true) {
williamr@2:     __adjust_heap(__first, __parent, __len, _Tp(*(__first + __parent)),
williamr@2:                   __comp);
williamr@2:     if (__parent == 0) return;
williamr@2:     __parent--;
williamr@2:   }
williamr@2: }
williamr@2: 
williamr@2: template <class _RandomAccessIterator, class _Compare>
williamr@2: void 
williamr@2: make_heap(_RandomAccessIterator __first, 
williamr@2:           _RandomAccessIterator __last, _Compare __comp)
williamr@2: {
williamr@2:   __make_heap(__first, __last, __comp,
williamr@2:               _STLP_VALUE_TYPE(__first, _RandomAccessIterator), _STLP_DISTANCE_TYPE(__first, _RandomAccessIterator));
williamr@2: }
williamr@2: 
williamr@2: _STLP_END_NAMESPACE
williamr@2: 
williamr@2: #endif /*  _STLP_HEAP_C */
williamr@2: 
williamr@2: // Local Variables:
williamr@2: // mode:C++
williamr@2: // End: