find_tcmalloc.py

# -*- python -*-

# stdlib imports ---
import os
import os.path as osp

# waf imports ---
import waflib.Utils
import waflib.Logs as msg
from waflib.Configure import conf

#
_heptooldir = osp.dirname(osp.abspath(__file__))

def options(ctx):
    ctx.load('hwaf-base', tooldir=_heptooldir)
    ctx.add_option(
        '--with-tcmalloc',
        default=None,
        help="Look for TCMalloc at the given path")
    return

def configure(ctx):
    ctx.load('hwaf-base', tooldir=_heptooldir)
    return

@conf
def find_tcmalloc(ctx, **kwargs):
    
    ctx.load('hwaf-base', tooldir=_heptooldir)

    if not ctx.env.HWAF_FOUND_C_COMPILER:
        ctx.fatal('load a C compiler first')
        pass

    if not ctx.env.HWAF_FOUND_CXX_COMPILER:
        ctx.fatal('load a C++ compiler first')
        pass

    path_list = waflib.Utils.to_list(kwargs.get('path_list', []))
    if getattr(ctx.options, 'with_tcmalloc', None):
        path_list.append(
            osp.join(ctx.options.with_tcmalloc, "bin")
            )
        pass

    kwargs['mandatory']=kwargs.get('mandatory', False)
    kwargs['path_list']=path_list

    
    ctx.check_with(
        ctx.check_cfg,
        "tcmalloc",
        package="libtcmalloc",
        uselib_store="tcmalloc",
        args='--cflags --libs',
        **kwargs)

    # test libtcmalloc
    version = ctx.check_cxx(
        msg="Checking libtcmalloc version",
        okmsg="ok",
        fragment='''\
        #include "gperftools/tcmalloc.h"
        #include <iostream>

        int main(int /*argc*/, char** /*argv*/) {
          std::cout << TC_VERSION_MAJOR << "." << TC_VERSION_MINOR << std::flush;
          if (TC_VERSION_PATCH != "") {
            std::cout << "." << TC_VERSION_PATCH << std::flush;
          }
          return 0;
        }
        ''',
        use="tcmalloc",
        define_name = "HWAF_TCMALLOC_VERSION",
        define_ret = True,
        execute  = True,
        mandatory= kwargs['mandatory'],
        )
    ctx.start_msg("tcmalloc version")
    ctx.end_msg(version)

    ctx.env.TCMALLOC_VERSION = version

    ctx.env.HWAF_FOUND_TCMALLOC = 1
    return

## EOF ##