author | wenzelm |
Sat, 15 Aug 2020 13:45:25 +0200 | |
changeset 72157 | d1ca82e27cbc |
parent 72152 | 3fa75db844f5 |
child 72159 | 40b5ee5889d2 |
permissions | -rw-r--r-- |
71849 | 1 |
/* Title: Pure/System/scala.scala |
43744
2c7e1565b4a3
some support to invoke Scala methods under program control;
wenzelm
parents:
diff
changeset
|
2 |
Author: Makarius |
2c7e1565b4a3
some support to invoke Scala methods under program control;
wenzelm
parents:
diff
changeset
|
3 |
|
71849 | 4 |
Support for Scala at runtime. |
43744
2c7e1565b4a3
some support to invoke Scala methods under program control;
wenzelm
parents:
diff
changeset
|
5 |
*/ |
2c7e1565b4a3
some support to invoke Scala methods under program control;
wenzelm
parents:
diff
changeset
|
6 |
|
2c7e1565b4a3
some support to invoke Scala methods under program control;
wenzelm
parents:
diff
changeset
|
7 |
package isabelle |
2c7e1565b4a3
some support to invoke Scala methods under program control;
wenzelm
parents:
diff
changeset
|
8 |
|
2c7e1565b4a3
some support to invoke Scala methods under program control;
wenzelm
parents:
diff
changeset
|
9 |
|
71864 | 10 |
import java.io.{File => JFile, StringWriter, PrintWriter} |
56730
e723f041b6d0
tuned signature -- separate pool for JFuture tasks, which can be canceled;
wenzelm
parents:
56667
diff
changeset
|
11 |
|
71868 | 12 |
import scala.tools.nsc.{GenericRunnerSettings, ConsoleWriter, NewLinePrintWriter} |
71864 | 13 |
import scala.tools.nsc.interpreter.IMain |
43744
2c7e1565b4a3
some support to invoke Scala methods under program control;
wenzelm
parents:
diff
changeset
|
14 |
|
2c7e1565b4a3
some support to invoke Scala methods under program control;
wenzelm
parents:
diff
changeset
|
15 |
|
71849 | 16 |
object Scala |
43744
2c7e1565b4a3
some support to invoke Scala methods under program control;
wenzelm
parents:
diff
changeset
|
17 |
{ |
72152 | 18 |
/** demo functions **/ |
19 |
||
20 |
def echo(arg: String): String = arg |
|
21 |
||
22 |
def sleep(seconds: String): String = |
|
23 |
{ |
|
24 |
val t = |
|
25 |
seconds match { |
|
26 |
case Value.Double(s) => Time.seconds(s) |
|
27 |
case _ => error("Malformed argument: " + quote(seconds)) |
|
28 |
} |
|
29 |
val t0 = Time.now() |
|
30 |
t.sleep |
|
31 |
val t1 = Time.now() |
|
32 |
(t1 - t0).toString |
|
33 |
} |
|
34 |
||
35 |
||
36 |
||
71870 | 37 |
/** compiler **/ |
71850 | 38 |
|
71864 | 39 |
object Compiler |
71850 | 40 |
{ |
71868 | 41 |
def context( |
42 |
error: String => Unit = Exn.error, |
|
43 |
jar_dirs: List[JFile] = Nil): Context = |
|
71864 | 44 |
{ |
45 |
def find_jars(dir: JFile): List[String] = |
|
46 |
File.find_files(dir, file => file.getName.endsWith(".jar")). |
|
47 |
map(File.absolute_name) |
|
48 |
||
49 |
val class_path = |
|
50 |
for { |
|
71882 | 51 |
prop <- List("isabelle.scala.classpath", "java.class.path") |
71864 | 52 |
path = System.getProperty(prop, "") if path != "\"\"" |
53 |
elem <- space_explode(JFile.pathSeparatorChar, path) |
|
54 |
} yield elem |
|
55 |
||
71868 | 56 |
val settings = new GenericRunnerSettings(error) |
57 |
settings.classpath.value = |
|
58 |
(class_path ::: jar_dirs.flatMap(find_jars)).mkString(JFile.pathSeparator) |
|
71850 | 59 |
|
71868 | 60 |
new Context(settings) |
71864 | 61 |
} |
71850 | 62 |
|
71868 | 63 |
def default_print_writer: PrintWriter = |
64 |
new NewLinePrintWriter(new ConsoleWriter, true) |
|
65 |
||
66 |
class Context private [Compiler](val settings: GenericRunnerSettings) |
|
71864 | 67 |
{ |
71869 | 68 |
override def toString: String = settings.toString |
69 |
||
71868 | 70 |
def interpreter( |
71 |
print_writer: PrintWriter = default_print_writer, |
|
72 |
class_loader: ClassLoader = null): IMain = |
|
73 |
{ |
|
74 |
new IMain(settings, print_writer) |
|
75 |
{ |
|
76 |
override def parentClassLoader: ClassLoader = |
|
77 |
if (class_loader == null) super.parentClassLoader |
|
78 |
else class_loader |
|
79 |
} |
|
80 |
} |
|
71850 | 81 |
|
71868 | 82 |
def toplevel(source: String): List[String] = |
83 |
{ |
|
84 |
val out = new StringWriter |
|
85 |
val interp = interpreter(new PrintWriter(out)) |
|
86 |
val rep = new interp.ReadEvalPrint |
|
87 |
val ok = interp.withLabel("\u0001") { rep.compile(source) } |
|
88 |
out.close |
|
71864 | 89 |
|
71868 | 90 |
val Error = """(?s)^\S* error: (.*)$""".r |
91 |
val errors = |
|
92 |
space_explode('\u0001', Library.strip_ansi_color(out.toString)). |
|
93 |
collect({ case Error(msg) => "Scala error: " + Library.trim_line(msg) }) |
|
94 |
||
95 |
if (!ok && errors.isEmpty) List("Error") else errors |
|
96 |
} |
|
71864 | 97 |
} |
71850 | 98 |
} |
99 |
||
71888 | 100 |
def toplevel_yxml(source: String): String = |
71872 | 101 |
{ |
71888 | 102 |
val errors = |
71889
8dbefe849666
omit pointless memoing: Scala compiler is rather bulky anyway;
wenzelm
parents:
71888
diff
changeset
|
103 |
try { Compiler.context().toplevel(source) } |
71888 | 104 |
catch { case ERROR(msg) => List(msg) } |
71872 | 105 |
locally { import XML.Encode._; YXML.string_of_body(list(string)(errors)) } |
106 |
} |
|
107 |
||
71850 | 108 |
|
109 |
||
71871
28def00726ca
more robust isabelle.Functions --- avoid Java reflection with unclear class/object treatment;
wenzelm
parents:
71870
diff
changeset
|
110 |
/** invoke Scala functions from ML **/ |
28def00726ca
more robust isabelle.Functions --- avoid Java reflection with unclear class/object treatment;
wenzelm
parents:
71870
diff
changeset
|
111 |
|
28def00726ca
more robust isabelle.Functions --- avoid Java reflection with unclear class/object treatment;
wenzelm
parents:
71870
diff
changeset
|
112 |
/* registered functions */ |
71849 | 113 |
|
71873 | 114 |
sealed case class Fun(name: String, apply: String => String) |
71881 | 115 |
{ |
116 |
override def toString: String = name |
|
117 |
} |
|
43748 | 118 |
|
71871
28def00726ca
more robust isabelle.Functions --- avoid Java reflection with unclear class/object treatment;
wenzelm
parents:
71870
diff
changeset
|
119 |
lazy val functions: List[Fun] = |
28def00726ca
more robust isabelle.Functions --- avoid Java reflection with unclear class/object treatment;
wenzelm
parents:
71870
diff
changeset
|
120 |
Isabelle_System.services.collect { case c: Isabelle_Scala_Functions => c.functions.toList }.flatten |
43748 | 121 |
|
122 |
||
71871
28def00726ca
more robust isabelle.Functions --- avoid Java reflection with unclear class/object treatment;
wenzelm
parents:
71870
diff
changeset
|
123 |
/* invoke function */ |
43748 | 124 |
|
125 |
object Tag extends Enumeration |
|
43744
2c7e1565b4a3
some support to invoke Scala methods under program control;
wenzelm
parents:
diff
changeset
|
126 |
{ |
65638 | 127 |
val NULL, OK, ERROR, FAIL, INTERRUPT = Value |
43748 | 128 |
} |
43744
2c7e1565b4a3
some support to invoke Scala methods under program control;
wenzelm
parents:
diff
changeset
|
129 |
|
71871
28def00726ca
more robust isabelle.Functions --- avoid Java reflection with unclear class/object treatment;
wenzelm
parents:
71870
diff
changeset
|
130 |
def function(name: String, arg: String): (Tag.Value, String) = |
71873 | 131 |
functions.find(fun => fun.name == name) match { |
132 |
case Some(fun) => |
|
133 |
Exn.capture { fun.apply(arg) } match { |
|
43748 | 134 |
case Exn.Res(null) => (Tag.NULL, "") |
135 |
case Exn.Res(res) => (Tag.OK, res) |
|
56667
65e84b0ef974
more abstract Exn.Interrupt and POSIX return code;
wenzelm
parents:
56387
diff
changeset
|
136 |
case Exn.Exn(Exn.Interrupt()) => (Tag.INTERRUPT, "") |
44158 | 137 |
case Exn.Exn(e) => (Tag.ERROR, Exn.message(e)) |
43748 | 138 |
} |
71871
28def00726ca
more robust isabelle.Functions --- avoid Java reflection with unclear class/object treatment;
wenzelm
parents:
71870
diff
changeset
|
139 |
case None => (Tag.FAIL, "Unknown Isabelle/Scala function: " + quote(name)) |
43748 | 140 |
} |
52111
1fd184eaa310
explicit management of Session.Protocol_Handlers, with protocol state and functions;
wenzelm
parents:
49470
diff
changeset
|
141 |
|
1fd184eaa310
explicit management of Session.Protocol_Handlers, with protocol state and functions;
wenzelm
parents:
49470
diff
changeset
|
142 |
|
72157 | 143 |
/* protocol handler */ |
144 |
||
145 |
class Handler extends Session.Protocol_Handler |
|
146 |
{ |
|
147 |
private var session: Session = null |
|
148 |
private var futures = Map.empty[String, Future[Unit]] |
|
52111
1fd184eaa310
explicit management of Session.Protocol_Handlers, with protocol state and functions;
wenzelm
parents:
49470
diff
changeset
|
149 |
|
72157 | 150 |
override def init(init_session: Session): Unit = |
151 |
synchronized { session = init_session } |
|
52111
1fd184eaa310
explicit management of Session.Protocol_Handlers, with protocol state and functions;
wenzelm
parents:
49470
diff
changeset
|
152 |
|
72157 | 153 |
override def exit(): Unit = synchronized |
154 |
{ |
|
155 |
for ((id, future) <- futures) cancel(id, future) |
|
156 |
futures = Map.empty |
|
157 |
} |
|
65219 | 158 |
|
72157 | 159 |
private def result(id: String, tag: Scala.Tag.Value, res: String): Unit = |
160 |
synchronized |
|
161 |
{ |
|
162 |
if (futures.isDefinedAt(id)) { |
|
163 |
session.protocol_command("Scala.result", id, tag.id.toString, res) |
|
164 |
futures -= id |
|
165 |
} |
|
166 |
} |
|
65220 | 167 |
|
72157 | 168 |
private def cancel(id: String, future: Future[Unit]) |
169 |
{ |
|
170 |
future.cancel |
|
171 |
result(id, Scala.Tag.INTERRUPT, "") |
|
172 |
} |
|
173 |
||
174 |
private def invoke_scala(msg: Prover.Protocol_Output): Boolean = synchronized |
|
56387 | 175 |
{ |
72157 | 176 |
msg.properties match { |
177 |
case Markup.Invoke_Scala(name, id) => |
|
178 |
futures += (id -> |
|
179 |
Future.fork { |
|
180 |
val (tag, res) = Scala.function(name, msg.text) |
|
181 |
result(id, tag, res) |
|
182 |
}) |
|
183 |
true |
|
184 |
case _ => false |
|
56387 | 185 |
} |
52111
1fd184eaa310
explicit management of Session.Protocol_Handlers, with protocol state and functions;
wenzelm
parents:
49470
diff
changeset
|
186 |
} |
1fd184eaa310
explicit management of Session.Protocol_Handlers, with protocol state and functions;
wenzelm
parents:
49470
diff
changeset
|
187 |
|
72157 | 188 |
private def cancel_scala(msg: Prover.Protocol_Output): Boolean = synchronized |
189 |
{ |
|
190 |
msg.properties match { |
|
191 |
case Markup.Cancel_Scala(id) => |
|
192 |
futures.get(id) match { |
|
193 |
case Some(future) => cancel(id, future) |
|
194 |
case None => |
|
195 |
} |
|
196 |
true |
|
197 |
case _ => false |
|
198 |
} |
|
199 |
} |
|
52111
1fd184eaa310
explicit management of Session.Protocol_Handlers, with protocol state and functions;
wenzelm
parents:
49470
diff
changeset
|
200 |
|
72157 | 201 |
val functions = |
202 |
List( |
|
203 |
Markup.Invoke_Scala.name -> invoke_scala, |
|
204 |
Markup.Cancel_Scala.name -> cancel_scala) |
|
52111
1fd184eaa310
explicit management of Session.Protocol_Handlers, with protocol state and functions;
wenzelm
parents:
49470
diff
changeset
|
205 |
} |
1fd184eaa310
explicit management of Session.Protocol_Handlers, with protocol state and functions;
wenzelm
parents:
49470
diff
changeset
|
206 |
} |
71871
28def00726ca
more robust isabelle.Functions --- avoid Java reflection with unclear class/object treatment;
wenzelm
parents:
71870
diff
changeset
|
207 |
|
28def00726ca
more robust isabelle.Functions --- avoid Java reflection with unclear class/object treatment;
wenzelm
parents:
71870
diff
changeset
|
208 |
|
28def00726ca
more robust isabelle.Functions --- avoid Java reflection with unclear class/object treatment;
wenzelm
parents:
71870
diff
changeset
|
209 |
/* registered functions */ |
28def00726ca
more robust isabelle.Functions --- avoid Java reflection with unclear class/object treatment;
wenzelm
parents:
71870
diff
changeset
|
210 |
|
28def00726ca
more robust isabelle.Functions --- avoid Java reflection with unclear class/object treatment;
wenzelm
parents:
71870
diff
changeset
|
211 |
class Isabelle_Scala_Functions(val functions: Scala.Fun*) extends Isabelle_System.Service |
28def00726ca
more robust isabelle.Functions --- avoid Java reflection with unclear class/object treatment;
wenzelm
parents:
71870
diff
changeset
|
212 |
|
28def00726ca
more robust isabelle.Functions --- avoid Java reflection with unclear class/object treatment;
wenzelm
parents:
71870
diff
changeset
|
213 |
class Functions extends Isabelle_Scala_Functions( |
72152 | 214 |
Scala.Fun("echo", Scala.echo), |
215 |
Scala.Fun("sleep", Scala.sleep), |
|
71888 | 216 |
Scala.Fun("scala_toplevel", Scala.toplevel_yxml), |
71871
28def00726ca
more robust isabelle.Functions --- avoid Java reflection with unclear class/object treatment;
wenzelm
parents:
71870
diff
changeset
|
217 |
Scala.Fun("check_bibtex_database", Bibtex.check_database_yxml)) |