xevion/fusion-fission-choices
1
0
Fork 0
mirror of https://github.com/Xevion/fusion-fission-choices.git synced 2025-12-10 14:07:25 -06:00
Code Issues Packages Projects Releases Wiki Activity

minor documentation, noNegatives config option

Browse Source
This commit is contained in:
Xevion
2019-08-11 03:00:22 -05:00
parent d09c76fb33
commit ee0889029b
2 changed files with 18 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
config.json
View File

@@ -121,5 +121,6 @@
}, },
"minimumScore": 1, "minimumScore": 1,
"reverseSorting": false, "reverseSorting": false,
"noNegatives" : true,
"fullPrint" : true "fullPrint" : true
} }

25
main.py
View File

@@ -3,11 +3,11 @@ import os, sys, json, re, mendeleev as mdv
path = os.path.join(sys.path[0], 'config.json') path = os.path.join(sys.path[0], 'config.json')
config = json.load(open(path, 'r')) config = json.load(open(path, 'r'))
def score(item): def score(element):
return config['elements'][str(item[0])] + config['elements'][str(item[1])], item return config['elements'][str(element)]
def arrange(item): def scoreSum(item):
return (max(item), min(item)) return (score(item[0]) + score(item[1])), item
def formatting(e1, e2): def formatting(e1, e2):
e1, e2 = mdv.element(e1), mdv.element(e2) e1, e2 = mdv.element(e1), mdv.element(e2)
@@ -16,6 +16,9 @@ def formatting(e1, e2):
e2.symbol if not config['fullPrint'] else e2.name, e2.symbol if not config['fullPrint'] else e2.name,
e2.atomic_number) e2.atomic_number)
def hasNegatives(item):
return score(item[0]) < 0 or score(item[1]) < 0
def getElement(element): def getElement(element):
if re.match(r'\d+', element): if re.match(r'\d+', element):
return mdv.element(int(element)) return mdv.element(int(element))
@@ -29,11 +32,17 @@ def getElement(element):
def bestSelection(select): def bestSelection(select):
select = select.atomic_number select = select.atomic_number
# Create a set, rearrange by (larger, smaller), then remove duplicates
posi = [(x, select - x) for x in range(1, select)] posi = [(x, select - x) for x in range(1, select)]
posi = list(map(arrange, posi)) posi = list(map(lambda item : (max(item), min(item)), posi))
posi = list(set(posi)) posi = list(dict.fromkeys(posi))
posi = list(filter(lambda item : score(item)[0] >= config['minimumScore'], posi)) # Filter out element tuples that don't have both with scores
posi.sort(key=lambda item : score(item), reverse=not config['reverseSorting']) if config['noNegatives']:
posi = list(filter(lambda item : hasNegatives(item), posi))
# Filter out elements that do not meet the minimum set
posi = list(filter(lambda item : scoreSum(item)[0] >= config['minimumScore'], posi))
# Sort the elements by the sum of their scores
posi.sort(key=lambda item : scoreSum(item), reverse=not config['reverseSorting'])
string1 = '[Best Elements for Element {}, {}]'.format(select, mdv.element(select).name) string1 = '[Best Elements for Element {}, {}]'.format(select, mdv.element(select).name)
string2 = '\n'.join([formatting(element1, element2) for element1, element2 in posi]) string2 = '\n'.join([formatting(element1, element2) for element1, element2 in posi])