support djh2 effects using a new EFFECTS midi track

Author: shockdude

Sharktooth/Mub/MubExport.cs

@@ -22,6 +22,9 @@ public void Export(string path)
             MidiEventCollection mid = new MidiEventCollection(1, DELTA_TICKS_PER_QUARTER);
             mid.AddTrack(CreateTempoTrack());
             mid.AddTrack(CreateTrack());
+            List<MidiEvent> effectsTrack = CreateEffectsTrack();
+            if (effectsTrack != null)
+                mid.AddTrack(effectsTrack);
 
             MidiFile.Export(path, mid);
         }
@@ -51,6 +54,10 @@ private List<MidiEvent> CreateTrack()
                 long start = (long)(entry.Start * ticksPerMeasure);
                 long end = (long)(start + (entry.Length * ticksPerMeasure));
 
+                // DJH2 effect type - 0x05FFFFFF through 0x06000009
+                // Should not be added to the notes track
+                if ((entry.Modifier & 0xFF000000) == 0x06000000 || entry.Modifier == 0x05FFFFFF) continue;
+
                 if ((entry.Modifier & 0xFFFFFF) == 0xFFFFFF)
                 {
                     // Text event?
@@ -87,5 +94,46 @@ private List<MidiEvent> CreateTrack()
             track.Add(new MetaEvent(MetaEventType.EndTrack, 0, track.Last().AbsoluteTime));
             return track;
         }
+
+        private List<MidiEvent> CreateEffectsTrack()
+        {
+            List<MidiEvent> effects = null;
+
+            int ticksPerMeasure = DELTA_TICKS_PER_QUARTER * 4; // Assume 4/4
+            foreach (var entry in _mub.Entries)
+            {
+                long start = (long)(entry.Start * ticksPerMeasure);
+                long end = (long)(start + (entry.Length * ticksPerMeasure));
+
+                if ((entry.Modifier & 0xFF000000) == 0x06000000)
+                {
+                    if (effects == null)
+                    {
+                        effects = new List<MidiEvent>();
+                        effects.Add(new NAudio.Midi.TextEvent("EFFECTS", MetaEventType.SequenceTrackName, 0));
+                    }
+                    effects.Add(new NoteEvent(start, 1, MidiCommandCode.NoteOn, entry.Modifier & 0xFF, 100));
+                    effects.Add(new NoteEvent(end, 1, MidiCommandCode.NoteOff, entry.Modifier & 0xFF, 100));
+                    continue;
+                }
+            }
+
+            if (effects != null)
+            {
+                effects.Sort((x, y) =>
+                {
+                    if (x.AbsoluteTime < y.AbsoluteTime)
+                        return -1;
+                    else if (x.AbsoluteTime > y.AbsoluteTime)
+                        return 1;
+                    else
+                        return 0;
+                });
+
+                // Adds end track
+                effects.Add(new MetaEvent(MetaEventType.EndTrack, 0, effects.Last().AbsoluteTime));
+            }
+            return effects;
+        }
     }
 }

Sharktooth/Mub/MubImport.cs

@@ -62,6 +62,29 @@ public Mub ExportToMub()
                     textNote.Text));
             }
 
+            // DJ Hero 2 effects
+            var effectTrack = existingTracks.Keys
+                .Where(x => x == "EFFECTS")
+                .Select(x => existingTracks[x])
+                .FirstOrDefault();
+
+            if (effectTrack != null)
+            {
+                var effects = effectTrack
+                    .Where(x => x is NoteOnEvent)
+                    .Select(x => x as NoteOnEvent);
+
+                foreach (var effect in effects)
+                {
+                    if (effect.Velocity <= 0)
+                        continue;
+
+                    mubNotes.Add(new MubEntry((effect.AbsoluteTime / (DeltaTicksPerQuarter * 4)),
+                        effect.NoteNumber + 0x06_00_00_00,
+                        (effect.NoteLength / (DeltaTicksPerQuarter * 4))));
+                }
+            }
+
             return new Mub()
             {
                 Version = 2,