#!/usr/bin/env python
# -*- coding: utf-8 -*-

GENRES = [
    "country",
    "classical",
    "electronic",
    "experimental",
    "folk",
    "hiphop",
    "international",
    "jazz",
    "metal",
    "pop",
    "rock",
]

GENRES_VEC = [0] * len(GENRES)

COUNTRY = {
    "contemporarybluegrass",
    "country",
}

CLASSICAL = {
    "classical",
}

ELECTRONIC = {
    "ambient",
    "club/dance",
    "dance",
    "downbeat",
    "dubstep",
    "edm",
    "downtempo",
    "electronic",
    "electronica",
    "house",
    "idm",
    "instrumental",
    "lo-fi",
    "techno",
}

EXPERIMENTAL = {
    "avant-garde",
    "experimental",
}

FOLK = {
    "folk"
}

HIPHOP = {
    "dirtysouth",
    "hip-hop",
    "trip-hop",
    "hiphop",
    "rap"
}

INTERNATIONAL = {
    "africantraditions",
    "americana",
    "celtic",
    "centralamerican",
    "global",
    "dancehall",
    "irish",
    "latin",
    "latindance",
    "welsh",
    "westafrican",
    "world",
}

JAZZ = {
    "afrobeat",
    "blues",
    "funk",
    "jazz",
}

METAL = {
    "metal",
}

POP = {
    "adultalternative",
    "adultcontemporary",
    "gospel",
    "neo-soul",
    "pop",
    "r&b",
    "reggae",
    "retro-soul",
    "soul",
}

ROCK = {
    "alternative",
    "emo",
    "garage",
    "grunge",
    "indie",
    "indie/alternative",
    "neo-prog",
    "newwave",
    "punk",
    "rock",
}

AMBIGUOUS = {
    "pop/rock",
    "...",
    "live",
    "singer-songwriter",
    "singer/songwriter",
    "comedy",
    "comedy/spoken",
}
