config.py

#!/usr/bin/env python3

import pathlib
import shutil

import os

part = os.environ.get('PART')
if not part:
    part = '4'

print(f'Part {part}')

chromosome = '22'

# Path to source treeseq file.
#treeseq = 'sgdp_chr22.trees'
treeseq = 'hgdp_tgp_sgdp_chr22_q.dated.trees'
#treeseq = 'hgdp_1kg_sgdp_high_cov_ancients_chr22.dated.trees'

years_per_generation = 25 # 25 years per generation in latest awohns code.

# Different treeseqs store location data differently.
# Modify database.Database.get_location() to add more choices.
#treeseq_type = 'sgdp'

# This needs to be a valid SQL table name, since that what it's going to become.
# See the order module to add choices.
#order = 'average_longitude'
#order = 'traveller_45e_30s_105w_40s'
#order = 'traveller_19e_39s_99w_19n'
#order = 'traveller_roundtrip_19e_39s_max480'
#order = 'traveller_19e_39s_99w_19n_max1000'
#order = 'traveller_roundtrip_19e_39s_max1000_byspread'
#order = 'traveller_roundtrip_19e_39s_max480_byspread'
#order = 'total_distance_to_average_location'
#order = 'two_world_30w_65s_169w_65n_max480'
#order = 'two_world_30w_0s_169w_65n_max480'
#order = 'two_world_jaccard5_30w_30s_169w_65n_max480'
#order = 'two_world_jaccard20_30w_30s_169w_65n_max480'
#order = 'two_world_jaccard20_30w_30s_169w_65n_max120'
#order = 'two_world_jaccard20_30w_30s_169w_65n_max480_group_limit'
#order = 'two_world_jaccard20_30w_30s_169w_65n_max480_round5'
#order = 'two_world_jaccard20_30w_30s_169w_65n_max120_round5'
#order = 'two_world_jaccard20_30w_30s_169w_65n_max360_round25'
#order = 'two_world_jaccard20_30w_30s_169w_65n_max360_round5'
#order = 'two_world_jaccard20_30w_30s_169w_65n_max240_group_limit_average_times'
order = 'two_world_jaccard20_30w_30s_169w_65n_max480_group_limit_average_times'


# Video parameters.
width = 2560
height = 1440
resolution = (width, height)
# 8 fps gives a Youtube viewer a range of 2fps to 16fps, which seems to be
# a reasonable range for this particular movie.
video_framerate = 8
#video_framerate = 2
audio_samplerate = 48000

def w(value):
    'Scale 2560-based x values to whatever the actual width is.'
    return value * width / 2560

def h(value):
    'Scale 1440-based y values to whatever the actual height is.'
    return value * height / 1440

# Required executables.
#ffmpeg = '/usr/bin/ffmpeg'
ffmpeg = shutil.which('ffmpeg') or r'C:\Program Files\ffmpeg\bin\ffmpeg.exe'
#ffmpeg = r'C:\Program Files\ffmpeg\bin\ffmpeg.exe'
#inkscape = '/usr/bin/inkscape'
inkscape = shutil.which('inkscape') or r'C:\Program Files\Inkscape\bin\inkscape.exe'
timidity = shutil.which('timidity') or r'C:\Program Files (x86)\TiMidity\timidity.exe'

treeseq_path = pathlib.Path(treeseq)

# We have to look for a couple of things in the code directory.
code = pathlib.Path(__file__).parent

# Use the basename of the treeseq file as our working directory.
working = pathlib.Path('.')/treeseq_path.stem
data = working/'data'
images = working/'layers'
audio = working/'audio'
movie = working/'movie'

# Collect a list of folders to create at the end.
folders = [data, images, audio, movie]

database_path = data/f'{treeseq_path.stem}.sqlite'
database_readonly_uri = f'file:{database_path.as_posix()}?mode=ro'
#weighted_average_locations = False
weighted_average_locations = True

# Adds some shadows to text and lines to make lines a little cleaner when
# movies are resized to arbitrary sizes by the viewer, or when high-resolution
# movies are viewed on low-resolution screens.  It's a hack, but it seems to
# kinda work.
shadows = True

#map_image = 'blue_marble'
#map_projection = 'natural_earth_2'
#map_rotation = (-150, 0)  # I like either -11 or -150.
#map_format = 'png'
#map_center = (w(1280), h(720-36))
#map_width = w(2048)
#map_height = h(1024)
#map_scale = 1

map_image = 'visionscarto_lightgrey'
#map_image = 'visionscarto_graticules'
map_projection = 'bertin'
map_rotation = (-16.5, -42)
map_format = 'svg'
map_center = (w(1280), h(720-36))
map_height = h(1024)
# These are the specific visionscarto dimensions - 700x475.
map_scale = map_height/475
map_width = map_scale*700


# Map images have to be manually created for now.
# I've been using G.Projector and GIMP.
default_world_map = code/'maps'/f'{map_image}_{map_projection}_{map_rotation[0]}_{map_rotation[1]}.{map_format}'

layer_names = [
    'world_map',
    'chromosome_map',
    'chromosome_position',
    'worldwide_frequency',
    'variant_histogram',
    'clef',
    #'note',
    'max_local',
    'graticule_vertices',
    'population_histogram',
    'legend_population_histogram',
    'traces',
    'average_location',
    'all_average_locations',
    'local_frequencies',
    'legend_frequency',
    'legend_position',
    'variant',
    'populations',
    'caption',
    'citation',
    'date',
    'background',
    'foreground',
]

background_layers = [
    # For Natural Earth 2 PNG we put world map here to avoid large file sizes.
    #'world_map',
    #'graticule_vertices',
    'chromosome_map',
    #'max_local',
    #'clef',
    'legend_frequency',
    'legend_position',
]

