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
23 import nitni
::nitni_utilities
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 AMethPropdef 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
, "FFI Error: two languages identified as possible handlers.")
73 if not found
then toolcontext
.error
(n
.location
, "FFI Error: 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 var ffi_language_assignation_phase
: FFILanguageAssignationPhase
103 ffi_language_assignation_phase
.languages
.add
(self)
106 # Is this `block` written in this language?
107 fun identify_language
(block
: AExternCodeBlock ): Bool is abstract
109 # Generate wrapper code for this module/header code block
110 fun compile_module_block
(block
: AExternCodeBlock, ecc
: CCompilationUnit, mmodule
: MModule) is abstract
112 # Generate wrapper code for this extern method
113 fun compile_extern_method
(block
: AExternCodeBlock, m
: AMethPropdef,
114 ecc
: CCompilationUnit, nmodule
: MModule) is abstract
116 # Generate wrapper code for this extern class
117 fun compile_extern_class
(block
: AExternCodeBlock, m
: AClassdef,
118 ecc
: CCompilationUnit, mmodule
: MModule) is abstract
120 # Get the foreign type of this extern class definition
121 fun get_ftype
(block
: AExternCodeBlock, m
: AClassdef): ForeignType is abstract
123 # Complete compilation of generated code
124 fun compile_to_files
(mmodule
: MModule, directory
: String) do end
128 # Returns the content of this node without both of the surrounding "
129 fun without_quotes
: String
131 assert text
.length
>= 2
132 return text
.substring
(1, text
.length-2
)
136 redef class TExternCodeSegment
137 # Returns the content of this node without the surrounding `{ and `}
138 fun without_guard
: String
140 assert text
.length
>= 4
141 return text
.substring
(2, text
.length-4
)
145 redef class CCompilationUnit
146 fun write_as_impl
(mmodule
: MModule, compdir
: String)
148 var base_name
= "{mmodule.c_name}._ffi"
150 var h_file
= "{base_name}.h"
151 var guard
= "{mmodule.c_name.to_upper}_NIT_H"
152 write_header_to_file
(mmodule
, "{compdir}/{h_file}", new Array[String], guard
)
154 var c_file
= "{base_name}.c"
155 write_body_to_file
(mmodule
, "{compdir}/{c_file}", ["<stdlib.h>", "<stdio.h>", "\"{h_file}\
""])
157 files
.add
( "{compdir}/{c_file}" )
160 fun write_header_to_file
(mmodule
: MModule, file
: String, includes
: Array[String], guard
: String)
162 var stream
= new FileWriter.open
( file
)
165 var module_info
= "/*\n\tExtern implementation of Nit module {mmodule.name}\n*/\n"
167 stream
.write
( module_info
)
169 stream
.write
( "#ifndef {guard}\n" )
170 stream
.write
( "#define {guard}\n\n" )
172 for incl
in includes
do stream
.write
( "#include {incl}\n" )
174 compile_header_core
( stream
)
176 # header file guard close
177 stream
.write
( "#endif\n" )
181 fun write_body_to_file
(mmodule
: MModule, file
: String, includes
: Array[String])
183 var stream
= new FileWriter.open
(file
)
185 var module_info
= "/*\n\tExtern implementation of Nit module {mmodule.name}\n*/\n"
187 stream
.write
( module_info
)
188 for incl
in includes
do stream
.write
( "#include {incl}\n" )
190 compile_body_core
( stream
)
197 fun ctype
: String do return "void*"