| 1 | // Copyright (c) 2017 Google Inc. |
| 2 | // |
| 3 | // Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | // you may not use this file except in compliance with the License. |
| 5 | // You may obtain a copy of the License at |
| 6 | // |
| 7 | // http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | // |
| 9 | // Unless required by applicable law or agreed to in writing, software |
| 10 | // distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | // See the License for the specific language governing permissions and |
| 13 | // limitations under the License. |
| 14 | |
| 15 | #ifndef SOURCE_OPT_DEAD_VARIABLE_ELIMINATION_H_ |
| 16 | #define SOURCE_OPT_DEAD_VARIABLE_ELIMINATION_H_ |
| 17 | |
| 18 | #include <climits> |
| 19 | #include <unordered_map> |
| 20 | |
| 21 | #include "source/opt/decoration_manager.h" |
| 22 | #include "source/opt/mem_pass.h" |
| 23 | |
| 24 | namespace spvtools { |
| 25 | namespace opt { |
| 26 | |
| 27 | class DeadVariableElimination : public MemPass { |
| 28 | public: |
| 29 | const char* name() const override { return "eliminate-dead-variables" ; } |
| 30 | Status Process() override; |
| 31 | |
| 32 | IRContext::Analysis GetPreservedAnalyses() override { |
| 33 | return IRContext::kAnalysisDefUse | IRContext::kAnalysisConstants | |
| 34 | IRContext::kAnalysisTypes; |
| 35 | } |
| 36 | |
| 37 | private: |
| 38 | // Deletes the OpVariable instruction who result id is |result_id|. |
| 39 | void DeleteVariable(uint32_t result_id); |
| 40 | |
| 41 | // Keeps track of the number of references of an id. Once that value is 0, it |
| 42 | // is safe to remove the corresponding instruction. |
| 43 | // |
| 44 | // Note that the special value kMustKeep is used to indicate that the |
| 45 | // instruction cannot be deleted for reasons other that is being explicitly |
| 46 | // referenced. |
| 47 | std::unordered_map<uint32_t, size_t> reference_count_; |
| 48 | |
| 49 | // Special value used to indicate that an id cannot be safely deleted. |
| 50 | enum { kMustKeep = INT_MAX }; |
| 51 | }; |
| 52 | |
| 53 | } // namespace opt |
| 54 | } // namespace spvtools |
| 55 | |
| 56 | #endif // SOURCE_OPT_DEAD_VARIABLE_ELIMINATION_H_ |
| 57 | |