1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225
|
#! /usr/bin/env python
# encoding: utf-8
import os,re
import TaskGen,Utils,Task,Build
from Logs import error
from TaskGen import taskgen,feature,before,after,extension
EXT_MLL=['.mll']
EXT_MLY=['.mly']
EXT_MLI=['.mli']
EXT_MLC=['.c']
EXT_ML=['.ml']
open_re=re.compile('^\s*open\s+([a-zA-Z]+)(;;){0,1}$',re.M)
foo=re.compile(r"""(\(\*)|(\*\))|("(\\.|[^"\\])*"|'(\\.|[^'\\])*'|.[^()*"'\\]*)""",re.M)
def filter_comments(txt):
meh=[0]
def repl(m):
if m.group(1):meh[0]+=1
elif m.group(2):meh[0]-=1
elif not meh[0]:return m.group(0)
return''
return foo.sub(repl,txt)
def scan(self):
node=self.inputs[0]
code=filter_comments(node.read(self.env))
global open_re
names=[]
import_iterator=open_re.finditer(code)
if import_iterator:
for import_match in import_iterator:
names.append(import_match.group(1))
found_lst=[]
raw_lst=[]
for name in names:
nd=None
for x in self.incpaths:
nd=x.find_resource(name.lower()+'.ml')
if not nd:nd=x.find_resource(name+'.ml')
if nd:
found_lst.append(nd)
break
else:
raw_lst.append(name)
return(found_lst,raw_lst)
native_lst=['native','all','c_object']
bytecode_lst=['bytecode','all']
class ocaml_taskgen(TaskGen.task_gen):
def __init__(self,*k,**kw):
TaskGen.task_gen.__init__(self,*k,**kw)
def init_ml(self):
Utils.def_attrs(self,type='all',incpaths_lst=[],bld_incpaths_lst=[],mlltasks=[],mlytasks=[],mlitasks=[],native_tasks=[],bytecode_tasks=[],linktasks=[],bytecode_env=None,native_env=None,compiled_tasks=[],includes='',uselib='',are_deps_set=0)
def init_envs_ml(self):
self.islibrary=getattr(self,'islibrary',False)
global native_lst,bytecode_lst
self.native_env=None
if self.type in native_lst:
self.native_env=self.env.copy()
if self.islibrary:self.native_env['OCALINKFLAGS']='-a'
self.bytecode_env=None
if self.type in bytecode_lst:
self.bytecode_env=self.env.copy()
if self.islibrary:self.bytecode_env['OCALINKFLAGS']='-a'
if self.type=='c_object':
self.native_env.append_unique('OCALINKFLAGS_OPT','-output-obj')
def apply_incpaths_ml(self):
inc_lst=self.includes.split()
lst=self.incpaths_lst
for dir in inc_lst:
node=self.path.find_dir(dir)
if not node:
error("node not found: "+str(dir))
continue
self.bld.rescan(node)
if not node in lst:lst.append(node)
self.bld_incpaths_lst.append(node)
def apply_vars_ml(self):
for i in self.incpaths_lst:
if self.bytecode_env:
app=self.bytecode_env.append_value
app('OCAMLPATH','-I')
app('OCAMLPATH',i.srcpath(self.env))
app('OCAMLPATH','-I')
app('OCAMLPATH',i.bldpath(self.env))
if self.native_env:
app=self.native_env.append_value
app('OCAMLPATH','-I')
app('OCAMLPATH',i.bldpath(self.env))
app('OCAMLPATH','-I')
app('OCAMLPATH',i.srcpath(self.env))
varnames=['INCLUDES','OCAMLFLAGS','OCALINKFLAGS','OCALINKFLAGS_OPT']
for name in self.uselib.split():
for vname in varnames:
cnt=self.env[vname+'_'+name]
if cnt:
if self.bytecode_env:self.bytecode_env.append_value(vname,cnt)
if self.native_env:self.native_env.append_value(vname,cnt)
def apply_link_ml(self):
if self.bytecode_env:
ext=self.islibrary and'.cma'or'.run'
linktask=self.create_task('ocalink')
linktask.bytecode=1
linktask.set_outputs(self.path.find_or_declare(self.target+ext))
linktask.obj=self
linktask.env=self.bytecode_env
self.linktasks.append(linktask)
if self.native_env:
if self.type=='c_object':ext='.o'
elif self.islibrary:ext='.cmxa'
else:ext=''
linktask=self.create_task('ocalinkx')
linktask.set_outputs(self.path.find_or_declare(self.target+ext))
linktask.obj=self
linktask.env=self.native_env
self.linktasks.append(linktask)
self.compiled_tasks.append(linktask)
def mll_hook(self,node):
mll_task=self.create_task('ocamllex',node,node.change_ext('.ml'),env=self.native_env)
self.mlltasks.append(mll_task)
self.allnodes.append(mll_task.outputs[0])
def mly_hook(self,node):
mly_task=self.create_task('ocamlyacc',node,[node.change_ext('.ml'),node.change_ext('.mli')],env=self.native_env)
self.mlytasks.append(mly_task)
self.allnodes.append(mly_task.outputs[0])
task=self.create_task('ocamlcmi',mly_task.outputs[1],mly_task.outputs[1].change_ext('.cmi'),env=self.native_env)
def mli_hook(self,node):
task=self.create_task('ocamlcmi',node,node.change_ext('.cmi'),env=self.native_env)
self.mlitasks.append(task)
def mlc_hook(self,node):
task=self.create_task('ocamlcc',node,node.change_ext('.o'),env=self.native_env)
self.compiled_tasks.append(task)
def ml_hook(self,node):
if self.native_env:
task=self.create_task('ocamlx',node,node.change_ext('.cmx'),env=self.native_env)
task.obj=self
task.incpaths=self.bld_incpaths_lst
self.native_tasks.append(task)
if self.bytecode_env:
task=self.create_task('ocaml',node,node.change_ext('.cmo'),env=self.bytecode_env)
task.obj=self
task.bytecode=1
task.incpaths=self.bld_incpaths_lst
self.bytecode_tasks.append(task)
def compile_may_start(self):
if not getattr(self,'flag_deps',''):
self.flag_deps=1
if getattr(self,'bytecode',''):alltasks=self.obj.bytecode_tasks
else:alltasks=self.obj.native_tasks
self.signature()
tree=self.generator.bld
env=self.env
for node in self.inputs:
lst=tree.node_deps[self.unique_id()]
for depnode in lst:
for t in alltasks:
if t==self:continue
if depnode in t.inputs:
self.set_run_after(t)
delattr(self,'cache_sig')
self.signature()
return Task.Task.runnable_status(self)
b=Task.simple_task_type
cls=b('ocamlx','${OCAMLOPT} ${OCAMLPATH} ${OCAMLFLAGS} ${INCLUDES} -c -o ${TGT} ${SRC}',color='GREEN',shell=False)
cls.runnable_status=compile_may_start
cls.scan=scan
b=Task.simple_task_type
cls=b('ocaml','${OCAMLC} ${OCAMLPATH} ${OCAMLFLAGS} ${INCLUDES} -c -o ${TGT} ${SRC}',color='GREEN',shell=False)
cls.runnable_status=compile_may_start
cls.scan=scan
b('ocamlcmi','${OCAMLC} ${OCAMLPATH} ${INCLUDES} -o ${TGT} -c ${SRC}',color='BLUE',before="ocaml ocamlcc ocamlx")
b('ocamlcc','cd ${TGT[0].bld_dir(env)} && ${OCAMLOPT} ${OCAMLFLAGS} ${OCAMLPATH} ${INCLUDES} -c ${SRC[0].abspath(env)}',color='GREEN')
b('ocamllex','${OCAMLLEX} ${SRC} -o ${TGT}',color='BLUE',before="ocamlcmi ocaml ocamlcc")
b('ocamlyacc','${OCAMLYACC} -b ${TGT[0].bld_base(env)} ${SRC}',color='BLUE',before="ocamlcmi ocaml ocamlcc")
def link_may_start(self):
if not getattr(self,'order',''):
if getattr(self,'bytecode',0):alltasks=self.obj.bytecode_tasks
else:alltasks=self.obj.native_tasks
seen=[]
pendant=[]+alltasks
while pendant:
task=pendant.pop(0)
if task in seen:continue
for x in task.run_after:
if not x in seen:
pendant.append(task)
break
else:
seen.append(task)
self.inputs=[x.outputs[0]for x in seen]
self.order=1
return Task.Task.runnable_status(self)
act=b('ocalink','${OCAMLC} -o ${TGT} ${INCLUDES} ${OCALINKFLAGS} ${SRC}',color='YELLOW',after="ocaml ocamlcc")
act.runnable_status=link_may_start
act=b('ocalinkx','${OCAMLOPT} -o ${TGT} ${INCLUDES} ${OCALINKFLAGS_OPT} ${SRC}',color='YELLOW',after="ocamlx ocamlcc")
act.runnable_status=link_may_start
def detect(conf):
opt=conf.find_program('ocamlopt',var='OCAMLOPT')
occ=conf.find_program('ocamlc',var='OCAMLC')
if(not opt)or(not occ):
conf.fatal('The objective caml compiler was not found:\ninstall it or make it available in your PATH')
v=conf.env
v['OCAMLC']=occ
v['OCAMLOPT']=opt
v['OCAMLLEX']=conf.find_program('ocamllex',var='OCAMLLEX')
v['OCAMLYACC']=conf.find_program('ocamlyacc',var='OCAMLYACC')
v['OCAMLFLAGS']=''
v['OCAMLLIB']=Utils.cmd_output(conf.env['OCAMLC']+' -where').strip()+os.sep
v['LIBPATH_OCAML']=Utils.cmd_output(conf.env['OCAMLC']+' -where').strip()+os.sep
v['CPPPATH_OCAML']=Utils.cmd_output(conf.env['OCAMLC']+' -where').strip()+os.sep
v['LIB_OCAML']='camlrun'
feature('ocaml')(init_ml)
feature('ocaml')(init_envs_ml)
after('init_ml')(init_envs_ml)
feature('ocaml')(apply_incpaths_ml)
before('apply_vars_ml')(apply_incpaths_ml)
after('init_envs_ml')(apply_incpaths_ml)
feature('ocaml')(apply_vars_ml)
before('apply_core')(apply_vars_ml)
feature('ocaml')(apply_link_ml)
after('apply_core')(apply_link_ml)
extension(EXT_MLL)(mll_hook)
extension(EXT_MLY)(mly_hook)
extension(EXT_MLI)(mli_hook)
extension(EXT_MLC)(mlc_hook)
extension(EXT_ML)(ml_hook)
|