1 # This file is part of NIT ( http://www.nitlanguage.org ).
3 # Copyright 2012 Alexis Laferrière <alexis.laf@xymus.net>
5 # Licensed under the Apache License, Version 2.0 (the "License");
6 # you may not use this file except in compliance with the License.
7 # You may obtain a copy of the License at
9 # http://www.apache.org/licenses/LICENSE-2.0
11 # Unless required by applicable law or agreed to in writing, software
12 # distributed under the License is distributed on an "AS IS" BASIS,
13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 # See the License for the specific language governing permissions and
15 # limitations under the License.
17 # Tools and utilities for language targetted FFI implementations
25 redef class ToolContext
26 var ffi_language_assignation_phase
: Phase = new FFILanguageAssignationPhase(self, null)
29 class FFILanguageAssignationPhase
32 # All supported languages
33 var languages
= new Array[FFILanguage]
35 redef fun process_nmodule
(nmodule
)
37 for block
in nmodule
.n_extern_code_blocks
do
38 verify_foreign_code_on_node
( block
)
42 redef fun process_npropdef
(npropdef
)
44 if npropdef
isa AExternPropdef then
45 var code_block
= npropdef
.n_extern_code_block
46 if code_block
!= null then
47 verify_foreign_code_on_node
( code_block
)
52 redef fun process_nclassdef
(nclassdef
)
54 if nclassdef
isa AStdClassdef and nclassdef
.n_extern_code_block
!= null then
55 verify_foreign_code_on_node
( nclassdef
.n_extern_code_block
.as(not null) )
59 private fun verify_foreign_code_on_node
(n
: AExternCodeBlock)
63 var identified
= v
.identify_language
(n
)
65 if found
and identified
then
66 toolcontext
.error
(n
.location
, "Two languages identified as possible handlers.")
73 if not found
then toolcontext
.error
(n
.location
, "Unsupported language.")
78 var ffi_files
= new Array[ExternFile]
81 redef class AExternCodeBlock
82 fun language_name
: nullable String do
83 if n_in_language
== null then return null
84 return n_in_language
.n_string
.without_quotes
86 fun language_name_lowered
: nullable String do
87 if language_name
== null then return null
88 return language_name
.to_lower
91 fun code
: String do return n_extern_code_segment
.without_guard
93 var language
: nullable FFILanguage = null
96 # Visitor for a specific languages. Works kinda like a `Phase` and is executed
99 init(ffi_language_assignation_phase
: FFILanguageAssignationPhase)
101 ffi_language_assignation_phase
.languages
.add
(self)
104 # Is this `block` written in this language?
105 fun identify_language
(block
: AExternCodeBlock ): Bool is abstract
107 # Generate wrapper code for this module/header code block
108 fun compile_module_block
(block
: AExternCodeBlock, ecc
: CCompilationUnit, nmodule
: AModule) is abstract
110 # Generate wrapper code for this extern method
111 fun compile_extern_method
(block
: AExternCodeBlock, m
: AExternPropdef,
112 ecc
: CCompilationUnit, nmodule
: AModule) is abstract
114 # Generate wrapper code for this extern class
115 fun compile_extern_class
(block
: AExternCodeBlock, m
: AClassdef,
116 ecc
: CCompilationUnit, nmodule
: AModule) is abstract
118 # Get the foreign type of this extern class definition
119 fun get_ftype
(block
: AExternCodeBlock, m
: AClassdef): ForeignType is abstract
121 # Generate the code to offer this callback if foreign code
122 fun compile_callback
(callback
: NitniCallback, nmodule
: AModule,
123 mainmmodule
: MModule, ecc
: CCompilationUnit) is abstract
125 # Complete compilation of generated code
126 fun compile_to_files
(amodule
: AModule, directory
: String) do end
130 # Returns the content of this node without both of the surrounding "
131 fun without_quotes
: String
133 assert text
.length
>= 2
134 return text
.substring
(1, text
.length-2
)
138 redef class TExternCodeSegment
139 # Returns the content of this node without the surrounding `{ and `}
140 fun without_guard
: String
142 assert text
.length
>= 4
143 return text
.substring
(2, text
.length-4
)
147 # An extern file to compile
149 # The filename of the file
152 fun makefile_rule_name
: String is abstract
153 fun makefile_rule_content
: String is abstract
156 redef class CCompilationUnit
157 fun write_as_impl
( amodule
: AModule, compdir
: String )
159 var base_name
= "{amodule.mmodule.name}._ffi"
161 var h_file
= "{base_name}.h"
162 var guard
= "{amodule.cname.to_s.to_upper}_NIT_H"
163 write_header_to_file
( amodule
, "{compdir}/{h_file}", new Array[String], guard
)
165 var c_file
= "{base_name}.c"
166 write_body_to_file
( amodule
, "{compdir}/{c_file}", ["<stdlib.h>", "<stdio.h>", "\"{h_file}\
""] )
168 files
.add
( "{compdir}/{c_file}" )
171 fun write_header_to_file
(amodule
: AModule, file
: String, includes
: Array[String], guard
: String)
173 var stream
= new OFStream.open
( file
)
176 var module_info
= "/*\n\tExtern implementation of Nit module {amodule.mmodule.name}\n*/\n"
178 stream
.write
( module_info
)
180 stream
.write
( "#ifndef {guard}\n" )
181 stream
.write
( "#define {guard}\n\n" )
183 for incl
in includes
do stream
.write
( "#include {incl}\n" )
185 compile_header_core
( stream
)
187 # header file guard close
188 stream
.write
( "#endif\n" )
192 fun write_body_to_file
(amodule
: AModule, file
: String, includes
: Array[String])
194 var stream
= new OFStream.open
(file
)
196 var module_info
= "/*\n\tExtern implementation of Nit module {amodule.mmodule.name}\n*/\n"
198 stream
.write
( module_info
)
199 for incl
in includes
do stream
.write
( "#include {incl}\n" )
201 compile_body_core
( stream
)
208 fun ctype
: String do return "void*"