foreground_layers = [
    # For Bertin SVG we put world map here to get cleaner traces overlays.
    'world_map',
    #'graticule_vertices',
    'chromosome_position',
    'variant_histogram',
    #'note',
    'clef',
    'legend_population_histogram',
    'population_histogram',
    'traces',
    'average_location',
    'local_frequencies',
    'variant',
    'populations',
    'caption',
    'citation',
    'date',
]

movie_layers = [
    #'world_map',
    #'chromosome_map',
    #'chromosome_position',
    #'clef',
    #'note',
    #'traces',
    #'average_location',
    #'local_frequencies',
    #'legend_frequency',
    #'legend_position',
    #'variant',
    #'caption',
    #'date',
    'background',
    'foreground',
]


map_layers = set([
    'average_location',
    'local_frequencies',
    'world_map',
    'max_local',
    'graticule_vertices',
    'all_average_locations',
    'background',
    'traces',
    'background',
    'foreground',
])
shadow_layers = set([
    'note',
    'variant',
    'populations',
    'caption',
    'citation',
    'date',
    'worldwide_frequency',
    'chromosome_map',
    #'legend_population_histogram',
    'legend_frequency',
    'legend_position',
    'clef',
    'background',
    'foreground',
])
order_layers = set([
    'count',
    'variant_histogram',
    'population_histogram',
    'traces',
    'background',
    'foreground',
])

class Stub():
    pass

layers = Stub()

mapfolder = f'{map_projection}_{map_rotation[0]}_{map_rotation[1]}'
shadowfolder = ('shadows' if shadows else 'noshadows')
orderfolder = order

for name in layer_names:
    layer = Stub()
    setattr(layers, name, layer)

    basefolder = images
    if name in map_layers:
        basefolder /= mapfolder
        if name not in ['foreground', 'background']:
            layer.center = map_center
            layer.height = map_height
            layer.width = map_width
    if name in shadow_layers:
        basefolder /= shadowfolder
    if name in order_layers:
        basefolder /= orderfolder

    if name in background_layers or name in ['background', 'world_map']:
        frame_suffix = ''
    #elif name == 'note':
    elif name == 'clef':
        frame_suffix = '.%s'
    else:
        frame_suffix = '.%08d'

    if name == 'foreground':
        layer.svg = basefolder/f'{name}-{part}'/'svg'/f'{name}{frame_suffix}.svg'
        layer.png = basefolder/f'{name}-{part}'/'png'/f'{name}{frame_suffix}.png'
    else:
        layer.svg = basefolder/name/'svg'/f'{name}{frame_suffix}.svg'
        layer.png = basefolder/name/'png'/f'{name}{frame_suffix}.png'
    layer.concat = movie/order/f'{name}.concat'

    folders.append(layer.svg.parent)
    folders.append(layer.png.parent)
    folders.append(layer.concat.parent)

## This is getting messy.  Clean up once final decisions are made.
#shadowfolder = images/('shadows' if shadows else 'noshadows')
#shadoworderfolder = images/('shadows' if shadows else 'noshadows')/order
#mapfolder = images/f'{map_projection}_{map_rotation[0]}_{map_rotation[1]}'
#maporderfolder = images/f'{map_projection}_{map_rotation[0]}_{map_rotation[1]}'/order
#finalfolder = images/f'{map_projection}_{map_rotation[0]}_{map_rotation[1]}'/('shadows' if shadows else 'noshadows')/order
#
#layers = Stub()
#for name in layer_names:
#    layer = Stub()
#    setattr(layers, name, layer)
#    if name in ['average_location', 'local_frequencies']:
#        layer.svg = images/mapfolder/name/'svg'/f'{name}.%08d.svg'
#        layer.png = images/mapfolder/name/'png'/f'{name}.%08d.png'
#        layer.center = map_center
#        layer.height = map_height
#        layer.width = map_width
#    elif name in ['world_map', 'max_local', 'graticule_vertices', 'all_average_locations', 'background']:
#        layer.svg = images/mapfolder/name/'svg'/f'{name}.svg'
#        layer.png = images/mapfolder/name/'png'/f'{name}.png'
#        layer.center = map_center
#        layer.height = map_height
#        layer.width = map_width
#    elif name in ['note']:
#        # Clef indexing on note name.
#        layer.svg = images/shadowfolder/name/'svg'/f'{name}.%s.svg'
#        layer.png = images/shadowfolder/name/'png'/f'{name}.%s.png'
#    elif name in ['variant', 'caption', 'date', 'worldwide_frequency']:
#        # Only use shadows for black lines/text.
#        layer.svg = images/shadowfolder/name/'svg'/f'{name}.%08d.svg'
#        layer.png = images/shadowfolder/name/'png'/f'{name}.%08d.png'
#    elif name in ['chromosome_map', 'legend_frequency', 'legend_position', 'clef']:
#        # Only use shadows for black lines/text.
#        layer.svg = images/shadowfolder/name/'svg'/f'{name}.svg'
#        layer.png = images/shadowfolder/name/'png'/f'{name}.png'
#    elif name in ['traces']:
#        layer.svg = images/mapfolder/orderfolder/name/'svg'/f'{name}.%08d.svg'
#        layer.png = images/mapfolder/orderfolder/name/'png'/f'{name}.%08d.png'
#        layer.center = map_center
#        layer.height = map_height
#        layer.width = map_width
#    elif name in ['final']:
#        layer.svg = images/mapfolder/orderfolder/shadowfolder/name/'svg'/f'{name}.%08d.svg'
#        layer.png = images/mapfolder/orderfolder/shadowfolder/name/'png'/f'{name}.%08d.png'
#    elif name in ['count']:
#        layer.svg = images/shadowfolder/orderfolder/name/'svg'/f'{name}.%08d.svg'
#        layer.png = images/shadowfolder/orderfolder/name/'png'/f'{name}.%08d.png'
#    else:
#        layer.svg = images/name/'svg'/f'{name}.%08d.svg'
#        layer.png = images/name/'png'/f'{name}.%08d.png'
#    # ffmpeg concat files, used so that we only have to generate the frames
#    # once and can assemble them in arbitrary order to make the movie.
#    layer.concat = movie/order/f'{name}.concat'
#    folders.append(layer.svg.parent)
#    folders.append(layer.png.parent)
#    folders.append(layer.concat.parent)

