process {
    publishDir = [
        path: { 
            "${task.process.tokenize(':')[-1].toLowerCase()}" == "multiqc" ?
                "${params.output}${params.fs}${params.pipeline.toLowerCase()}-${task.process.tokenize(':')[-1].toLowerCase()}" :
                "${params.output}${params.fs}${task.process.tokenize(':')[-1].toLowerCase()}"
        },
        mode: params.publish_dir_mode,
        overwrite: params.publish_dir_overwrite,
        saveAs: { filename -> filename.equals('versions.yml') ? null : filename }
    ]

    errorStrategy = {
        ![0].contains(task.exitStatus) ? dynamic_retry(task.attempt, 10) : 'finish'
    }

    maxRetries = 5

    withLabel: 'process_femto' {
        cpus = 2
    }

    withLabel: 'process_pico' {
        cpus = 2
    }

    withLabel: 'process_nano' {
        cpus = 4
    }

    withLabel: 'process_micro' {
        cpus = 8
    }

    withLabel: 'process_only_mem_low' {
        cpus = 2
    }

    withLabel: 'process_only_mem_medium' {
        cpus = 2
    }

    withLabel: 'process_only_mem_high' {
        cpus = 2
    }

    withLabel: 'process_low' {
        cpus = 8
    }

    withLabel: 'process_medium' {
        cpus = 8
    }

    withLabel: 'process_high' {
        cpus = 8
    }

    withLabel: 'process_higher' {
        cpus = 8
    }

    withLabel: 'process_gigantic' {
        cpus = 8
    }
}

if ( (params.input || params.metadata ) && params.pipeline ) {
    try {
        includeConfig "${params.workflowsconf}${params.fs}process${params.fs}${params.pipeline}.process.config"
    } catch (Exception e) {
        System.err.println('-'.multiply(params.linewidth) + "\n" +
            "\033[0;31m${params.cfsanpipename} - ERROR\033[0m\n" +
            '-'.multiply(params.linewidth) + "\n" + "\033[0;31mCould not load " +
            "default pipeline's process configuration. Please provide a pipeline \n" +
            "name using the --pipeline option.\n\033[0m" + '-'.multiply(params.linewidth) + "\n")
        System.exit(1)
    }
}

// Function will return after sleeping for some time. 
// Sleep time increases exponentially by task attempt.
def dynamic_retry(task_retry_num, factor_by) {
    // sleep(Math.pow(2, task_retry_num.toInteger()) * factor_by.toInteger() as long)
    sleep(Math.pow(1.27, task_retry_num.toInteger()) as long)
    return 'retry'
}