author | wenzelm |
Thu, 10 Dec 2009 16:11:07 +0100 | |
changeset 34047 | 2af94d45597f |
parent 34046 | 8e743ca417b9 |
child 34108 | 54d48ca8708f |
permissions | -rw-r--r-- |
27931 | 1 |
/* Title: Pure/General/xml.scala |
2 |
Author: Makarius |
|
3 |
||
27947 | 4 |
Simple XML tree values. |
27931 | 5 |
*/ |
6 |
||
7 |
package isabelle |
|
8 |
||
27947 | 9 |
import org.w3c.dom.{Node, Document} |
10 |
import javax.xml.parsers.DocumentBuilderFactory |
|
11 |
||
12 |
||
29203 | 13 |
object XML |
14 |
{ |
|
27947 | 15 |
/* datatype representation */ |
16 |
||
27931 | 17 |
type Attributes = List[(String, String)] |
18 |
||
34046 | 19 |
sealed abstract class Tree { |
29204 | 20 |
override def toString = { |
21 |
val s = new StringBuilder |
|
22 |
append_tree(this, s) |
|
23 |
s.toString |
|
24 |
} |
|
29203 | 25 |
} |
29204 | 26 |
case class Elem(name: String, attributes: Attributes, body: List[Tree]) extends Tree |
27 |
case class Text(content: String) extends Tree |
|
29203 | 28 |
|
33999 | 29 |
def elem(name: String, body: List[Tree]) = Elem(name, Nil, body) |
30 |
def elem(name: String) = Elem(name, Nil, Nil) |
|
31 |
||
29203 | 32 |
|
33 |
/* string representation */ |
|
34 |
||
35 |
private def append_text(text: String, s: StringBuilder) { |
|
34005 | 36 |
if (text == null) s ++ text |
37 |
else { |
|
38 |
for (c <- text.elements) c match { |
|
39 |
case '<' => s ++ "<" |
|
40 |
case '>' => s ++ ">" |
|
41 |
case '&' => s ++ "&" |
|
42 |
case '"' => s ++ """ |
|
43 |
case '\'' => s ++ "'" |
|
44 |
case _ => s + c |
|
45 |
} |
|
29203 | 46 |
} |
47 |
} |
|
48 |
||
49 |
private def append_elem(name: String, atts: Attributes, s: StringBuilder) { |
|
34005 | 50 |
s ++ name |
29203 | 51 |
for ((a, x) <- atts) { |
34005 | 52 |
s ++ " "; s ++ a; s ++ "=\""; append_text(x, s); s ++ "\"" |
29203 | 53 |
} |
54 |
} |
|
55 |
||
29204 | 56 |
private def append_tree(tree: Tree, s: StringBuilder) { |
29203 | 57 |
tree match { |
58 |
case Elem(name, atts, Nil) => |
|
34005 | 59 |
s ++ "<"; append_elem(name, atts, s); s ++ "/>" |
29203 | 60 |
case Elem(name, atts, ts) => |
34005 | 61 |
s ++ "<"; append_elem(name, atts, s); s ++ ">" |
29203 | 62 |
for (t <- ts) append_tree(t, s) |
34005 | 63 |
s ++ "</"; s ++ name; s ++ ">" |
29203 | 64 |
case Text(text) => append_text(text, s) |
65 |
} |
|
66 |
} |
|
27941 | 67 |
|
68 |
||
27942 | 69 |
/* iterate over content */ |
27941 | 70 |
|
71 |
private type State = Option[(String, List[Tree])] |
|
72 |
||
73 |
private def get_next(tree: Tree): State = tree match { |
|
74 |
case Elem(_, _, body) => get_nexts(body) |
|
75 |
case Text(content) => Some(content, Nil) |
|
76 |
} |
|
77 |
private def get_nexts(trees: List[Tree]): State = trees match { |
|
78 |
case Nil => None |
|
79 |
case t :: ts => get_next(t) match { |
|
80 |
case None => get_nexts(ts) |
|
28007 | 81 |
case Some((s, r)) => Some((s, r ++ ts)) |
27941 | 82 |
} |
83 |
} |
|
84 |
||
85 |
def content(tree: Tree) = new Iterator[String] { |
|
86 |
private var state = get_next(tree) |
|
87 |
def hasNext() = state.isDefined |
|
88 |
def next() = state match { |
|
89 |
case Some((s, rest)) => { state = get_nexts(rest); s } |
|
90 |
case None => throw new NoSuchElementException("next on empty iterator") |
|
91 |
} |
|
92 |
} |
|
93 |
||
27947 | 94 |
|
33953 | 95 |
/* document object model (W3C DOM) */ |
27948
2638b611d3ce
renamed DOM to document, add xml version and optional stylesheets;
wenzelm
parents:
27947
diff
changeset
|
96 |
|
34047 | 97 |
def get_data(node: Node): Option[XML.Tree] = |
98 |
node.getUserData(Markup.DATA) match { |
|
99 |
case tree: XML.Tree => Some(tree) |
|
100 |
case _ => None |
|
101 |
} |
|
102 |
||
33953 | 103 |
def document_node(doc: Document, tree: Tree): Node = |
104 |
{ |
|
27952 | 105 |
def DOM(tr: Tree): Node = tr match { |
34046 | 106 |
case Elem(Markup.DATA, Nil, List(data, t)) => |
107 |
val node = DOM(t) |
|
108 |
node.setUserData(Markup.DATA, data, null) |
|
109 |
node |
|
110 |
case Elem(name, atts, ts) => |
|
111 |
if (name == Markup.DATA) |
|
112 |
error("Malformed data element: " + tr.toString) |
|
27947 | 113 |
val node = doc.createElement(name) |
114 |
for ((name, value) <- atts) node.setAttribute(name, value) |
|
27952 | 115 |
for (t <- ts) node.appendChild(DOM(t)) |
27947 | 116 |
node |
117 |
case Text(txt) => doc.createTextNode(txt) |
|
118 |
} |
|
33953 | 119 |
DOM(tree) |
120 |
} |
|
121 |
||
122 |
def document(tree: Tree, styles: String*): Document = |
|
123 |
{ |
|
124 |
val doc = DocumentBuilderFactory.newInstance.newDocumentBuilder.newDocument |
|
125 |
doc.appendChild(doc.createProcessingInstruction("xml", "version=\"1.0\"")) |
|
126 |
||
127 |
for (style <- styles) { |
|
128 |
doc.appendChild(doc.createProcessingInstruction("xml-stylesheet", |
|
129 |
"href=\"" + style + "\" type=\"text/css\"")) |
|
130 |
} |
|
27947 | 131 |
val root_elem = tree match { |
33953 | 132 |
case Elem(_, _, _) => document_node(doc, tree) |
133 |
case Text(_) => document_node(doc, (Elem(Markup.ROOT, Nil, List(tree)))) |
|
27947 | 134 |
} |
135 |
doc.appendChild(root_elem) |
|
136 |
doc |
|
137 |
} |
|
27931 | 138 |
} |