font_large = h(72)
font_medium = h(54)
font_small = h(54)
font_tiny = h(32)
font_tinier = h(24)
font_miniscule = h(16)

layers.world_map.scale = map_scale

layers.chromosome_map.sections = [
    {'axis': 'width', 'start': (1248, 1280), 'end': (1120, 1320)},
    {'axis': 'width', 'start': (1120, 1280), 'end': (0, 1440)},
    {'axis': 'height', 'start': (0, 1280), 'end': (160, 160)},
    {'axis': 'width', 'start': (0, 160), 'end': (2560, 0)},
    {'axis': 'height', 'start': (2560, 160), 'end': (2400, 1280)},
    {'axis': 'width', 'start': (2560, 1280), 'end': (1440, 1440)},
    {'axis': 'width', 'start': (1440, 1280), 'end': (1312, 1320)},
]
layers.chromosome_map.radius = 8

layers.max_local.max_radius = h(16)
layers.max_local.stroke_width = 0
layers.max_local.style = 'stroke:none;fill:pink;fill-opacity:1.0;'

layers.graticule_vertices.max_radius = 1
layers.graticule_vertices.stroke_width = 0
layers.graticule_vertices.style = 'stroke:none;fill:pink;fill-opacity:1.0;'
layers.graticule_vertices.spacing = 10 # degrees

layers.average_location.max_radius = h(8)
layers.average_location.stroke_width = 0
layers.average_location.style = 'stroke:none;fill:black;fill-opacity:1.0;'

layers.local_frequencies.max_radius = h(32)
layers.local_frequencies.min_radius = h(4)
layers.local_frequencies.fixed_radius = False
# We need stroke width separated out from the style for calculations.
layers.local_frequencies.stroke_width = 8
#layers.local_frequencies.style = 'stroke:red;stroke-opacity:1.0;fill:red;fill-opacity:0.6;'
#layers.local_frequencies.style = 'stroke:orange;stroke-opacity:1.0;fill:orange;fill-opacity:0.7;'
layers.local_frequencies.style = 'stroke:red;stroke-opacity:1.0;fill:red;fill-opacity:0.5;'
layers.local_frequencies.shrink_style = 'stroke:#208cff;stroke-opacity:1.0;fill:#208cff;fill-opacity:0.5;'
layers.local_frequencies.shrink_below_sample_count = 20
layers.local_frequencies.shrink_factor = 0.5
#layers.local_frequencies.color = '#ff0000ff'

layers.population_histogram.center = (w(1280), h(1045))
layers.population_histogram.width = w(840)
layers.population_histogram.height = h(430)
layers.population_histogram.style = 'stroke:none;fill:#ff00fd;'
layers.population_histogram.bar_width = 4
# FIXME: bar_height_[min|max] are not yet used.  See graph.py.
layers.population_histogram.bar_height_max = 244
layers.population_histogram.bar_height_min = 4
layers.population_histogram.deque_length = 480
layers.population_histogram.scale_frequencies = [.05, .1, .2, .4, .6, .8]
layers.population_histogram.scale_style = 'font-family:sans-serif;stroke:#ff00fd;fill:#ff00fd;'
layers.population_histogram.font_size = font_miniscule

#layers.traces.style = 'stroke:#ffe800;stroke-width:5;stroke-linecap:round;fill:none;'
layers.traces.style = 'stroke:#ecc669;stroke-width:5;stroke-linecap:round;fill:none;'
# How long to keep a trace around before removing it.
layers.traces.deque_length = 480
layers.traces.start_order = 579_840
layers.traces.prefer_pacific = False

layers.worldwide_frequency.center = (w(220), h(220))
layers.worldwide_frequency.max_radius = h(100)
# We need stroke width separated out from the style for calculations.
layers.worldwide_frequency.stroke_width = 4
layers.worldwide_frequency.style = 'stroke:red;stroke-opacity:1.0;fill:red;fill-opacity:0.6;'
layers.worldwide_frequency.backing_style = 'stroke:none;fill:grey;fill-opacity:0.4;'
layers.worldwide_frequency.font_size = font_medium
layers.worldwide_frequency.text_style = 'font-family:sans-serif;'

layers.variant_histogram.center = (w(2180), h(683))
layers.variant_histogram.width = w(360)
layers.variant_histogram.height = h(270)
layers.variant_histogram.style = 'stroke:#ffffff;stroke-width:2px;fill:#29efff;'
layers.variant_histogram.bar_width = 720
layers.variant_histogram.bar_height = 22.5
layers.variant_histogram.deque_length = 480
layers.variant_histogram.scale_frequencies = [0, .1, .2]
layers.variant_histogram.scale_style = 'font-family:sans-serif;stroke:#29efff;fill:#29efff;'
layers.variant_histogram.font_size = font_miniscule

# We're measuring clef from top left instead of centre.
# Maybe it would be easier for ffmpeg setup to use centre for all layers?
# In which case change this position to a center measurement.
layers.clef.center = (w(2204), h(686))
#layers.clef.scale = h(280/18000)
layers.clef.width = h(360)
layers.clef.height = h(360)

layers.variant.center = (w(1966), h(430))
layers.variant.font_size = font_small
layers.variant.width = font_small * 24
layers.variant.height = font_small * 4
layers.variant.style = 'font-family:sans-serif;'

