TPT_gene.py

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

import MoNeT_MGDrivE as monet
import TPT_gene_LDR as LDR
import TPT_gene_HUM as HUM
import TPT_gene_INC as INC
import TPT_aux as aux

###############################################################################
# Drive
###############################################################################
def driveSelector(DRIVE, TYPE, popSize=(100*12000), humSize=350e3):
    ###########################################################################
    if (TYPE == 'HUM'):
        (aggD, yRange, folder) = HUM.driveParameters(TYPE, popSize=humSize)
    elif (TYPE == 'INC'):
        (aggD, yRange, folder) = INC.driveParameters(TYPE, popSize=aux.INC_SIZE)
    else:
        if (DRIVE == 'LDR'):
            (aggD, yRange, folder) = LDR.driveParameters(TYPE, popSize)
    ###########################################################################
    if TYPE == 'ECO':
        colors = monet.COLEN
    elif TYPE == 'HLT':
        colors = ['#f2008955', '#0d47a155', '#d6d6d611']
    elif TYPE == 'TRS':
        colors = monet.COLTN
    elif TYPE == 'WLD':
        colors = monet.COLWN
    elif TYPE == 'HUM':
        colors = ['#274c7700', '#6f2dbdAA', '#dee2ff55']
    elif TYPE == 'INC':
        colors = ['#8CD9FF00', '#16db6544', '#dee2ff55']
    ###########################################################################
    geneDict = {
        'gDict': aggD, 'yRange': yRange, 'colors': colors, 'folder': folder
    }
    return geneDict


def maleFemaleSelector(AOI):
    if (AOI == 'HLT') or (AOI == 'INC'):
        MF = (False, True)
    elif (AOI == 'HUM'):
        MF = (False, True)
    else:
        MF = (True, True)
    return MF


def colorSelector(AOI):
    if AOI == 'ECO':
        colors = monet.COLEO
    elif AOI == 'HLT':
        colors = monet.COLHO
    elif AOI == 'TRS':
        colors = monet.COLTO
    elif AOI == 'WLD':
        colors = monet.COLWO
    return colors