@Override
protected ClassBytes[] getByteCode(final ClassNames className, final String sourcecode, boolean debug)
    throws CompileException, IOException, ClassNotFoundException, ClassTransformationException {
  StringReader reader = new StringReader(sourcecode);
  Scanner scanner = new Scanner((String) null, reader);
  Java.CompilationUnit compilationUnit = new Parser(scanner).parseCompilationUnit();
  ClassFile[] classFiles = new UnitCompiler(compilationUnit, compilationClassLoader)
                                .compileUnit(debug, debug, debug);

  ClassBytes[] byteCodes = new ClassBytes[classFiles.length];
  for(int i = 0; i < classFiles.length; i++){
    ClassFile file = classFiles[i];
    byteCodes[i] = new ClassBytes(file.getThisClassName(), file.toByteArray());
  }
  return byteCodes;
}
 

private CompilationUnit get(Class<?> c) throws IOException {
    URL u = getSourceURL(c);
    try (Reader reader = Resources.asCharSource(u, UTF_8).openStream()) {
      String body = CharStreams.toString(reader);

      // TODO: Hack to remove annotations so Janino doesn't choke. Need to reconsider this problem...
      body = body.replaceAll("@\\w+(?:\\([^\\\\]*?\\))?", "");
      for(Replacement r : REPLACERS){
        body = r.apply(body);
      }
//       System.out.println("original");
      // System.out.println(body);;
      // System.out.println("decompiled");
      // System.out.println(decompile(c));

      try {
        return new Parser(new Scanner(null, new StringReader(body))).parseCompilationUnit();
      } catch (CompileException e) {
        logger.warn("Failure while parsing function class:\n{}", body, e);
        return null;
      }

    }

  }
 

private ClassFile[] doCompile(final String sourceCode)
    throws CompileException, IOException, ClassNotFoundException {
  StringReader reader = new StringReader(sourceCode);
  Scanner scanner = new Scanner((String) null, reader);
  Java.CompilationUnit compilationUnit = new Parser(scanner).parseCompilationUnit();
  return new UnitCompiler(compilationUnit, compilationClassLoader)
                                .compileUnit(this.debug, this.debug, this.debug);
}
 

/**
 * Using class name generates path to class source code (*.java),
 * reads its content as string and parses it into {@link org.codehaus.janino.Java.CompilationUnit}.
 *
 * @param clazz function class
 * @return compilation unit
 * @throws IOException if did not find class or could not load it
 */
@VisibleForTesting
CompilationUnit convertToCompilationUnit(Class<?> clazz) throws IOException {
  String path = clazz.getName();
  path = path.replaceFirst("\\$.*", "");
  path = path.replace(".", DrillFileUtils.SEPARATOR);
  path = "/" + path + ".java";

  logger.trace("Loading function code from the {}", path);
  try (InputStream is = clazz.getResourceAsStream(path)) {
    if (is == null) {
      throw new IOException(String.format(
          "Failure trying to locate source code for class %s, tried to read on classpath location %s", clazz.getName(),
          path));
    }
    String body = IO.toString(is);

    // TODO: Hack to remove annotations so Janino doesn't choke. Need to reconsider this problem...
    body = body.replaceAll("@\\w+(?:\\([^\\\\]*?\\))?", "");
    try {
      return new Parser(new Scanner(null, new StringReader(body))).parseCompilationUnit();
    } catch (CompileException e) {
        throw new IOException(String.format("Failure while loading class %s.", clazz.getName()), e);
    }

  }

}
 

public byte[] compile(String name, String source) throws Exception {
    Parser p = new Parser(new Scanner(name, new StringReader(source)));
    UnitCompiler uc = new UnitCompiler(p.parseCompilationUnit(), CL);
    return uc.compileUnit(DebuggingInformation.ALL)[0].toByteArray();
}
 

public byte[] compile(String name, String source) throws Exception {
    Parser p = new Parser(new Scanner(name, new StringReader(source)));
    UnitCompiler uc = new UnitCompiler(p.parseCompilationUnit(), CL);
    return uc.compileUnit(DebuggingInformation.ALL)[0].toByteArray();
}