# Do we need width and height here?  Hopefully not?
layers.populations.font_size = font_miniscule
layers.populations.style = 'font-family:Raleway;font-weight:800;'
layers.populations.line_spacing = 1.25
layers.populations.source = {
    'SGDP': {
        'title': 'Simons Genome Diversity Project',
        'top': h(196),
        'left': w(180),
        'column': {
            0: w(0),
            1: w(125),
            2: w(275),
            3: w(405),
        },
    },
    'HGDP': {
        'title': 'Human Genome Diversity Project',
        'top': h(1008),
        'left': w(180),
        'column': {
            0: w(0),
            1: w(160),
            2: w(260),
            3: w(380),
            4: w(530),
        },
    },
    '1KG': {
        'title': '1000 Genomes Project',
        'top': h(1048),
        'left': w(1860),
        'column': {
            0: w(0),
            1: w(180),
            2: w(340),
        },
    },
}
# I'm sure there's a less manual way to do this, but in the interest of
# getting it done...
# Ideally with an automatic method we wouldn't want to be tied to population
# IDs, which can change between treesequence DBs.  Source and name should,
# in theory, be more reliable.
layers.populations.position = {
    ('SGDP', 'Abkhasian'): (0, 0, 'Abkhasian', 'Europe'),
    ('SGDP', 'Adygei'): (0, 1, 'Adygei', 'Europe'),
    ('SGDP', 'Albanian'): (0, 2, 'Albanian', 'Europe'),
    ('SGDP', 'Aleut'): (0, 3, 'Aleut', 'Central Asia'),
    ('SGDP', 'Altaian'): (0, 4, 'Altaian', 'Central Asia'),
    ('SGDP', 'Ami'): (0, 5, 'Ami', 'East Asia'),
    ('SGDP', 'Armenian'): (0, 6, 'Armenian', 'Europe'),
    ('SGDP', 'Atayal'): (0, 7, 'Atayal', 'East Asia'),
    ('SGDP', 'Australian'): (0, 8, 'Australian', 'Oceania'),
    ('SGDP', 'Balochi'): (0, 9, 'Balochi', 'South Asia'),
    ('SGDP', 'BantuHerero'): (0, 10, 'Bantu Herero', 'Africa'),
    ('SGDP', 'BantuKenya'): (0, 11, 'Bantu Kenya', 'Africa'),
    ('SGDP', 'BantuTswana'): (0, 12, 'Bantu Tswana', 'Africa'),
    ('SGDP', 'Basque'): (0, 13, 'Basque', 'Europe'),
    ('SGDP', 'BedouinB'): (0, 14, 'Bedouin', 'Middle East'),
    ('SGDP', 'Bengali'): (0, 15, 'Bengali', 'South Asia'),
    ('SGDP', 'Bergamo'): (0, 16, 'Bergamo', 'Europe'),
    ('SGDP', 'Biaka'): (0, 17, 'Biaka', 'Africa'),
    ('SGDP', 'Bougainville'): (0, 18, 'Bougainville', 'Oceania'),
    ('SGDP', 'Brahmin'): (0, 19, 'Brahmin', 'South Asia'),
    ('SGDP', 'Brahui'): (0, 20, 'Brahui', 'South Asia'),
    ('SGDP', 'Bulgarian'): (0, 21, 'Bulgarian', 'Europe'),
    ('SGDP', 'Burmese'): (0, 22, 'Burmese', 'East Asia'),
    ('SGDP', 'Burusho'): (0, 23, 'Burusho', 'South Asia'),
    ('SGDP', 'Cambodian'): (0, 24, 'Cambodian', 'East Asia'),
    ('SGDP', 'Chane'): (0, 25, 'Chane', 'America'),
    ('SGDP', 'Chechen'): (0, 26, 'Chechen', 'Europe'),
    ('SGDP', 'Chukchi'): (0, 27, 'Chukchi', 'Central Asia'),
    ('SGDP', 'Crete'): (0, 28, 'Crete', 'Europe'),
    ('SGDP', 'Czech'): (0, 29, 'Czech', 'Europe'),
    ('SGDP', 'Dai'): (0, 30, 'Dai', 'East Asia'),
    ('SGDP', 'Daur'): (0, 31, 'Daur', 'East Asia'),
    ('SGDP', 'Dinka'): (0, 32, 'Dinka', 'Africa'),
    ('SGDP', 'Druze'): (0, 33, 'Druze', 'Middle East'),
    ('SGDP', 'Dusun'): (0, 34, 'Dusun', 'Oceania'),
    ('SGDP', 'English'): (0, 35, 'English', 'Europe'),
    ('SGDP', 'Esan'): (0, 36, 'Esan', 'Africa'),
    ('SGDP', 'Eskimo_Chaplin'): (1, 0, 'Eskimo Chaplin', 'Central Asia'),
    ('SGDP', 'Eskimo_Naukan'): (1, 1, 'Eskimo Naukan', 'Central Asia'),
    ('SGDP', 'Eskimo_Sireniki'): (1, 2, 'Eskimo Sireniki', 'Central Asia'),
    ('SGDP', 'Estonian'): (1, 3, 'Estonian', 'Europe'),
    ('SGDP', 'Even'): (1, 4, 'Even', 'Central Asia'),
    ('SGDP', 'Finnish'): (1, 5, 'Finnish', 'Europe'),
    ('SGDP', 'French'): (1, 6, 'French', 'Europe'),
    ('SGDP', 'Gambian'): (1, 7, 'Gambian', 'Africa'),
    ('SGDP', 'Georgian'): (1, 8, 'Georgian', 'Europe'),
    ('SGDP', 'Greek'): (1, 9, 'Greek', 'Europe'),
    ('SGDP', 'Han'): (1, 10, 'Han', 'East Asia'),
    ('SGDP', 'Hawaiian'): (1, 11, 'Hawaiian', 'Oceania'),
    ('SGDP', 'Hazara'): (1, 12, 'Hazara', 'South Asia'),
    ('SGDP', 'Hezhen'): (1, 13, 'Hezhen', 'East Asia'),
    ('SGDP', 'Hungarian'): (1, 14, 'Hungarian', 'Europe'),
    ('SGDP', 'Icelandic'): (1, 15, 'Icelandic', 'Europe'),
    ('SGDP', 'Igorot'): (1, 16, 'Igorot', 'Oceania'),
    ('SGDP', 'Iranian'): (1, 17, 'Iranian', 'Middle East'),
    ('SGDP', 'Iraqi_Jew'): (1, 18, 'Iraqi Jew', 'Middle East'),
    ('SGDP', 'Irula'): (1, 19, 'Irula', 'South Asia'),
    ('SGDP', 'Itelman'): (1, 20, 'Itelman', 'Central Asia'),
    ('SGDP', 'Japanese'): (1, 21, 'Japanese', 'East Asia'),
    ('SGDP', 'Jordanian'): (1, 22, 'Jordanian', 'Middle East'),
    ('SGDP', 'Ju_hoan_North'): (1, 23, 'Ju hoan North', 'Africa'),
    ('SGDP', 'Kalash'): (1, 24, 'Kalash', 'South Asia'),
    ('SGDP', 'Kapu'): (1, 25, 'Kapu', 'South Asia'),
    ('SGDP', 'Karitiana'): (1, 26, 'Karitiana', 'America'),
    ('SGDP', 'Khomani_San'): (1, 27, 'Khomani San', 'Africa'),
    ('SGDP', 'Khonda_Dora'): (1, 28, 'Khonda Dora', 'South Asia'),
    ('SGDP', 'Kinh'): (1, 29, 'Kinh', 'East Asia'),
    ('SGDP', 'Korean'): (1, 30, 'Korean', 'East Asia'),
    ('SGDP', 'Kusunda'): (1, 31, 'Kusunda', 'South Asia'),
    ('SGDP', 'Kyrgyz'): (1, 32, 'Kyrgyz', 'Central Asia'),
    ('SGDP', 'Lahu'): (1, 33, 'Lahu', 'East Asia'),
    ('SGDP', 'Lezgin'): (1, 34, 'Lezgin', 'Europe'),
    ('SGDP', 'Luhya'): (1, 35, 'Luhya', 'Africa'),
    ('SGDP', 'Luo'): (1, 36, 'Luo', 'Africa'),
    ('SGDP', 'Madiga'): (2, 0, 'Madiga', 'South Asia'),
    ('SGDP', 'Makrani'): (2, 1, 'Makrani', 'South Asia'),
    ('SGDP', 'Mala'): (2, 2, 'Mala', 'South Asia'),
    ('SGDP', 'Mandenka'): (2, 3, 'Mandenka', 'Africa'),
    ('SGDP', 'Mansi'): (2, 4, 'Mansi', 'Central Asia'),
    ('SGDP', 'Maori'): (2, 5, 'Maori', 'Oceania'),
    ('SGDP', 'Masai'): (2, 6, 'Masai', 'Africa'),
    ('SGDP', 'Mayan'): (2, 7, 'Mayan', 'America'),
    ('SGDP', 'Mbuti'): (2, 8, 'Mbuti', 'Africa'),
    ('SGDP', 'Mende'): (2, 9, 'Mende', 'Africa'),
    ('SGDP', 'Miao'): (2, 10, 'Miao', 'East Asia'),
    ('SGDP', 'Mixe'): (2, 11, 'Mixe', 'America'),
    ('SGDP', 'Mixtec'): (2, 12, 'Mixtec', 'America'),
    ('SGDP', 'Mongola'): (2, 13, 'Mongola', 'Central Asia'),
    ('SGDP', 'Mozabite'): (2, 14, 'Mozabite', 'Middle East'),
    ('SGDP', 'Naxi'): (2, 15, 'Naxi', 'East Asia'),
    ('SGDP', 'North_Ossetian'): (2, 16, 'North Ossetian', 'Europe'),
    ('SGDP', 'Norwegian'): (2, 17, 'Norwegian', 'Europe'),
    ('SGDP', 'Orcadian'): (2, 18, 'Orcadian', 'Europe'),
    ('SGDP', 'Oroqen'): (2, 19, 'Oroqen', 'East Asia'),
    ('SGDP', 'Palestinian'): (2, 20, 'Palestinian', 'Middle East'),
    ('SGDP', 'Papuan'): (2, 21, 'Papuan', 'Oceania'),
    ('SGDP', 'Pathan'): (2, 22, 'Pathan', 'South Asia'),
    ('SGDP', 'Piapoco'): (2, 23, 'Piapoco', 'America'),
    ('SGDP', 'Pima'): (2, 24, 'Pima', 'America'),
    ('SGDP', 'Polish'): (2, 25, 'Polish', 'Europe'),
    ('SGDP', 'Punjabi'): (2, 26, 'Punjabi', 'South Asia'),
    ('SGDP', 'Quechua'): (2, 27, 'Quechua', 'America'),
    ('SGDP', 'Relli'): (2, 28, 'Relli', 'South Asia'),
    ('SGDP', 'Russian'): (2, 29, 'Russian', 'Europe'),
    ('SGDP', 'Saami'): (2, 30, 'Saami', 'Europe'),
    ('SGDP', 'Saharawi'): (2, 31, 'Saharawi', 'Middle East'),
    ('SGDP', 'Samaritan'): (2, 32, 'Samaritan', 'Middle East'),
    ('SGDP', 'Sardinian'): (2, 33, 'Sardinian', 'Europe'),
    ('SGDP', 'She'): (2, 34, 'She', 'East Asia'),
    ('SGDP', 'Sindhi'): (2, 35, 'Sindhi', 'South Asia'),
    ('SGDP', 'Somali'): (2, 36, 'Somali', 'Africa'),
    ('SGDP', 'Spanish'): (3, 0, 'Spanish', 'Europe'),
    ('SGDP', 'Surui'): (3, 1, 'Surui', 'America'),
    ('SGDP', 'Tajik'): (3, 2, 'Tajik', 'Central Asia'),
    ('SGDP', 'Thai'): (3, 3, 'Thai', 'East Asia'),
    ('SGDP', 'Tlingit'): (3, 4, 'Tlingit', 'Central Asia'),
    ('SGDP', 'Tu'): (3, 5, 'Tu', 'East Asia'),
    ('SGDP', 'Tubalar'): (3, 6, 'Tubalar', 'Central Asia'),
    ('SGDP', 'Tujia'): (3, 7, 'Tujia', 'East Asia'),
    ('SGDP', 'Turkish'): (3, 8, 'Turkish', 'Middle East'),
    ('SGDP', 'Tuscan'): (3, 9, 'Tuscan', 'Europe'),
    ('SGDP', 'Ulchi'): (3, 10, 'Ulchi', 'Central Asia'),
    ('SGDP', 'Uygur'): (3, 11, 'Uygur', 'Central Asia'),
    ('SGDP', 'Xibo'): (3, 12, 'Xibo', 'East Asia'),
    ('SGDP', 'Yadava'): (3, 13, 'Yadava', 'South Asia'),
    ('SGDP', 'Yakut'): (3, 14, 'Yakut', 'Central Asia'),
    ('SGDP', 'Yemenite_Jew'): (3, 15, 'Yemenite Jew', 'Middle East'),
    ('SGDP', 'Yi'): (3, 16, 'Yi', 'East Asia'),
    ('SGDP', 'Yoruba'): (3, 17, 'Yoruba', 'Africa'),
    ('SGDP', 'Zapotec'): (3, 18, 'Zapotec', 'America'),
    ('HGDP', 'Adygei'): (0, 0, 'Adygei', 'Europe'),
    ('HGDP', 'Balochi'): (0, 1, 'Balochi', 'South Asia'),
    ('HGDP', 'BantuKenya'): (0, 2, 'Bantu Kenya', 'Africa'),
    ('HGDP', 'BantuSouthAfrica'): (0, 3, 'Bantu South Africa', 'Africa'),
    ('HGDP', 'Basque'): (0, 4, 'Basque', 'Europe'),
    ('HGDP', 'Bedouin'): (0, 5, 'Bedouin', 'Middle East'),
    ('HGDP', 'BergamoItalian'): (0, 6, 'Bergamo Italian', 'Europe'),
    ('HGDP', 'Biaka'): (0, 7, 'Biaka', 'Africa'),
    ('HGDP', 'Bougainville'): (0, 8, 'Bougainville', 'Oceania'),
    ('HGDP', 'Brahui'): (0, 9, 'Brahui', 'South Asia'),
    ('HGDP', 'Burusho'): (0, 10, 'Burusho', 'South Asia'),
    ('HGDP', 'Cambodian'): (1, 0, 'Cambodian', 'East Asia'),
    ('HGDP', 'Colombian'): (1, 1, 'Colombian', 'America'),
    ('HGDP', 'Dai'): (1, 2, 'Dai', 'East Asia'),
    ('HGDP', 'Daur'): (1, 3, 'Daur', 'East Asia'),
    ('HGDP', 'Druze'): (1, 4, 'Druze', 'Middle East'),
    ('HGDP', 'French'): (1, 5, 'French', 'Europe'),
    ('HGDP', 'Han'): (1, 6, 'Han', 'East Asia'),
    ('HGDP', 'Hazara'): (1, 7, 'Hazara', 'South Asia'),
    ('HGDP', 'Hezhen'): (1, 8, 'Hezhen', 'East Asia'),
    ('HGDP', 'Japanese'): (1, 9, 'Japanese', 'East Asia'),
    ('HGDP', 'Kalash'): (1, 10, 'Kalash', 'South Asia'),
    ('HGDP', 'Karitiana'): (2, 0, 'Karitiana', 'America'),
    ('HGDP', 'Lahu'): (2, 1, 'Lahu', 'East Asia'),
    ('HGDP', 'Makrani'): (2, 2, 'Makrani', 'South Asia'),
    ('HGDP', 'Mandenka'): (2, 3, 'Mandenka', 'Africa'),
    ('HGDP', 'Maya'): (2, 4, 'Maya', 'America'),
    ('HGDP', 'Mbuti'): (2, 5, 'Mbuti', 'Africa'),
    ('HGDP', 'Miao'): (2, 6, 'Miao', 'East Asia'),
    ('HGDP', 'Mongolian'): (2, 7, 'Mongolian', 'Central Asia'),
    ('HGDP', 'Mozabite'): (2, 8, 'Mozabite', 'Middle East'),
    ('HGDP', 'Naxi'): (2, 9, 'Naxi', 'East Asia'),
    ('HGDP', 'NorthernHan'): (2, 10, 'Northern Han', 'East Asia'),
    ('HGDP', 'Orcadian'): (3, 0, 'Orcadian', 'Europe'),
    ('HGDP', 'Oroqen'): (3, 1, 'Oroqen', 'East Asia'),
    ('HGDP', 'Palestinian'): (3, 2, 'Palestinian', 'Middle East'),
    ('HGDP', 'PapuanHighlands'): (3, 3, 'Papuan Highlands', 'Oceania'),
    ('HGDP', 'PapuanSepik'): (3, 4, 'Papuan Sepik', 'Oceania'),
    ('HGDP', 'Pathan'): (3, 5, 'Pathan', 'South Asia'),
    ('HGDP', 'Pima'): (3, 6, 'Pima', 'America'),
    ('HGDP', 'Russian'): (3, 7, 'Russian', 'Europe'),
    ('HGDP', 'San'): (3, 8, 'San', 'Africa'),
    ('HGDP', 'Sardinian'): (3, 9, 'Sardinian', 'Europe'),
    ('HGDP', 'She'): (3, 10, 'She', 'East Asia'),
    ('HGDP', 'Sindhi'): (4, 0, 'Sindhi', 'South Asia'),
    ('HGDP', 'Surui'): (4, 1, 'Surui', 'America'),
    ('HGDP', 'Tu'): (4, 2, 'Tu', 'East Asia'),
    ('HGDP', 'Tujia'): (4, 3, 'Tujia', 'East Asia'),
    ('HGDP', 'Tuscan'): (4, 4, 'Tuscan', 'Europe'),
    ('HGDP', 'Uygur'): (4, 5, 'Uygur', 'Central Asia'),
    ('HGDP', 'Xibo'): (4, 6, 'Xibo', 'East Asia'),
    ('HGDP', 'Yakut'): (4, 7, 'Yakut', 'Central Asia'),
    ('HGDP', 'Yi'): (4, 8, 'Yi', 'East Asia'),
    ('HGDP', 'Yoruba'): (4, 9, 'Yoruba', 'Africa'),
    ('1KG', 'ASW'): (0, 0, 'African American', 'Africa'),
    ('1KG', 'ACB'): (0, 1, 'African Caribbean', 'Africa'),
    ('1KG', 'BEB'): (0, 2, 'Bengali', 'South Asia'),
    ('1KG', 'GBR'): (0, 3, 'British', 'Europe'),
    ('1KG', 'CLM'): (0, 4, 'Colombian', 'America'),
    ('1KG', 'CDX'): (0, 5, 'Dai Chinese', 'East Asia'),
    ('1KG', 'ESN'): (0, 6, 'Esan', 'Africa'),
    ('1KG', 'FIN'): (0, 7, 'Finnish', 'Europe'),
    ('1KG', 'GWD'): (0, 8, 'Gambian Mandinka', 'Africa'),
    ('1KG', 'GIH'): (1, 0, 'Gujarati', 'South Asia'),
    ('1KG', 'CHB'): (1, 1, 'Han Chinese', 'East Asia'),
    ('1KG', 'IBS'): (1, 2, 'Iberian', 'Europe'),
    ('1KG', 'JPT'): (1, 3, 'Japanese', 'East Asia'),
    ('1KG', 'KHV'): (1, 4, 'Kinh Vietnamese', 'East Asia'),
    ('1KG', 'LWK'): (1, 5, 'Luhya', 'Africa'),
    ('1KG', 'MSL'): (1, 6, 'Mende', 'Africa'),
    ('1KG', 'MXL'): (1, 7, 'Mexican Ancestry', 'America'),
    ('1KG', 'PEL'): (1, 8, 'Peruvian', 'America'),
    ('1KG', 'PUR'): (2, 0, 'Puerto Rican', 'America'),
    ('1KG', 'PJL'): (2, 1, 'Punjabi', 'South Asia'),
    ('1KG', 'CHS'): (2, 2, 'Southern Han Chinese', 'East Asia'),
    ('1KG', 'STU'): (2, 3, 'Tamil', 'South Asia'),
    ('1KG', 'ITU'): (2, 4, 'Telugu', 'South Asia'),
    ('1KG', 'TSI'): (2, 5, 'Toscani', 'Europe'),
    ('1KG', 'CEU'): (2, 6, 'Utah European', 'Europe'),
    ('1KG', 'YRI'): (2, 7, 'Yoruba', 'Africa'),
}

