v50 Steam/Premium information for editors
- v50 information can now be added to pages in the main namespace. v0.47 information can still be found in the DF2014 namespace. See here for more details on the new versioning policy.
- Use this page to report any issues related to the migration.
This notice may be cached—the current version can be found here.
Difference between revisions of "User:Button/ModestMod"
Jump to navigation
Jump to search
(→Some related LNP pseudocode: stuff I need to comment) |
(→Some related LNP pseudocode: comments) |
||
Line 70: | Line 70: | ||
<pre> | <pre> | ||
def _load_ascii_conversions(): | def _load_ascii_conversions(): | ||
+ | """Load ASCII conversions from file. | ||
+ | |||
+ | The file is structured as a .property file, with a new conversion on each line. | ||
+ | |||
+ | Each line of the file should be in the format " '%'=# ", where % is a single character, and # is the character's position on code page 437 (the basis of the Dwarf Fortress tileset). | ||
+ | |||
+ | Comments are not allowed in this file. | ||
+ | |||
+ | These conversions will later be used by the function escape_problematic_literals. | ||
+ | """ | ||
global ascii_codes | global ascii_codes | ||
global verbose | global verbose | ||
Line 77: | Line 87: | ||
ascii_codes = {} | ascii_codes = {} | ||
if properties['ascii'][1] == None: | if properties['ascii'][1] == None: | ||
− | + | print("Undefined ascii conversion file. Please add an 'ascii' property in",runconfig,".") | |
else: | else: | ||
for line in open(properties['ascii'][1]): | for line in open(properties['ascii'][1]): | ||
Line 84: | Line 94: | ||
continue | continue | ||
elif '=' not in real_line: | elif '=' not in real_line: | ||
− | + | if verbose: | |
+ | print('ASCII conversion file contains the improperly-formatted line ',real_line,'.') | ||
else: | else: | ||
point = real_line.rindex('=') | point = real_line.rindex('=') | ||
if real_line[:point] in ascii_codes.keys(): | if real_line[:point] in ascii_codes.keys(): | ||
− | + | if verbose: | |
+ | print('Duplicate entry for ascii replacement ',real_line[:point]) | ||
else: | else: | ||
ascii_codes[real_line[:point]] = real_line[point+1:] | ascii_codes[real_line[:point]] = real_line[point+1:] | ||
Line 95: | Line 107: | ||
def write_modified_raws(graphics_to_apply): | def write_modified_raws(graphics_to_apply): | ||
+ | """Write the full modified raws to the raw output directory. | ||
+ | |||
+ | graphics_to_apply is a dict of type string:dict{string:BoundNode}. The top-level string keys are filenames. Each filename's value is a dict representing the top-level, relevant tags in the target raw file of that name. The inner key is the full tag (without brackets), and the BoundNode is the top node of the tree that that tag begins. | ||
+ | |||
+ | Unless you have way too much time on your hands, I recommend you generate graphics_to_apply using the bind_graphics_to_targets function. | ||
+ | |||
+ | The actual writing is done in this way: first, the function walks the target raw source directory structure, and duplicates it in the raw output directory. Then it walks the target raw source files, looking for each filename in graphics_to_apply's keyset, or one of the properties graphics_overwrite or graphics_ignore. | ||
+ | * If it finds the filename in graphics_overwrite, it copies the file directly from the corresponding place in the graphics source directory. | ||
+ | * If it finds the filename in graphics_ignore, or doesn't find the filename at all, it copies the file directly from the corresponding place in the target raw source directory. | ||
+ | * If it finds the filename in graphics_to_apply's keyset, it opens the file in the target raw source directory, creates and opens a corresponding file in the raw output directory, and walks through the target raw source file a line at a time, constructing the modified file. | ||
+ | """ | ||
if len(properties['output']) < 2: | if len(properties['output']) < 2: | ||
print("Undefined output directory. Please add an 'output' property in",runconfig,".") | print("Undefined output directory. Please add an 'output' property in",runconfig,".") | ||
Line 101: | Line 124: | ||
print("Writing modified raws...") | print("Writing modified raws...") | ||
for root, dirs, files in os.walk(properties['target'][1]): | for root, dirs, files in os.walk(properties['target'][1]): | ||
+ | # Create directories so we don't have any issues later on | ||
for dir in dirs: | for dir in dirs: | ||
targetdir = os.path.join(root,dir) | targetdir = os.path.join(root,dir) | ||
Line 168: | Line 192: | ||
if verbose: | if verbose: | ||
print("Adding tag",tag_node._tag,"at line",linecount,".") | print("Adding tag",tag_node._tag,"at line",linecount,".") | ||
− | line_to_write = "[" + tag_node._tag + "] (BAMM)\n" | + | line_to_write = "[" + tag_node._tag + "]\n"# (BAMM)\n" |
targetfile.writelines(line_to_write) | targetfile.writelines(line_to_write) | ||
Line 180: | Line 204: | ||
def _walk_rawfiles_into_tagnode_collection(directory): | def _walk_rawfiles_into_tagnode_collection(directory): | ||
+ | """Load the graphics-relevant content of raw files into memory. | ||
+ | |||
+ | directory is a directory containing the raw files you want to load into memory. None of the files may have duplicate names, even if they're in different sub-folders. | ||
+ | |||
+ | The function returns a dictionary of string:dict{string:TagNode}. The outer key is a filename which contains some graphics-relevant content. The inner key is the tag corresponding with a top-level TagNode in that file, and maps to that TagNode. | ||
+ | |||
+ | This format is the expected input format of both parameters of bind_graphics_to_targets(graphics_nodes,target_nodes). | ||
+ | """ | ||
global verbose | global verbose | ||
node_collection = {} | node_collection = {} | ||
for root, dirs, files in os.walk(directory): | for root, dirs, files in os.walk(directory): | ||
for rawfile in files: | for rawfile in files: | ||
+ | # Only look at .txt files | ||
if '.txt' not in rawfile: | if '.txt' not in rawfile: | ||
if verbose: | if verbose: | ||
Line 219: | Line 252: | ||
def bind_graphics_to_targets(graphics_nodes,targets_nodes): | def bind_graphics_to_targets(graphics_nodes,targets_nodes): | ||
+ | """Associate two collections of TagNodes with each other, in preparation for merging. | ||
+ | |||
+ | graphics_nodes and targets_nodes are dicts of type string:dict{string:TagNode}, where the outer key is a file name, the inner keys are tags associated with top-level TagNodes, and the TagNodes are of course those same top-level TagNodes. These string:string:TagNodes are expected to be produced by _walk_rawfiles_into_tagnode_collection. | ||
+ | |||
+ | The function returns a similarly-formatted dict of BoundNodes, in the format string:dict{string:BoundNode}, where the outer key is a file name, the inner key is the tag associated with the target_nodes of top level BoundNodes, and the BoundNodes are, as usual, those BoundNodes. | ||
+ | |||
+ | The BoundNodes in the return dict are generated from the two arguments, by matching up TagNodes from graphics_nodes and targets_nodes based on their filenames, templates, and parents. If a TagNode in either argument has no corresponding TagNode in the other, it is dropped if it has children or non-graphical content. If neither of those is the case, it's saved as an "additional" graphics tag to be added or removed in the conversion. | ||
+ | """ | ||
global verbose | global verbose | ||
if verbose: | if verbose: | ||
Line 224: | Line 265: | ||
to_return = {} | to_return = {} | ||
for filename in targets_nodes.keys(): | for filename in targets_nodes.keys(): | ||
+ | # Files which both share, both have found graphics in, and haven't been put in the return dict yet. | ||
if filename in graphics_nodes.keys() and filename not in to_return.keys(): | if filename in graphics_nodes.keys() and filename not in to_return.keys(): | ||
if verbose: | if verbose: | ||
Line 229: | Line 271: | ||
to_return[filename] = {} | to_return[filename] = {} | ||
for top_level_tag in targets_nodes[filename].keys(): | for top_level_tag in targets_nodes[filename].keys(): | ||
+ | # Top level tags which both share, both have found graphics in, and haven't been put in the return dict yet. | ||
if top_level_tag in graphics_nodes[filename] and top_level_tag not in to_return[filename].keys(): | if top_level_tag in graphics_nodes[filename] and top_level_tag not in to_return[filename].keys(): | ||
+ | # Create new BoundNode tree and put it in the return dict. | ||
to_return[filename][top_level_tag] = BoundNode(targets_nodes[filename][top_level_tag],graphics_nodes[filename][top_level_tag]) | to_return[filename][top_level_tag] = BoundNode(targets_nodes[filename][top_level_tag],graphics_nodes[filename][top_level_tag]) | ||
− | |||
if verbose: | if verbose: | ||
print(filename,"tags bound.") | print(filename,"tags bound.") | ||
Line 239: | Line 282: | ||
def tags(line): | def tags(line): | ||
+ | """Return an ordered list of all the tags in this line, without brackets, with literals escaped if necessary.""" | ||
processed_line = escape_problematic_literals(line) | processed_line = escape_problematic_literals(line) | ||
to_return = [] # list of strings, a la split() | to_return = [] # list of strings, a la split() | ||
Line 249: | Line 293: | ||
def path_compatible(full_path,allowed_paths): | def path_compatible(full_path,allowed_paths): | ||
+ | """Return True if full_path regex matches anything in allowed_paths, or False otherwise.""" | ||
# TODO use regexes | # TODO use regexes | ||
to_return = True | to_return = True | ||
Line 265: | Line 310: | ||
class TreeNode(): | class TreeNode(): | ||
+ | """Parent class for the other Node classes. | ||
+ | |||
+ | Contains default implementations of common Tree functionality. | ||
+ | |||
+ | Members: | ||
+ | self._parent = the parent TreeNode of this TreeNode. Any given subclass of TreeNode should only have TreeNodes of its own type as _parent. | ||
+ | self._tag = The string that this node represents. This should be overridden and re-defined by subclasses. | ||
+ | self._children = A dict of type string:TreeNode, where the key is the child's ._tag property. | ||
+ | """ | ||
+ | |||
def __init__(self,parent=None): | def __init__(self,parent=None): | ||
self._parent = parent | self._parent = parent | ||
Line 273: | Line 328: | ||
def add_child(self, child_node): | def add_child(self, child_node): | ||
+ | """Add a child to this TreeNode.""" | ||
self._children[child_node._tag] = child_node | self._children[child_node._tag] = child_node | ||
def find_match(self, tag): | def find_match(self, tag): | ||
+ | """Find a TreeNode related to this TreeNode which has the given tag. | ||
+ | |||
+ | tag is a value corresponding to the ._tag field of TreeNode you're calling this on. REMEMBER: Different TreeNode subclasses have different formats of ._tags! | ||
+ | |||
+ | The return value is a TreeNode with the specified ._tag value, or None. | ||
+ | |||
+ | The TreeNodes searched are this node's immediate children, and its direct ancestors' immediate children. | ||
+ | """ | ||
curr_node = self | curr_node = self | ||
matching_node = None | matching_node = None | ||
out_of_parents = False | out_of_parents = False | ||
− | while matching_node | + | while matching_node is None and not out_of_parents: |
#matching_node = curr_node.get_template_match(tag)[0] | #matching_node = curr_node.get_template_match(tag)[0] | ||
#if matching_node == None: | #if matching_node == None: | ||
matching_node = curr_node.get_child(tag) | matching_node = curr_node.get_child(tag) | ||
− | if curr_node._parent | + | if curr_node._parent is None: |
out_of_parents = True | out_of_parents = True | ||
else: | else: | ||
Line 290: | Line 354: | ||
def get_child(self, tag): | def get_child(self, tag): | ||
+ | """Return an immediate child with the given tag, or None if no such child exists.""" | ||
if tag in self._children.keys(): | if tag in self._children.keys(): | ||
return self._children[tag] | return self._children[tag] | ||
Line 296: | Line 361: | ||
class TemplateNode(TreeNode): | class TemplateNode(TreeNode): | ||
+ | """An implementation of TreeNode specialized for holding tag templates. | ||
+ | |||
+ | Members: | ||
+ | self._parent = the parent TemplateNode of this TemplateNode. | ||
+ | self._tag = the string representing a valid format of raw tag. Literals are themselves; colons separate tokens; $s indicate identifiers; ?s indicate graphics information; and &s indicate non-identifier, non-graphics information. For more fluidly-formatted tags, the following format indicates more than one tag: | ||
+ | x(#,#) , where the x is a $, ? or &, the #s are a min/max pair, and the max is optional. | ||
+ | self._is_graphics_tag | ||
+ | """ | ||
− | #self._tag #string | + | #self._tag # string |
− | #self._children | + | #self._children # dict of TemplateNodes |
− | #self._childref | + | #self._childref # dict of lists of TemplateNodes where the key is the first token |
− | #self._parent | + | #self._parent # TemplateNode |
− | #self._is_graphics_tag | + | #self._is_graphics_tag # Boolean |
#string does not contain the character '|'. | #string does not contain the character '|'. | ||
Line 308: | Line 381: | ||
self._is_graphics_tag = False | self._is_graphics_tag = False | ||
self._childref = {} | self._childref = {} | ||
− | |||
global template_tree | global template_tree | ||
− | if parent | + | if parent is None: |
self._parent = None | self._parent = None | ||
template_tree = self | template_tree = self | ||
Line 520: | Line 592: | ||
return False | return False | ||
else: | else: | ||
− | return True | + | return True</pre> |
− | </pre> |
Revision as of 22:03, 21 May 2015
This is a directory page for my Modest Mod work.
I feel a little bad for clogging up the wiki with this stuff, but sometimes I get bored and I'm not at my home computer.
Plans
- Moving the no-fear-of-commitment orientation mods out into a pair of creature variations, in c_variation_modest. May make a Module applying them to sapients as well. Eternal Lovers are kind of frustrating.
- Could probably just make the plant growths that I currently have wrapping tree nuts, edible or pressable or whatever in their own right, and never have to deal with the seeds. OTOH, that might mess up grinding them to paste/pressing them? And would still need the hulling reaction for plantable legumes.
- Could change implementation of all plants, including those brewed 'whole', into growths. Then would be able to share reactions better, and do seed/no seed on a purely FAKESEED basis.
- Probably not worth it in vanilla, but maybe fore plant features, since we'll be needing to mill "whole plants" (e.g. cassava, whip vines) and also to mill seedheads.
- Need to move Plump Helmets back to a usable plant instead of a fruit. Save it as fruit for the Basic Adventurer's Pack.
- Need to look at the expanded creature variation replacement tokens in preparation for the vampire anti-sobriety module.
Accelerated
- Need to look at the old Accelerated Modest Mod and see how much of it is applicable to the current version.
- Need to see about thinning down plant materials, like a lot.
- Need to remove useless plant growths, like cherry blossoms etc. Leaves? Or would that be too ugly?
- Could probably reduce grass to a single species? Or would that make pandas shitty?
Plant Features
- Could probably make grain use the fruit(like?) reactions for brewing, milling, etc.
- Kahlua & Chajun & possibly the plant products that are used to make them could give a temporary [NOSLEEP] tag? Followed by a 'crash' of a drowsiness syndrome?
- Jellies? Preserves? Juices?
- More variation in material values
Basic Adventurer's Pack
This needs to be a basic pack. Don't get cocky.
Must include
- Plump helmet to fruit
- Backpacks, bags, quivers, waterskins.
- Tanning
- Stone knife
- Spinning & weaving
- Spindle
- Tanning
- Bone ammo
- Stone chisel
- Arrow bamboo?
- Trophy jewelry/crafts for trading
Might want to include
- Leather armor.
- Bone & shell armor.
- Bone bows, crossbows, blowguns
- Bamboo?
- Plant fiber thread. Rope reeds, pig tail stalks, others?
- For the merge with Plant Features, abaca/banana/pineapple leaves, cotton bolls, etc. A lot of the plant fabric stuff is Plant Features-necessary.
Probably going too far
- Enable a whole bunch of sapients playable as outsiders. Not just the ones in the gear-already-exists size range.
- Hahaha, fairies as a playable outsider race |D That would be silly
- Cooking. Just biscuits? Or full-on recipes?
- Flour? Oil? Good use for mortar & pestle...
- Maybe a balanced meal would require a veg, a protein, & a sugar/starch, & would provide minor (temporary) attribute boosts?
- Self-buff "cheats"?
Raws
- creature_subterranean
- c_variation_default
- body_modest_pearl
- c_variation_modest
- plant_crops
- plant_garden
- plant_standard
- plant_new_trees
def _load_ascii_conversions(): """Load ASCII conversions from file. The file is structured as a .property file, with a new conversion on each line. Each line of the file should be in the format " '%'=# ", where % is a single character, and # is the character's position on code page 437 (the basis of the Dwarf Fortress tileset). Comments are not allowed in this file. These conversions will later be used by the function escape_problematic_literals. """ global ascii_codes global verbose if verbose: print("Loading ASCII conversions...") if ascii_codes == None: ascii_codes = {} if properties['ascii'][1] == None: print("Undefined ascii conversion file. Please add an 'ascii' property in",runconfig,".") else: for line in open(properties['ascii'][1]): real_line = line.strip() if len(real_line) == 0: continue elif '=' not in real_line: if verbose: print('ASCII conversion file contains the improperly-formatted line ',real_line,'.') else: point = real_line.rindex('=') if real_line[:point] in ascii_codes.keys(): if verbose: print('Duplicate entry for ascii replacement ',real_line[:point]) else: ascii_codes[real_line[:point]] = real_line[point+1:] if verbose: print("ASCII conversions loaded.") def write_modified_raws(graphics_to_apply): """Write the full modified raws to the raw output directory. graphics_to_apply is a dict of type string:dict{string:BoundNode}. The top-level string keys are filenames. Each filename's value is a dict representing the top-level, relevant tags in the target raw file of that name. The inner key is the full tag (without brackets), and the BoundNode is the top node of the tree that that tag begins. Unless you have way too much time on your hands, I recommend you generate graphics_to_apply using the bind_graphics_to_targets function. The actual writing is done in this way: first, the function walks the target raw source directory structure, and duplicates it in the raw output directory. Then it walks the target raw source files, looking for each filename in graphics_to_apply's keyset, or one of the properties graphics_overwrite or graphics_ignore. * If it finds the filename in graphics_overwrite, it copies the file directly from the corresponding place in the graphics source directory. * If it finds the filename in graphics_ignore, or doesn't find the filename at all, it copies the file directly from the corresponding place in the target raw source directory. * If it finds the filename in graphics_to_apply's keyset, it opens the file in the target raw source directory, creates and opens a corresponding file in the raw output directory, and walks through the target raw source file a line at a time, constructing the modified file. """ if len(properties['output']) < 2: print("Undefined output directory. Please add an 'output' property in",runconfig,".") else: if verbose: print("Writing modified raws...") for root, dirs, files in os.walk(properties['target'][1]): # Create directories so we don't have any issues later on for dir in dirs: targetdir = os.path.join(root,dir) targetdir = properties['output'][1] + targetdir[len(properties['target'][1]):] if not os.path.exists(targetdir): if verbose: print("Creating output directory",dir) os.mkdir(targetdir) for file in files: targetpath = os.path.join(root,file) targetpath = properties['output'][1] + targetpath[len(properties['target'][1]):] if path_compatible(targetpath,properties['graphics_overwrite'][1:]): if verbose: print("Skipping",file,": graphics overwrite TBI.") pass elif path_compatible(targetpath,properties['graphics_ignore'][1:]): if verbose: print("Skipping",file,": graphics ignore TBI.") pass elif file not in graphics_to_apply.keys(): if verbose: print("No graphics to apply to",file,". Copying from target source...") targetpath = shutil.copyfile(os.path.join(root,file),targetpath) if verbose: print(file,"copied.") else: if verbose: print("Merging graphics into",file,"...") curr_dict = graphics_to_apply[file] curr_node = None targetfile = open(targetpath,'wt',encoding='cp437') sourcefile = open(os.path.join(root,file),'rt',encoding='cp437') linecount = 0 for line in sourcefile: linecount = linecount + 1 modified_line = escape_problematic_literals(line) additional = [] for tag in tags(line): matching_node = None if tag in curr_dict.keys(): matching_node = curr_dict[tag] elif curr_node is not None: matching_node = curr_node.find_match(tag) if matching_node is not None: curr_node = matching_node matching_node.pop_self() if matching_node.is_there_a_difference(): merged_tag = matching_node.get_merged() if merged_tag is not None: replacement = matching_node.get_merged() if verbose: print("Replacing",tag,"with",replacement,"at line",linecount,".") modified_line = modified_line.replace(tag,replacement) else: if verbose: print("Removing tag",tag,"at line",linecount,".") to_remove = "[" + tag + "]" modified_line = modified_line.replace(to_remove,"") #modified_line = modified_line[:-1] + " (BAMM)\n" additional.extend(matching_node.pop_addl()) targetfile.writelines(modified_line) for tag_node in additional: linecount = linecount + 1 if verbose: print("Adding tag",tag_node._tag,"at line",linecount,".") line_to_write = "[" + tag_node._tag + "]\n"# (BAMM)\n" targetfile.writelines(line_to_write) targetfile.flush() if verbose: print("Finished outputting",file,".") targetfile.close() sourcefile.close() if verbose: print("All files written.") def _walk_rawfiles_into_tagnode_collection(directory): """Load the graphics-relevant content of raw files into memory. directory is a directory containing the raw files you want to load into memory. None of the files may have duplicate names, even if they're in different sub-folders. The function returns a dictionary of string:dict{string:TagNode}. The outer key is a filename which contains some graphics-relevant content. The inner key is the tag corresponding with a top-level TagNode in that file, and maps to that TagNode. This format is the expected input format of both parameters of bind_graphics_to_targets(graphics_nodes,target_nodes). """ global verbose node_collection = {} for root, dirs, files in os.walk(directory): for rawfile in files: # Only look at .txt files if '.txt' not in rawfile: if verbose: print("Skipping file",rawfile,"...") continue if verbose: print("Loading graphics tags from",rawfile,"...") global template_tree # curr_template_node keeps track of what format of tag we've most recently seen, and thus what's valid next curr_template_node = template_tree # curr_real_node keeps track of the tag we stored that corresponds to the most local instance of curr_template_node. curr_real_node = None tarpath = os.path.join(root, rawfile) openfile = open(tarpath,encoding='cp437') for line in openfile: for tag in tags(line): matching_node = curr_template_node.find_match(tag) if matching_node != None: curr_template_node = matching_node if curr_real_node == None or matching_node._tag in template_tree._children: curr_real_node = TagNode(rawfile,matching_node,tag) else: while curr_real_node != None and matching_node._tag not in curr_real_node._template._children: curr_real_node = curr_real_node._parent curr_real_node = TagNode(rawfile,matching_node,tag,curr_real_node) if rawfile not in node_collection: node_collection[rawfile] = { } if curr_real_node._parent == None: node_collection[rawfile][tag] = curr_real_node openfile.close() if verbose: print("Finished processing",rawfile,".") return node_collection def bind_graphics_to_targets(graphics_nodes,targets_nodes): """Associate two collections of TagNodes with each other, in preparation for merging. graphics_nodes and targets_nodes are dicts of type string:dict{string:TagNode}, where the outer key is a file name, the inner keys are tags associated with top-level TagNodes, and the TagNodes are of course those same top-level TagNodes. These string:string:TagNodes are expected to be produced by _walk_rawfiles_into_tagnode_collection. The function returns a similarly-formatted dict of BoundNodes, in the format string:dict{string:BoundNode}, where the outer key is a file name, the inner key is the tag associated with the target_nodes of top level BoundNodes, and the BoundNodes are, as usual, those BoundNodes. The BoundNodes in the return dict are generated from the two arguments, by matching up TagNodes from graphics_nodes and targets_nodes based on their filenames, templates, and parents. If a TagNode in either argument has no corresponding TagNode in the other, it is dropped if it has children or non-graphical content. If neither of those is the case, it's saved as an "additional" graphics tag to be added or removed in the conversion. """ global verbose if verbose: print("Binding graphics source tags to target tags...") to_return = {} for filename in targets_nodes.keys(): # Files which both share, both have found graphics in, and haven't been put in the return dict yet. if filename in graphics_nodes.keys() and filename not in to_return.keys(): if verbose: print("Binding tags for",filename,"...") to_return[filename] = {} for top_level_tag in targets_nodes[filename].keys(): # Top level tags which both share, both have found graphics in, and haven't been put in the return dict yet. if top_level_tag in graphics_nodes[filename] and top_level_tag not in to_return[filename].keys(): # Create new BoundNode tree and put it in the return dict. to_return[filename][top_level_tag] = BoundNode(targets_nodes[filename][top_level_tag],graphics_nodes[filename][top_level_tag]) if verbose: print(filename,"tags bound.") if verbose: print("Tag binding complete.") return to_return def tags(line): """Return an ordered list of all the tags in this line, without brackets, with literals escaped if necessary.""" processed_line = escape_problematic_literals(line) to_return = [] # list of strings, a la split() while '[' in processed_line and ']' in processed_line and processed_line.index('[') < processed_line.rindex(']'): if processed_line.index(']') < processed_line.index('['): processed_line = processed_line[processed_line.index('['):] to_return.append(processed_line[processed_line.index('[')+1:processed_line.index(']')]) processed_line = processed_line[processed_line.index(']')+1:] return to_return def path_compatible(full_path,allowed_paths): """Return True if full_path regex matches anything in allowed_paths, or False otherwise.""" # TODO use regexes to_return = True full_path = full_path.replace("/",os.pathsep) full_path_tokens = full_path.split(os.pathsep) for possible_path in allowed_paths: possibility = possible_path.split('/') match_tuples = [] for ii in range(0,len(possibility)): if possibility[ii] == '*': continue elif possibility[ii] not in full_path_tokens[ii:]: break else: match_tuples.append((ii,full_path_tokens[ii:].index(possibility[ii]))) class TreeNode(): """Parent class for the other Node classes. Contains default implementations of common Tree functionality. Members: self._parent = the parent TreeNode of this TreeNode. Any given subclass of TreeNode should only have TreeNodes of its own type as _parent. self._tag = The string that this node represents. This should be overridden and re-defined by subclasses. self._children = A dict of type string:TreeNode, where the key is the child's ._tag property. """ def __init__(self,parent=None): self._parent = parent self._children = {} self._tag = None #if parent != None: # parent.add_child(self) def add_child(self, child_node): """Add a child to this TreeNode.""" self._children[child_node._tag] = child_node def find_match(self, tag): """Find a TreeNode related to this TreeNode which has the given tag. tag is a value corresponding to the ._tag field of TreeNode you're calling this on. REMEMBER: Different TreeNode subclasses have different formats of ._tags! The return value is a TreeNode with the specified ._tag value, or None. The TreeNodes searched are this node's immediate children, and its direct ancestors' immediate children. """ curr_node = self matching_node = None out_of_parents = False while matching_node is None and not out_of_parents: #matching_node = curr_node.get_template_match(tag)[0] #if matching_node == None: matching_node = curr_node.get_child(tag) if curr_node._parent is None: out_of_parents = True else: curr_node = curr_node._parent return matching_node def get_child(self, tag): """Return an immediate child with the given tag, or None if no such child exists.""" if tag in self._children.keys(): return self._children[tag] else: return None class TemplateNode(TreeNode): """An implementation of TreeNode specialized for holding tag templates. Members: self._parent = the parent TemplateNode of this TemplateNode. self._tag = the string representing a valid format of raw tag. Literals are themselves; colons separate tokens; $s indicate identifiers; ?s indicate graphics information; and &s indicate non-identifier, non-graphics information. For more fluidly-formatted tags, the following format indicates more than one tag: x(#,#) , where the x is a $, ? or &, the #s are a min/max pair, and the max is optional. self._is_graphics_tag """ #self._tag # string #self._children # dict of TemplateNodes #self._childref # dict of lists of TemplateNodes where the key is the first token #self._parent # TemplateNode #self._is_graphics_tag # Boolean #string does not contain the character '|'. def __init__(self, parent, string=""): TreeNode.__init__(self,parent) self._is_graphics_tag = False self._childref = {} global template_tree if parent is None: self._parent = None template_tree = self else: if template_tree == None: self._parent = TemplateNode(None, "") else: self._parent = parent self._tag = string parent.add_child(self) def add_child(self, node): if node._tag in self._children.keys(): return self._children[node._tag] else: self._children[node._tag] = node first_token = node._tag.split(':')[0] if first_token not in self._childref.keys(): self._childref[first_token] = [] self._childref[first_token].append(node) return node def get_child(self, tag): global verbose if tag in self._children.keys(): return self._children[tag] else: return_possibilities = [] first_token = tag.split(':')[0] if first_token in self._childref: for child in self._childref[first_token]: return_node = child.get_template_match(tag) if return_node != None: return_possibilities.append(child) if len(return_possibilities) == 1: return return_possibilities[0] elif len(return_possibilities) == 0: return None else: # TODO error handling if verbose: print("Found more than one matching child. Matching children are:") for poss in return_possibilities: print(poss) return return_possibilities[0] else: return None class TagNode(TreeNode): def __init__(self,filename,template,tag,parent=None): TreeNode.__init__(self, parent) self._tag = tag self._filename = filename self._template = template self._children = {} self._pat_children = {} self._pattern = None self._pattern = self.get_pattern() global graphics_tokens_by_file if parent != None: parent.add_child(self) def add_child(self, child_tag_node): self._children[child_tag_node._tag] = child_tag_node self._pat_children[child_tag_node.get_pattern()] = child_tag_node def apply_graphics(self, graphics_node): global verbose if graphics_node == None: return None tags = self._tag.split(':') graphics = graphics_node._tag.split(':') tag_template = self._template.get_template_match(self._tag)[0] merged = [] graphics_template = self._template.get_template_match(graphics_node._tag)[0] # print("Graphics cannot be applied from",graphics_node._tag,"onto",self._tag) for ii in range(0,len(tag_template)): if tag_template[ii] != graphics_template[ii]: if verbose: print("Graphics cannot be applied from",graphics_node._tag,"onto",self._tag,"because their templates do not match.") elif tag_template[ii] != '&' and tag_template[ii] != '?': if tags[ii] != graphics[ii]: if verbose: print("Tags are not compatible because token",ii,"does not match. Target:",tags[ii]," Graphics:",graphics[ii]) else: merged.append(tags[ii]) elif tag_template[ii] == '&': merged.append(tags[ii]) elif tag_template[ii] == '?': merged.append(graphics[ii]) else: if verbose: print("This block should never be reached. Big problem in TagNode.apply_graphics.") return ":".join(merged) # Too much work # def compatible_with(self, graphics_tag_node): # return self._template.resolve(self._tag,graphics_tag_node._tag) != None def get_pattern(self): global verbose if self._pattern == None: to_return = self._tag.split(':') tag_tokens = self._tag.split(':') template_possibilities = self._template.get_template_match(self._tag) if len(template_possibilities) != 1: if verbose: print("Tag",self._tag,"has",len(template_possibilities),"possible configurations:") for config in template_possibilities: print(config,', ') else: for ii in range(0,len(tag_tokens)): if template_possibilities[0][ii] in [tag_tokens[ii],'$']: to_return[ii] = tag_tokens[ii] elif template_possibilities[0][ii] in ['?','&']: to_return[ii] = template_possibilities[0][ii] elif verbose: print("Tag does not match its own template!! Tag:",self._tag,"; Template:",self._template._template_tag) self._pattern = ":".join(to_return) return self._pattern def aligns_with(self,other_tag): return self._template == other_tag._template and self.get_pattern() == other_tag.get_pattern() def is_graphics_tag(self): return self._template._is_graphics_tag class BoundNode(TreeNode): def __init__(self,target_node,graphics_node,parent=None): TreeNode.__init__(self, parent) self._tag = target_node._tag self._popped_children = {} self._additional = [] self._are_addl_popped = False self._target_node = target_node self._graphics_node = graphics_node if parent != None: parent.add_child(self) else: self.create_child_nodes() def add_child(self, child_node): self._children[child_node._target_node._tag]=child_node self._popped_children[child_node._target_node._tag]=False def create_child_nodes(self): # Children with pattern keys in both target & graphics for shared_key in set(self._target_node._pat_children.keys()).intersection(set(self._graphics_node._pat_children.keys())): new_node = BoundNode(self._target_node._pat_children[shared_key],self._graphics_node._pat_children[shared_key],self) self.add_child(new_node) new_node.create_child_nodes() # Children with pattern keys in target but not in graphics for target_key in set(self._target_node._pat_children.keys()) - set(self._graphics_node._pat_children.keys()): if self._target_node._pat_children[target_key].is_graphics_tag() and not ('&' in self._target_node._template._tag or '$' in self._target_node._template._tag): self.add_child(BoundNode(self._target_node._pat_children[target_key],None,self)) # Children with pattern keys in graphics but not in target for graphics_key in set(self._graphics_node._pat_children.keys()) - set(self._target_node._pat_children.keys()): graphics_in_question = self._graphics_node._pat_children[graphics_key] if graphics_in_question.is_graphics_tag(): self._additional.append(graphics_in_question) # End def are_all_children_popped(self): to_return = True for child in self._popped_children.keys(): to_return &= self._popped_children[child] return to_return def pop_addl(self): to_return = [] # If this hasn't had its additionals popped and is ready to pop if (not self._are_addl_popped): self._are_addl_popped = True to_return.extend(self._additional) if self._parent != None: to_return.extend(self._parent.pop_addl()) return to_return def get_merged(self): return self._target_node.apply_graphics(self._graphics_node) def pop_child(self, target_tag): if target_tag not in self._children.keys(): return self else: if self._popped_children[target_tag] and verbose: print("Popping tag that has already been popped:",target_tag,"child of",self._tag) self._popped_children[target_tag] = True return self._children[target_tag] def pop_self(self): if self._parent != None: also_self = self._parent.pop_child(self._tag) if also_self != self and verbose: print("Big problem: Bound Node with _tag",self._tag,"is not its parent's ._children[",self._tag,"]") return self def is_there_a_difference(self): if self._graphics_node == None: return True elif self._target_node._tag == self._graphics_node._tag: return False else: return True