public DestructifyReductionOpportunity(DeclarationStmt declaration, TranslationUnit tu, BlockStmt block, VisitationDepth depth) { super(depth); assert declaration.getVariablesDeclaration().getNumDecls() == 1; this.declaration = declaration; this.tu = tu; this.block = block; this.originalNumVariables = declaration.getVariablesDeclaration().getNumDecls(); }
@Override void applyReductionImpl() { if (!variablesDeclaration.getDeclInfos().contains(variableDeclInfo)) { // The declaration must have been removed by another reduction opportunity return; } for (int i = 0; i < variablesDeclaration.getNumDecls(); i++) { if (variablesDeclaration.getDeclInfo(i) == variableDeclInfo) { variablesDeclaration.removeDeclInfo(i); return; } } throw new RuntimeException("Should be unreachable."); }
private boolean isEmptyAndUnreferencedDeclaration(DeclarationStmt stmt) { if (stmt.getVariablesDeclaration().getNumDecls() != 0) { return false; } return !StructUtils.declaresReferencedStruct(tu, stmt.getVariablesDeclaration()); }
static boolean isStructifiedDeclaration(DeclarationStmt declarationStmt) { return declarationStmt.getVariablesDeclaration().getNumDecls() == 1 && declarationStmt.getVariablesDeclaration().getDeclInfo(0).getName() .startsWith(Constants.STRUCTIFICATION_STRUCT_PREFIX); }
private Optional<String> extractLoopCounter(ForStmt loop) { if (!(loop.getInit() instanceof DeclarationStmt)) { return Optional.empty(); } final DeclarationStmt init = (DeclarationStmt) loop.getInit(); if (init.getVariablesDeclaration().getNumDecls() != 1) { return Optional.empty(); } return Optional.of(init.getVariablesDeclaration().getDeclInfo(0).getName()); }
private void sweep(TranslationUnit tu) { final List<Declaration> oldTopLevelDecls = new ArrayList<>(); oldTopLevelDecls.addAll(tu.getTopLevelDeclarations()); for (Declaration decl : oldTopLevelDecls) { if (!(decl instanceof VariablesDeclaration)) { continue; } final VariablesDeclaration variablesDeclaration = (VariablesDeclaration) decl; int index = 0; while (index < variablesDeclaration.getNumDecls()) { if (unusedGlobals.contains(variablesDeclaration.getDeclInfo(index))) { variablesDeclaration.removeDeclInfo(index); } else { index++; } } if (variablesDeclaration.getNumDecls() == 0 && !isUsedStructType(variablesDeclaration.getBaseType())) { tu.removeTopLevelDeclaration(variablesDeclaration); } } }
@Override public void visitVariablesDeclaration(VariablesDeclaration variablesDeclaration) { if (variablesDeclaration.getNumDecls() == 0) { if (!StructUtils.declaresReferencedStruct(tu, variablesDeclaration)) { addOpportunity(new GlobalVariablesDeclarationReductionOpportunity( tu, variablesDeclaration, getVistitationDepth())); } } }
public StructificationOpportunity(DeclarationStmt declToTransform, BlockStmt block, TranslationUnit tu, ShadingLanguageVersion shadingLanguageVersion) { assert declToTransform.getVariablesDeclaration().getNumDecls() == 1 : "Only solo declarations can be structified"; assert !declToTransform.getVariablesDeclaration().getDeclInfo(0).hasArrayInfo() : "Structification of arrays not supported"; this.declToTransform = declToTransform; this.block = block; this.tu = tu; this.shadingLanguageVersion = shadingLanguageVersion; }
private void addGlobalInitializers(FunctionDefinition mainFunction) { assert mainFunction.getPrototype().getName().equals("main"); for (int i = globalsToBeReInitialized.size() - 1; i >= 0; i--) { for (int j = globalsToBeReInitialized.get(i).getNumDecls() - 1; j >= 0; j--) { final VariableDeclInfo vdi = globalsToBeReInitialized.get(i).getDeclInfo(j); if (!(vdi.getInitializer() instanceof ScalarInitializer)) { throw new RuntimeException("Only know how to deal with scalar initializers at present."); } mainFunction.getBody().insertStmt(0, new ExprStmt(new BinaryExpr(new VariableIdentifierExpr(vdi.getName()), ((ScalarInitializer) vdi.getInitializer()).getExpr(), BinOp.ASSIGN))); vdi.setInitializer(null); } } }
&& ((VariablesDeclaration) decl).getBaseType().hasQualifier(TypeQualifier.UNIFORM)) { final VariablesDeclaration variablesDeclaration = (VariablesDeclaration) decl; for (int i = 0; i < variablesDeclaration.getNumDecls(); i++) { final VariableDeclInfo declInfo = variablesDeclaration.getDeclInfo(i); if (!declInfo.getName().equals(uniformName)) { break; if (variablesDeclaration.getNumDecls() > 0) {
if (initDeclaration.getVariablesDeclaration().getNumDecls() != 1) { return Optional.empty();
private boolean isLoopLimiterBlock(Stmt stmt) { // Identifies when a block starts with a loop-limiter declaration, in which case the whole // block can go. We are really careful about otherwise removing loop-limiters, so this is // the chance to do it! if (!(stmt instanceof BlockStmt)) { return false; } final BlockStmt blockStmt = (BlockStmt) stmt; if (blockStmt.getNumStmts() == 0) { return false; } final Stmt firstStmt = blockStmt.getStmt(0); if (!(firstStmt instanceof DeclarationStmt)) { return false; } final DeclarationStmt declarationStmt = (DeclarationStmt) firstStmt; if (declarationStmt.getVariablesDeclaration().getNumDecls() == 0) { return false; } return isLooplimiter(declarationStmt.getVariablesDeclaration().getDeclInfo(0).getName()); }
.getExpr())).getRhs(); } else if (init instanceof DeclarationStmt && ((DeclarationStmt) init).getVariablesDeclaration().getNumDecls() == 1 && ((DeclarationStmt) init).getVariablesDeclaration().getDeclInfo(0) .getInitializer() instanceof ScalarInitializer) {
@Override public void visitDeclarationStmt(DeclarationStmt declarationStmt) { super.visitDeclarationStmt(declarationStmt); // Currently we only structify solo declarations if (declarationStmt.getVariablesDeclaration().getNumDecls() != 1) { return; } // We don't currently structify arrays if (declarationStmt.getVariablesDeclaration().getDeclInfo(0).hasArrayInfo()) { return; } final Type baseType = declarationStmt.getVariablesDeclaration().getBaseType(); if (hasQualifiers(baseType)) { return; } // TODO: For simplicity, at present we do not structify non-basic types. The issue is that // if a struct S is to be structified, we need to declare the structs that enclose S *after* // S is declared, which is a bit fiddly (currently they all go at the top of the translation // unit). if (!(baseType.getWithoutQualifiers() instanceof BasicType)) { return; } opportunities.add(new StructificationOpportunity(declarationStmt, currentBlock(), tu, shadingLanguageVersion)); }
&& ((VariablesDeclaration) decl).getBaseType().hasQualifier(TypeQualifier.UNIFORM)) { final VariablesDeclaration variablesDeclaration = (VariablesDeclaration) decl; if (variablesDeclaration.getNumDecls() == 0) { assert variablesDeclaration.getNumDecls() == 1;
@Override public void applyReductionImpl() { if (originalNumVariables != declaration.getVariablesDeclaration().getNumDecls()) { // Something else changed how many declarations there are, so bail out. return; } final StructifiedVariableInfo originalVariableInfo = findOriginalVariableInfo(); // First, replace all occurrences in the translation unit deStructify(originalVariableInfo); // Now change the declaration declaration.getVariablesDeclaration().getDeclInfo(0).setName(originalVariableInfo.getName()); declaration.getVariablesDeclaration().setBaseType(originalVariableInfo.getType()); declaration.getVariablesDeclaration().getDeclInfo(0).setInitializer(originalVariableInfo .getInitializer().orElse(null)); }