layers.legend_frequency.center = (w(1910), h(930))
layers.legend_frequency.font_size = font_small
layers.legend_frequency.frequencies = [1.0, 0.5, 0.1]
layers.legend_frequency.title = '%s-%s samples'
layers.legend_frequency.orientation = 'horizontal'
layers.legend_frequency.legends = [
    {'title': (0, 9), 'shrink': True},
    {'title': (10, 200), 'shrink': False},
]
layers.legend_frequency.width = font_small * len(layers.legend_frequency.frequencies) * len(layers.legend_frequency.legends) * 6
layers.legend_frequency.height = font_small * 4
layers.legend_frequency.style = 'font-family:sans-serif;'
layers.legend_frequency.line_spacing = 1.2

layers.legend_position.center = (w(2310), h(260))
layers.legend_position.font_size = font_small
layers.legend_position.width = font_small * 24
layers.legend_position.height = font_small * 4
layers.legend_position.style = 'font-family:sans-serif;'

layers.caption.center = (w(1280), h(1200))
layers.caption.font_size = font_large
layers.caption.width = width
layers.caption.height = font_large * 2
layers.caption.style = 'font-family:sans-serif;'
layers.caption.line_spacing = 1.125
layers.caption.srt = code/'captions'/f'captions.{part}.srt'
layers.caption.typing = None

