diff --git a/xmake/core/package/package.lua b/xmake/core/package/package.lua index 00ae72768d1..740e6f4f110 100644 --- a/xmake/core/package/package.lua +++ b/xmake/core/package/package.lua @@ -2639,6 +2639,20 @@ function _instance:check_mxxsnippets(snippets, opt) return sandbox_module.import("lib.detect.check_mxxsnippets", {anonymous = true})(snippets, opt) end +-- check the given fortran snippets? +-- +-- @param snippets the snippets +-- @param opt the argument options, e.g. {configs = {defines = ""}} +-- +-- @return true or false, errors +-- +function _instance:check_fcsnippets(snippets, opt) + opt = opt or {} + opt.target = self + opt.configs = self:_generate_build_configs(opt.configs, {sourcekind = "fc"}) + return sandbox_module.import("lib.detect.check_fcsnippets", {anonymous = true})(snippets, opt) +end + -- the current mode is belong to the given modes? function package._api_is_mode(interp, ...) return config.is_mode(...) diff --git a/xmake/modules/lib/detect/check_fcsnippets.lua b/xmake/modules/lib/detect/check_fcsnippets.lua new file mode 100644 index 00000000000..97fb63c6b07 --- /dev/null +++ b/xmake/modules/lib/detect/check_fcsnippets.lua @@ -0,0 +1,161 @@ +--!A cross-platform build utility based on Lua +-- +-- Licensed under the Apache License, Version 2.0 (the "License"); +-- you may not use this file except in compliance with the License. +-- You may obtain a copy of the License at +-- +-- http://www.apache.org/licenses/LICENSE-2.0 +-- +-- Unless required by applicable law or agreed to in writing, software +-- distributed under the License is distributed on an "AS IS" BASIS, +-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +-- See the License for the specific language governing permissions and +-- limitations under the License. +-- +-- Copyright (C) 2015-present, TBOOX Open Source Group. +-- +-- @author ruki +-- @file check_fcsnippets.lua +-- + +-- imports +import("core.base.option") +import("core.tool.linker") +import("core.tool.compiler") + +-- make source code +function _sourcecode(snippets, opt) + local sourcecode = "" + for _, snippet in pairs(snippets) do + sourcecode = sourcecode .. "\n" .. snippet + end + sourcecode = sourcecode .. "\n" + return sourcecode +end + +-- check the given fortran snippets? +-- +-- @param snippets the snippets +-- @param opt the argument options +-- e.g. +-- { verbose = false, target = [target|option] +-- , configs = {defines = "xx", fcflags = ""} +-- , tryrun = true, output = true} +-- +-- funcs: +-- sigsetjmp +-- sigsetjmp((void*)0, 0) +-- sigsetjmp{sigsetjmp((void*)0, 0);} +-- sigsetjmp{int a = 0; sigsetjmp((void*)a, a);} +-- +-- @return true or false +-- +-- @code +-- local ok, output_or_errors = check_fcsnippets([[ +-- program hello +-- print *, "Hello World!" +-- end program hello +-- ]], {tryrun = true}) +-- @endcode +-- +function main(snippets, opt) + opt = opt or {} + snippets = snippets or {} + + -- get configs + local configs = opt.configs or opt.config + + -- get links + local links = {} + local target = opt.target + if configs and configs.links then + table.join2(links, configs.links) + end + if target and target:type() ~= "package" then + table.join2(links, target:get("links")) + end + if configs and configs.syslinks then + table.join2(links, configs.syslinks) + end + if target and target:type() ~= "package" then + table.join2(links, opt.target:get("syslinks")) + end + + -- make source code + local sourcecode = _sourcecode(snippets, opt) + + -- make the source file + -- @note we use fixed temporary filenames in order to better cache the compilation results for build_cache. + local tmpfile = os.tmpfile(sourcecode) + local sourcefile = tmpfile .. ".f90" + local objectfile = os.tmpfile() .. ".o" + local binaryfile = objectfile:gsub("%.o$", ".b") + if not os.isfile(sourcefile) then + io.writefile(sourcefile, sourcecode) + end + + -- @note cannot cache result, all conditions will be changed + -- attempt to compile it + local errors = nil + local ok, output = try + { + function () + if option.get("diagnosis") then + cprint("${dim}> %s", compiler.compcmd(sourcefile, objectfile, opt)) + end + opt = table.clone(opt) + opt.build_warnings = false + compiler.compile(sourcefile, objectfile, opt) + if #links > 0 or opt.tryrun or opt.binary_match then + if option.get("diagnosis") then + cprint("${dim}> %s", linker.linkcmd("binary", "fc", objectfile, binaryfile, opt)) + end + linker.link("binary", "fc", objectfile, binaryfile, opt) + end + if opt.tryrun then + if opt.output then + local output = os.iorun(binaryfile) + if output then + output = output:trim() + end + return true, output + else + os.vrun(binaryfile) + end + end + local binary_match = opt.binary_match + if binary_match then + local content = io.readfile(binaryfile, {encoding = "binary"}) + local match = type(binary_match) == "function" and binary_match(content) or content:match(binary_match) + if match ~= nil then + return true, match + end + end + return true + end, + catch { function (errs) errors = errs end } + } + + -- remove some files + os.tryrm(objectfile) + os.tryrm(binaryfile) + + -- trace + if opt.verbose or option.get("verbose") or option.get("diagnosis") then + if #links > 0 then + cprint("${dim}> checking for fortran links(%s)", table.concat(links, ", ")) + end + for idx_or_name, snippet in pairs(snippets) do + local name = idx_or_name + if type(name) == "number" then + name = snippet:sub(1, 16) + end + cprint("${dim}> checking for fortran snippet(%s)", name) + end + end + if errors and option.get("diagnosis") and #tostring(errors) > 0 then + cprint("${color.warning}checkinfo:${clear dim} %s", errors) + end + return ok, ok and output or errors +end +