arm_dice_distance.c 2.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192
  1. /* ----------------------------------------------------------------------
  2. * Project: CMSIS DSP Library
  3. * Title: arm_dice_distance.c
  4. * Description: Dice distance between two vectors
  5. *
  6. * $Date: 23 April 2021
  7. * $Revision: V1.9.0
  8. *
  9. * Target Processor: Cortex-M and Cortex-A cores
  10. * -------------------------------------------------------------------- */
  11. /*
  12. * Copyright (C) 2010-2021 ARM Limited or its affiliates. All rights reserved.
  13. *
  14. * SPDX-License-Identifier: Apache-2.0
  15. *
  16. * Licensed under the Apache License, Version 2.0 (the License); you may
  17. * not use this file except in compliance with the License.
  18. * You may obtain a copy of the License at
  19. *
  20. * www.apache.org/licenses/LICENSE-2.0
  21. *
  22. * Unless required by applicable law or agreed to in writing, software
  23. * distributed under the License is distributed on an AS IS BASIS, WITHOUT
  24. * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  25. * See the License for the specific language governing permissions and
  26. * limitations under the License.
  27. */
  28. #include "dsp/distance_functions.h"
  29. #include <limits.h>
  30. #include <math.h>
  31. extern void arm_boolean_distance_TT_TF_FT(const uint32_t *pA
  32. , const uint32_t *pB
  33. , uint32_t numberOfBools
  34. , uint32_t *cTT
  35. , uint32_t *cTF
  36. , uint32_t *cFT
  37. );
  38. /**
  39. * @ingroup groupDistance
  40. * @{
  41. */
  42. /**
  43. * @defgroup BoolDist Boolean Distances
  44. *
  45. * Distances between two vectors of boolean values.
  46. *
  47. * Booleans are packed in 32 bit words.
  48. * numberOfBooleans argument is the number of booleans and not the
  49. * number of words.
  50. *
  51. * Bits are packed in big-endian mode (because of behavior of numpy packbits in
  52. * in version < 1.17)
  53. */
  54. /**
  55. @addtogroup BoolDist
  56. @{
  57. */
  58. /**
  59. * @brief Dice distance between two vectors
  60. *
  61. * @param[in] pA First vector of packed booleans
  62. * @param[in] pB Second vector of packed booleans
  63. * @param[in] numberOfBools Number of booleans
  64. * @return distance
  65. *
  66. */
  67. float32_t arm_dice_distance(const uint32_t *pA, const uint32_t *pB, uint32_t numberOfBools)
  68. {
  69. uint32_t ctt=0,ctf=0,cft=0;
  70. arm_boolean_distance_TT_TF_FT(pA, pB, numberOfBools, &ctt, &ctf, &cft);
  71. return(1.0*(ctf + cft) / (2.0*ctt + cft + ctf));
  72. }
  73. /**
  74. * @} end of BoolDist group
  75. */
  76. /**
  77. * @} end of groupDistance group
  78. */