From 2794792c13f56dbb2bbabe2c640b9ee29117709f Mon Sep 17 00:00:00 2001 From: Kurt Jung Date: Sat, 24 Aug 2013 16:11:53 -0400 Subject: Added PageSize() and NewCustom() methods --- fpdf.go | 216 ++++++++++++++++++++++++++++++++++------------------------------ 1 file changed, 116 insertions(+), 100 deletions(-) (limited to 'fpdf.go') diff --git a/fpdf.go b/fpdf.go index b8422e2..d8718a3 100644 --- a/fpdf.go +++ b/fpdf.go @@ -45,24 +45,7 @@ func (b *fmtBuffer) printf(fmtStr string, args ...interface{}) { b.Buffer.WriteString(fmt.Sprintf(fmtStr, args...)) } -// New returns a pointer to a new Fpdf instance. Its methods are subsequently -// called to produce a single PDF document. -// -// orientationStr specifies the default page orientation. For portrait mode, -// specify "P" or "Portrait". For landscape mode, specify "L" or "Landscape". -// An empty string will be replaced with "P". -// -// unitStr specifies the unit of length used in size parameters for elements -// other than fonts, which are always measured in points. Specify "pt" for -// point, "mm" for millimeter, "cm" for centimeter, or "in" for inch. An empty -// string will be replaced with "mm". -// -// sizeStr specifies the page size. Acceptable values are "A3", "A4", "A5", -// "Letter", or "Legal". An empty string will be replaced with "A4". -// -// fontDirStr specifies the file system location in which font resources will -// be found. An empty string is replaced with ".". -func New(orientationStr, unitStr, sizeStr, fontDirStr string) (f *Fpdf) { +func fpdfNew(orientationStr, unitStr, sizeStr, fontDirStr string, size SizeType) (f *Fpdf) { f = new(Fpdf) if orientationStr == "" { orientationStr = "P" @@ -80,7 +63,7 @@ func New(orientationStr, unitStr, sizeStr, fontDirStr string) (f *Fpdf) { f.n = 2 f.pages = make([]*bytes.Buffer, 0, 8) f.pages = append(f.pages, bytes.NewBufferString("")) // pages[0] is unused (1-based) - f.pageSizes = make(map[int]sizeType) + f.pageSizes = make(map[int]SizeType) f.state = 0 f.fonts = make(map[string]fontDefType) f.fontFiles = make(map[string]fontFileType) @@ -125,16 +108,21 @@ func New(orientationStr, unitStr, sizeStr, fontDirStr string) (f *Fpdf) { f.err = fmt.Errorf("Incorrect unit %s", unitStr) return } + f.unitStr = unitStr // Page sizes - f.stdpageSizes = make(map[string]sizeType) - f.stdpageSizes["a3"] = sizeType{841.89, 1190.55} - f.stdpageSizes["a4"] = sizeType{595.28, 841.89} - f.stdpageSizes["a5"] = sizeType{420.94, 595.28} - f.stdpageSizes["letter"] = sizeType{612, 792} - f.stdpageSizes["legal"] = sizeType{612, 1008} - f.defPageSize = f.getpagesizestr(sizeStr) - if f.err != nil { - return + f.stdPageSizes = make(map[string]SizeType) + f.stdPageSizes["a3"] = SizeType{841.89, 1190.55} + f.stdPageSizes["a4"] = SizeType{595.28, 841.89} + f.stdPageSizes["a5"] = SizeType{420.94, 595.28} + f.stdPageSizes["letter"] = SizeType{612, 792} + f.stdPageSizes["legal"] = SizeType{612, 1008} + if size.Wd > 0 && size.Ht > 0 { + f.defPageSize = size + } else { + f.defPageSize = f.getpagesizestr(sizeStr) + if f.err != nil { + return + } } f.curPageSize = f.defPageSize // Page orientation @@ -142,13 +130,13 @@ func New(orientationStr, unitStr, sizeStr, fontDirStr string) (f *Fpdf) { switch orientationStr { case "p", "portrait": f.defOrientation = "P" - f.w = f.defPageSize.wd - f.h = f.defPageSize.ht + f.w = f.defPageSize.Wd + f.h = f.defPageSize.Ht // dbg("Assign h: %8.2f", f.h) case "l", "landscape": f.defOrientation = "L" - f.w = f.defPageSize.ht - f.h = f.defPageSize.wd + f.w = f.defPageSize.Ht + f.h = f.defPageSize.Wd default: f.err = fmt.Errorf("Incorrect orientation: %s", orientationStr) return @@ -185,6 +173,46 @@ func New(orientationStr, unitStr, sizeStr, fontDirStr string) (f *Fpdf) { return } +// InitType is used with NewCustom() to customize an Fpdf instance. +// OrientationStr, UnitStr, SizeStr and FontDirStr correspond to the arguments +// accepted by New(). If the Wd and Ht fields of Size are each greater than +// zero, Size will be used to set the default page size rather than SizeStr. +type InitType struct { + OrientationStr string + UnitStr string + SizeStr string + Size SizeType + FontDirStr string +} + +// NewCustom returns a pointer to a new Fpdf instance. Its methods are +// subsequently called to produce a single PDF document. NewCustom() is an +// alternative to New() that provides additional customization. +func NewCustom(init *InitType) (f *Fpdf) { + return fpdfNew(init.OrientationStr, init.UnitStr, init.SizeStr, init.FontDirStr, init.Size) +} + +// New returns a pointer to a new Fpdf instance. Its methods are subsequently +// called to produce a single PDF document. +// +// orientationStr specifies the default page orientation. For portrait mode, +// specify "P" or "Portrait". For landscape mode, specify "L" or "Landscape". +// An empty string will be replaced with "P". +// +// unitStr specifies the unit of length used in size parameters for elements +// other than fonts, which are always measured in points. Specify "pt" for +// point, "mm" for millimeter, "cm" for centimeter, or "in" for inch. An empty +// string will be replaced with "mm". +// +// sizeStr specifies the page size. Acceptable values are "A3", "A4", "A5", +// "Letter", or "Legal". An empty string will be replaced with "A4". +// +// fontDirStr specifies the file system location in which font resources will +// be found. An empty string is replaced with ".". +func New(orientationStr, unitStr, sizeStr, fontDirStr string) (f *Fpdf) { + return fpdfNew(orientationStr, unitStr, sizeStr, fontDirStr, SizeType{0, 0}) +} + // Returns true if no processing errors have occurred. func (f *Fpdf) Ok() bool { return f.err == nil @@ -407,7 +435,7 @@ func (f *Fpdf) open() { // an invalid document. func (f *Fpdf) Close() { if f.err == nil { - if f.clipActive { + if f.clipNest > 0 { f.err = fmt.Errorf("Clip procedure must be explicitly ended") } } @@ -436,13 +464,27 @@ func (f *Fpdf) Close() { return } +// Returns the width and height of the specified page in the units established +// in New(). These return values are followed by the unit of measure itself. If +// pageNum is zero or otherwise out of bounds, it returns the default page +// size, that is, the size of the page that would be added by AddPage(). +func (f *Fpdf) PageSize(pageNum int) (wd, ht float64, unitStr string) { + sz, ok := f.pageSizes[pageNum] + if ok { + sz.Wd, sz.Ht = sz.Wd/f.k, sz.Ht/f.k + } else { + sz = f.defPageSize // user units + } + return sz.Wd, sz.Ht, f.unitStr +} + // Adds a new page with non-default orientation or size. See AddPage() for more // details. // // See New() for a description of orientationStr. // // size specifies the size of the new page in the units established in New(). -func (f *Fpdf) AddPageFormat(orientationStr string, size sizeType) { +func (f *Fpdf) AddPageFormat(orientationStr string, size SizeType) { if f.err != nil { return } @@ -948,17 +990,6 @@ func (f *Fpdf) RadialGradient(x, y, w, h float64, r1, g1, b1 int, r2, g2, b2 int f.gradientClipEnd() } -func (f *Fpdf) setClipActive() bool { - if f.err == nil { - if f.clipActive { - f.err = fmt.Errorf("Clipping operation already active") - } else { - f.clipActive = true - } - } - return f.err == nil -} - // Begins a rectangular clipping operation. The rectangle is of width w and // height h. Its upper left corner is positioned at point (x, y). outline is // true to draw a border with the current draw color and line width centered on @@ -969,9 +1000,7 @@ func (f *Fpdf) setClipActive() bool { // // See tutorial 14 for an example of this function. func (f *Fpdf) ClipRect(x, y, w, h float64, outline bool) { - if !f.setClipActive() { - return - } + f.clipNest++ f.outf("q %.2f %.2f %.2f %.2f re W %s", x*f.k, (f.h-y)*f.k, w*f.k, -h*f.k, strIf(outline, "S", "n")) } @@ -986,9 +1015,7 @@ func (f *Fpdf) ClipRect(x, y, w, h float64, outline bool) { // // See tutorial 14 for an example of this function. func (f *Fpdf) ClipText(x, y float64, txtStr string, outline bool) { - if !f.setClipActive() { - return - } + f.clipNest++ f.outf("q BT %.2f %.2f Td %d Tr (%s) Tj ET", x*f.k, (f.h-y)*f.k, intIf(outline, 5, 7), f.escape(txtStr)) } @@ -1009,9 +1036,7 @@ func (f *Fpdf) clipArc(x1, y1, x2, y2, x3, y3 float64) { // // See tutorial 14 for an example of this function. func (f *Fpdf) ClipRoundedRect(x, y, w, h, r float64, outline bool) { - if !f.setClipActive() { - return - } + f.clipNest++ k := f.k hp := f.h myArc := (4.0 / 3.0) * (math.Sqrt2 - 1.0) @@ -1045,9 +1070,7 @@ func (f *Fpdf) ClipRoundedRect(x, y, w, h, r float64, outline bool) { // // See tutorial 14 for an example of this function. func (f *Fpdf) ClipEllipse(x, y, rx, ry float64, outline bool) { - if !f.setClipActive() { - return - } + f.clipNest++ lx := (4.0 / 3.0) * rx * (math.Sqrt2 - 1) ly := (4.0 / 3.0) * ry * (math.Sqrt2 - 1) k := f.k @@ -1095,16 +1118,14 @@ func (f *Fpdf) ClipCircle(x, y, r float64, outline bool) { // ClipEnd() to restore unclipped operations. // // See tutorial 14 for an example of this function. -func (f *Fpdf) ClipPolygon(points []pointType, outline bool) { - if !f.setClipActive() { - return - } +func (f *Fpdf) ClipPolygon(points []PointType, outline bool) { + f.clipNest++ var s fmtBuffer h := f.h k := f.k s.printf("q ") for j, pt := range points { - s.printf("%.2f %.2f %s ", pt.x*k, (h-pt.y)*k, strIf(j == 0, "m", "l")) + s.printf("%.2f %.2f %s ", pt.X*k, (h-pt.Y)*k, strIf(j == 0, "m", "l")) } s.printf("h W %s", strIf(outline, "S", "n")) f.out(s.String()) @@ -1112,14 +1133,14 @@ func (f *Fpdf) ClipPolygon(points []pointType, outline bool) { // Ends a clipping operation that was started with a call to ClipRect(), // ClipRoundedRect(), ClipText(), ClipEllipse(), ClipCircle() or ClipPolygon(). -// Only one clipping operation can be active at a time, and the document cannot -// be successfully output while a clipping operation is active. +// Clipping operations can be nested. The document cannot be successfully +// output while a clipping operation is active. // // See tutorial 14 for an example of this function. func (f *Fpdf) ClipEnd() { if f.err == nil { - if f.clipActive { - f.clipActive = false + if f.clipNest > 0 { + f.clipNest-- f.out("Q") } else { f.err = fmt.Errorf("Error attempting to end clip operation") @@ -1965,45 +1986,40 @@ func (f *Fpdf) Output(w io.Writer) error { if err != nil { f.err = err } - dump("pdf.txt", f.stdpageSizes, + dump("pdf.txt", f.stdPageSizes, f.defPageSize, f.curPageSize, f.pageSizes) return f.err } -func (f *Fpdf) getpagesizestr(sizeStr string) (size sizeType) { +func (f *Fpdf) getpagesizestr(sizeStr string) (size SizeType) { if f.err != nil { return } sizeStr = strings.ToLower(sizeStr) // dbg("Size [%s]", sizeStr) - if sizeStr == "" { - // dbg("not found %s", sizeStr) - size = f.defPageSize - } else { - var ok bool - size, ok = f.stdpageSizes[sizeStr] - if ok { - // dbg("found %s", sizeStr) - size.wd /= f.k - size.ht /= f.k + var ok bool + size, ok = f.stdPageSizes[sizeStr] + if ok { + // dbg("found %s", sizeStr) + size.Wd /= f.k + size.Ht /= f.k - } else { - f.err = fmt.Errorf("Unknown page size %s", sizeStr) - } + } else { + f.err = fmt.Errorf("Unknown page size %s", sizeStr) } return } -func (f *Fpdf) _getpagesize(size sizeType) sizeType { - if size.wd > size.ht { - size.wd, size.ht = size.ht, size.wd +func (f *Fpdf) _getpagesize(size SizeType) SizeType { + if size.Wd > size.Ht { + size.Wd, size.Ht = size.Ht, size.Wd } return size } -func (f *Fpdf) beginpage(orientationStr string, size sizeType) { +func (f *Fpdf) beginpage(orientationStr string, size SizeType) { if f.err != nil { return } @@ -2020,14 +2036,14 @@ func (f *Fpdf) beginpage(orientationStr string, size sizeType) { } else { orientationStr = strings.ToUpper(orientationStr[0:1]) } - if orientationStr != f.curOrientation || size.wd != f.curPageSize.wd || size.ht != f.curPageSize.ht { + if orientationStr != f.curOrientation || size.Wd != f.curPageSize.Wd || size.Ht != f.curPageSize.Ht { // New size or orientation if orientationStr == "P" { - f.w = size.wd - f.h = size.ht + f.w = size.Wd + f.h = size.Ht } else { - f.w = size.ht - f.h = size.wd + f.w = size.Ht + f.h = size.Wd } f.wPt = f.w * f.k f.hPt = f.h * f.k @@ -2035,8 +2051,8 @@ func (f *Fpdf) beginpage(orientationStr string, size sizeType) { f.curOrientation = orientationStr f.curPageSize = size } - if orientationStr != f.defOrientation || size.wd != f.defPageSize.wd || size.ht != f.defPageSize.ht { - f.pageSizes[f.page] = sizeType{f.wPt, f.hPt} + if orientationStr != f.defOrientation || size.Wd != f.defPageSize.Wd || size.Ht != f.defPageSize.Ht { + f.pageSizes[f.page] = SizeType{f.wPt, f.hPt} } return } @@ -2411,7 +2427,7 @@ func (f *Fpdf) outf(fmtStr string, args ...interface{}) { func (f *Fpdf) putpages() { var wPt, hPt float64 - var pageSize sizeType + var pageSize SizeType // var linkList []linkType var ok bool nb := f.page @@ -2428,11 +2444,11 @@ func (f *Fpdf) putpages() { } } if f.defOrientation == "P" { - wPt = f.defPageSize.wd * f.k - hPt = f.defPageSize.ht * f.k + wPt = f.defPageSize.Wd * f.k + hPt = f.defPageSize.Ht * f.k } else { - wPt = f.defPageSize.ht * f.k - hPt = f.defPageSize.wd * f.k + wPt = f.defPageSize.Ht * f.k + hPt = f.defPageSize.Wd * f.k } for n := 1; n <= nb; n++ { // Page @@ -2441,7 +2457,7 @@ func (f *Fpdf) putpages() { f.out("/Parent 1 0 R") pageSize, ok = f.pageSizes[n] if ok { - f.outf("/MediaBox [0 0 %.2f %.2f]", pageSize.wd, pageSize.ht) + f.outf("/MediaBox [0 0 %.2f %.2f]", pageSize.Wd, pageSize.Ht) } f.out("/Resources 2 0 R") // Links @@ -2455,11 +2471,11 @@ func (f *Fpdf) putpages() { annots.printf("/A <>>>", f.textstring(pl.linkStr)) } else { l := f.links[pl.link] - var sz sizeType + var sz SizeType var h float64 sz, ok = f.pageSizes[l.page] if ok { - h = sz.ht + h = sz.Ht } else { h = hPt } -- cgit v1.2.1-24-ge1ad