# The following is how these files were generated from memory dumps

import numpy as np
unk1_lut = np.fromfile('unk1_values.bin', dtype=np.int32)
unk2_lut = np.fromfile('unk2_values.bin', dtype=np.int32)
unk1_rlut = {v: i for i, v in enumerate(unk1_values)}
unk2_rlut = {v: i for i, v in enumerate(unk2_values)}
notetype_lut = ['e', 't', 'h', 'b', 's', 'x']

def make_rgtx(filename, data):
  # Near-lossless (float rounding) representation of the memory dumps.
  # Once the meaning of unk1 and unk2 are determined, it should be possible to interpret a chart 100% correctly from this.
  out_filename = filename.partition('.')[0] + '.rgtx'
  lines = []
  if len(data.shape) != 2:
    data = data.reshape([-1,20])
  times = data[:,:2].view(dtype=np.float32).sum(1)
  durations = data[:,2].view(dtype=np.float32)
  cols = data[:,3]
  ntypes = [notetype_lut[t] for t in data[:,4]]
  sids = data[:,5]
  stypes = data[:,6]
  unk1s = [unk1_rlut[i] for i in data[:,7]]
  unk2s = [unk2_rlut[i] for i in data[:,8]]
  for i in range(len(times)):
    timestr = f'{times[i]:.7g}'
    notestr = f':{cols[i]}{ntypes[i]}{durations[i]:.7g},{sids[i]},{stypes[i]:x},{unk1s[i]:x},{unk2s[i]:x}'
    if i==0 or times[i]!=times[i-1]:
      lines.append(timestr + notestr)
    else:
      lines[-1] = lines[-1] + notestr
  with open(out_filename, 'w') as f:
    f.write('\n'.join(lines))


def make_rgts(filename, data):
  # Simplified format that excludes durations for non-holds and slideIDs for non-slides
  # Also excludes the unk1 and unk2 values for now which may discard important information
  out_filename = filename.partition('.')[0] + '.rgts'
  lines = []
  if len(data.shape) != 2:
    data = data.reshape([-1,20])
  times = data[:,:2].view(dtype=np.float32).sum(1)
  durations = data[:,2].view(dtype=np.float32)
  cols = data[:,3]
  ntypes = [notetype_lut[t] for t in data[:,4]]
  sids = data[:,5]
  stypes = data[:,6]
  for i in range(len(times)):
    timestr = f'{times[i]:.7g}'
    notestr = f':{cols[i]}{ntypes[i]}'
    if ntypes[i]=='h':
      notestr += f'{durations[i]:.7g}'
    elif ntypes[i] in ('s', 'e'):
      notestr += f'{stypes[i]:x}{sids[i]}'  # No comma, stype is exactly one character
    if i==0 or times[i]!=times[i-1]:
      lines.append(timestr + notestr)
    else:
      lines[-1] = lines[-1] + notestr
  with open(out_filename, 'w') as f:
    f.write('\n'.join(lines))


def make_files_from_dump(filename):
  data = np.fromfile(filename, dtype=np.int32)
  make_rgtx(filename, data)
  make_rgts(filename, data)

# If you load this file in interactive mode (python -i example.py) you should be able to run the above function to convert any new memory dumps of the MMF format