1f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// 2f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// Copyright 2012 Francisco Jerez 3f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// 4f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// Permission is hereby granted, free of charge, to any person obtaining a 5f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// copy of this software and associated documentation files (the "Software"), 6f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// to deal in the Software without restriction, including without limitation 7f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// the rights to use, copy, modify, merge, publish, distribute, sublicense, 8f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// and/or sell copies of the Software, and to permit persons to whom the 9f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// Software is furnished to do so, subject to the following conditions: 10f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// 11f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// The above copyright notice and this permission notice shall be included in 12f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// all copies or substantial portions of the Software. 13f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// 14f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 15f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 16f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 17f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, 18f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF 19f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE 20f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// SOFTWARE. 21f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// 22f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 23f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org#include <sstream> 24f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 25f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org#include "core/compiler.hpp" 26f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 27f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org#include "tgsi/tgsi_parse.h" 28f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org#include "tgsi/tgsi_text.h" 29f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org#include "util/u_memory.h" 30f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 31f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgusing namespace clover; 32f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 33f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgnamespace { 34f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org void 35f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org read_header(const std::string &header, module &m) { 36f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org std::istringstream ls(header); 37f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org std::string line; 38f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 39f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org while (getline(ls, line)) { 40f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org std::istringstream ts(line); 41f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org std::string name, tok; 42f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org module::size_t offset; 43f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org compat::vector<module::argument> args; 44f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 45f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org if (!(ts >> name)) 46f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org continue; 47f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 48f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org if (!(ts >> offset)) 49f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org throw build_error("invalid kernel start address"); 50f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 51f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org while (ts >> tok) { 52f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org if (tok == "scalar") 53f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org args.push_back({ module::argument::scalar, 4 }); 54f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org else if (tok == "global") 55f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org args.push_back({ module::argument::global, 4 }); 56f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org else if (tok == "local") 57f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org args.push_back({ module::argument::local, 4 }); 58f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org else if (tok == "constant") 59f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org args.push_back({ module::argument::constant, 4 }); 60f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org else if (tok == "image2d_rd") 61f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org args.push_back({ module::argument::image2d_rd, 4 }); 62f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org else if (tok == "image2d_wr") 63f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org args.push_back({ module::argument::image2d_wr, 4 }); 64f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org else if (tok == "image3d_rd") 65f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org args.push_back({ module::argument::image3d_rd, 4 }); 66f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org else if (tok == "image3d_wr") 67f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org args.push_back({ module::argument::image3d_wr, 4 }); 68f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org else if (tok == "sampler") 69f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org args.push_back({ module::argument::sampler, 0 }); 70f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org else 71f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org throw build_error("invalid kernel argument"); 72f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org } 73f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 74f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org m.syms.push_back({ name, 0, offset, args }); 75f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org } 76f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org } 77f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 78f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org void 79f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org read_body(const char *source, module &m) { 80f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org tgsi_token prog[1024]; 81f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 82f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org if (!tgsi_text_translate(source, prog, Elements(prog))) 83f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org throw build_error("translate failed"); 84f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 85f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org unsigned sz = tgsi_num_tokens(prog) * sizeof(tgsi_token); 86f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org m.secs.push_back({ 0, module::section::text, sz, { (char *)prog, sz } }); 87f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org } 88f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org} 89f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 90f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgmodule 91f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgclover::compile_program_tgsi(const compat::string &source) { 92f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org const char *body = source.find("COMP\n"); 93f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org module m; 94f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 95f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org read_header({ source.begin(), body }, m); 96f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org read_body(body, m); 97f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org 98f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org return m; 99f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org} 100