Implement ceiling power of two

src/__tests__/index.spec.ts

@@ -2,6 +2,7 @@ import {describe, it, expect} from 'vitest';
 import {
   arraysEqual,
   binomial,
+  ceilPow2,
   circleDifference,
   circleDistance,
   clamp,
@@ -152,3 +153,29 @@ describe('Pitch distance with circle equivalence', () => {
     expect(diff).toBeCloseTo(247.741);
   });
 });
+
+describe('Ceiling power of two', () => {
+  it('works with small values', () => {
+    const x = 1 + Math.random() * (2 ** 30 - 2);
+    const p2 = ceilPow2(x);
+    expect(x).toBeLessThanOrEqual(p2);
+    expect(p2).toBeLessThan(2 * x);
+    expect(Math.log2(p2)).toBeCloseTo(Math.round(Math.log2(p2)));
+  });
+
+  it('works with tiny values', () => {
+    const x = Math.random();
+    const p2 = ceilPow2(x);
+    expect(x).toBeLessThanOrEqual(p2);
+    expect(p2).toBeLessThan(2 * x);
+    expect(Math.log2(p2)).toBeCloseTo(Math.round(Math.log2(p2)));
+  });
+
+  it('works with large values', () => {
+    const x = 2 ** 31 + Math.random() * 2 ** 37;
+    const p2 = ceilPow2(x);
+    expect(x).toBeLessThanOrEqual(p2);
+    expect(p2).toBeLessThan(2 * x);
+    expect(Math.log2(p2)).toBeCloseTo(Math.round(Math.log2(p2)));
+  });
+});

src/index.ts

@@ -284,3 +284,18 @@ export function circleDifference(a: number, b: number, equaveCents = 1200.0) {
 export function circleDistance(a: number, b: number, equaveCents = 1200.0) {
   return Math.abs(circleDifference(a, b, equaveCents));
 }
+
+/**
+ * Calculate the smallest power of two greater or equal to the input value.
+ * @param x Value to compare to.
+ * @returns Smallest `2**n` such that `x <= 2**n`.
+ */
+export function ceilPow2(x: number) {
+  if (x >= 1 && x < 0x40000000) {
+    return 1 << (32 - Math.clz32(x - 1));
+  }
+  if (x <= 0) {
+    return 0;
+  }
+  return 2 ** Math.ceil(Math.log2(x));
+}