| 1 | import os |
|---|
| 2 | import glob |
|---|
| 3 | |
|---|
| 4 | def generate(env): |
|---|
| 5 | def SGlob(pattern): |
|---|
| 6 | path = env.GetBuildPath('SConscript').replace('SConscript', '') |
|---|
| 7 | return [ i.replace(path, '') for i in glob.glob(path + pattern) ] |
|---|
| 8 | env.SGlob = SGlob |
|---|
| 9 | |
|---|
| 10 | def AddCustomPath(path=""): |
|---|
| 11 | if not len(path) or not os.path.exists(path): |
|---|
| 12 | return |
|---|
| 13 | env.PrependUnique(CPPPATH = [os.path.join(path, "include")]) |
|---|
| 14 | env.PrependUnique(LIBPATH = [os.path.join(path, "lib")]) |
|---|
| 15 | env.AddCustomPath = AddCustomPath |
|---|
| 16 | |
|---|
| 17 | def AddCustomPackage(pkgname=None): |
|---|
| 18 | if pkgname is None: |
|---|
| 19 | return |
|---|
| 20 | pkgroot = env.get("%sroot" % pkgname, None) |
|---|
| 21 | pkgincd = env.get("%sincdir" % pkgname, None) |
|---|
| 22 | pkglibd = env.get("%slibdir" % pkgname, None) |
|---|
| 23 | incd = None |
|---|
| 24 | libd = None |
|---|
| 25 | if pkgroot is not None: |
|---|
| 26 | incd = os.path.join(pkgroot, "include") |
|---|
| 27 | libd = os.path.join(pkgroot, "lib") |
|---|
| 28 | else: |
|---|
| 29 | if pkgincd is not None: |
|---|
| 30 | incd = pkgincd |
|---|
| 31 | if pkglibd is not None: |
|---|
| 32 | libd = pkglibd |
|---|
| 33 | if incd is not None: |
|---|
| 34 | if not os.path.exists(incd): |
|---|
| 35 | print "Custom %s include dir '%s' not found" % (pkgname, incd) |
|---|
| 36 | env.Exit(1) |
|---|
| 37 | env.PrependUnique(CPPPATH = [incd]) |
|---|
| 38 | if libd is not None: |
|---|
| 39 | if not os.path.exists(libd): |
|---|
| 40 | print "Custom %s lib dir '%s' not found" % (pkgname, libd) |
|---|
| 41 | env.Exit(1) |
|---|
| 42 | env.PrependUnique(LIBPATH = [libd]) |
|---|
| 43 | |
|---|
| 44 | env.AddCustomPackage = AddCustomPackage |
|---|
| 45 | |
|---|
| 46 | def CheckFortran(conf): |
|---|
| 47 | |
|---|
| 48 | if not conf.env.has_key("FORTRAN"): |
|---|
| 49 | # auto-detect fortran |
|---|
| 50 | detect_fortran = conf.env.Detect(['gfortran', 'g77', 'f77']) |
|---|
| 51 | if not detect_fortran: |
|---|
| 52 | print "No fortran compiler found. Specify FORTRAN and f2clib." |
|---|
| 53 | conf.env.Exit(1) |
|---|
| 54 | conf.env["FORTRAN"] = detect_fortran |
|---|
| 55 | fdict = {'gfortran': 'gfortran', 'g77': 'g2c', 'f77': 'f2c'} |
|---|
| 56 | f2clib = conf.env.get("f2clib", fdict[detect_fortran]) |
|---|
| 57 | if not conf.CheckLib(f2clib): |
|---|
| 58 | env.Exit(1) |
|---|
| 59 | else: |
|---|
| 60 | if not conf.env.has_key("f2clib"): |
|---|
| 61 | print "A custom fortran compiler also needs f2clib defined" |
|---|
| 62 | env.Exit(1) |
|---|
| 63 | else: |
|---|
| 64 | if not conf.CheckLib(env["f2clib"]): |
|---|
| 65 | env.Exit(1) |
|---|
| 66 | if conf.env["FORTRAN"].startswith("g77"): |
|---|
| 67 | fflags = ["-Wno-globals", "-fno-second-underscore"] |
|---|
| 68 | conf.env.Append(SHFORTRANFLAGS=fflags) |
|---|
| 69 | conf.env.Append(FORTRANFLAGS=fflags) |
|---|
| 70 | env.CheckFortran = CheckFortran |
|---|
| 71 | |
|---|
| 72 | def WalkDirTree(targetroot, sourceroot, sources): |
|---|
| 73 | ifiles = [] |
|---|
| 74 | ofiles = [] |
|---|
| 75 | for s in sources: |
|---|
| 76 | if os.path.isdir(os.path.join(sourceroot ,s)): |
|---|
| 77 | for d,ld,f in os.walk(os.path.join(sourceroot ,s)): |
|---|
| 78 | for fl in f: |
|---|
| 79 | ifile = os.path.join(d, fl) |
|---|
| 80 | ifiles.append(ifile) |
|---|
| 81 | ofile = ifile.replace(sourceroot, targetroot) |
|---|
| 82 | ofiles.append(ofile) |
|---|
| 83 | return ofiles, ifiles |
|---|
| 84 | env.WalkDirTree = WalkDirTree |
|---|
| 85 | |
|---|
| 86 | def null_action(target, source, env): return 0 |
|---|
| 87 | |
|---|
| 88 | def message(target, source, env): |
|---|
| 89 | return "%s" % target[0] |
|---|
| 90 | env.MessageAction = env.Action(null_action, message) |
|---|
| 91 | |
|---|
| 92 | def exists(env): |
|---|
| 93 | try: |
|---|
| 94 | import os |
|---|
| 95 | import glob |
|---|
| 96 | except ImportError: |
|---|
| 97 | return False |
|---|
| 98 | else: |
|---|
| 99 | return True |
|---|
