# -*- Mode: python; indent-tabs-mode: nil; tab-width: 40 -*-
# vim: set filetype=python:
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.


@deprecated_option(env='MOZ_JEMALLOC4')
def moz_jemalloc4(value):
    die('MOZ_JEMALLOC4 is deprecated')


option('--enable-jemalloc', env='MOZ_MEMORY',
       help='Replace memory allocator with jemalloc')


@depends('--enable-jemalloc', target, build_project, c_compiler)
def jemalloc(value, target, build_project, c_compiler):
    if value.origin != 'default':
        return bool(value) or None

    if build_project == 'js':
        return True

    if target.kernel == 'Darwin' and target.cpu == 'x86_64':
        # Don't enable by default on 32-bits OSX. See bug 702250.
        return True

    if target.kernel == 'WINNT' and c_compiler.type in ('msvc', 'clang-cl'):
        return True

    if target.kernel == 'Linux':
        return True

    if value and target.kernel not in ('WINNT', 'Linux', 'Darwin', 'kFreeBSD',
                                       'FreeBSD', 'NetBSD'):
        die('--enable-jemalloc is not supported on %s', target.kernel)


set_config('MOZ_MEMORY', jemalloc)
set_define('MOZ_MEMORY', jemalloc)
add_old_configure_assignment('MOZ_MEMORY', jemalloc)


# Because --enable-jemalloc doesn't use a default because of the dependency
# on the target, we can't use a js_option for it to propagate to js/src
# through the old-configure.
@depends(jemalloc)
def jemalloc_for_old_configure(jemalloc):
    return '--%s-jemalloc' % ('enable' if jemalloc else 'disable')


add_old_configure_arg(jemalloc_for_old_configure)


option('--enable-replace-malloc',
       help='Enable ability to dynamically replace the malloc implementation')


@depends('--enable-replace-malloc', jemalloc, milestone, build_project)
def replace_malloc(value, jemalloc, milestone, build_project):
    # Enable on central for the debugging opportunities it adds.
    if value and not jemalloc:
        die('--enable-replace-malloc requires --enable-jemalloc')
    if value.origin != 'default':
        return bool(value) or None
    if milestone.is_nightly and jemalloc and build_project != 'js':
        return True


set_config('MOZ_REPLACE_MALLOC', replace_malloc)
set_define('MOZ_REPLACE_MALLOC', replace_malloc)
add_old_configure_assignment('MOZ_REPLACE_MALLOC', replace_malloc)