我们还需要最后一个改进,之前的Element实现不够完善,只支持同样高度和同样宽度的Element使用above和beside函数,比如下面的代码将无法正常工作,因为组合元素的第二行比第一行要长:
new ArrayElement(Array("hello")) above new ArrayElement(Array("world!"))
与之相似的,下面的表达式也不能正常工作,因为第一个ArrayElement高度为二,而第二个的高度只是一
new ArrayElement(Array("one", "two")) beside new ArrayElement(Array("one"))
下面代码展示了一个私有帮助方法,widen,能够带个宽度做参数并返回那个宽度的Element。结果包含了这个Element的内容,居中,左侧和右侧留需带的空格以获得需要的宽度。这段代码还展示了一个类似的方法,heighten,能在竖直方向执行同样的功能。widen方法被above调用以确保Element堆叠在一起有同样的宽度。类似的,heighten方法被beside调用以确保靠在一起的元素具有同样的高度。有了这些改变,布局库函数可以使用了
abstract class Element { def contents: Array[String] def height: Int = contents.length def width: Int = if (height == 0) 0 else contents(0).length def above(that: Element) :Element = Element.elem(this.contents ++ that.contents) def beside(that: Element) :Element = { Element.elem( for( (line1,line2) <- this.contents zip that.contents ) yield line1+line2 ) } def widen(w: Int): Element = if (w <= width) this else { val left = Element.elem(' ', (w - width) / 2, height) var right = Element.elem(' ', w - width - left.width, height) left beside this beside right } def heighten(h: Int): Element = if (h <= height) this else { val top = Element.elem(' ', width, (h - height) / 2) var bot = Element.elem(' ', width, h - height - top.height) top above this above bot } override def toString = contents mkString "\n" }