unipdf/pdf/core/utils.go
Gunnsteinn Hall f358b3e26f Refactoring in progress
Currently compiles with tests, but not fully functional yet.
2016-09-09 15:02:52 +00:00

114 lines
2.7 KiB
Go

/*
* This file is subject to the terms and conditions defined in
* file 'LICENSE.md', which is part of this source code package.
*/
package core
import (
"sort"
"github.com/unidoc/unidoc/common"
)
/*
* Inspect object types.
* Go through all objects in the cross ref table and detect the types.
* Mostly for debugging purposes and inspecting odd PDF files.
*/
func (this *PdfParser) Inspect() {
common.Log.Debug("--------INSPECT ----------")
common.Log.Debug("Xref table:")
objTypes := map[string]int{}
objCount := 0
failedCount := 0
keys := []int{}
for k, _ := range this.xrefs {
keys = append(keys, k)
}
sort.Ints(keys)
i := 0
for _, k := range keys {
xref := this.xrefs[k]
if xref.objectNumber == 0 {
continue
}
objCount++
common.Log.Debug("==========")
common.Log.Debug("Looking up object number: %d", xref.objectNumber)
o, err := this.LookupByNumber(xref.objectNumber)
if err != nil {
common.Log.Error("Fail to lookup obj %d (%s)", xref.objectNumber, err)
failedCount++
continue
}
common.Log.Debug("obj: %s", o)
iobj, isIndirect := o.(*PdfIndirectObject)
if isIndirect {
common.Log.Debug("IND OOBJ %d: %s", xref.objectNumber, iobj)
dict, isDict := iobj.PdfObject.(*PdfObjectDictionary)
if isDict {
ot, ok := (*dict)["Type"].(*PdfObjectName)
if ok {
otype := string(*ot)
common.Log.Debug("---> Obj type: %s", otype)
_, isDefined := objTypes[otype]
if isDefined {
objTypes[otype]++
} else {
objTypes[otype] = 1
}
}
}
} else if sobj, isStream := o.(*PdfObjectStream); isStream {
if otype, ok := (*(sobj.PdfObjectDictionary))["Type"].(*PdfObjectName); ok {
common.Log.Debug("--> Stream object type: %s", *otype)
k := string(*otype)
if _, isDefined := objTypes[k]; isDefined {
objTypes[k]++
} else {
objTypes[k] = 1
}
}
} else { // Direct.
dict, isDict := o.(*PdfObjectDictionary)
if isDict {
ot, isName := (*dict)["Type"].(*PdfObjectName)
if isName {
otype := string(*ot)
common.Log.Debug("AAA obj type %s", otype)
objTypes[otype]++
}
}
common.Log.Debug("DIR OOBJ %d: %s", xref.objectNumber, o)
}
i++
}
common.Log.Debug("--------EOF INSPECT ----------")
common.Log.Debug("=======")
common.Log.Debug("Object count: %d", objCount)
common.Log.Debug("Failed lookup: %d", failedCount)
for t, c := range objTypes {
common.Log.Debug("%s: %d", t, c)
}
common.Log.Debug("=======")
if len(this.xrefs) < 1 {
common.Log.Error("This document is invalid (xref table missing!)")
return
}
fontObjs, ok := objTypes["Font"]
if !ok || fontObjs < 2 {
common.Log.Debug("This document is probably scanned!")
} else {
common.Log.Debug("This document is valid for extraction!")
}
}