AVLTree.h
上传用户:xjjlds
上传日期:2015-12-05
资源大小:22823k
文件大小:18k
源码类别:

多媒体编程

开发平台:

Visual C++

  1. /*
  2.      File:       AVLTree.h
  3.  
  4.      Contains:   Prototypes for routines which create, destroy, allow for
  5.  
  6.      Version:    Technology: 
  7.                  Release:    QuickTime 6.0.2
  8.  
  9.      Copyright:  (c) 1999-2001 by Apple Computer, Inc., all rights reserved.
  10.  
  11.      Bugs?:      For bug reports, consult the following page on
  12.                  the World Wide Web:
  13.  
  14.                      http://developer.apple.com/bugreporter/
  15.  
  16. */
  17. #ifndef __AVLTREE__
  18. #define __AVLTREE__
  19. #ifndef __MACTYPES__
  20. #include "MacTypes.h"
  21. #endif
  22. #ifndef __MIXEDMODE__
  23. #include "MixedMode.h"
  24. #endif
  25. /* The visit stage for AVLWalk() walkProcs */
  26. #if PRAGMA_ONCE
  27. #pragma once
  28. #endif
  29. #ifdef __cplusplus
  30. extern "C" {
  31. #endif
  32. #if PRAGMA_IMPORT
  33. #pragma import on
  34. #endif
  35. #if PRAGMA_STRUCT_ALIGN
  36.     #pragma options align=mac68k
  37. #elif PRAGMA_STRUCT_PACKPUSH
  38.     #pragma pack(push, 2)
  39. #elif PRAGMA_STRUCT_PACK
  40.     #pragma pack(2)
  41. #endif
  42. typedef UInt16 AVLVisitStage;
  43. enum {
  44.     kAVLPreOrder                = 0,
  45.     kAVLInOrder                 = 1,
  46.     kAVLPostOrder               = 2
  47. };
  48. /* The order the tree is walked or disposed of. */
  49. typedef UInt16 AVLOrder;
  50. enum {
  51.     kLeftToRight                = 0,
  52.     kRightToLeft                = 1
  53. };
  54. /* The type of the node being passed to a callback proc. */
  55. typedef UInt16 AVLNodeType;
  56. enum {
  57.     kAVLIsTree                  = 0,
  58.     kAVLIsLeftBranch            = 1,
  59.     kAVLIsRightBranch           = 2,
  60.     kAVLIsLeaf                  = 3,
  61.     kAVLNullNode                = 4
  62. };
  63. enum {
  64.     errItemAlreadyInTree        = -960,
  65.     errNotValidTree             = -961,
  66.     errItemNotFoundInTree       = -962,
  67.     errCanNotInsertWhileWalkProcInProgress = -963,
  68.     errTreeIsLocked             = -964
  69. };
  70. /*  The structure of a tree.  It's opaque; don't assume it's 36 bytes in size.*/
  71. struct AVLTreeStruct {
  72.     OSType                          signature;
  73.     unsigned long                   privateStuff[8];
  74. };
  75. typedef struct AVLTreeStruct            AVLTreeStruct;
  76. typedef AVLTreeStruct *                 AVLTreePtr;
  77. /*
  78.     Every tree must have a function which compares the data for two items and returns < 0, 0, or >0
  79.     for the items - < 0 if the first item is 'before' the second item according to some criteria,
  80.     == 0 if the two items are identical according to the criteria, or > 0 if the first item is
  81.     'after' the second item according to the criteria.  The comparison function is also passed the
  82.     node type, but most of the time this can be ignored.
  83. */
  84. typedef CALLBACK_API( SInt32 , AVLCompareItemsProcPtr )(AVLTreePtr tree, const void *i1, const void *i2, AVLNodeType nd_typ);
  85. /*
  86.     Every tree must have a itemSizeProc; this routine gets passed a pointer to the item's data and
  87.     returns the size of the data.  If a tree contains records of a fixed size, this function can
  88.     just return sizeof( that-struct ); otherwise it should calculate the size of the item based on
  89.     the data for the item.
  90. */
  91. typedef CALLBACK_API( UInt32 , AVLItemSizeProcPtr )(AVLTreePtr tree, const void *itemPtr);
  92. /*
  93.     A tree may have an optional disposeItemProc, which gets called whenever an item is removed
  94.     from the tree ( via AVLRemove() or when AVLDispose() deletes all of the items in the tree ).
  95.     This might be useful if the nodes in the tree own 'resources'  ( like, open files ) which
  96.     should be released before the item is removed.
  97. */
  98. typedef CALLBACK_API( void , AVLDisposeItemProcPtr )(AVLTreePtr tree, const void *dataP);
  99. /*
  100.     The common way to iterate across all of the items in a tree is via AVLWalk(), which takes
  101.     a walkProcPtr.  This function will get called for every item in the tree three times, as
  102.     the tree is being walked across.  First, the walkProc will get called with visitStage ==
  103.     kAVLPreOrder, at which point internally the node of the tree for the given data has just
  104.     been reached.  Later, this function will get called with visitStage == kAVLInOrder, and
  105.     lastly this function will get called with visitStage == kAVLPostOrder.
  106.     The 'minimum' item in the tree will get called with visitStage == kInOrder first, followed
  107.     by the 'next' item in the tree, up until the last item in the tree structure is called.
  108.     In general, you'll only care about calls to this function when visitStage == kAVLInOrder.
  109. */
  110. typedef CALLBACK_API( OSErr , AVLWalkProcPtr )(AVLTreePtr tree, const void *dataP, AVLVisitStage visitStage, AVLNodeType node, UInt32 level, SInt32 balance, void *refCon);
  111. typedef STACK_UPP_TYPE(AVLCompareItemsProcPtr)                  AVLCompareItemsUPP;
  112. typedef STACK_UPP_TYPE(AVLItemSizeProcPtr)                      AVLItemSizeUPP;
  113. typedef STACK_UPP_TYPE(AVLDisposeItemProcPtr)                   AVLDisposeItemUPP;
  114. typedef STACK_UPP_TYPE(AVLWalkProcPtr)                          AVLWalkUPP;
  115. #if OPAQUE_UPP_TYPES
  116.     EXTERN_API(AVLCompareItemsUPP)
  117.     NewAVLCompareItemsUPP          (AVLCompareItemsProcPtr  userRoutine);
  118.     EXTERN_API(AVLItemSizeUPP)
  119.     NewAVLItemSizeUPP              (AVLItemSizeProcPtr      userRoutine);
  120.     EXTERN_API(AVLDisposeItemUPP)
  121.     NewAVLDisposeItemUPP           (AVLDisposeItemProcPtr   userRoutine);
  122.     EXTERN_API(AVLWalkUPP)
  123.     NewAVLWalkUPP                  (AVLWalkProcPtr          userRoutine);
  124.     EXTERN_API(void)
  125.     DisposeAVLCompareItemsUPP      (AVLCompareItemsUPP      userUPP);
  126.     EXTERN_API(void)
  127.     DisposeAVLItemSizeUPP          (AVLItemSizeUPP          userUPP);
  128.     EXTERN_API(void)
  129.     DisposeAVLDisposeItemUPP       (AVLDisposeItemUPP       userUPP);
  130.     EXTERN_API(void)
  131.     DisposeAVLWalkUPP              (AVLWalkUPP              userUPP);
  132.     EXTERN_API(SInt32)
  133.     InvokeAVLCompareItemsUPP       (AVLTreePtr              tree,
  134.                                     const void *            i1,
  135.                                     const void *            i2,
  136.                                     AVLNodeType             nd_typ,
  137.                                     AVLCompareItemsUPP      userUPP);
  138.     EXTERN_API(UInt32)
  139.     InvokeAVLItemSizeUPP           (AVLTreePtr              tree,
  140.                                     const void *            itemPtr,
  141.                                     AVLItemSizeUPP          userUPP);
  142.     EXTERN_API(void)
  143.     InvokeAVLDisposeItemUPP        (AVLTreePtr              tree,
  144.                                     const void *            dataP,
  145.                                     AVLDisposeItemUPP       userUPP);
  146.     EXTERN_API(OSErr)
  147.     InvokeAVLWalkUPP               (AVLTreePtr              tree,
  148.                                     const void *            dataP,
  149.                                     AVLVisitStage           visitStage,
  150.                                     AVLNodeType             node,
  151.                                     UInt32                  level,
  152.                                     SInt32                  balance,
  153.                                     void *                  refCon,
  154.                                     AVLWalkUPP              userUPP);
  155. #else
  156.     enum { uppAVLCompareItemsProcInfo = 0x00002FF0 };               /* pascal 4_bytes Func(4_bytes, 4_bytes, 4_bytes, 2_bytes) */
  157.     enum { uppAVLItemSizeProcInfo = 0x000003F0 };                   /* pascal 4_bytes Func(4_bytes, 4_bytes) */
  158.     enum { uppAVLDisposeItemProcInfo = 0x000003C0 };                /* pascal no_return_value Func(4_bytes, 4_bytes) */
  159.     enum { uppAVLWalkProcInfo = 0x000FEBE0 };                       /* pascal 2_bytes Func(4_bytes, 4_bytes, 2_bytes, 2_bytes, 4_bytes, 4_bytes, 4_bytes) */
  160.     #define NewAVLCompareItemsUPP(userRoutine)                      (AVLCompareItemsUPP)NewRoutineDescriptor((ProcPtr)(userRoutine), uppAVLCompareItemsProcInfo, GetCurrentArchitecture())
  161.     #define NewAVLItemSizeUPP(userRoutine)                          (AVLItemSizeUPP)NewRoutineDescriptor((ProcPtr)(userRoutine), uppAVLItemSizeProcInfo, GetCurrentArchitecture())
  162.     #define NewAVLDisposeItemUPP(userRoutine)                       (AVLDisposeItemUPP)NewRoutineDescriptor((ProcPtr)(userRoutine), uppAVLDisposeItemProcInfo, GetCurrentArchitecture())
  163.     #define NewAVLWalkUPP(userRoutine)                              (AVLWalkUPP)NewRoutineDescriptor((ProcPtr)(userRoutine), uppAVLWalkProcInfo, GetCurrentArchitecture())
  164.     #define DisposeAVLCompareItemsUPP(userUPP)                      DisposeRoutineDescriptor(userUPP)
  165.     #define DisposeAVLItemSizeUPP(userUPP)                          DisposeRoutineDescriptor(userUPP)
  166.     #define DisposeAVLDisposeItemUPP(userUPP)                       DisposeRoutineDescriptor(userUPP)
  167.     #define DisposeAVLWalkUPP(userUPP)                              DisposeRoutineDescriptor(userUPP)
  168.     #define InvokeAVLCompareItemsUPP(tree, i1, i2, nd_typ, userUPP)  (SInt32)CALL_FOUR_PARAMETER_UPP((userUPP), uppAVLCompareItemsProcInfo, (tree), (i1), (i2), (nd_typ))
  169.     #define InvokeAVLItemSizeUPP(tree, itemPtr, userUPP)            (UInt32)CALL_TWO_PARAMETER_UPP((userUPP), uppAVLItemSizeProcInfo, (tree), (itemPtr))
  170.     #define InvokeAVLDisposeItemUPP(tree, dataP, userUPP)           CALL_TWO_PARAMETER_UPP((userUPP), uppAVLDisposeItemProcInfo, (tree), (dataP))
  171.     #define InvokeAVLWalkUPP(tree, dataP, visitStage, node, level, balance, refCon, userUPP)  (OSErr)CALL_SEVEN_PARAMETER_UPP((userUPP), uppAVLWalkProcInfo, (tree), (dataP), (visitStage), (node), (level), (balance), (refCon))
  172. #endif
  173. /* support for pre-Carbon UPP routines: NewXXXProc and CallXXXProc */
  174. #define NewAVLCompareItemsProc(userRoutine)                     NewAVLCompareItemsUPP(userRoutine)
  175. #define NewAVLItemSizeProc(userRoutine)                         NewAVLItemSizeUPP(userRoutine)
  176. #define NewAVLDisposeItemProc(userRoutine)                      NewAVLDisposeItemUPP(userRoutine)
  177. #define NewAVLWalkProc(userRoutine)                             NewAVLWalkUPP(userRoutine)
  178. #define CallAVLCompareItemsProc(userRoutine, tree, i1, i2, nd_typ) InvokeAVLCompareItemsUPP(tree, i1, i2, nd_typ, userRoutine)
  179. #define CallAVLItemSizeProc(userRoutine, tree, itemPtr)         InvokeAVLItemSizeUPP(tree, itemPtr, userRoutine)
  180. #define CallAVLDisposeItemProc(userRoutine, tree, dataP)        InvokeAVLDisposeItemUPP(tree, dataP, userRoutine)
  181. #define CallAVLWalkProc(userRoutine, tree, dataP, visitStage, node, level, balance, refCon) InvokeAVLWalkUPP(tree, dataP, visitStage, node, level, balance, refCon, userRoutine)
  182. /*
  183.     Create an AVL tree.  The compareItemsProc and the sizeItemProc are required; disposeItemProc is
  184.     optional and can be nil.  The refCon is stored with the list, and is passed back to the
  185.     compareItemsProc, sizeItemProc, and disposeItemsProc calls.  The allocation of the tree ( and all
  186.     nodes later added to the list with AVLInsert ) will be created in what is the current zone at the
  187.     time AVLInit() is called.  Always call AVLDispose() to dispose of a list created with AVLInit().
  188. */
  189. EXTERN_API( OSErr )
  190. AVLInit                         (UInt32                 flags,
  191.                                  AVLCompareItemsUPP     compareItemsProc,
  192.                                  AVLItemSizeUPP         sizeItemProc,
  193.                                  AVLDisposeItemUPP      disposeItemProc,
  194.                                  void *                 refCon,
  195.                                  AVLTreePtr *           tree)                               THREEWORDINLINE(0x303C, 0x0C01, 0xAA80);
  196. /*
  197.     Dispose of an AVL tree.  This will dispose of each item in the tree in the order specified,
  198.     call the tree's disposeProc proc for each item, and then dispose of the space allocated for
  199.     the tree itself.
  200. */
  201. EXTERN_API( OSErr )
  202. AVLDispose                      (AVLTreePtr *           tree,
  203.                                  AVLOrder               order)                              THREEWORDINLINE(0x303C, 0x0302, 0xAA80);
  204. /*
  205.     Iterate across all of the items in the tree, in the order specified.  kLeftToRight is
  206.     basically lowest-to-highest order, kRightToLeft is highest-to-lowest order.  For each
  207.     node in the tree, it will call the walkProc with three messages ( at the appropriate 
  208.     time ).  First, with kAVLPreOrder when the walking gets to this node in the tree,
  209.     before handling either the left or right subtree, secondly, with kAVLInOrder after
  210.     handling one subtree but before handling the other, and lastly with kAVLPostOrder after
  211.     handling both subtrees.  If you want to handle items in order, then only do something
  212.     if the visit stage is kAVLInOrder.  You can only call AVLRemove() from inside a walkProc
  213.     if visit stage is kAVLPostOrder ( because if you remove a node during the pre or in order
  214.     stages you will corrupt the list ) OR if you return a non-zero result from the walkProc
  215.     call which called AVLRemove() to immediately terminate the walkProc.  Do not call AVLInsert()
  216.     to insert a node into the tree from inside a walkProc.
  217.     The walkProc function gets called with the AVLTreePtr, a pointer to the data for the
  218.     current node ( which you can change in place as long as you do not affect the order within
  219.     the tree ), the visit stage, the type of the current node ( leaf node, right or left branch,
  220.     or full tree ), the level within the tree ( the root is level 1 ), the balance for the
  221.     current node, and the refCon passed to AVLWalk().  This refCon is different from the one passed
  222.     into AVLInit(); use AVLGetRefCon() to get that refCon if you want it inside a walkProc.
  223.     ( Most walkProcs will not care about the values for node type, level, or balance. )
  224. */
  225. EXTERN_API( OSErr )
  226. AVLWalk                         (AVLTreePtr             tree,
  227.                                  AVLWalkUPP             walkProc,
  228.                                  AVLOrder               order,
  229.                                  void *                 walkRefCon)                         THREEWORDINLINE(0x303C, 0x0703, 0xAA80);
  230. /*  Return  the number of items in the given tree.*/
  231. EXTERN_API( OSErr )
  232. AVLCount                        (AVLTreePtr             tree,
  233.                                  UInt32 *               count)                              THREEWORDINLINE(0x303C, 0x0804, 0xAA80);
  234. /*
  235.     Return the one-based index-th item from the tree by putting it's data at dataPtr
  236.     if dataPtr is non-nil, and it's size into *itemSize if itemSize is non-nil.
  237.     If index is out of range, return errItemNotFoundInTree.  ( Internally, this does
  238.     an AVLWalk(), so the tree can not be modified while this call is in progress ).
  239. */
  240. EXTERN_API( OSErr )
  241. AVLGetIndItem                   (AVLTreePtr             tree,
  242.                                  UInt32                 index,
  243.                                  void *                 dataPtr,
  244.                                  UInt32 *               itemSize)                           THREEWORDINLINE(0x303C, 0x0805, 0xAA80);
  245. /*
  246.     Insert the given item into the tree.  This will call the tree's sizeItemProc
  247.     to determine how big the item at data is, and then will make a copy of the
  248.     item and insert it into the tree in the appropriate place.  If an item already
  249.     exists in the tree with the same key ( so that the compareItemsUPP returns 0
  250.     when asked to compare this item to an existing one ), then it will return
  251.     errItemNotFoundInTree.
  252. */
  253. EXTERN_API( OSErr )
  254. AVLInsert                       (AVLTreePtr             tree,
  255.                                  const void *           data)                               THREEWORDINLINE(0x303C, 0x0406, 0xAA80);
  256. /*
  257.     Remove any item from the tree with the given key.  If dataPtr != nil, then
  258.     copy the item's data to dataPtr before removing it from the tree.  Before
  259.     removing the item, call the tree's disposeItemProc to let it release anything
  260.     used by the data in the tree.  It is not necessary to fill in a complete
  261.     record for key, only that the compareItemsProc return 0 when asked to compare
  262.     the data at key with the node in the tree to be deleted.  If the item cannot
  263.     be found in the tree, this will return errItemNotFoundInTree.
  264. */
  265. EXTERN_API( OSErr )
  266. AVLRemove                       (AVLTreePtr             tree,
  267.                                  const void *           key,
  268.                                  void *                 dataPtr,
  269.                                  UInt32 *               itemSize)                           THREEWORDINLINE(0x303C, 0x0807, 0xAA80);
  270. /*
  271.     Find the item in the tree with the given key, and return it's data in
  272.     dataPtr ( if dataPtr != nil ), and it's size in *itemSize ( if itemSize
  273.     != nil ).  It is not necessary to fill in a complete record for key,
  274.     only that the compareItemsProc return 0 when asked to compare the data
  275.     at key with the node in the tree to be deleted.  If the item cannot
  276.     be found in the tree, this will return errItemNotFoundInTree.
  277. */
  278. EXTERN_API( OSErr )
  279. AVLFind                         (AVLTreePtr             tree,
  280.                                  const void *           key,
  281.                                  void *                 dataPtr,
  282.                                  UInt32 *               itemSize)                           THREEWORDINLINE(0x303C, 0x0808, 0xAA80);
  283. /*
  284.     Get the refCon for the given tree ( set in AVLInit ) and return it.
  285.     If the given tree is invalid, then return nil.
  286. */
  287. EXTERN_API( OSErr )
  288. AVLGetRefcon                    (AVLTreePtr             tree,
  289.                                  void **                refCon)                             THREEWORDINLINE(0x303C, 0x0409, 0xAA80);
  290. #if PRAGMA_STRUCT_ALIGN
  291.     #pragma options align=reset
  292. #elif PRAGMA_STRUCT_PACKPUSH
  293.     #pragma pack(pop)
  294. #elif PRAGMA_STRUCT_PACK
  295.     #pragma pack()
  296. #endif
  297. #ifdef PRAGMA_IMPORT_OFF
  298. #pragma import off
  299. #elif PRAGMA_IMPORT
  300. #pragma import reset
  301. #endif
  302. #ifdef __cplusplus
  303. }
  304. #endif
  305. #endif /* __AVLTREE__ */