1 //===- SPIRVBasicBlock.h � SPIR-V Basic Block --------------------*- C++ -*-===// 2 // 3 // The LLVM/SPIRV Translator 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 // Copyright (c) 2014 Advanced Micro Devices, Inc. All rights reserved. 9 // 10 // Permission is hereby granted, free of charge, to any person obtaining a 11 // copy of this software and associated documentation files (the "Software"), 12 // to deal with the Software without restriction, including without limitation 13 // the rights to use, copy, modify, merge, publish, distribute, sublicense, 14 // and/or sell copies of the Software, and to permit persons to whom the 15 // Software is furnished to do so, subject to the following conditions: 16 // 17 // Redistributions of source code must retain the above copyright notice, 18 // this list of conditions and the following disclaimers. 19 // Redistributions in binary form must reproduce the above copyright notice, 20 // this list of conditions and the following disclaimers in the documentation 21 // and/or other materials provided with the distribution. 22 // Neither the names of Advanced Micro Devices, Inc., nor the names of its 23 // contributors may be used to endorse or promote products derived from this 24 // Software without specific prior written permission. 25 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 26 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 27 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 28 // CONTRIBUTORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 29 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 30 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS WITH 31 // THE SOFTWARE. 32 // 33 //===----------------------------------------------------------------------===// 34 /// \file 35 /// 36 /// This file defines Basic Block class for SPIR-V. 37 /// 38 //===----------------------------------------------------------------------===// 39 40 #ifndef SPIRVBASICBLOCK_HPP_ 41 #define SPIRVBASICBLOCK_HPP_ 42 43 #include "SPIRVValue.h" 44 #include <algorithm> 45 46 namespace SPIRV{ 47 class SPIRVFunction; 48 class SPIRVInstruction; 49 class SPIRVDecoder; 50 class SPIRVBasicBlock: public SPIRVValue { 51 52 public: 53 SPIRVBasicBlock(SPIRVId TheId, SPIRVFunction *Func); 54 SPIRVBasicBlock()55 SPIRVBasicBlock():SPIRVValue(OpLabel), ParentF(NULL){ 56 setAttr(); 57 } 58 59 SPIRVDecoder getDecoder(std::istream &IS); getParent()60 SPIRVFunction *getParent() const { return ParentF;} getNumInst()61 size_t getNumInst() const { return InstVec.size();} getInst(size_t I)62 SPIRVInstruction *getInst(size_t I) const { return InstVec[I];} getPrevious(const SPIRVInstruction * I)63 SPIRVInstruction *getPrevious(const SPIRVInstruction *I) const { 64 auto Loc = find(I); 65 if (Loc == InstVec.end() || Loc == InstVec.begin()) 66 return nullptr; 67 return *(--Loc); 68 } getNext(const SPIRVInstruction * I)69 SPIRVInstruction *getNext(const SPIRVInstruction *I) const { 70 auto Loc = find(I); 71 if (Loc == InstVec.end()) 72 return nullptr; 73 ++Loc; 74 if (Loc == InstVec.end()) 75 return nullptr; 76 return *Loc; 77 } 78 79 void setScope(SPIRVEntry *Scope); setParent(SPIRVFunction * F)80 void setParent(SPIRVFunction *F) { ParentF = F;} 81 SPIRVInstruction *addInstruction(SPIRVInstruction *I); 82 setAttr()83 void setAttr() { setHasNoType();} 84 _SPIRV_DCL_ENCDEC 85 void encodeChildren(spv_ostream &)const; validate()86 void validate()const { 87 SPIRVValue::validate(); 88 assert(ParentF && "Invalid parent function"); 89 } 90 91 private: 92 SPIRVFunction *ParentF; 93 typedef std::vector<SPIRVInstruction *> SPIRVInstructionVector; 94 SPIRVInstructionVector InstVec; 95 find(const SPIRVInstruction * Inst)96 SPIRVInstructionVector::const_iterator find(const SPIRVInstruction *Inst) 97 const { 98 return std::find(InstVec.begin(), InstVec.end(), Inst); 99 } 100 }; 101 102 typedef SPIRVBasicBlock SPIRVLabel; 103 } 104 105 #endif 106