| author | wenzelm | 
| Mon, 02 Jan 2017 11:42:15 +0100 | |
| changeset 64748 | 155bf8632104 | 
| parent 62820 | 5c678ee5d34a | 
| child 65130 | 695930882487 | 
| permissions | -rw-r--r-- | 
| 36683 | 1 | /* Title: Pure/General/pretty.scala | 
| 2 | Author: Makarius | |
| 3 | ||
| 36687 | 4 | Generic pretty printing module. | 
| 36683 | 5 | */ | 
| 6 | ||
| 7 | package isabelle | |
| 8 | ||
| 9 | ||
| 10 | object Pretty | |
| 11 | {
 | |
| 61871 | 12 | /* XML constructors */ | 
| 13 | ||
| 61874 | 14 | val space: XML.Body = List(XML.Text(Symbol.space)) | 
| 15 | def spaces(n: Int): XML.Body = | |
| 16 | if (n == 0) Nil | |
| 17 | else if (n == 1) space | |
| 18 | else List(XML.Text(Symbol.spaces(n))) | |
| 48704 
85a3de10567d
tuned signature -- make Pretty less dependent on Symbol;
 wenzelm parents: 
45666diff
changeset | 19 | |
| 61871 | 20 | def block(consistent: Boolean, indent: Int, body: XML.Body): XML.Tree = | 
| 21 | XML.Elem(Markup.Block(consistent, indent), body) | |
| 22 | def block(indent: Int, body: XML.Body): XML.Tree = block(false, indent, body) | |
| 23 | def block(body: XML.Body): XML.Tree = block(2, body) | |
| 48704 
85a3de10567d
tuned signature -- make Pretty less dependent on Symbol;
 wenzelm parents: 
45666diff
changeset | 24 | |
| 61871 | 25 | def brk(width: Int, indent: Int = 0): XML.Tree = | 
| 26 | XML.Elem(Markup.Break(width, indent), spaces(width)) | |
| 27 | ||
| 28 |   val fbrk: XML.Tree = XML.Text("\n")
 | |
| 29 | ||
| 30 | val Separator: XML.Body = List(XML.elem(Markup.SEPARATOR, space), fbrk) | |
| 31 | def separate(ts: List[XML.Tree]): XML.Body = Library.separate(Separator, ts.map(List(_))).flatten | |
| 61864 | 32 | |
| 48704 
85a3de10567d
tuned signature -- make Pretty less dependent on Symbol;
 wenzelm parents: 
45666diff
changeset | 33 | |
| 51492 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 34 | /* text metric -- standardized to width of space */ | 
| 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 35 | |
| 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 36 | abstract class Metric | 
| 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 37 |   {
 | 
| 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 38 | val unit: Double | 
| 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 39 | def apply(s: String): Double | 
| 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 40 | } | 
| 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 41 | |
| 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 42 | object Metric_Default extends Metric | 
| 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 43 |   {
 | 
| 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 44 | val unit = 1.0 | 
| 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 45 | def apply(s: String): Double = s.length.toDouble | 
| 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 46 | } | 
| 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 47 | |
| 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 48 | |
| 36687 | 49 | /* markup trees with physical blocks and breaks */ | 
| 50 | ||
| 62820 | 51 | private def force_nat(i: Int): Int = i max 0 | 
| 52 | ||
| 61874 | 53 |   private sealed abstract class Tree { def length: Double }
 | 
| 54 | private case class Block( | |
| 61871 | 55 | markup: Option[(Markup, Option[XML.Body])], | 
| 56 | consistent: Boolean, indent: Int, body: List[Tree], length: Double) extends Tree | |
| 61874 | 57 | private case class Break(force: Boolean, width: Int, indent: Int) extends Tree | 
| 61871 | 58 |   { def length: Double = width.toDouble }
 | 
| 61874 | 59 | private case class Str(string: String, length: Double) extends Tree | 
| 36683 | 60 | |
| 61874 | 61 | private val FBreak = Break(true, 1, 0) | 
| 36683 | 62 | |
| 61874 | 63 | private def make_block( | 
| 61871 | 64 | markup: Option[(Markup, Option[XML.Body])], | 
| 65 | consistent: Boolean, | |
| 66 | indent: Int, | |
| 67 | body: List[Tree]): Tree = | |
| 61883 
c0f34fe6aa61
clarified length of block with pre-existant forced breaks;
 wenzelm parents: 
61875diff
changeset | 68 |   {
 | 
| 62785 
70b9c7d4ed7f
more robust pretty printing: permissive treatment of bad values;
 wenzelm parents: 
61883diff
changeset | 69 | val indent1 = force_nat(indent) | 
| 
70b9c7d4ed7f
more robust pretty printing: permissive treatment of bad values;
 wenzelm parents: 
61883diff
changeset | 70 | |
| 61883 
c0f34fe6aa61
clarified length of block with pre-existant forced breaks;
 wenzelm parents: 
61875diff
changeset | 71 | def body_length(prts: List[Tree], len: Double): Double = | 
| 
c0f34fe6aa61
clarified length of block with pre-existant forced breaks;
 wenzelm parents: 
61875diff
changeset | 72 |     {
 | 
| 
c0f34fe6aa61
clarified length of block with pre-existant forced breaks;
 wenzelm parents: 
61875diff
changeset | 73 | val (line, rest) = | 
| 
c0f34fe6aa61
clarified length of block with pre-existant forced breaks;
 wenzelm parents: 
61875diff
changeset | 74 |         Library.take_prefix[Tree]({ case Break(true, _, _) => false case _ => true }, prts)
 | 
| 
c0f34fe6aa61
clarified length of block with pre-existant forced breaks;
 wenzelm parents: 
61875diff
changeset | 75 |       val len1 = ((0.0 /: line) { case (l, t) => l + t.length }) max len
 | 
| 
c0f34fe6aa61
clarified length of block with pre-existant forced breaks;
 wenzelm parents: 
61875diff
changeset | 76 |       rest match {
 | 
| 
c0f34fe6aa61
clarified length of block with pre-existant forced breaks;
 wenzelm parents: 
61875diff
changeset | 77 | case Break(true, _, ind) :: rest1 => | 
| 62785 
70b9c7d4ed7f
more robust pretty printing: permissive treatment of bad values;
 wenzelm parents: 
61883diff
changeset | 78 | body_length(Break(false, indent1 + ind, 0) :: rest1, len1) | 
| 61883 
c0f34fe6aa61
clarified length of block with pre-existant forced breaks;
 wenzelm parents: 
61875diff
changeset | 79 | case Nil => len1 | 
| 
c0f34fe6aa61
clarified length of block with pre-existant forced breaks;
 wenzelm parents: 
61875diff
changeset | 80 | } | 
| 
c0f34fe6aa61
clarified length of block with pre-existant forced breaks;
 wenzelm parents: 
61875diff
changeset | 81 | } | 
| 62785 
70b9c7d4ed7f
more robust pretty printing: permissive treatment of bad values;
 wenzelm parents: 
61883diff
changeset | 82 | Block(markup, consistent, indent1, body, body_length(body, 0.0)) | 
| 61883 
c0f34fe6aa61
clarified length of block with pre-existant forced breaks;
 wenzelm parents: 
61875diff
changeset | 83 | } | 
| 36683 | 84 | |
| 61874 | 85 | |
| 86 | /* formatted output */ | |
| 87 | ||
| 88 | private sealed case class Text(tx: XML.Body = Nil, pos: Double = 0.0, nl: Int = 0) | |
| 89 |   {
 | |
| 90 | def newline: Text = copy(tx = fbrk :: tx, pos = 0.0, nl = nl + 1) | |
| 91 | def string(s: String, len: Double): Text = | |
| 92 | copy(tx = if (s == "") tx else XML.Text(s) :: tx, pos = pos + len) | |
| 93 | def blanks(wd: Int): Text = string(Symbol.spaces(wd), wd.toDouble) | |
| 94 | def content: XML.Body = tx.reverse | |
| 95 | } | |
| 96 | ||
| 97 | private def break_dist(trees: List[Tree], after: Double): Double = | |
| 98 |     trees match {
 | |
| 99 | case (_: Break) :: _ => 0.0 | |
| 100 | case t :: ts => t.length + break_dist(ts, after) | |
| 101 | case Nil => after | |
| 36689 
379f5b1e7f91
replaced slightly odd fbreak markup by plain "\n", which also coincides with regular linebreaks produced outside the ML pretty engine;
 wenzelm parents: 
36687diff
changeset | 102 | } | 
| 
379f5b1e7f91
replaced slightly odd fbreak markup by plain "\n", which also coincides with regular linebreaks produced outside the ML pretty engine;
 wenzelm parents: 
36687diff
changeset | 103 | |
| 61874 | 104 | private def force_break(tree: Tree): Tree = | 
| 105 |     tree match { case Break(false, wd, ind) => Break(true, wd, ind) case _ => tree }
 | |
| 106 | private def force_all(trees: List[Tree]): List[Tree] = trees.map(force_break(_)) | |
| 51570 
3633828d80fc
basic support for Pretty.item, which is considered as logical markup and interpreted in Isabelle/Scala, but ignored elsewhere (TTY, latex etc.);
 wenzelm parents: 
51569diff
changeset | 107 | |
| 61874 | 108 | private def force_next(trees: List[Tree]): List[Tree] = | 
| 109 |     trees match {
 | |
| 110 | case Nil => Nil | |
| 111 | case (t: Break) :: ts => force_break(t) :: ts | |
| 112 | case t :: ts => t :: force_next(ts) | |
| 113 | } | |
| 51570 
3633828d80fc
basic support for Pretty.item, which is considered as logical markup and interpreted in Isabelle/Scala, but ignored elsewhere (TTY, latex etc.);
 wenzelm parents: 
51569diff
changeset | 114 | |
| 51493 
59d8a1031c00
allow fractional pretty margin -- avoid premature rounding;
 wenzelm parents: 
51492diff
changeset | 115 | private val margin_default = 76.0 | 
| 36820 | 116 | |
| 51493 
59d8a1031c00
allow fractional pretty margin -- avoid premature rounding;
 wenzelm parents: 
51492diff
changeset | 117 | def formatted(input: XML.Body, margin: Double = margin_default, | 
| 51492 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 118 | metric: Metric = Metric_Default): XML.Body = | 
| 36687 | 119 |   {
 | 
| 120 | val breakgain = margin / 20 | |
| 51569 
4e084727faae
maintain integer indentation during formatting -- it needs to be implemented by repeated spaces eventually;
 wenzelm parents: 
51568diff
changeset | 121 | val emergencypos = (margin / 2).round.toInt | 
| 36687 | 122 | |
| 61874 | 123 | def make_tree(inp: XML.Body): List[Tree] = | 
| 124 |       inp flatMap {
 | |
| 125 | case XML.Wrapped_Elem(markup, body1, body2) => | |
| 126 | List(make_block(Some(markup, Some(body1)), false, 0, make_tree(body2))) | |
| 127 | case XML.Elem(markup, body) => | |
| 128 |           markup match {
 | |
| 129 | case Markup.Block(consistent, indent) => | |
| 130 | List(make_block(None, consistent, indent, make_tree(body))) | |
| 131 | case Markup.Break(width, indent) => | |
| 62785 
70b9c7d4ed7f
more robust pretty printing: permissive treatment of bad values;
 wenzelm parents: 
61883diff
changeset | 132 | List(Break(false, force_nat(width), force_nat(indent))) | 
| 61874 | 133 | case Markup(Markup.ITEM, _) => | 
| 134 | List(make_block(None, false, 2, | |
| 135 | make_tree(XML.elem(Markup.BULLET, space) :: space ::: body))) | |
| 136 | case _ => | |
| 137 | List(make_block(Some((markup, None)), false, 0, make_tree(body))) | |
| 138 | } | |
| 139 | case XML.Text(text) => | |
| 140 | Library.separate(FBreak, split_lines(text).map(s => Str(s, metric(s)))) | |
| 36817 
ed97e877ff2d
more precise pretty printing based on actual font metrics;
 wenzelm parents: 
36763diff
changeset | 141 | } | 
| 
ed97e877ff2d
more precise pretty printing based on actual font metrics;
 wenzelm parents: 
36763diff
changeset | 142 | |
| 61871 | 143 | def format(trees: List[Tree], blockin: Int, after: Double, text: Text): Text = | 
| 36687 | 144 |       trees match {
 | 
| 145 | case Nil => text | |
| 146 | ||
| 61871 | 147 | case Block(markup, consistent, indent, body, blen) :: ts => | 
| 51569 
4e084727faae
maintain integer indentation during formatting -- it needs to be implemented by repeated spaces eventually;
 wenzelm parents: 
51568diff
changeset | 148 | val pos1 = (text.pos + indent).ceil.toInt | 
| 36687 | 149 | val pos2 = pos1 % emergencypos | 
| 61868 | 150 | val blockin1 = if (pos1 < emergencypos) pos1 else pos2 | 
| 61864 | 151 | val d = break_dist(ts, after) | 
| 152 | val body1 = if (consistent && text.pos + blen > margin - d) force_all(body) else body | |
| 61871 | 153 | val btext = | 
| 154 |             markup match {
 | |
| 155 | case None => format(body1, blockin1, d, text) | |
| 156 | case Some((m, markup_body)) => | |
| 157 | val btext0 = format(body1, blockin1, d, text.copy(tx = Nil)) | |
| 158 | val elem = | |
| 159 |                   markup_body match {
 | |
| 160 | case None => XML.Elem(m, btext0.content) | |
| 161 | case Some(b) => XML.Wrapped_Elem(m, b, btext0.content) | |
| 162 | } | |
| 163 | btext0.copy(tx = elem :: text.tx) | |
| 164 | } | |
| 61864 | 165 | val ts1 = if (text.nl < btext.nl) force_next(ts) else ts | 
| 36687 | 166 | format(ts1, blockin, after, btext) | 
| 167 | ||
| 61871 | 168 | case Break(force, wd, ind) :: ts => | 
| 169 | if (!force && | |
| 170 | text.pos + wd <= ((margin - break_dist(ts, after)) max (blockin + breakgain))) | |
| 36687 | 171 | format(ts, blockin, after, text.blanks(wd)) | 
| 61862 
e2a9e46ac0fb
support pretty break indent, like underlying ML systems;
 wenzelm parents: 
55551diff
changeset | 172 | else format(ts, blockin, after, text.newline.blanks(blockin + ind)) | 
| 36687 | 173 | |
| 61871 | 174 | case Str(s, len) :: ts => format(ts, blockin, after, text.string(s, len)) | 
| 36687 | 175 | } | 
| 61874 | 176 | format(make_tree(input), 0, 0.0, Text()).content | 
| 36687 | 177 | } | 
| 36734 
d9b10c173330
Pretty.formatted operates directly on XML trees, treating XML.Elem like a pro-forma block of indentation 0, like the ML version;
 wenzelm parents: 
36689diff
changeset | 178 | |
| 51493 
59d8a1031c00
allow fractional pretty margin -- avoid premature rounding;
 wenzelm parents: 
51492diff
changeset | 179 | def string_of(input: XML.Body, margin: Double = margin_default, | 
| 51492 
eaa1c4cc1106
more explicit Pretty.Metric, with clear distinction of unit (space width) vs. average char width (for visual adjustments) -- NB: Pretty formatting works via full space characters (despite a981a5c8a505 and 70f7483df9cb);
 wenzelm parents: 
51470diff
changeset | 180 | metric: Metric = Metric_Default): String = | 
| 49416 
1053a564dd25
some actual rich text markup via XML.content_markup;
 wenzelm parents: 
49414diff
changeset | 181 | XML.content(formatted(input, margin, metric)) | 
| 36683 | 182 | } |