import re from optparse import OptionParser import os.path from reportlab.pdfgen import canvas from reportlab.lib.pagesizes import LETTER,A4 from reportlab.lib.units import cm from reportlab.platypus import Paragraph from reportlab.lib.styles import getSampleStyleSheet from reportlab.pdfbase.ttfonts import TTFont from reportlab.pdfbase import pdfmetrics def split(l,n): i = 0 while i < len(l) - n: yield l[i:i+n] i += n yield l[i:] class Card: def __init__(self,name,cardset,types,cost,description,potcost=0): self.name = name self.cardset = cardset self.types = types self.cost = cost self.potcost = potcost self.description = description self.extra = "" def getType(self): return DominionTabs.cardTypes[self.types] def __repr__(self): return '"' + self.name + '"' def toString(self): return self.name + ' ' + self.cardset + ' ' + '-'.join(self.types) + ' ' + `self.cost` + ' ' + self.description + ' ' + self.extra class CardType: def __init__(self, typeNames, tabImageFile, tabTextHeightOffset=0, tabCostHeightOffset=-1): self.typeNames = typeNames self.tabImageFile = tabImageFile self.tabTextHeightOffset = tabTextHeightOffset self.tabCostHeightOffset = tabCostHeightOffset def getTypeNames(self): return self.typeNames def getTabImageFile(self): return self.tabImageFile def getNoCoinTabImageFile(self): return ''.join(os.path.splitext(self.tabImageFile)[0] + '_nc' + os.path.splitext(self.tabImageFile)[1]) def getTabTextHeightOffset(self): return self.tabTextHeightOffset def getTabCostHeightOffset(self): return self.tabCostHeightOffset class DominionTabs: cardTypes = [ CardType(('Action',), 'action.png'), CardType(('Action','Attack'), 'action.png'), CardType(('Action','Attack','Prize'), 'action.png'), CardType(('Action','Reaction'), 'reaction.png'), CardType(('Action','Victory'), 'action-victory.png'), CardType(('Action','Duration'), 'duration.png'), CardType(('Action','Looter'), 'action.png'), CardType(('Action','Prize'), 'action.png'), CardType(('Action','Ruins'), 'ruins.png', 0, 1), CardType(('Action','Shelter'), 'shelter.png', 0, 1), CardType(('Action','Attack','Looter'), 'action.png'), CardType(('Reaction',), 'reaction.png'), CardType(('Reaction','Shelter'), 'shelter.png', 0, 1), CardType(('Treasure',), 'treasure.png',3,0), CardType(('Treasure','Victory'), 'treasure-victory.png'), CardType(('Treasure','Prize'), 'treasure.png',3,0), CardType(('Treasure','Reaction'), 'treasure-reaction.png', 0, 1), CardType(('Victory',), 'victory.png'), CardType(('Victory','Reaction'), 'victory-reaction.png', 0, 1), CardType(('Victory','Shelter'), 'shelter.png', 0, 1), CardType(('Curse',), 'curse.png',3), CardType(('Expansion',), 'expansion.png',4) ] cardTypes = dict(((c.getTypeNames(),c) for c in cardTypes)) setImages = { 'dominion' : 'base_set.png', 'intrigue' : 'intrigue_set.png', 'seaside' : 'seaside_set.png', 'prosperity' : 'prosperity_set.png', 'alchemy' : 'alchemy_set.png', 'cornucopia' : 'cornucopia_set.png', 'hinterlands' : 'hinterlands_set.png', 'dark ages' : 'dark_ages_set.png', 'dark ages extras' : 'dark_ages_set.png', 'guilds' : 'guilds_set.png' } promoImages = { 'walled village' : 'walled_village_set.png', 'stash' : 'stash_set.png', 'governor' : 'governor_set.png', 'black market' : 'black_market_set.png', 'envoy' : 'envoy_set.png' } def __init__(self): self.filedir = os.path.dirname(__file__) def add_inline_images(self, text, fontsize): path = os.path.join(self.filedir,'images') replace = '' % (path,fontsize*1.2) text = re.sub('(\d)\s(c|C)oin(s)?', replace,text) replace = '' % (path,fontsize*1.2) text = re.sub('\?\s(c|C)oin(s)?', replace,text) replace = '' % (path,fontsize*1.5) text = re.sub('\', replace,text) return text def drawOutline(self, x, y, rightSide, isBack=False): #draw outline or cropmarks self.canvas.saveState() self.canvas.setLineWidth(self.options.linewidth) cropmarksright = (x == self.numTabsHorizontal-1) cropmarksleft = (x == 0) if rightSide and not self.options.sameside: self.canvas.translate(self.tabWidth,0) self.canvas.scale(-1,1) if not self.options.cropmarks and not isBack: #don't draw outline on back, in case lines don't line up with front self.canvas.lines(self.tabOutline) elif self.options.cropmarks: cmw = 0.5*cm mirror = cropmarksright and not rightSide or cropmarksleft and rightSide if mirror: self.canvas.saveState() self.canvas.translate(self.tabWidth,0) self.canvas.scale(-1,1) if cropmarksleft or cropmarksright: self.canvas.line(-2*cmw,0,-cmw,0) self.canvas.line(-2*cmw,self.tabBaseHeight,-cmw,self.tabBaseHeight) if y > 0: self.canvas.line(-2*cmw,self.tabHeight,-cmw,self.tabHeight) if mirror: self.canvas.restoreState() if y == 0: self.canvas.line(self.tabWidth,-2*cmw,self.tabWidth,-cmw) self.canvas.line(self.tabWidth-self.tabLabelWidth,-2*cmw,self.tabWidth-self.tabLabelWidth,-cmw) if x == 0: self.canvas.line(0,-2*cmw,0,-cmw) elif y == self.numTabsVertical-1: self.canvas.line(self.tabWidth,self.tabHeight+cmw,self.tabWidth,self.tabHeight+2*cmw) self.canvas.line(self.tabWidth-self.tabLabelWidth, self.tabHeight+cmw, self.tabWidth-self.tabLabelWidth, self.tabHeight+2*cmw) if x == 0: self.canvas.line(0,self.tabHeight+cmw,0,self.tabHeight+2*cmw) self.canvas.restoreState() def drawTab(self, card, rightSide): #draw tab flap self.canvas.saveState() if not rightSide or self.options.sameside: self.canvas.translate(self.tabWidth-self.tabLabelWidth, self.tabHeight-self.tabLabelHeight) else: self.canvas.translate(0,self.tabHeight-self.tabLabelHeight) textWidth = 85 textHeight = self.tabLabelHeight/2-7+card.getType().getTabTextHeightOffset() self.canvas.drawImage(os.path.join(self.filedir,'images',card.getType().getNoCoinTabImageFile()),1,0, self.tabLabelWidth-2,self.tabLabelHeight-1, preserveAspectRatio=False,anchor='n',mask='auto') if card.getType().getTypeNames() != ('Expansion',): textInset = 22 costHeight = textHeight + card.getType().getTabCostHeightOffset() potHeight = 3 + card.getType().getTabTextHeightOffset() potSize = 11 self.canvas.drawImage(os.path.join(self.filedir,'images','coin_small.png'),4,costHeight-5,16,16,preserveAspectRatio=True,mask='auto') if card.potcost: self.canvas.drawImage(os.path.join(self.filedir,'images','potion.png'),21,potHeight,potSize,potSize,preserveAspectRatio=True,mask=[255,255,255,255,255,255]) textInset += potSize textWidth -= potSize setImageHeight = potHeight self.canvas.setFont('MinionPro-Bold',12) cost = str(card.cost) if 'Prize' in card.types: cost += '*' costWidthOffset = 12 self.canvas.drawCentredString(costWidthOffset,costHeight,cost) else: textInset = 13 setImageHeight = 3 + card.getType().getTabTextHeightOffset() #set image setImage = DominionTabs.setImages.get(card.cardset, None) if not setImage: setImage = DominionTabs.promoImages.get(card.name.lower(), None) if setImage: self.canvas.drawImage(os.path.join(self.filedir,'images',setImage), self.tabLabelWidth-20, setImageHeight, 14, 12, mask='auto') elif setImage == None and card.cardset != 'base' and card.getType().getTypeNames() != ('Expansion',): print 'warning, no set image for set "%s" card "%s"' % (card.cardset, card.name) DominionTabs.setImages[card.cardset] = 0 DominionTabs.promoImages[card.name.lower()] = 0 fontSize = 12 name = card.name.upper() name_parts = name.partition(' / ') if name_parts[1]: name_parts = (name_parts[0] + ' /', name_parts[2]) else: name_parts = name.split() width = pdfmetrics.stringWidth(name,'MinionPro-Regular',fontSize) while width > textWidth and fontSize > 8: fontSize -= 1 #print 'decreasing font size for tab of',name,'now',fontSize width = pdfmetrics.stringWidth(name,'MinionPro-Regular',fontSize) tooLong = width > textWidth #if tooLong: # print name #self.canvas.drawString(tabLabelWidth/2+8,tabLabelHeight/2-7,name[0]) w = 0 for i,n in enumerate(name_parts): self.canvas.setFont('MinionPro-Regular',fontSize) h = textHeight if tooLong: if i == 0: h += h/2 else: h -= h/2 self.canvas.drawString(textInset+w,h,n[0]) w += pdfmetrics.stringWidth(n[0],'MinionPro-Regular',fontSize) #self.canvas.drawString(tabLabelWidth/2+8+w,tabLabelHeight/2-7,name[1:]) self.canvas.setFont('MinionPro-Regular',fontSize-2) self.canvas.drawString(textInset+w,h,n[1:]) w += pdfmetrics.stringWidth(n[1:],'MinionPro-Regular',fontSize-2) w += pdfmetrics.stringWidth(' ','MinionPro-Regular',fontSize) if tooLong: w = 0 self.canvas.restoreState() def drawText(self, card, useExtra=False): #draw text if useExtra and card.extra: descriptions = (card.extra,) else: if self.options.read_yaml: descriptions = re.split("\n",card.description) else: descriptions = re.split("--+",card.description) height = 0 for d in descriptions: s = getSampleStyleSheet()['BodyText'] s.fontName = "Times-Roman" dmod = self.add_inline_images(d,s.fontSize) p = Paragraph(dmod,s) textHeight = self.tabHeight - self.tabLabelHeight + 0.2*cm textWidth = self.tabWidth - cm w,h = p.wrap(textWidth,textHeight) while h > textHeight: s.fontSize -= 1 s.leading -= 1 #print 'decreasing fontsize on description for',card.name,'now',s.fontSize dmod = self.add_inline_images(d,s.fontSize) p = Paragraph(dmod,s) w,h = p.wrap(textWidth,textHeight) p.drawOn(self.canvas,cm/2.0,textHeight-height-h-0.5*cm) height += h + 0.2*cm def drawDivider(self,card,x,y,useExtra=False): #figure out whether the tab should go on the right side or not if self.numTabsHorizontal % 2 == 0: rightSide = x%2 == 1 else: rightSide = useExtra #apply the transforms to get us to the corner of the current card self.canvas.resetTransforms() self.canvas.translate(self.horizontalMargin,self.verticalMargin) if useExtra: self.canvas.translate(self.options.back_offset,0) self.canvas.translate(x*self.totalTabWidth,y*self.totalTabHeight) #actual drawing if not self.options.tabs_only: self.drawOutline(x, y, rightSide, useExtra) self.drawTab(card, rightSide) if not self.options.tabs_only: self.drawText(card, useExtra) def read_card_extras(self,fname,cards): f = open(fname) cardName = re.compile("^:::(?P[ \w\-/']*)") extras = {} currentCard = "" extra = "" for line in f: m = cardName.match(line) if m: if currentCard: #print 'found',currentCard #print extra #print '------------------' extras[currentCard] = extra currentCard = m.groupdict()["name"] extra = "" if not self.options.expansions and currentCard and (currentCard not in (c.name for c in cards)): print currentCard + ' has extra description, but is not in cards' else: extra += line.strip() if currentCard and extra: extras[currentCard] = extra.strip() for c in cards: if not c.name in extras: print c.name + ' missing from extras' else: c.extra = extras[c.name] #print c.name + ' ::: ' + extra def add_definition_line(self,card,line): #try to figure out if this a 'basic action' like +X Cards or +Y Actions baseaction = re.compile("^\s*(\+\d+\s+\w+)(?:[,.;])") m = baseaction.match(line) prefix = '' while m: prefix += line[m.start(1):m.end(1)] + '----' line = line[m.end():] m = baseaction.match(line) line = prefix + line #this is a messy way to preserve the way the card spec file indicates separation #and add our own via '---' sequences. Needs to be completely replaced, probably #at the same time as changing the card spec file format. if not card.description.strip().endswith(';')\ and not card.description.strip().endswith('---')\ and not line.startswith('-'): card.description += '----' + line else: card.description += line def read_card_defs(self,fname,fileobject=None): cards = [] f = open(fname) carddef = re.compile("^\d+\t+(?P[\w\-'/ ]+)\t+(?P[\w ]+)\t+(?P[-\w ]+)\t+\$(?P\d+)( (?P\d)+P)?\t+(?P.*)") currentCard = None for line in f: line = line.strip() m = carddef.match(line) if m: if m.groupdict()["potioncost"]: potcost = int(m.groupdict()["potioncost"]) else: potcost = 0 currentCard = Card(m.groupdict()["name"], m.groupdict()["set"].lower(), tuple([t.strip() for t in m.groupdict()["type"].split("-")]), int(m.groupdict()["cost"]), '', potcost) self.add_definition_line(currentCard,m.groupdict()["description"]) cards.append(currentCard) elif line: assert currentCard self.add_definition_line(currentCard,line) #print currentCard #print '----' return cards def drawSetNames(self, pageCards): #print sets for this page self.canvas.saveState() try: # calculate the text height, font size, and orientation maxFontsize = 12 minFontsize = 6 fontname = 'MinionPro-Regular' font = pdfmetrics.getFont(fontname) fontHeightRelative = (font.face.ascent + font.face.descent) / 1000 canFit = False layouts = [{'rotation': 0, 'minMarginHeight': self.minVerticalMargin, 'totalMarginHeight': self.verticalMargin, 'width': self.paperwidth}, {'rotation': 90, 'minMarginHeight': self.minHorizontalMargin, 'totalMarginHeight': self.horizontalMargin, 'width': self.paperheight}] for layout in layouts: availableMargin = layout['totalMarginHeight'] - layout['minMarginHeight'] fontsize = availableMargin / fontHeightRelative fontsize = min(maxFontsize, fontsize) if fontsize >= minFontsize: canFit = True break if not canFit: import warnings warnings.warn("Not enough space to display set names") return self.canvas.setFont(fontname,fontsize) sets = [] for c in pageCards: setTitle = c.cardset.title() if setTitle not in sets: sets.append(setTitle) xPos = layout['width'] / 2 yPos = layout['minMarginHeight'] + availableMargin / 2 if layout['rotation']: self.canvas.rotate(layout['rotation']) yPos = -yPos self.canvas.drawCentredString(xPos,yPos,'/'.join(sets)) finally: self.canvas.restoreState() def drawDividers(self,cards): cards = split(cards,self.numTabsVertical*self.numTabsHorizontal) for pageCards in cards: if self.options.order != "global": self.drawSetNames(pageCards) for i,card in enumerate(pageCards): #print card x = i % self.numTabsHorizontal y = i / self.numTabsHorizontal self.canvas.saveState() self.drawDivider(card,x,self.numTabsVertical-1-y) self.canvas.restoreState() self.canvas.showPage() if self.options.order != "global": self.drawSetNames(pageCards) for i,card in enumerate(pageCards): #print card x = (self.numTabsHorizontal-1-i) % self.numTabsHorizontal y = i / self.numTabsHorizontal self.canvas.saveState() self.drawDivider(card,x,self.numTabsVertical-1-y,useExtra=True) self.canvas.restoreState() self.canvas.showPage() @staticmethod def parse_opts(argstring): parser = OptionParser() parser.add_option("--back_offset",type="int",dest="back_offset",default=0, help="Points to offset the back page to the right; needed for some printers") parser.add_option("--orientation",type="choice",choices=["horizontal","vertical"],dest="orientation",default="horizontal", help="horizontal or vertical, default:horizontal") parser.add_option("--sleeved",action="store_true",dest="sleeved",help="use --size=sleeved instead") parser.add_option("--size",type="string",dest="size",default='normal', help="'<%f>x<%f>' (size in cm), or 'normal' = '9.1x5.9', or 'sleeved' = '9.4x6.15'") parser.add_option("--minmargin",type="string",dest="minmargin",default="1x1", help="'<%f>x<%f>' (size in cm, left/right, top/bottom), default: 1x1") parser.add_option("--papersize",type="string",dest="papersize",default=None, help="'<%f>x<%f>' (size in cm), or 'A4', or 'LETTER'") parser.add_option("--samesidelabels",action="store_true",dest="sameside", help="force all label tabs to be on the same side") parser.add_option("--expansions",action="append",type="string", help="subset of dominion expansions to produce tabs for") parser.add_option("--cropmarks",action="store_true",dest="cropmarks", help="print crop marks on both sides, rather than tab outlines on one") parser.add_option("--linewidth",type="float",default=.1, help="width of lines for card outlines/crop marks") parser.add_option("--read_yaml", action="store_true",dest="read_yaml", help="read yaml version of card definitions and extras") parser.add_option("--write_yaml", action="store_true",dest="write_yaml", help="write yaml version of card definitions and extras") parser.add_option("--tabs-only", action="store_true", dest="tabs_only", help="draw only tabs to be printed on labels, no divider outlines") parser.add_option("--order", type="choice", choices=["expansion","global"], dest="order", help="sort order for the cards, whether by expansion or globally alphabetical") parser.add_option("--expansion_dividers", action="store_true", dest="expansion_dividers", help="add dividers describing each expansion set") return parser.parse_args(argstring) def main(self,argstring): options,args = DominionTabs.parse_opts(argstring) fname = None if args: fname = args[0] return self.generate(options,fname) def generate(self,options,f): self.options = options size = self.options.size.upper() if size == 'SLEEVED' or self.options.sleeved: dominionCardWidth, dominionCardHeight = (9.4*cm, 6.15*cm) print 'Using sleeved card size, %.2fcm x %.2fcm' % (dominionCardWidth/cm,dominionCardHeight/cm) elif size in ['NORMAL','UNSLEEVED']: dominionCardWidth, dominionCardHeight = (9.1*cm, 5.9*cm) print 'Using normal card size, %.2fcm x%.2fcm' % (dominionCardWidth/cm,dominionCardHeight/cm) else: x, y = size.split ("X", 1) dominionCardWidth, dominionCardHeight = (float (x) * cm, float (y) * cm) print 'Using custom card size, %.2fcm x %.2fcm' % (dominionCardWidth/cm,dominionCardHeight/cm) papersize = None if not self.options.papersize: if os.path.exists("/etc/papersize"): papersize = open ("/etc/papersize").readline().upper() else: papersize = self.options.papersize.upper() if papersize == 'A4': print "Using A4 sized paper." self.paperwidth, self.paperheight = A4 else: print "Using letter sized paper." self.paperwidth, self.paperheight = LETTER if self.options.orientation == "vertical": self.tabWidth, self.tabBaseHeight = dominionCardHeight, dominionCardWidth else: self.tabWidth, self.tabBaseHeight = dominionCardWidth, dominionCardHeight fixedMargins = False if self.options.tabs_only: #fixed for Avery 8867 for now minmarginwidth=0.76*cm minmarginheight=1.27*cm self.tabLabelHeight = 1.27*cm self.tabLabelWidth = 4.44*cm self.tabBaseHeight = 0 self.tabWidth = self.tabLabelWidth self.horizontalBorderSpace = 0.76*cm self.verticalBorderSpace = 0.01*cm fixedMargins = True else: minmarginwidth, minmarginheight = self.options.minmargin.split ("x", 1) minmarginwidth, minmarginheight = float (minmarginwidth) * cm, float (minmarginheight) * cm self.tabLabelHeight = 0.9*cm self.tabLabelWidth = 4*cm self.horizontalBorderSpace = 0*cm self.verticalBorderSpace = 0*cm self.tabHeight = self.tabBaseHeight + self.tabLabelHeight self.totalTabWidth = self.tabWidth + self.horizontalBorderSpace self.totalTabHeight = self.tabHeight + self.verticalBorderSpace print "Paper dimensions: %fcm (w) x %fcm (h)" % (self.paperwidth / cm, self.paperheight / cm) print "Tab dimensions: %fcm (w) x %fcm (h)" % (self.totalTabWidth / cm, self.totalTabHeight / cm) numTabsVerticalP = int ((self.paperheight - 2*minmarginheight) / self.totalTabHeight) numTabsHorizontalP = int ((self.paperwidth - 2*minmarginwidth) / self.totalTabWidth) numTabsVerticalL = int ((self.paperwidth - 2*minmarginwidth) / self.totalTabHeight) numTabsHorizontalL = int ((self.paperheight - 2*minmarginheight) / self.totalTabWidth) if numTabsVerticalL * numTabsHorizontalL > numTabsVerticalP * numTabsHorizontalP and not fixedMargins: self.numTabsVertical, self.numTabsHorizontal\ = numTabsVerticalL, numTabsHorizontalL self.paperheight, self.paperwidth = self.paperwidth, self.paperheight self.minHorizontalMargin = minmarginheight self.minVerticalMargin = minmarginwidth else: self.numTabsVertical, self.numTabsHorizontal\ = numTabsVerticalP, numTabsHorizontalP self.minHorizontalMargin = minmarginwidth self.minVerticalMargin = minmarginheight if not fixedMargins: #dynamically max margins self.horizontalMargin = (self.paperwidth-self.numTabsHorizontal*self.totalTabWidth)/2 self.verticalMargin = (self.paperheight-self.numTabsVertical*self.totalTabHeight)/2 else: self.horizontalMargin = minmarginwidth self.verticalMargin = minmarginheight print "Margins: %fcm h, %fcm v\n" % (self.horizontalMargin / cm, self.verticalMargin / cm) self.tabOutline = [(0,0,self.tabWidth,0), (self.tabWidth,0,self.tabWidth,self.tabHeight), (self.tabWidth,self.tabHeight, self.tabWidth-self.tabLabelWidth,self.tabHeight), (self.tabWidth-self.tabLabelWidth, self.tabHeight,self.tabWidth-self.tabLabelWidth, self.tabBaseHeight), (self.tabWidth-self.tabLabelWidth, self.tabBaseHeight,0,self.tabBaseHeight), (0,self.tabBaseHeight,0,0)] try: dirn = os.path.join(self.filedir,'fonts') pdfmetrics.registerFont(TTFont('MinionPro-Regular',os.path.join(dirn,'MinionPro-Regular.ttf'))) pdfmetrics.registerFont(TTFont('MinionPro-Bold',os.path.join(dirn,'MinionPro-Bold.ttf'))) except: raise pdfmetrics.registerFont(TTFont('MinionPro-Regular','OptimusPrincepsSemiBold.ttf')) pdfmetrics.registerFont(TTFont('MinionPro-Bold','OptimusPrinceps.ttf')) if options.read_yaml: import yaml cardfile = open("cards.yaml","r") cards = yaml.load(cardfile) else: cards = self.read_card_defs(os.path.join(self.filedir,"dominion_cards.txt")) if self.options.expansions: self.options.expansions = [o.lower() for o in self.options.expansions] cards=[c for c in cards if c.cardset in self.options.expansions] self.read_card_extras(os.path.join(self.filedir,"dominion_card_extras.txt"),cards) if options.expansion_dividers: cardnamesByExpansion = {} for c in cards: cardnamesByExpansion.setdefault(c.cardset,[]).append(c.name.strip()) for exp,names in cardnamesByExpansion.iteritems(): c = Card(exp, exp, ("Expansion",), None, ' | '.join(sorted(names))) cards.append(c) #print '%d cards read' % len(cards) sets = {} types = {} for c in cards: sets[c.cardset] = sets.get(c.cardset,0) + 1 types[c.types] = types.get(c.types,0) + 1 c.description = re.sub('----+','\n',c.description) c.description = re.sub('\n\s*\n','\n',c.description) if options.write_yaml: import yaml out = yaml.dump(cards) open('cards.yaml','w').write(out) if options.order == "global": sf = lambda x,y: cmp(x.name,y.name) else: sf = lambda x,y: cmp((x.cardset,x.name),(y.cardset,y.name)) cards.sort(cmp=sf) if not f: f = "dominion_tabs.pdf" self.canvas = canvas.Canvas(f, pagesize=(self.paperwidth, self.paperheight)) self.drawDividers(cards) self.canvas.save() if __name__=='__main__': import sys tabs = DominionTabs() tabs.main(sys.argv[1:])