stt/stt.py

#!/usr/bin/env python3

# include standard modules
import argparse
import os
import sys
import re
import json

mydir = os.path.dirname(os.path.realpath(__file__))

# Lookup value is value of note for ( string * 1000 ) + value of fret
lookuptable = {
    57:   ["Cr", "x"],
    49:   ["Cr", "x"],
    "XX": ["Hh", "x"],
    51:   ["Ri", "x"],
    3048: ["HT", "o"],
    3047: ["MT", "o"],
    4038: ["Sn", "o"],
    3043: ["FT", "o"],
    5035: ["Bd", "o"],
    44:   ["Hf", "x"]
}

### DrumBurp notation   Wikipedia notation             Wikipedia notation
### Cr - Crash          Drums                          Cymbals
### Hh - HiHat          |-o-| Strike                   |-x-| Strike cymbal or hi-hat
### Ri - Ride           |-O-| Accent                   |-X-| Strike loose hi-hat, or hit crash hard
### HT - High Tom       |-g-| Ghost note               |-o-| Open hi-hat
### MT - Mid Tom        |-f-| Flam                     |-#-| Choke cymbal (grab cymbal with hand after striking it)
### Sn - Snare          |-d-| Drag                     |-s-| Splash cymbal
### FT - Floor Tom      |-b-| Soft one-handed roll     |-c-| China cymbal
### Bd - Kick           |-B-| Accented one-handed roll |-b-| Bell of ride
### Hf - Foot pedal     |-@-| Snare rim                |-x-| Click hi-hat with foot
### DrumBurp:  https://whatang.org/
### Wikipedia: https://en.wikipedia.org/wiki/Drum_tablature

def commandline():
    # initiate the parser
    parser = argparse.ArgumentParser(description="Make ASCII tabs from songsterr tabs.")
    # option without argument via 'store_true'
    parser.add_argument("-V", "--version", help="show program version", action="store_true")
    # option with argument
    parser.add_argument("--width", "-w", help="set output width in measures (default: 2)")
    parser.add_argument("--input", "-i", help="set input file")
    parser.add_argument("--url", "-u", help="set input url")
    parser.add_argument("--exclude", "-x", help="exclude unused instruments", action="store_true")

    # read arguments from the command line
    args = parser.parse_args()

    # handle arguments
    # check for --version or -V
    if args.version:
        print("Version 0.1")
    # check for --width
    if args.width:
        print("set output width to %s" % args.width)
    else:
        args.width = 2
    if args.url:
        print("url not handled yet %s" % args.width)
    if args.input and args.url:
        print("Give either a url, or an input file, not both.")
        exit()
    if not args.input and not args.url:
        print("Give either a url, or an input file.")
        exit()

    return args


def get_json(html):
    result = re.search(r'<script id="state" type="application/json">(?P<json>.*?)</script', html, re.MULTILINE)
    if result:
        data = json.loads(result.group("json"))
    else:
        print("Can't find the magic")
    return data

def print_meta(jsondata):
    print( "Artist: " + jsondata["meta"]["artist"] )
    print( "Title: " + jsondata["meta"]["title"] )
    print( "Instrument: " + jsondata["data"]["part"]["instrument"] )
    print( "BPM: " + str( jsondata["data"]["part"]["measures"][0]["voices"][0]["beats"][0]["tempo"]["bpm"] ) )
    print("")

    return


def get_skip(tp, typelength):
    if tp < typelength:
        skip = int( ( typelength / tp ) )
    else:
        skip = 1

    return skip

def get_typelength(jsondata):
    typelength = 0
    for i in range(len(jsondata["data"]["part"]["measures"])):
        for j in range(len(jsondata["data"]["part"]["measures"][i]["voices"][0]["beats"])):
            beat = jsondata["data"]["part"]["measures"][i]["voices"][0]["beats"][j]
            if typelength < beat["type"]:
                typelength = beat["type"]
    return typelength

def parse_instruments(jsondata):
# In the end I want an array of measures for each instrument
# That'll make for easy printing


    used_instr = {
        "Cr": False,
        "Hh": False,
        "Ri": False,
        "HT": False,
        "MT": False,
        "Sn": False,
        "FT": False,
        "Bd": False,
        "Hf": False
    }

    instruments = {
        "Cr": [],
        "Hh": [],
        "Ri": [],
        "HT": [],
        "MT": [],
        "Sn": [],
        "FT": [],
        "Bd": [],
        "Hf": []
    }

    typelength = get_typelength(jsondata)
    #print("Typelength: " + str(typelength))
    #print("")


    for measurecnt in range(len(jsondata["data"]["part"]["measures"])):
        count = 0

        # add one measure to each instrument
        for instr in ["Cr", "Hh", "Ri", "HT", "MT", "Sn", "FT", "Bd", "Hf"]:
            instruments[instr].append([])
            for i in range(typelength):
                instruments[instr][-1].append("-")

        for beatcnt in range(len(jsondata["data"]["part"]["measures"][measurecnt]["voices"][0]["beats"])):
            beat = jsondata["data"]["part"]["measures"][measurecnt]["voices"][0]["beats"][beatcnt]
            skip = get_skip(beat["type"], typelength)

            for note in beat["notes"]:
                if note != {'rest': True}:
                    lookupval = ( note["string"] * 1000 ) + note["fret"]
                    inst, marker = lookuptable[lookupval]
                    #print(measurecnt, inst, count)
                    if inst != None:
                        instruments[inst][-1][count] = marker
                        used_instr[inst] = True
                    else:
                        print("Unhandled: " + note + " in measure " + str(measurecnt+1))
                        print("Add: " + str(( note["string"] * 1000 ) + note["fret"]) + " to lookuptable")
            count += skip

    return instruments, used_instr

def print_instruments(instruments, used_instr):
    measuresnr = len(instruments["Cr"])
    width = int(args.width)

    rows = ( measuresnr / width )
    count = 0
    while count < measuresnr:
        for instr in ["Cr", "Hh", "Ri", "HT", "MT", "Sn", "FT", "Bd", "Hf"]:
            if not args.exclude or used_instr[instr]:
                sys.stdout.write(instr + " |")
                for i in range(count, count+width):
                    if i < measuresnr:
                        sys.stdout.write("".join(instruments[instr][i]) + "|")
                print("")
        print("")
        count += width

args = commandline()

with open(args.input, 'r') as content_file:
    content = content_file.read()

jsondata = get_json(content)
print_meta(jsondata)

instr, used_instr = parse_instruments(jsondata)
print_instruments(instr, used_instr)