Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 1 | // Copyright (c) 2015 The Khronos Group Inc. |
| 2 | // |
| 3 | // Permission is hereby granted, free of charge, to any person obtaining a |
| 4 | // copy of this software and/or associated documentation files (the |
| 5 | // "Materials"), to deal in the Materials without restriction, including |
| 6 | // without limitation the rights to use, copy, modify, merge, publish, |
| 7 | // distribute, sublicense, and/or sell copies of the Materials, and to |
| 8 | // permit persons to whom the Materials are furnished to do so, subject to |
| 9 | // the following conditions: |
| 10 | // |
| 11 | // The above copyright notice and this permission notice shall be included |
| 12 | // in all copies or substantial portions of the Materials. |
| 13 | // |
| 14 | // MODIFICATIONS TO THIS FILE MAY MEAN IT NO LONGER ACCURATELY REFLECTS |
| 15 | // KHRONOS STANDARDS. THE UNMODIFIED, NORMATIVE VERSIONS OF KHRONOS |
| 16 | // SPECIFICATIONS AND HEADER INFORMATION ARE LOCATED AT |
| 17 | // https://www.khronos.org/registry/ |
| 18 | // |
| 19 | // THE MATERIALS ARE PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, |
| 20 | // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF |
| 21 | // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. |
| 22 | // IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY |
| 23 | // CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, |
| 24 | // TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE |
| 25 | // MATERIALS OR THE USE OR OTHER DEALINGS IN THE MATERIALS. |
| 26 | |
David Neto | 36b0c0f | 2015-09-16 18:32:54 -0400 | [diff] [blame] | 27 | #include "text.h" |
| 28 | |
Dejan Mircevski | ba569fb | 2015-09-11 16:34:49 -0400 | [diff] [blame] | 29 | #include <algorithm> |
| 30 | #include <cassert> |
Andrew Woloszyn | 13804e5 | 2015-09-22 15:50:33 -0400 | [diff] [blame] | 31 | #include <cctype> |
Dejan Mircevski | ba569fb | 2015-09-11 16:34:49 -0400 | [diff] [blame] | 32 | #include <cstdio> |
| 33 | #include <cstdlib> |
David Neto | 36b0c0f | 2015-09-16 18:32:54 -0400 | [diff] [blame] | 34 | #include <cstring> |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 35 | #include <memory> |
Dejan Mircevski | ba569fb | 2015-09-11 16:34:49 -0400 | [diff] [blame] | 36 | #include <string> |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 37 | #include <sstream> |
David Neto | 36b0c0f | 2015-09-16 18:32:54 -0400 | [diff] [blame] | 38 | #include <unordered_map> |
| 39 | #include <vector> |
Dejan Mircevski | ba569fb | 2015-09-11 16:34:49 -0400 | [diff] [blame] | 40 | |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 41 | #include "binary.h" |
David Neto | 36b0c0f | 2015-09-16 18:32:54 -0400 | [diff] [blame] | 42 | #include "bitwisecast.h" |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 43 | #include "diagnostic.h" |
| 44 | #include "ext_inst.h" |
David Neto | b5dc8fc | 2015-10-06 16:22:00 -0400 | [diff] [blame] | 45 | #include "instruction.h" |
David Neto | 36b0c0f | 2015-09-16 18:32:54 -0400 | [diff] [blame] | 46 | #include <libspirv/libspirv.h> |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 47 | #include "opcode.h" |
| 48 | #include "operand.h" |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 49 | #include "text_handler.h" |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 50 | |
Andrew Woloszyn | 13804e5 | 2015-09-22 15:50:33 -0400 | [diff] [blame] | 51 | bool spvIsValidIDCharacter(const char value) { |
| 52 | return value == '_' || 0 != ::isalnum(value); |
| 53 | } |
| 54 | |
| 55 | // Returns true if the given string represents a valid ID name. |
Dejan Mircevski | 14c4b10 | 2015-09-29 17:07:21 -0400 | [diff] [blame] | 56 | bool spvIsValidID(const char* textValue) { |
| 57 | const char* c = textValue; |
Andrew Woloszyn | 13804e5 | 2015-09-22 15:50:33 -0400 | [diff] [blame] | 58 | for (; *c != '\0'; ++c) { |
| 59 | if (!spvIsValidIDCharacter(*c)) { |
| 60 | return false; |
| 61 | } |
| 62 | } |
| 63 | // If the string was empty, then the ID also is not valid. |
| 64 | return c != textValue; |
| 65 | } |
| 66 | |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 67 | // Text API |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 68 | |
Dejan Mircevski | 903f9d6 | 2015-09-28 17:04:39 -0400 | [diff] [blame] | 69 | spv_result_t spvTextToLiteral(const char* textValue, spv_literal_t* pLiteral) { |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 70 | bool isSigned = false; |
David Neto | affa696 | 2015-08-24 15:33:14 -0400 | [diff] [blame] | 71 | int numPeriods = 0; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 72 | bool isString = false; |
| 73 | |
David Neto | affa696 | 2015-08-24 15:33:14 -0400 | [diff] [blame] | 74 | const size_t len = strlen(textValue); |
David Neto | 98290a2 | 2015-08-24 16:27:02 -0400 | [diff] [blame] | 75 | if (len == 0) return SPV_FAILED_MATCH; |
| 76 | |
David Neto | affa696 | 2015-08-24 15:33:14 -0400 | [diff] [blame] | 77 | for (uint64_t index = 0; index < len; ++index) { |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 78 | switch (textValue[index]) { |
| 79 | case '0': |
| 80 | case '1': |
| 81 | case '2': |
| 82 | case '3': |
| 83 | case '4': |
| 84 | case '5': |
| 85 | case '6': |
| 86 | case '7': |
| 87 | case '8': |
| 88 | case '9': |
| 89 | break; |
| 90 | case '.': |
David Neto | affa696 | 2015-08-24 15:33:14 -0400 | [diff] [blame] | 91 | numPeriods++; |
| 92 | break; |
| 93 | case '-': |
| 94 | if (index == 0) { |
| 95 | isSigned = true; |
| 96 | } else { |
| 97 | isString = true; |
| 98 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 99 | break; |
| 100 | default: |
| 101 | isString = true; |
Lei Zhang | e78a7c1 | 2015-09-10 17:07:21 -0400 | [diff] [blame] | 102 | index = len; // break out of the loop too. |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 103 | break; |
| 104 | } |
| 105 | } |
| 106 | |
David Neto | affa696 | 2015-08-24 15:33:14 -0400 | [diff] [blame] | 107 | pLiteral->type = spv_literal_type_t(99); |
| 108 | |
Lei Zhang | e78a7c1 | 2015-09-10 17:07:21 -0400 | [diff] [blame] | 109 | if (isString || numPeriods > 1 || (isSigned && len == 1)) { |
David Neto | 98290a2 | 2015-08-24 16:27:02 -0400 | [diff] [blame] | 110 | if (len < 2 || textValue[0] != '"' || textValue[len - 1] != '"') |
| 111 | return SPV_FAILED_MATCH; |
Andrew Woloszyn | 3e69cd1 | 2015-10-14 12:44:19 -0400 | [diff] [blame] | 112 | bool escaping = false; |
| 113 | size_t write_index = 0; |
| 114 | for(const char* val = textValue + 1; val != textValue + len - 1; ++val) { |
| 115 | if ((*val == '\\') && (!escaping)) { |
| 116 | escaping = true; |
| 117 | } else { |
| 118 | // Have to save space for the null-terminator |
| 119 | if (write_index >= sizeof(pLiteral->value.str) - 1) |
| 120 | return SPV_ERROR_OUT_OF_MEMORY; |
| 121 | pLiteral->value.str[write_index] = *val; |
| 122 | escaping = false; |
| 123 | ++write_index; |
| 124 | } |
| 125 | } |
| 126 | |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 127 | pLiteral->type = SPV_LITERAL_TYPE_STRING; |
Andrew Woloszyn | 3e69cd1 | 2015-10-14 12:44:19 -0400 | [diff] [blame] | 128 | pLiteral->value.str[write_index] = '\0'; |
David Neto | affa696 | 2015-08-24 15:33:14 -0400 | [diff] [blame] | 129 | } else if (numPeriods == 1) { |
| 130 | double d = std::strtod(textValue, nullptr); |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 131 | float f = (float)d; |
| 132 | if (d == (double)f) { |
| 133 | pLiteral->type = SPV_LITERAL_TYPE_FLOAT_32; |
| 134 | pLiteral->value.f = f; |
| 135 | } else { |
| 136 | pLiteral->type = SPV_LITERAL_TYPE_FLOAT_64; |
| 137 | pLiteral->value.d = d; |
| 138 | } |
| 139 | } else if (isSigned) { |
| 140 | int64_t i64 = strtoll(textValue, nullptr, 10); |
| 141 | int32_t i32 = (int32_t)i64; |
| 142 | if (i64 == (int64_t)i32) { |
| 143 | pLiteral->type = SPV_LITERAL_TYPE_INT_32; |
| 144 | pLiteral->value.i32 = i32; |
| 145 | } else { |
| 146 | pLiteral->type = SPV_LITERAL_TYPE_INT_64; |
| 147 | pLiteral->value.i64 = i64; |
| 148 | } |
| 149 | } else { |
| 150 | uint64_t u64 = strtoull(textValue, nullptr, 10); |
| 151 | uint32_t u32 = (uint32_t)u64; |
| 152 | if (u64 == (uint64_t)u32) { |
| 153 | pLiteral->type = SPV_LITERAL_TYPE_UINT_32; |
| 154 | pLiteral->value.u32 = u32; |
| 155 | } else { |
| 156 | pLiteral->type = SPV_LITERAL_TYPE_UINT_64; |
| 157 | pLiteral->value.u64 = u64; |
| 158 | } |
| 159 | } |
| 160 | |
| 161 | return SPV_SUCCESS; |
| 162 | } |
| 163 | |
Dejan Mircevski | 14c4b10 | 2015-09-29 17:07:21 -0400 | [diff] [blame] | 164 | namespace { |
| 165 | |
| 166 | /// Parses an immediate integer from text, guarding against overflow. If |
| 167 | /// successful, adds the parsed value to pInst, advances the context past it, |
| 168 | /// and returns SPV_SUCCESS. Otherwise, leaves pInst alone, emits diagnostics, |
| 169 | /// and returns SPV_ERROR_INVALID_TEXT. |
| 170 | spv_result_t encodeImmediate(libspirv::AssemblyContext* context, |
| 171 | const char* text, spv_instruction_t* pInst) { |
| 172 | assert(*text == '!'); |
David Neto | 78e677b | 2015-10-05 13:28:46 -0400 | [diff] [blame] | 173 | uint32_t parse_result; |
| 174 | if (auto error = |
| 175 | context->parseNumber(text + 1, false, &parse_result, |
| 176 | "Invalid immediate integer: !")) |
| 177 | return error; |
| 178 | context->binaryEncodeU32(parse_result, pInst); |
Dejan Mircevski | 14c4b10 | 2015-09-29 17:07:21 -0400 | [diff] [blame] | 179 | context->seekForward(strlen(text)); |
| 180 | return SPV_SUCCESS; |
| 181 | } |
| 182 | |
| 183 | } // anonymous namespace |
| 184 | |
Andrew Woloszyn | 13804e5 | 2015-09-22 15:50:33 -0400 | [diff] [blame] | 185 | /// @brief Translate an Opcode operand to binary form |
| 186 | /// |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 187 | /// @param[in] grammar the grammar to use for compilation |
| 188 | /// @param[in, out] context the dynamic compilation info |
Andrew Woloszyn | 13804e5 | 2015-09-22 15:50:33 -0400 | [diff] [blame] | 189 | /// @param[in] type of the operand |
| 190 | /// @param[in] textValue word of text to be parsed |
Andrew Woloszyn | 13804e5 | 2015-09-22 15:50:33 -0400 | [diff] [blame] | 191 | /// @param[out] pInst return binary Opcode |
| 192 | /// @param[in,out] pExpectedOperands the operand types expected |
Andrew Woloszyn | 13804e5 | 2015-09-22 15:50:33 -0400 | [diff] [blame] | 193 | /// |
| 194 | /// @return result code |
Dejan Mircevski | 14c4b10 | 2015-09-29 17:07:21 -0400 | [diff] [blame] | 195 | spv_result_t spvTextEncodeOperand(const libspirv::AssemblyGrammar& grammar, |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 196 | libspirv::AssemblyContext* context, |
| 197 | const spv_operand_type_t type, |
| 198 | const char* textValue, |
| 199 | spv_instruction_t* pInst, |
| 200 | spv_operand_pattern_t* pExpectedOperands) { |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 201 | // NOTE: Handle immediate int in the stream |
| 202 | if ('!' == textValue[0]) { |
Dejan Mircevski | 14c4b10 | 2015-09-29 17:07:21 -0400 | [diff] [blame] | 203 | if (auto error = encodeImmediate(context, textValue, pInst)) { |
| 204 | return error; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 205 | } |
Dejan Mircevski | 897bff9 | 2015-09-29 10:38:18 -0400 | [diff] [blame] | 206 | *pExpectedOperands = |
| 207 | spvAlternatePatternFollowingImmediate(*pExpectedOperands); |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 208 | return SPV_SUCCESS; |
| 209 | } |
| 210 | |
| 211 | switch (type) { |
David Neto | b14a727 | 2015-09-25 13:56:09 -0400 | [diff] [blame] | 212 | case SPV_OPERAND_TYPE_EXECUTION_SCOPE: |
David Neto | e3f70b9 | 2015-08-27 13:50:05 -0400 | [diff] [blame] | 213 | case SPV_OPERAND_TYPE_ID: |
Andrew Woloszyn | 537e776 | 2015-09-29 11:28:34 -0400 | [diff] [blame] | 214 | case SPV_OPERAND_TYPE_TYPE_ID: |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 215 | case SPV_OPERAND_TYPE_ID_IN_OPTIONAL_TUPLE: |
| 216 | case SPV_OPERAND_TYPE_OPTIONAL_ID: |
David Neto | b14a727 | 2015-09-25 13:56:09 -0400 | [diff] [blame] | 217 | case SPV_OPERAND_TYPE_MEMORY_SEMANTICS: |
| 218 | case SPV_OPERAND_TYPE_RESULT_ID: { |
Lei Zhang | abafd5e | 2015-08-21 11:52:29 -0400 | [diff] [blame] | 219 | if ('%' == textValue[0]) { |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 220 | textValue++; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 221 | } else { |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 222 | return context->diagnostic() << "Expected id to start with %."; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 223 | } |
Andrew Woloszyn | 13804e5 | 2015-09-22 15:50:33 -0400 | [diff] [blame] | 224 | if (!spvIsValidID(textValue)) { |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 225 | return context->diagnostic() << "Invalid ID " << textValue; |
Andrew Woloszyn | 13804e5 | 2015-09-22 15:50:33 -0400 | [diff] [blame] | 226 | } |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 227 | const uint32_t id = context->spvNamedIdAssignOrGet(textValue); |
Andrew Woloszyn | 537e776 | 2015-09-29 11:28:34 -0400 | [diff] [blame] | 228 | if (type == SPV_OPERAND_TYPE_TYPE_ID) pInst->resultTypeId = id; |
David Neto | b5dc8fc | 2015-10-06 16:22:00 -0400 | [diff] [blame] | 229 | spvInstructionAddWord(pInst, id); |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 230 | } break; |
| 231 | case SPV_OPERAND_TYPE_LITERAL_NUMBER: { |
| 232 | // NOTE: Special case for extension instruction lookup |
| 233 | if (OpExtInst == pInst->opcode) { |
| 234 | spv_ext_inst_desc extInst; |
Dejan Mircevski | 14c4b10 | 2015-09-29 17:07:21 -0400 | [diff] [blame] | 235 | if (grammar.lookupExtInst(pInst->extInstType, textValue, &extInst)) { |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 236 | return context->diagnostic() << "Invalid extended instruction name '" |
| 237 | << textValue << "'."; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 238 | } |
David Neto | b5dc8fc | 2015-10-06 16:22:00 -0400 | [diff] [blame] | 239 | spvInstructionAddWord(pInst, extInst->ext_inst); |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 240 | |
| 241 | // Prepare to parse the operands for the extended instructions. |
| 242 | spvPrependOperandTypes(extInst->operandTypes, pExpectedOperands); |
| 243 | |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 244 | return SPV_SUCCESS; |
| 245 | } |
Lei Zhang | 6d41581 | 2015-09-15 13:36:21 -0400 | [diff] [blame] | 246 | } // Fall through for the general case. |
Lei Zhang | b41d150 | 2015-09-14 15:22:23 -0400 | [diff] [blame] | 247 | case SPV_OPERAND_TYPE_MULTIWORD_LITERAL_NUMBER: |
David Neto | 561dc4e | 2015-09-25 14:23:29 -0400 | [diff] [blame] | 248 | case SPV_OPERAND_TYPE_LITERAL_NUMBER_IN_OPTIONAL_TUPLE: |
| 249 | case SPV_OPERAND_TYPE_OPTIONAL_LITERAL_NUMBER: { |
David Neto | 78e677b | 2015-10-05 13:28:46 -0400 | [diff] [blame] | 250 | libspirv::IdType expected_type = libspirv::kUnknownType; |
Andrew Woloszyn | 537e776 | 2015-09-29 11:28:34 -0400 | [diff] [blame] | 251 | // The encoding for OpConstant, OpSpecConstant and OpSwitch all |
| 252 | // depend on either their own result-id or the result-id of |
| 253 | // one of their parameters. |
| 254 | if (OpConstant == pInst->opcode || OpSpecConstant == pInst->opcode) { |
| 255 | // Special cases for encoding possibly non-32-bit literals here. |
David Neto | 78e677b | 2015-10-05 13:28:46 -0400 | [diff] [blame] | 256 | expected_type = |
Andrew Woloszyn | 537e776 | 2015-09-29 11:28:34 -0400 | [diff] [blame] | 257 | context->getTypeOfTypeGeneratingValue(pInst->resultTypeId); |
David Neto | 78e677b | 2015-10-05 13:28:46 -0400 | [diff] [blame] | 258 | if (!libspirv::isScalarFloating(expected_type) && |
| 259 | !libspirv::isScalarIntegral(expected_type)) { |
Andrew Woloszyn | 537e776 | 2015-09-29 11:28:34 -0400 | [diff] [blame] | 260 | spv_opcode_desc d; |
| 261 | const char* opcode_name = "opcode"; |
| 262 | if (SPV_SUCCESS == grammar.lookupOpcode(pInst->opcode, &d)) { |
| 263 | opcode_name = d->name; |
| 264 | } |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 265 | return context->diagnostic() |
| 266 | << "Type for " << opcode_name |
| 267 | << " must be a scalar floating point or integer type"; |
Andrew Woloszyn | 537e776 | 2015-09-29 11:28:34 -0400 | [diff] [blame] | 268 | } |
| 269 | } else if (pInst->opcode == OpSwitch) { |
David Neto | 78e677b | 2015-10-05 13:28:46 -0400 | [diff] [blame] | 270 | // We need to know the type of the selector. |
| 271 | expected_type = context->getTypeOfValueInstruction(pInst->words[1]); |
| 272 | if (!libspirv::isScalarIntegral(expected_type)) { |
| 273 | context->diagnostic() |
| 274 | << "The selector operand for OpSwitch must be the result" |
| 275 | " of an instruction that generates an integer scalar"; |
| 276 | return SPV_ERROR_INVALID_TEXT; |
Andrew Woloszyn | 537e776 | 2015-09-29 11:28:34 -0400 | [diff] [blame] | 277 | } |
| 278 | } |
Andrew Woloszyn | 537e776 | 2015-09-29 11:28:34 -0400 | [diff] [blame] | 279 | |
David Neto | 78e677b | 2015-10-05 13:28:46 -0400 | [diff] [blame] | 280 | if (auto error = context->binaryEncodeNumericLiteral( |
| 281 | textValue, spvOperandIsOptional(type), expected_type, pInst)) { |
| 282 | return error; |
| 283 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 284 | } break; |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 285 | case SPV_OPERAND_TYPE_LITERAL_STRING: |
| 286 | case SPV_OPERAND_TYPE_OPTIONAL_LITERAL_STRING: { |
Lei Zhang | 6d41581 | 2015-09-15 13:36:21 -0400 | [diff] [blame] | 287 | spv_literal_t literal = {}; |
| 288 | spv_result_t error = spvTextToLiteral(textValue, &literal); |
| 289 | if (error != SPV_SUCCESS) { |
| 290 | if (error == SPV_ERROR_OUT_OF_MEMORY) return error; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 291 | if (spvOperandIsOptional(type)) return SPV_FAILED_MATCH; |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 292 | return context->diagnostic() << "Invalid literal string '" << textValue |
| 293 | << "'."; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 294 | } |
Lei Zhang | 6d41581 | 2015-09-15 13:36:21 -0400 | [diff] [blame] | 295 | if (literal.type != SPV_LITERAL_TYPE_STRING) { |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 296 | return context->diagnostic(SPV_FAILED_MATCH) |
| 297 | << "Expected literal string, found literal number '" << textValue |
| 298 | << "'."; |
Lei Zhang | 6d41581 | 2015-09-15 13:36:21 -0400 | [diff] [blame] | 299 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 300 | |
| 301 | // NOTE: Special case for extended instruction library import |
| 302 | if (OpExtInstImport == pInst->opcode) { |
Lei Zhang | 6d41581 | 2015-09-15 13:36:21 -0400 | [diff] [blame] | 303 | pInst->extInstType = spvExtInstImportTypeGet(literal.value.str); |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 304 | } |
| 305 | |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 306 | if (context->binaryEncodeString(literal.value.str, pInst)) |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 307 | return SPV_ERROR_INVALID_TEXT; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 308 | } break; |
David Neto | 36b0c0f | 2015-09-16 18:32:54 -0400 | [diff] [blame] | 309 | case SPV_OPERAND_TYPE_FP_FAST_MATH_MODE: |
| 310 | case SPV_OPERAND_TYPE_FUNCTION_CONTROL: |
| 311 | case SPV_OPERAND_TYPE_LOOP_CONTROL: |
David Neto | ee1b3bb | 2015-09-18 11:19:18 -0400 | [diff] [blame] | 312 | case SPV_OPERAND_TYPE_OPTIONAL_IMAGE: |
David Neto | 5bf88fc | 2015-09-17 17:06:10 -0400 | [diff] [blame] | 313 | case SPV_OPERAND_TYPE_OPTIONAL_MEMORY_ACCESS: |
David Neto | 36b0c0f | 2015-09-16 18:32:54 -0400 | [diff] [blame] | 314 | case SPV_OPERAND_TYPE_SELECTION_CONTROL: { |
| 315 | uint32_t value; |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 316 | if (grammar.parseMaskOperand(type, textValue, &value)) { |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 317 | return context->diagnostic() << "Invalid " << spvOperandTypeStr(type) |
| 318 | << " '" << textValue << "'."; |
David Neto | 36b0c0f | 2015-09-16 18:32:54 -0400 | [diff] [blame] | 319 | } |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 320 | if (auto error = context->binaryEncodeU32(value, pInst)) return error; |
David Neto | 5bf88fc | 2015-09-17 17:06:10 -0400 | [diff] [blame] | 321 | // Prepare to parse the operands for this logical operand. |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 322 | grammar.prependOperandTypesForMask(type, value, pExpectedOperands); |
Dejan Mircevski | 903f9d6 | 2015-09-28 17:04:39 -0400 | [diff] [blame] | 323 | } break; |
| 324 | case SPV_OPERAND_TYPE_OPTIONAL_CIV: { |
| 325 | auto error = spvTextEncodeOperand( |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 326 | grammar, context, SPV_OPERAND_TYPE_OPTIONAL_LITERAL_NUMBER, textValue, |
| 327 | pInst, pExpectedOperands); |
Dejan Mircevski | 903f9d6 | 2015-09-28 17:04:39 -0400 | [diff] [blame] | 328 | if (error == SPV_FAILED_MATCH) { |
| 329 | // It's not a literal number -- is it a literal string? |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 330 | error = spvTextEncodeOperand(grammar, context, |
| 331 | SPV_OPERAND_TYPE_OPTIONAL_LITERAL_STRING, |
| 332 | textValue, pInst, pExpectedOperands); |
Dejan Mircevski | 903f9d6 | 2015-09-28 17:04:39 -0400 | [diff] [blame] | 333 | } |
| 334 | if (error == SPV_FAILED_MATCH) { |
| 335 | // It's not a literal -- is it an ID? |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 336 | error = |
| 337 | spvTextEncodeOperand(grammar, context, SPV_OPERAND_TYPE_OPTIONAL_ID, |
| 338 | textValue, pInst, pExpectedOperands); |
Dejan Mircevski | 903f9d6 | 2015-09-28 17:04:39 -0400 | [diff] [blame] | 339 | } |
| 340 | if (error) { |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 341 | return context->diagnostic(error) |
| 342 | << "Invalid word following !<integer>: " << textValue; |
Dejan Mircevski | 903f9d6 | 2015-09-28 17:04:39 -0400 | [diff] [blame] | 343 | } |
| 344 | if (pExpectedOperands->empty()) { |
| 345 | pExpectedOperands->push_back(SPV_OPERAND_TYPE_OPTIONAL_CIV); |
| 346 | } |
David Neto | 36b0c0f | 2015-09-16 18:32:54 -0400 | [diff] [blame] | 347 | } break; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 348 | default: { |
| 349 | // NOTE: All non literal operands are handled here using the operand |
| 350 | // table. |
| 351 | spv_operand_desc entry; |
Dejan Mircevski | 14c4b10 | 2015-09-29 17:07:21 -0400 | [diff] [blame] | 352 | if (grammar.lookupOperand(type, textValue, strlen(textValue), &entry)) { |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 353 | return context->diagnostic() << "Invalid " << spvOperandTypeStr(type) |
| 354 | << " '" << textValue << "'."; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 355 | } |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 356 | if (context->binaryEncodeU32(entry->value, pInst)) { |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 357 | return context->diagnostic() << "Invalid " << spvOperandTypeStr(type) |
| 358 | << " '" << textValue << "'."; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 359 | } |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 360 | |
| 361 | // Prepare to parse the operands for this logical operand. |
| 362 | spvPrependOperandTypes(entry->operandTypes, pExpectedOperands); |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 363 | } break; |
| 364 | } |
| 365 | return SPV_SUCCESS; |
| 366 | } |
| 367 | |
Dejan Mircevski | f79519c | 2015-09-11 00:43:11 -0400 | [diff] [blame] | 368 | namespace { |
| 369 | |
| 370 | /// Encodes an instruction started by !<integer> at the given position in text. |
| 371 | /// |
| 372 | /// Puts the encoded words into *pInst. If successful, moves position past the |
| 373 | /// instruction and returns SPV_SUCCESS. Otherwise, returns an error code and |
| 374 | /// leaves position pointing to the error in text. |
| 375 | spv_result_t encodeInstructionStartingWithImmediate( |
Dejan Mircevski | 14c4b10 | 2015-09-29 17:07:21 -0400 | [diff] [blame] | 376 | const libspirv::AssemblyGrammar& grammar, |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 377 | libspirv::AssemblyContext* context, spv_instruction_t* pInst) { |
Dejan Mircevski | f79519c | 2015-09-11 00:43:11 -0400 | [diff] [blame] | 378 | std::string firstWord; |
| 379 | spv_position_t nextPosition = {}; |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 380 | auto error = context->getWord(firstWord, &nextPosition); |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 381 | if (error) return context->diagnostic(error) << "Internal Error"; |
Dejan Mircevski | f79519c | 2015-09-11 00:43:11 -0400 | [diff] [blame] | 382 | |
Dejan Mircevski | 14c4b10 | 2015-09-29 17:07:21 -0400 | [diff] [blame] | 383 | if ((error = encodeImmediate(context, firstWord.c_str(), pInst))) { |
| 384 | return error; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 385 | } |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 386 | while (context->advance() != SPV_END_OF_STREAM) { |
Dejan Mircevski | f79519c | 2015-09-11 00:43:11 -0400 | [diff] [blame] | 387 | // A beginning of a new instruction means we're done. |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 388 | if (context->isStartOfNewInst()) return SPV_SUCCESS; |
Dejan Mircevski | f79519c | 2015-09-11 00:43:11 -0400 | [diff] [blame] | 389 | |
| 390 | // Otherwise, there must be an operand that's either a literal, an ID, or |
| 391 | // an immediate. |
| 392 | std::string operandValue; |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 393 | if ((error = context->getWord(operandValue, &nextPosition))) |
| 394 | return context->diagnostic(error) << "Internal Error"; |
Dejan Mircevski | f79519c | 2015-09-11 00:43:11 -0400 | [diff] [blame] | 395 | |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 396 | if (operandValue == "=") |
| 397 | return context->diagnostic() << firstWord << " not allowed before =."; |
Dejan Mircevski | e3a19c0 | 2015-09-11 15:03:54 -0400 | [diff] [blame] | 398 | |
Dejan Mircevski | f79519c | 2015-09-11 00:43:11 -0400 | [diff] [blame] | 399 | // Needed to pass to spvTextEncodeOpcode(), but it shouldn't ever be |
| 400 | // expanded. |
| 401 | spv_operand_pattern_t dummyExpectedOperands; |
| 402 | error = spvTextEncodeOperand( |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 403 | grammar, context, SPV_OPERAND_TYPE_OPTIONAL_CIV, operandValue.c_str(), |
| 404 | pInst, &dummyExpectedOperands); |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 405 | if (error) return error; |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 406 | context->setPosition(nextPosition); |
Dejan Mircevski | f79519c | 2015-09-11 00:43:11 -0400 | [diff] [blame] | 407 | } |
| 408 | return SPV_SUCCESS; |
| 409 | } |
Dejan Mircevski | f79519c | 2015-09-11 00:43:11 -0400 | [diff] [blame] | 410 | } // anonymous namespace |
| 411 | |
Andrew Woloszyn | 13804e5 | 2015-09-22 15:50:33 -0400 | [diff] [blame] | 412 | /// @brief Translate single Opcode and operands to binary form |
| 413 | /// |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 414 | /// @param[in] grammar the grammar to use for compilation |
| 415 | /// @param[in, out] context the dynamic compilation info |
Andrew Woloszyn | 13804e5 | 2015-09-22 15:50:33 -0400 | [diff] [blame] | 416 | /// @param[in] text stream to translate |
| 417 | /// @param[in] format the assembly syntax format of text |
Andrew Woloszyn | 13804e5 | 2015-09-22 15:50:33 -0400 | [diff] [blame] | 418 | /// @param[out] pInst returned binary Opcode |
| 419 | /// @param[in,out] pPosition in the text stream |
Andrew Woloszyn | 13804e5 | 2015-09-22 15:50:33 -0400 | [diff] [blame] | 420 | /// |
| 421 | /// @return result code |
Dejan Mircevski | 14c4b10 | 2015-09-29 17:07:21 -0400 | [diff] [blame] | 422 | spv_result_t spvTextEncodeOpcode(const libspirv::AssemblyGrammar& grammar, |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 423 | libspirv::AssemblyContext* context, |
| 424 | spv_assembly_syntax_format_t format, |
| 425 | spv_instruction_t* pInst) { |
Dejan Mircevski | f79519c | 2015-09-11 00:43:11 -0400 | [diff] [blame] | 426 | // Check for !<integer> first. |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 427 | if ('!' == context->peek()) { |
| 428 | return encodeInstructionStartingWithImmediate(grammar, context, pInst); |
Dejan Mircevski | f79519c | 2015-09-11 00:43:11 -0400 | [diff] [blame] | 429 | } |
| 430 | |
Lei Zhang | dfc5008 | 2015-08-21 11:50:55 -0400 | [diff] [blame] | 431 | // An assembly instruction has two possible formats: |
Lei Zhang | 06efdc5 | 2015-09-10 14:00:00 -0400 | [diff] [blame] | 432 | // 1(CAF): <opcode> <operand>..., e.g., "OpTypeVoid %void". |
| 433 | // 2(AAF): <result-id> = <opcode> <operand>..., e.g., "%void = OpTypeVoid". |
Lei Zhang | dfc5008 | 2015-08-21 11:50:55 -0400 | [diff] [blame] | 434 | |
Lei Zhang | 06efdc5 | 2015-09-10 14:00:00 -0400 | [diff] [blame] | 435 | std::string firstWord; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 436 | spv_position_t nextPosition = {}; |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 437 | spv_result_t error = context->getWord(firstWord, &nextPosition); |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 438 | if (error) return context->diagnostic() << "Internal Error"; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 439 | |
Lei Zhang | 06efdc5 | 2015-09-10 14:00:00 -0400 | [diff] [blame] | 440 | std::string opcodeName; |
Lei Zhang | dfc5008 | 2015-08-21 11:50:55 -0400 | [diff] [blame] | 441 | std::string result_id; |
| 442 | spv_position_t result_id_position = {}; |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 443 | if (context->startsWithOp()) { |
Lei Zhang | 06efdc5 | 2015-09-10 14:00:00 -0400 | [diff] [blame] | 444 | opcodeName = firstWord; |
| 445 | } else { |
| 446 | // If the first word of this instruction is not an opcode, we must be |
| 447 | // processing AAF now. |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 448 | if (SPV_ASSEMBLY_SYNTAX_FORMAT_ASSIGNMENT != format) { |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 449 | return context->diagnostic() |
| 450 | << "Expected <opcode> at the beginning of an instruction, found '" |
| 451 | << firstWord << "'."; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 452 | } |
Lei Zhang | 06efdc5 | 2015-09-10 14:00:00 -0400 | [diff] [blame] | 453 | |
| 454 | result_id = firstWord; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 455 | if ('%' != result_id.front()) { |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 456 | return context->diagnostic() |
| 457 | << "Expected <opcode> or <result-id> at the beginning " |
| 458 | "of an instruction, found '" |
| 459 | << result_id << "'."; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 460 | } |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 461 | result_id_position = context->position(); |
Lei Zhang | 06efdc5 | 2015-09-10 14:00:00 -0400 | [diff] [blame] | 462 | |
| 463 | // The '=' sign. |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 464 | context->setPosition(nextPosition); |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 465 | if (context->advance()) |
| 466 | return context->diagnostic() << "Expected '=', found end of stream."; |
Lei Zhang | dfc5008 | 2015-08-21 11:50:55 -0400 | [diff] [blame] | 467 | std::string equal_sign; |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 468 | error = context->getWord(equal_sign, &nextPosition); |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 469 | if ("=" != equal_sign) |
| 470 | return context->diagnostic() << "'=' expected after result id."; |
Lei Zhang | dfc5008 | 2015-08-21 11:50:55 -0400 | [diff] [blame] | 471 | |
| 472 | // The <opcode> after the '=' sign. |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 473 | context->setPosition(nextPosition); |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 474 | if (context->advance()) |
| 475 | return context->diagnostic() << "Expected opcode, found end of stream."; |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 476 | error = context->getWord(opcodeName, &nextPosition); |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 477 | if (error) |
| 478 | return context->diagnostic(error) << "Internal Error"; |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 479 | if (!context->startsWithOp()) { |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 480 | return context->diagnostic() << "Invalid Opcode prefix '" << opcodeName |
| 481 | << "'."; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 482 | } |
Lei Zhang | dfc5008 | 2015-08-21 11:50:55 -0400 | [diff] [blame] | 483 | } |
| 484 | |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 485 | // NOTE: The table contains Opcode names without the "Op" prefix. |
Dejan Mircevski | 903f9d6 | 2015-09-28 17:04:39 -0400 | [diff] [blame] | 486 | const char* pInstName = opcodeName.data() + 2; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 487 | |
| 488 | spv_opcode_desc opcodeEntry; |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 489 | error = grammar.lookupOpcode(pInstName, &opcodeEntry); |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 490 | if (error) { |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 491 | return context->diagnostic(error) << "Invalid Opcode name '" |
| 492 | << context->getWord() << "'"; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 493 | } |
Lei Zhang | 06efdc5 | 2015-09-10 14:00:00 -0400 | [diff] [blame] | 494 | if (SPV_ASSEMBLY_SYNTAX_FORMAT_ASSIGNMENT == format) { |
| 495 | // If this instruction has <result-id>, check it follows AAF. |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 496 | if (opcodeEntry->hasResult && result_id.empty()) { |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 497 | return context->diagnostic() |
| 498 | << "Expected <result-id> at the beginning of an " |
| 499 | "instruction, found '" |
| 500 | << firstWord << "'."; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 501 | } |
Lei Zhang | 06efdc5 | 2015-09-10 14:00:00 -0400 | [diff] [blame] | 502 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 503 | pInst->opcode = opcodeEntry->opcode; |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 504 | context->setPosition(nextPosition); |
David Neto | b5dc8fc | 2015-10-06 16:22:00 -0400 | [diff] [blame] | 505 | // Reserve the first word for the instruction. |
| 506 | spvInstructionAddWord(pInst, 0); |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 507 | |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 508 | // Maintains the ordered list of expected operand types. |
| 509 | // For many instructions we only need the {numTypes, operandTypes} |
| 510 | // entries in opcodeEntry. However, sometimes we need to modify |
| 511 | // the list as we parse the operands. This occurs when an operand |
| 512 | // has its own logical operands (such as the LocalSize operand for |
| 513 | // ExecutionMode), or for extended instructions that may have their |
| 514 | // own operands depending on the selected extended instruction. |
| 515 | spv_operand_pattern_t expectedOperands( |
| 516 | opcodeEntry->operandTypes, |
| 517 | opcodeEntry->operandTypes + opcodeEntry->numTypes); |
Lei Zhang | dfc5008 | 2015-08-21 11:50:55 -0400 | [diff] [blame] | 518 | |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 519 | while (!expectedOperands.empty()) { |
| 520 | const spv_operand_type_t type = expectedOperands.front(); |
| 521 | expectedOperands.pop_front(); |
| 522 | |
| 523 | // Expand optional tuples lazily. |
Lei Zhang | e78a7c1 | 2015-09-10 17:07:21 -0400 | [diff] [blame] | 524 | if (spvExpandOperandSequenceOnce(type, &expectedOperands)) continue; |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 525 | |
| 526 | if (type == SPV_OPERAND_TYPE_RESULT_ID && !result_id.empty()) { |
| 527 | // Handle the <result-id> for value generating instructions. |
| 528 | // We've already consumed it from the text stream. Here |
| 529 | // we inject its words into the instruction. |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 530 | spv_position_t temp_pos = context->position(); |
| 531 | error = spvTextEncodeOperand(grammar, context, SPV_OPERAND_TYPE_RESULT_ID, |
| 532 | result_id.c_str(), pInst, nullptr); |
| 533 | result_id_position = context->position(); |
| 534 | // Because we are injecting we have to reset the position afterwards. |
| 535 | context->setPosition(temp_pos); |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 536 | if (error) return error; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 537 | } else { |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 538 | // Find the next word. |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 539 | error = context->advance(); |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 540 | if (error == SPV_END_OF_STREAM) { |
| 541 | if (spvOperandIsOptional(type)) { |
Lei Zhang | e78a7c1 | 2015-09-10 17:07:21 -0400 | [diff] [blame] | 542 | // This would have been the last potential operand for the |
| 543 | // instruction, |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 544 | // and we didn't find one. We're finished parsing this instruction. |
| 545 | break; |
| 546 | } else { |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 547 | return context->diagnostic() |
| 548 | << "Expected operand, found end of stream."; |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 549 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 550 | } |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 551 | assert(error == SPV_SUCCESS && "Somebody added another way to fail"); |
| 552 | |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 553 | if (context->isStartOfNewInst()) { |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 554 | if (spvOperandIsOptional(type)) { |
| 555 | break; |
| 556 | } else { |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 557 | return context->diagnostic() |
| 558 | << "Expected operand, found next instruction instead."; |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 559 | } |
| 560 | } |
| 561 | |
| 562 | std::string operandValue; |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 563 | error = context->getWord(operandValue, &nextPosition); |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 564 | if (error) return context->diagnostic(error) << "Internal Error"; |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 565 | |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 566 | error = spvTextEncodeOperand(grammar, context, type, operandValue.c_str(), |
| 567 | pInst, &expectedOperands); |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 568 | |
| 569 | if (error == SPV_FAILED_MATCH && spvOperandIsOptional(type)) |
| 570 | return SPV_SUCCESS; |
| 571 | |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 572 | if (error) return error; |
David Neto | 78c3b43 | 2015-08-27 13:03:52 -0400 | [diff] [blame] | 573 | |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 574 | context->setPosition(nextPosition); |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 575 | } |
| 576 | } |
| 577 | |
Andrew Woloszyn | 537e776 | 2015-09-29 11:28:34 -0400 | [diff] [blame] | 578 | if (spvOpcodeGeneratesType(pInst->opcode)) { |
| 579 | if (context->recordTypeDefinition(pInst) != SPV_SUCCESS) { |
| 580 | return SPV_ERROR_INVALID_TEXT; |
| 581 | } |
| 582 | } else if (opcodeEntry->hasType) { |
| 583 | // SPIR-V dictates that if an instruction has both a return value and a |
| 584 | // type ID then the type id is first, and the return value is second. |
| 585 | assert(opcodeEntry->hasResult && |
| 586 | "Unknown opcode: has a type but no result."); |
| 587 | context->recordTypeIdForValue(pInst->words[2], pInst->words[1]); |
| 588 | } |
| 589 | |
David Neto | b5dc8fc | 2015-10-06 16:22:00 -0400 | [diff] [blame] | 590 | if (pInst->words.size() > SPV_LIMIT_INSTRUCTION_WORD_COUNT_MAX) { |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 591 | return context->diagnostic() |
| 592 | << "Instruction too long: " << pInst->words.size() |
| 593 | << " words, but the limit is " |
| 594 | << SPV_LIMIT_INSTRUCTION_WORD_COUNT_MAX; |
David Neto | b5dc8fc | 2015-10-06 16:22:00 -0400 | [diff] [blame] | 595 | } |
| 596 | |
| 597 | pInst->words[0] = spvOpcodeMake(pInst->words.size(), opcodeEntry->opcode); |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 598 | |
| 599 | return SPV_SUCCESS; |
| 600 | } |
| 601 | |
David Neto | c978643 | 2015-09-01 18:05:14 -0400 | [diff] [blame] | 602 | namespace { |
| 603 | |
| 604 | // Translates a given assembly language module into binary form. |
| 605 | // If a diagnostic is generated, it is not yet marked as being |
| 606 | // for a text-based input. |
Dejan Mircevski | 14c4b10 | 2015-09-29 17:07:21 -0400 | [diff] [blame] | 607 | spv_result_t spvTextToBinaryInternal(const libspirv::AssemblyGrammar& grammar, |
| 608 | const spv_text text, |
| 609 | spv_assembly_syntax_format_t format, |
| 610 | spv_binary* pBinary, |
| 611 | spv_diagnostic* pDiagnostic) { |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 612 | if (!pDiagnostic) return SPV_ERROR_INVALID_DIAGNOSTIC; |
| 613 | libspirv::AssemblyContext context(text, pDiagnostic); |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 614 | if (!text->str || !text->length) |
| 615 | return context.diagnostic() << "Text stream is empty."; |
| 616 | |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 617 | if (!grammar.isValid()) { |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 618 | return SPV_ERROR_INVALID_TABLE; |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 619 | } |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 620 | if (!pBinary) return SPV_ERROR_INVALID_POINTER; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 621 | |
| 622 | // NOTE: Ensure diagnostic is zero initialised |
| 623 | *pDiagnostic = {}; |
| 624 | |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 625 | std::vector<spv_instruction_t> instructions; |
| 626 | |
David Neto | ac508b0 | 2015-10-09 15:48:09 -0400 | [diff] [blame] | 627 | if (context.advance()) return context.diagnostic() << "Text stream is empty."; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 628 | |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 629 | spv_ext_inst_type_t extInstType = SPV_EXT_INST_TYPE_NONE; |
| 630 | while (context.hasText()) { |
David Neto | b5dc8fc | 2015-10-06 16:22:00 -0400 | [diff] [blame] | 631 | instructions.push_back({}); |
| 632 | spv_instruction_t& inst = instructions.back(); |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 633 | inst.extInstType = extInstType; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 634 | |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 635 | if (spvTextEncodeOpcode(grammar, &context, format, &inst)) { |
| 636 | return SPV_ERROR_INVALID_TEXT; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 637 | } |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 638 | extInstType = inst.extInstType; |
| 639 | |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 640 | if (context.advance()) break; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 641 | } |
| 642 | |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 643 | size_t totalSize = SPV_INDEX_INSTRUCTION; |
Dejan Mircevski | 903f9d6 | 2015-09-28 17:04:39 -0400 | [diff] [blame] | 644 | for (auto& inst : instructions) { |
David Neto | b5dc8fc | 2015-10-06 16:22:00 -0400 | [diff] [blame] | 645 | totalSize += inst.words.size(); |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 646 | } |
| 647 | |
Dejan Mircevski | 903f9d6 | 2015-09-28 17:04:39 -0400 | [diff] [blame] | 648 | uint32_t* data = new uint32_t[totalSize]; |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 649 | if (!data) return SPV_ERROR_OUT_OF_MEMORY; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 650 | uint64_t currentIndex = SPV_INDEX_INSTRUCTION; |
Dejan Mircevski | 903f9d6 | 2015-09-28 17:04:39 -0400 | [diff] [blame] | 651 | for (auto& inst : instructions) { |
David Neto | b5dc8fc | 2015-10-06 16:22:00 -0400 | [diff] [blame] | 652 | memcpy(data + currentIndex, inst.words.data(), sizeof(uint32_t) * inst.words.size()); |
| 653 | currentIndex += inst.words.size(); |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 654 | } |
| 655 | |
| 656 | spv_binary binary = new spv_binary_t(); |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 657 | if (!binary) { |
| 658 | delete[] data; |
| 659 | return SPV_ERROR_OUT_OF_MEMORY; |
| 660 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 661 | binary->code = data; |
| 662 | binary->wordCount = totalSize; |
| 663 | |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 664 | spv_result_t error = spvBinaryHeaderSet(binary, context.getBound()); |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 665 | if (error) { |
| 666 | spvBinaryDestroy(binary); |
| 667 | return error; |
| 668 | } |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 669 | |
| 670 | *pBinary = binary; |
| 671 | |
| 672 | return SPV_SUCCESS; |
| 673 | } |
| 674 | |
Lei Zhang | e78a7c1 | 2015-09-10 17:07:21 -0400 | [diff] [blame] | 675 | } // anonymous namespace |
David Neto | c978643 | 2015-09-01 18:05:14 -0400 | [diff] [blame] | 676 | |
Dejan Mircevski | 903f9d6 | 2015-09-28 17:04:39 -0400 | [diff] [blame] | 677 | spv_result_t spvTextToBinary(const char* input_text, |
Andrew Woloszyn | cfeac48 | 2015-09-09 13:04:32 -0400 | [diff] [blame] | 678 | const uint64_t input_text_size, |
David Neto | c978643 | 2015-09-01 18:05:14 -0400 | [diff] [blame] | 679 | const spv_opcode_table opcodeTable, |
| 680 | const spv_operand_table operandTable, |
| 681 | const spv_ext_inst_table extInstTable, |
Dejan Mircevski | 903f9d6 | 2015-09-28 17:04:39 -0400 | [diff] [blame] | 682 | spv_binary* pBinary, spv_diagnostic* pDiagnostic) { |
Lei Zhang | 06efdc5 | 2015-09-10 14:00:00 -0400 | [diff] [blame] | 683 | return spvTextWithFormatToBinary( |
| 684 | input_text, input_text_size, SPV_ASSEMBLY_SYNTAX_FORMAT_DEFAULT, |
| 685 | opcodeTable, operandTable, extInstTable, pBinary, pDiagnostic); |
| 686 | } |
| 687 | |
| 688 | spv_result_t spvTextWithFormatToBinary( |
Dejan Mircevski | 903f9d6 | 2015-09-28 17:04:39 -0400 | [diff] [blame] | 689 | const char* input_text, const uint64_t input_text_size, |
Lei Zhang | 06efdc5 | 2015-09-10 14:00:00 -0400 | [diff] [blame] | 690 | spv_assembly_syntax_format_t format, const spv_opcode_table opcodeTable, |
| 691 | const spv_operand_table operandTable, const spv_ext_inst_table extInstTable, |
Dejan Mircevski | 903f9d6 | 2015-09-28 17:04:39 -0400 | [diff] [blame] | 692 | spv_binary* pBinary, spv_diagnostic* pDiagnostic) { |
Andrew Woloszyn | cfeac48 | 2015-09-09 13:04:32 -0400 | [diff] [blame] | 693 | spv_text_t text = {input_text, input_text_size}; |
| 694 | |
Dejan Mircevski | 14c4b10 | 2015-09-29 17:07:21 -0400 | [diff] [blame] | 695 | libspirv::AssemblyGrammar grammar(operandTable, opcodeTable, extInstTable); |
Lei Zhang | 06efdc5 | 2015-09-10 14:00:00 -0400 | [diff] [blame] | 696 | spv_result_t result = |
Andrew Woloszyn | 71fc055 | 2015-09-24 10:26:51 -0400 | [diff] [blame] | 697 | spvTextToBinaryInternal(grammar, &text, format, pBinary, pDiagnostic); |
David Neto | c978643 | 2015-09-01 18:05:14 -0400 | [diff] [blame] | 698 | if (pDiagnostic && *pDiagnostic) (*pDiagnostic)->isTextSource = true; |
| 699 | |
| 700 | return result; |
| 701 | } |
| 702 | |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 703 | void spvTextDestroy(spv_text text) { |
Lei Zhang | 4005670 | 2015-09-11 14:31:27 -0400 | [diff] [blame] | 704 | if (!text) return; |
Kenneth Benzie (Benie) | 83e5a29 | 2015-05-22 18:26:19 +0100 | [diff] [blame] | 705 | if (text->str) { |
| 706 | delete[] text->str; |
| 707 | } |
| 708 | delete text; |
| 709 | } |