alx/procat2
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

markup: support rectangle and circle annotations

Browse Source
This commit is contained in:
Seth Ladygo
2020-02-28 17:15:22 -08:00
parent dc85d784ab
commit b625b4f16d
2 changed files with 53 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
markup/pdf.py
View File

@ -10,7 +10,7 @@ from pdfminer.pdfdocument import PDFDocument
from pdfminer.pdftypes import PDFObjRef, resolve1
from .utils import Rect, pdf_rect, ensure_dir, set_file_perms
from .img import write_inklist
from .img import write_inklist, write_square_or_circle
def make_product_box(obj, pagenum, mediabox):
@ -44,7 +44,19 @@ def make_ink_scribble(obj, pagenum, mediabox, workdir):
oid = obj['NM'].decode('utf-8')
png_path = os.path.join(workdir, f"export-page{pagenum:03d}-nm{oid}.png")
write_inklist(obj, png_path)
write_inklist(obj, mediabox, png_path)
return { 'page': pagenum,
'rect': Rect(*mediabox),
'objid': oid,
'image': png_path }
def make_square_or_circle_scribble(obj, pagenum, mediabox, workdir):
oid = obj['NM'].decode('utf-8')
png_path = os.path.join(workdir, f"export-page{pagenum:03d}-nm{oid}.png")
write_square_or_circle(obj, mediabox, png_path)
return { 'page': pagenum,
'rect': Rect(*mediabox),
@ -160,6 +172,13 @@ def is_inklist_annotation(anno):
return 'Subtype' in anno and anno["Subtype"] == LIT('Ink')
def is_square_or_circle_annotation(anno):
if 'Subtype' in anno:
if anno["Subtype"] == LIT('Square') or anno["Subtype"] == LIT('Circle'):
return True
return False
def parse_pdf(fname, workdir, debug=0):
PDFDocument.debug = debug
PDFParser.debug = debug
@ -192,10 +211,16 @@ def parse_pdf(fname, workdir, debug=0):
anno = resolve1(anno)
if is_inklist_annotation(anno):
scribbles.append(make_ink_scribble(anno, pagenum, mediabox, workdir))
elif is_square_or_circle_annotation(anno):
scribbles.append(make_square_or_circle_scribble(anno, pagenum, mediabox, workdir))
elif 'AAPL:AKExtras' in anno:
scribbles.append(make_aapl_scribble(anno, pagenum, mediabox, workdir))
elif 'ProCatName' in anno:
prod_boxes.append(make_product_box(anno, pagenum, mediabox))
elif anno['Subtype'] == LIT('FreeText'):
print('ignoring FreeText annotation')
elif anno['Subtype'] == LIT('Highlight'):
print('ignoring Highlight annotation')
else:
print('ignoring other annotation:')
print(anno)