@techoptio/react-native-live-pitch-detection 1.0.0 → 1.2.0

package/android/build.gradle

@@ -43,6 +43,7 @@ android {
       cmake {
         cppFlags "-O2 -frtti -fexceptions -Wall -fstack-protector-all"
         abiFilters (*reactNativeArchitectures())
+        arguments "-DANDROID_SUPPORT_FLEXIBLE_PAGE_SIZES=ON"
       }
     }
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@techoptio/react-native-live-pitch-detection",
-  "version": "1.0.0",
+  "version": "1.2.0",
   "description": "Live pitch detection (frequency, note and octave) from device microphone for React Native.",
   "main": "./lib/module/index.js",
   "types": "./lib/typescript/src/index.d.ts",
@@ -18,6 +18,7 @@
     "android",
     "ios",
     "cpp",
+    "shared",
     "*.podspec",
     "react-native.config.js",
     "!ios/build",

package/shared/ReactNativeLivePitchDetectionModule.cpp

@@ -0,0 +1,80 @@
+// Thanks to rnheroes/react-native-pitchy for the original implementation
+// https://github.com/rnheroes/react-native-pitchy/blob/main/cpp/react-native-pitchy.cpp
+
+#include "ReactNativeLivePitchDetectionModule.h"
+
+#include <vector>
+#include <cmath>
+#include <algorithm>
+#include <numeric>
+#include <limits>
+
+namespace techoptio { 
+    namespace reactnativelivepitchdetection {
+        double getVolumeDecibel(double rms)
+        {
+            return 20 * std::log10(rms);
+        }
+
+        double autoCorrelate(const std::vector<double> &buf, double sampleRate, double minVolume)
+        {
+            int SIZE = buf.size();
+            double rms = 0;
+
+            for (int i = 0; i < SIZE; ++i)
+            {
+                double val = buf[i];
+                rms += val * val;
+            }
+            rms = std::sqrt(rms / SIZE);
+            double decibel = getVolumeDecibel(rms);
+            if (decibel < minVolume)
+                return -1;
+
+            int r1 = 0, r2 = SIZE - 1;
+            double thres = 0.2;
+            for (int i = 0; i < SIZE / 2; ++i)
+                if (std::abs(buf[i]) < thres)
+                {
+                    r1 = i;
+                    break;
+                }
+            for (int i = 1; i < SIZE / 2; ++i)
+                if (std::abs(buf[SIZE - i]) < thres)
+                {
+                    r2 = SIZE - i;
+                    break;
+                }
+
+            std::vector<double> slicedBuf(buf.begin() + r1, buf.begin() + r2);
+            SIZE = slicedBuf.size();
+
+            std::vector<double> c(SIZE, 0);
+            for (int i = 0; i < SIZE; ++i)
+                for (int j = 0; j < SIZE - i; ++j)
+                    c[i] += slicedBuf[j] * slicedBuf[j + i];
+
+            int d = 0;
+            while (c[d] > c[d + 1])
+                d++;
+            double maxval = -1, maxpos = -1;
+            for (int i = d; i < SIZE; ++i)
+            {
+                if (c[i] > maxval)
+                {
+                    maxval = c[i];
+                    maxpos = i;
+                }
+            }
+            double T0 = maxpos;
+
+            double x1 = c[T0 - 1], x2 = c[T0], x3 = c[T0 + 1];
+            double a = (x1 + x3 - 2 * x2) / 2;
+            double b = (x3 - x1) / 2;
+            if (a)
+                T0 = T0 - b / (2 * a);
+
+            return sampleRate / T0;
+        }
+    }
+}

package/shared/ReactNativeLivePitchDetectionModule.h

@@ -0,0 +1,12 @@
+#ifndef REACT_NATIVE_LIVE_PITCH_DETECTION_MODULE_H
+#define REACT_NATIVE_LIVE_PITCH_DETECTION_MODULE_H
+
+#include <vector>
+
+namespace techoptio { 
+    namespace reactnativelivepitchdetection {
+        double autoCorrelate(const std::vector<double> &buf, double sampleRate, double minVolume);
+    }
+}
+
+#endif