layers.citation.center = (w(1280), h(1200))
layers.citation.font_size = font_tiny
layers.citation.width = width
layers.citation.height = font_tiny * 6
layers.citation.style = 'font-family:sans-serif;'
layers.citation.line_spacing = 1
layers.citation.srt = code/'citations'/f'citations.{part}.srt'
layers.citation.typing = None

layers.date.center = (w(1280), h(1374))
layers.date.font_size = font_medium
layers.date.width = width
layers.date.height = font_medium * 3
layers.date.style = 'font-family:sans-serif;'
layers.date.line_spacing = 1

#movie_time = 0 # Select a specific time.  0 = all times.
#movie_limit = 120 # Limit the number of frames created.  0 = no limit.

#movie_times = ((551, 480), (550, 480), (549, 480), (300, 480), (100, 480), (50, 480), (40, 480), (30, 480), (20, 480), (10, 480), (3, 480), (1, 480))
#movie_time_name = '-'.join(f'{time}_{limit}' for time, limit in movie_times)
#movie_laps = [(time, 1) for time in range(551, 1, -13)]
#movie_laps.append((1, 4))
#movie_time_name = 'lapsAAB'

#movie_times = {
#    'name': 'cite1',
#    'type': 'time_limit',
#    'time_limits': [
#        #(80_240, 480),
#        #(80_140, 480),
#        #(80_110, 480),
#        #(80_080, 480),
#        #(60_000, 480),
#        #(40_000, 480),
#        #(20_000, 480),
#        #(10_000, 480),
#        #(8_000, 480),
#        #(6_000, 480),
#        #(4_000, 480),
#        #(2_000, 480),
#        #(1_000, 480),
#        #(800, 480),
#        #(600, 480),
#        #(400, 480),
#        #(200, 480),
#        #(100, 480),
#        #(80, 480),
#        #(60, 480),
#        #(40, 480),
#        #(20, 480),
#        #(10, 480),
#        (1, 960),
#    ],
#}

