Line | |
---|
1 | from evolalg.base.step import Step |
---|
2 | import os |
---|
3 | from framsfiles import writer as framswriter |
---|
4 | |
---|
5 | |
---|
6 | class PopulationSave(Step): |
---|
7 | def __init__(self, path, fields, provider=None): |
---|
8 | self.path = path |
---|
9 | self.provider = provider |
---|
10 | self.fields = fields |
---|
11 | |
---|
12 | @staticmethod |
---|
13 | def ensure_dir(path): |
---|
14 | directory = os.path.dirname(path) |
---|
15 | if directory == "": |
---|
16 | return |
---|
17 | if not os.path.exists(directory): |
---|
18 | os.makedirs(directory) |
---|
19 | |
---|
20 | def call(self, population): |
---|
21 | PopulationSave.ensure_dir(self.path) |
---|
22 | provider = self.provider |
---|
23 | if provider is None: |
---|
24 | provider = population |
---|
25 | |
---|
26 | #TODO instead of "fitness", write all fields used as fitness source with their original names (e.g. "velocity", "vertpos" etc.). In evaluation, set all attributes we get from Framsticks so that here we get all original names and values. Or, even better, introduce a dict field in Individual and assign to it everything that we get from Framsticks on evaluation (and add a filtering ability, i.e. when None - save all, when a list of field names - save only the enumerated fields) |
---|
27 | with open(self.path, "w") as outfile: |
---|
28 | for ind in provider: |
---|
29 | keyval = {} |
---|
30 | for k in self.fields: # construct a dictionary with criteria names and their values |
---|
31 | keyval[k] = getattr(ind, self.fields[k]) |
---|
32 | outfile.write(framswriter.from_collection({"_classname": "org", **keyval})) |
---|
33 | outfile.write("\n") |
---|
34 | |
---|
35 | return population |
---|
Note: See
TracBrowser
for help on using the repository browser.