1 /* 2 * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. Oracle designates this 8 * particular file as subject to the "Classpath" exception as provided 9 * by Oracle in the LICENSE file that accompanied this code. 10 * 11 * This code is distributed in the hope that it will be useful, but WITHOUT 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 14 * version 2 for more details (a copy is included in the LICENSE file that 15 * accompanied this code). 16 * 17 * You should have received a copy of the GNU General Public License version 18 * 2 along with this work; if not, write to the Free Software Foundation, 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 20 * 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 22 * or visit www.oracle.com if you need additional information or have any 23 * questions. 24 */ 25 26 package jdk.internal.clang; 27 28 import java.foreign.NativeTypes; 29 import java.foreign.Scope; 30 import java.foreign.memory.LayoutType; 31 import java.foreign.memory.Pointer; 32 import java.io.IOException; 33 import java.nio.file.Path; 34 35 import clang.CXString_h.CXString; 36 import clang.Index_h.CXDiagnostic; 37 import clang.Index_h.CXToken; 38 import clang.Index_h.CXTokenKind; 39 import clang.Index_h.CXTranslationUnitImpl; 40 41 public class TranslationUnit { 42 private final Pointer<CXTranslationUnitImpl> tu; 43 private final Scope scope = Scope.globalScope().fork(); 44 45 TranslationUnit(Pointer<CXTranslationUnitImpl> tu) { 46 this.tu = tu; 47 } 48 49 public Cursor getCursor() { 50 return new Cursor(LibClang.lib.clang_getTranslationUnitCursor(tu)); 51 } 52 53 public Diagnostic[] getDiagnostics() { 54 final clang.Index_h lclang = LibClang.lib; 55 56 int cntDiags = lclang.clang_getNumDiagnostics(tu); 57 if (cntDiags == 0) { 58 return null; 59 } 60 61 Diagnostic[] rv = new Diagnostic[cntDiags]; 62 for (int i = 0; i < cntDiags; i++) { 63 @CXDiagnostic Pointer<Void> diag = lclang.clang_getDiagnostic(tu, i); 64 rv[i] = new Diagnostic(diag); 65 } 66 67 return rv; 68 } 69 70 public final void save(Path path) throws TranslationUnit.TranslationUnitSaveException { 71 try (Scope sc = Scope.globalScope().fork()) { 72 int res = LibClang.lib.clang_saveTranslationUnit(tu, 73 sc.allocateCString(path.toAbsolutePath().toString()), 0); 74 if (res != 0) { 75 throw new TranslationUnit.TranslationUnitSaveException(path); 76 } 77 } 78 } 79 80 public String[] tokens(SourceRange range) { 81 Tokens tokens = tokenize(range); 82 String rv[] = new String[tokens.size()]; 83 for (int i = 0; i < rv.length; i++) { 84 rv[i] = tokens.getToken(i).spelling(); 85 } 86 return rv; 87 } 88 89 public Tokens tokenize(SourceRange range) { 90 // FIXME: Use a shared scope now, we really want to use is StackScope 91 Pointer<Pointer<CXToken>> p = scope.allocate(LayoutType.ofStruct(CXToken.class).pointer()); 92 Pointer<Integer> pCnt = scope.allocate(NativeTypes.UINT); 93 LibClang.lib.clang_tokenize(tu, range.range, p, pCnt); 94 Tokens rv = new Tokens(p.get().cast(LayoutType.ofStruct(CXToken.class)), pCnt.get()); 95 return rv; 96 } 97 98 public void dispose() { 99 LibClang.lib.clang_disposeTranslationUnit(tu); 100 } 101 102 public class Tokens { 103 private final Pointer<CXToken> ar; 104 private final int size; 105 106 Tokens(Pointer<CXToken> ar, int size) { 107 this.ar = ar; 108 this.size = size; 109 } 110 111 public void dispose() { 112 LibClang.lib.clang_disposeTokens(tu, ar, size); 113 } 114 115 public int size() { 116 return size; 117 } 118 119 public Token getToken(int idx) { 120 Pointer<CXToken> p = ar.offset(idx); 121 return new Token(p.get()); 122 } 123 124 @Override 125 public String toString() { 126 StringBuilder sb = new StringBuilder(); 127 for (int i = 0; i < size; i++) { 128 CXString s = LibClang.lib.clang_getTokenSpelling(tu, ar.offset(i).get()); 129 sb.append("Token["); 130 sb.append(i); 131 sb.append("]="); 132 sb.append(LibClang.CXStrToString(s)); 133 sb.append("\n"); 134 } 135 return sb.toString(); 136 } 137 } 138 139 public class Token { 140 final CXToken token; 141 142 Token(CXToken token) { 143 this.token = token; 144 } 145 146 public @CXTokenKind int kind() { 147 return LibClang.lib.clang_getTokenKind(token); 148 } 149 150 public String spelling() { 151 CXString s = LibClang.lib.clang_getTokenSpelling( 152 tu, token); 153 return LibClang.CXStrToString(s); 154 } 155 156 public SourceLocation getLocation() { 157 return new SourceLocation(LibClang.lib.clang_getTokenLocation( 158 tu, token)); 159 } 160 161 public SourceRange getExtent() { 162 return new SourceRange(LibClang.lib.clang_getTokenExtent( 163 tu, token)); 164 } 165 } 166 167 public static class TranslationUnitSaveException extends IOException { 168 169 static final long serialVersionUID = 1L; 170 171 TranslationUnitSaveException(Path path) { 172 super("Cannot save translation unit to: " + path.toAbsolutePath()); 173 } 174 } 175 }