#all_movie_times = {
#    1: {
#        'name': 'part1',
#        'type': 'time_range',
#        'time_ranges': [
#            # These are in generations.  Inclusive on smaller number only.
#            #(999_999_999, 3000),
#            (999_999_999, 70000),
#        ],
#    },
#    2: {
#        'name': 'part2',
#        'type': 'time_range',
#        'time_ranges': [
#            # These are in generations.  Inclusive on smaller number only.
#            #(3000, 1000),
#            (3000, 2950),
#        ],
#    },
#    3: {
#        'name': 'part3',
#        'type': 'time_range',
#        'time_ranges': [
#            # These are in generations.  Inclusive on smaller number only.
#            #(1000, 400),
#            (1000, 995),
#        ],
#    },
#    4: {
#        'name': 'part4',
#        'type': 'time_range',
#        'time_ranges': [
#            # These are in generations.  Inclusive on smaller number only.
#            #(400, 40),
#            (400, 398),
#        ],
#    },
#    5: {
#        'name': 'part5',
#        'type': 'time_range',
#        'time_ranges': [
#            #(40, 0),
#            (40, 38),
#        ],
#    },
#}

all_movie_times = {
    # Inclusive on both numbers.
    '1': {
        'name': 'part1',
        'type': 'order_range',
        'order_ranges': [
            (0, 161_279),
            #(0, 2160),
        ],
    },
    '2': {
        'name': 'part2',
        'type': 'order_range',
        'order_ranges': [
            (161_280, 315_359),
            #(276_000, 276_960),
        ],
    },
    '3': {
        'name': 'part3',
        'type': 'order_range',
        'order_ranges': [
            (315_360, 455_039),
            #(555_360, 556_320),
        ],
    },
    '4': {
        'name': 'part4',
        'type': 'order_range',
        'order_ranges': [
            (455_040, 579_839),
            #(876_960, 877_920),
        ],
    },
    '5': {
        'name': 'part5',
        'type': 'order_range',
        'order_ranges': [
            (579_840, 999_999_999),
            #(1_162_560, 1_163_520),
        ],
    },
    't': {
        'name': 'teaser',
        'type': 'order_range',
        'order_ranges': [
            # 2 million to 100,000.
            (0, 479),
            (160_800, 161_279),

            # 100,000 to 25,000.
            (161_280, 161_759),
            (314_880, 315_359),

            # 25,000 to 10,000.
            (315_360, 315_839),
            (454_560, 455_039),

            # 10,000 to 1,000.
            (455_040, 455_519),
            (579_360, 579_839),

            # 1,000 on.
            (579_840, 580_319),
            (703_099, 703_578),
        ],
    },
    'x': {
        'name': 'test',
        'type': 'order_range',
        'order_ranges': [
            (0, 480),
        ],
    },
}

