@ugfoundation/swaralipi-js 0.3.0 → 0.3.1

package/dist/player/index.cjs

@@ -1066,34 +1066,72 @@ function utf8(s) {
 function u32be(n) {
   return [n >>> 24 & 255, n >>> 16 & 255, n >>> 8 & 255, n & 255];
 }
+var DRUM_NOTE = {
+  kick: 36,
+  snare: 38,
+  hihat: 42,
+  // closed hi-hat
+  tom1: 48,
+  tom2: 45,
+  tom3: 41
+};
+function trackChunk(raw, head) {
+  raw.sort((a, b) => a.tick - b.tick || Number(a.on) - Number(b.on));
+  const track = [...head];
+  let last = 0;
+  for (const e of raw) {
+    const status = (e.on ? 144 : 128) | e.ch & 15;
+    track.push(...vlq(e.tick - last), status, e.note & 127, e.vel & 127);
+    last = e.tick;
+  }
+  track.push(...vlq(0), 255, 47, 0);
+  return [77, 84, 114, 107, ...u32be(track.length), ...track];
+}
 function encodeMidi(schedule, opts) {
-  const raw = [];
+  const melody = [];
   for (const ev of schedule.events) {
     if (ev.midi === null) continue;
     const start = Math.round(ev.beat * PPQ);
     const end = Math.max(Math.round((ev.beat + ev.durationBeats) * PPQ), start + 1);
     if (ev.kan !== null) {
       const kOn = Math.max(start - Math.round(0.12 * PPQ), 0);
-      raw.push({ tick: kOn, on: true, note: ev.kan, vel: 70 });
-      raw.push({ tick: start, on: false, note: ev.kan, vel: 0 });
+      melody.push({ tick: kOn, on: true, note: ev.kan, vel: 70, ch: 0 });
+      melody.push({ tick: start, on: false, note: ev.kan, vel: 0, ch: 0 });
     }
-    raw.push({ tick: start, on: true, note: ev.midi, vel: 100 });
-    raw.push({ tick: end, on: false, note: ev.midi, vel: 0 });
+    melody.push({ tick: start, on: true, note: ev.midi, vel: 100, ch: 0 });
+    melody.push({ tick: end, on: false, note: ev.midi, vel: 0, ch: 0 });
   }
-  raw.sort((a, b) => a.tick - b.tick || Number(a.on) - Number(b.on));
-  const track = [];
-  if (opts.name) track.push(...vlq(0), 255, 3, ...vlq(utf8(opts.name).length), ...utf8(opts.name));
   const usPerBeat = Math.round(6e7 / opts.bpm);
-  track.push(...vlq(0), 255, 81, 3, usPerBeat >> 16 & 255, usPerBeat >> 8 & 255, usPerBeat & 255);
-  let last = 0;
-  for (const e of raw) {
-    track.push(...vlq(e.tick - last), e.on ? 144 : 128, e.note & 127, e.vel & 127);
-    last = e.tick;
+  const head = [];
+  if (opts.name) head.push(...vlq(0), 255, 3, ...vlq(utf8(opts.name).length), ...utf8(opts.name));
+  head.push(...vlq(0), 255, 81, 3, usPerBeat >> 16 & 255, usPerBeat >> 8 & 255, usPerBeat & 255);
+  const chunks = [trackChunk(melody, head)];
+  if (opts.drums && opts.drums.length > 0) {
+    const drum = [];
+    const totalBeats = Math.ceil(schedule.totalBeats);
+    for (let beat = 0; beat < totalBeats; beat++) {
+      const note = DRUM_NOTE[opts.drums[beat % opts.drums.length]];
+      const t = beat * PPQ;
+      drum.push({ tick: t, on: true, note, vel: 100, ch: 9 });
+      drum.push({ tick: t + Math.round(PPQ / 4), on: false, note, vel: 0, ch: 9 });
+    }
+    chunks.push(trackChunk(drum, [...vlq(0), 255, 3, ...vlq(5), 68, 114, 117, 109, 115]));
   }
-  track.push(...vlq(0), 255, 47, 0);
-  const header = [77, 84, 104, 100, ...u32be(6), 0, 0, 0, 1, PPQ >> 8 & 255, PPQ & 255];
-  const trackChunk = [77, 84, 114, 107, ...u32be(track.length), ...track];
-  return new Uint8Array([...header, ...trackChunk]);
+  const format = chunks.length > 1 ? 1 : 0;
+  const header = [
+    77,
+    84,
+    104,
+    100,
+    ...u32be(6),
+    0,
+    format,
+    chunks.length >> 8 & 255,
+    chunks.length & 255,
+    PPQ >> 8 & 255,
+    PPQ & 255
+  ];
+  return new Uint8Array([...header, ...chunks.flat()]);
 }
 function safeName2(title) {
   return (title ?? "").replace(/[\\/:*?"<>|]+/g, "").trim() || "swaralipi";
@@ -1101,7 +1139,10 @@ function safeName2(title) {
 function exportMidi(doc, options = {}) {
   const schedule = buildSchedule(doc, { tonic: options.tonic, baseOctave: options.baseOctave });
   const bpm = options.bpm ?? doc.meta.bpm ?? 90;
-  const bytes = encodeMidi(schedule, { bpm, name: options.name ?? doc.meta.title });
+  const taalId = doc.meta.taal?.id;
+  const taal = taalId !== void 0 && taalId in TAALS ? TAALS[taalId] : { numBeats: 4, sam: 0 };
+  const drums = options.drums ? drumPattern(taal) : void 0;
+  const bytes = encodeMidi(schedule, { bpm, name: options.name ?? doc.meta.title, drums });
   return {
     blob: new Blob([bytes], { type: "audio/midi" }),
     filename: `${safeName2(doc.meta.title)}.mid`,