movie_times = all_movie_times[part]

#movie_times = {
#    'name': 'full',
#    'type': None,
#}

layers.legend_population_histogram.center = (w(1280), h(1155))
layers.legend_population_histogram.font_size = font_small
layers.legend_population_histogram.width = w(840)
layers.legend_population_histogram.height = h(210)
layers.legend_population_histogram.style = 'font-family:sans-serif;fill:#ff00fd;stroke:#ff00fd;'
if part == 'x':
    population_histogram_buffer = 0
elif part == 't':
    population_histogram_buffer = 256
elif part == '1':
    population_histogram_buffer = 1680
else:
    population_histogram_buffer = 272
layers.legend_population_histogram.start_order = movie_times['order_ranges'][-1][0] + population_histogram_buffer
#layers.legend_population_histogram.start_order = 0


#audio_midi = audio/f'{order}_{movie_time}_{movie_limit}.midi'
#audio_wav = audio/f'{order}_{movie_time}_{movie_limit}.wav'
audio_midi = audio/f'{order}_{movie_times["name"]}.midi'
audio_wav = audio/f'{order}_{movie_times["name"]}.wav'
audio_midi_program = 46
audio_pipe = True

audio_notes = {
    ('A', 'C'): {'note': 'G5', 'index': 0},
    ('T', 'G'): {'note': 'F5', 'index': 1},
    ('G', 'C'): {'note': 'E5', 'index': 2},
    ('A', 'G'): {'note': 'D5', 'index': 3},
    ('T', 'C'): {'note': 'C5', 'index': 4},
    ('A', 'T'): {'note': 'B4', 'index': 5},
    ('C', 'T'): {'note': 'A4', 'index': 6},
    ('G', 'A'): {'note': 'G4', 'index': 7},
    ('C', 'G'): {'note': 'F4', 'index': 8},
    ('G', 'T'): {'note': 'E4', 'index': 9},
    ('C', 'A'): {'note': 'D4', 'index': 10},
    ('T', 'A'): {'note': 'C4', 'index': 11},
}

#movie_mp4 = movie/f'{map_image}_{map_projection}_{map_rotation[0]}_{map_rotation[1]}_{order}_{"shadows" if shadows else "noshadows"}_{movie_time}_{movie_limit}.mp4'
movie_mp4 = movie/f'{map_image}_{map_projection}_{map_rotation[0]}_{map_rotation[1]}_{order}_{"shadows" if shadows else "noshadows"}_{movie_times["name"]}.mp4'