feat: init project by moving these from my gist
This commit is contained in:
345
main.py
Normal file
345
main.py
Normal file
@@ -0,0 +1,345 @@
|
||||
import math
|
||||
import struct
|
||||
import os
|
||||
import sys
|
||||
|
||||
OneComponentList = []
|
||||
TwoComponentList = []
|
||||
ThreeComponentList = []
|
||||
|
||||
ResultList = []
|
||||
|
||||
class OneComponent(object):
|
||||
Value1 = 0.0
|
||||
|
||||
Value = 0.0
|
||||
def PrintCircuit(self):
|
||||
print(OutputAsHuman(self.Value1))
|
||||
|
||||
class TwoComponent(object):
|
||||
Value1 = 0.0
|
||||
Value2 = 0.0
|
||||
IsSeries = True
|
||||
|
||||
Value = 0.0
|
||||
def PrintCircuit(self):
|
||||
print("[{}] ┬ {}".format('S' if self.IsSeries else 'P', OutputAsHuman(self.Value1)))
|
||||
print(" └ {}".format(OutputAsHuman(self.Value2)))
|
||||
|
||||
class ThreeComponent(object):
|
||||
Value1 = 0.0
|
||||
Value2 = 0.0
|
||||
Value3 = 0.0
|
||||
IsSeries1 = True
|
||||
IsSeries2 = True
|
||||
|
||||
Value = 0.0
|
||||
def PrintCircuit(self):
|
||||
print("[{}] ┬ [{}] ┬ {}".format('S' if self.IsSeries2 else 'P', 'S' if self.IsSeries1 else 'P', OutputAsHuman(self.Value1)))
|
||||
print(" │ └ {}".format(OutputAsHuman(self.Value2)))
|
||||
print(" └ {}".format(OutputAsHuman(self.Value3)))
|
||||
|
||||
class ResultStruct(object):
|
||||
Subtraction = 0.0
|
||||
ComponentCount = 0
|
||||
CorrespondingClass = None
|
||||
|
||||
def LoadFromFile(name):
|
||||
f = open(name, 'r', encoding = 'utf-8')
|
||||
|
||||
# read file to get one elements state
|
||||
while True:
|
||||
cache = f.readline()
|
||||
if cache == '':
|
||||
break
|
||||
|
||||
(status, value) = InputAsHuman(cache.strip())
|
||||
if status:
|
||||
newobj = None
|
||||
newobj = OneComponent()
|
||||
newobj.Value = value
|
||||
newobj.Value1 = value
|
||||
OneComponentList.append(newobj)
|
||||
|
||||
# construct two component list
|
||||
length = len(OneComponentList)
|
||||
for i_index in range(length):
|
||||
for j_index in range(i_index, length):
|
||||
i = OneComponentList[i_index]
|
||||
j = OneComponentList[j_index]
|
||||
|
||||
# series
|
||||
newobj = TwoComponent()
|
||||
newobj.Value1 = i.Value
|
||||
newobj.Value2 = j.Value
|
||||
newobj.IsSeries = True
|
||||
newobj.Value = i.Value + j.Value
|
||||
TwoComponentList.append(newobj)
|
||||
|
||||
# parallel
|
||||
newobj = TwoComponent()
|
||||
newobj.Value1 = i.Value
|
||||
newobj.Value2 = j.Value
|
||||
newobj.IsSeries = False
|
||||
newobj.Value = (i.Value * j.Value) / (i.Value + j.Value)
|
||||
TwoComponentList.append(newobj)
|
||||
|
||||
# construct three component list
|
||||
for i in OneComponentList:
|
||||
for j in TwoComponentList:
|
||||
# series
|
||||
newobj = ThreeComponent()
|
||||
newobj.Value1 = j.Value1
|
||||
newobj.Value2 = j.Value2
|
||||
newobj.Value3 = i.Value
|
||||
newobj.IsSeries1 = j.IsSeries
|
||||
newobj.IsSeries2 = True
|
||||
newobj.Value = j.Value + i.Value
|
||||
ThreeComponentList.append(newobj)
|
||||
|
||||
# parallel
|
||||
newobj = ThreeComponent()
|
||||
newobj.Value1 = j.Value1
|
||||
newobj.Value2 = j.Value2
|
||||
newobj.Value3 = i.Value
|
||||
newobj.IsSeries1 = j.IsSeries
|
||||
newobj.IsSeries2 = False
|
||||
newobj.Value = (j.Value * i.Value) / (j.Value + i.Value)
|
||||
ThreeComponentList.append(newobj)
|
||||
|
||||
f.close()
|
||||
SaveAsCache(name)
|
||||
|
||||
def LoadFromCache(name):
|
||||
f = open(name + '.cache', 'rb')
|
||||
counter = 0
|
||||
(counter, ) = struct.unpack("I", f.read(4))
|
||||
for i in range(counter):
|
||||
newobj = OneComponent()
|
||||
newobj.Value1 = ReadDouble(f)
|
||||
newobj.Value = ReadDouble(f)
|
||||
OneComponentList.append(newobj)
|
||||
(counter, ) = struct.unpack("I", f.read(4))
|
||||
for i in range(counter):
|
||||
newobj = TwoComponent()
|
||||
newobj.Value1 = ReadDouble(f)
|
||||
newobj.Value2 = ReadDouble(f)
|
||||
newobj.IsSeries = ReadBoolean(f)
|
||||
newobj.Value = ReadDouble(f)
|
||||
TwoComponentList.append(newobj)
|
||||
(counter, ) = struct.unpack("I", f.read(4))
|
||||
for i in range(counter):
|
||||
newobj = ThreeComponent()
|
||||
newobj.Value1 = ReadDouble(f)
|
||||
newobj.Value2 = ReadDouble(f)
|
||||
newobj.Value3 = ReadDouble(f)
|
||||
newobj.IsSeries1 = ReadBoolean(f)
|
||||
newobj.IsSeries2 = ReadBoolean(f)
|
||||
newobj.Value = ReadDouble(f)
|
||||
ThreeComponentList.append(newobj)
|
||||
|
||||
f.close()
|
||||
|
||||
def SaveAsCache(name):
|
||||
# in cache, is series should follow resistor mode
|
||||
f = open(name + '.cache', 'wb')
|
||||
WriteInt(f, len(OneComponentList))
|
||||
for i in OneComponentList:
|
||||
WriteDouble(f, i.Value1)
|
||||
WriteDouble(f, i.Value)
|
||||
WriteInt(f, len(TwoComponentList))
|
||||
for i in TwoComponentList:
|
||||
WriteDouble(f, i.Value1)
|
||||
WriteDouble(f, i.Value2)
|
||||
WriteBoolean(f, i.IsSeries)
|
||||
WriteDouble(f, i.Value)
|
||||
WriteInt(f, len(ThreeComponentList))
|
||||
for i in ThreeComponentList:
|
||||
WriteDouble(f, i.Value1)
|
||||
WriteDouble(f, i.Value2)
|
||||
WriteDouble(f, i.Value3)
|
||||
WriteBoolean(f, i.IsSeries1)
|
||||
WriteBoolean(f, i.IsSeries2)
|
||||
WriteDouble(f, i.Value)
|
||||
|
||||
f.close()
|
||||
|
||||
def ReadDouble(fs):
|
||||
return struct.unpack("d", fs.read(8))[0]
|
||||
|
||||
def ReadInt(fs):
|
||||
return struct.unpack("I", fs.read(4))[0]
|
||||
|
||||
def ReadBoolean(fs):
|
||||
return struct.unpack("?", fs.read(1))[0]
|
||||
|
||||
def WriteDouble(fs, num):
|
||||
fs.write(struct.pack("d", num))
|
||||
|
||||
def WriteInt(fs, num):
|
||||
fs.write(struct.pack("I", num))
|
||||
|
||||
def WriteBoolean(fs, num):
|
||||
fs.write(struct.pack("?", num))
|
||||
|
||||
def OutputAsHuman(v):
|
||||
if v / 1e-12 < 1e3:
|
||||
return "{:e} n".format(v / 1e-12)
|
||||
if v / 1e-9 < 1e3:
|
||||
return "{:.4f} p".format(v / 1e-9)
|
||||
if v / 1e-6 < 1e3:
|
||||
return "{:.4f} u".format(v / 1e-6)
|
||||
if v / 1e-3 < 1e3:
|
||||
return "{:.4f} m".format(v / 1e-3)
|
||||
if v < 1e3:
|
||||
return "{:.4f}".format(v)
|
||||
if v / 1e3 < 1e3:
|
||||
return "{:.4f} k".format(v / 1e3)
|
||||
if v / 1e6 < 1e3:
|
||||
return "{:.4f} M".format(v / 1e6)
|
||||
|
||||
return "{:e}".format(v)
|
||||
|
||||
def InputAsHuman(strl):
|
||||
try:
|
||||
if strl.endswith('n'):
|
||||
return (True, float(strl[0:-1]) * 1e-12)
|
||||
if strl.endswith('p'):
|
||||
return (True, float(strl[0:-1]) * 1e-9)
|
||||
if strl.endswith('u'):
|
||||
return (True, float(strl[0:-1]) * 1e-6)
|
||||
if strl.endswith('m'):
|
||||
return (True, float(strl[0:-1]) * 1e-3)
|
||||
if strl.endswith('k'):
|
||||
return (True, float(strl[0:-1]) * 1e3)
|
||||
if strl.endswith('M'):
|
||||
return (True, float(strl[0:-1]) * 1e6)
|
||||
return (True, float(strl))
|
||||
except:
|
||||
return (False, 0.0)
|
||||
|
||||
def ValidCommandInput(valid_list):
|
||||
while True:
|
||||
cache = input()
|
||||
if cache not in valid_list:
|
||||
print('Wrong command, please input again.')
|
||||
else:
|
||||
return cache
|
||||
|
||||
def ValidNumberInput():
|
||||
while True:
|
||||
cache = input()
|
||||
(status, num) = InputAsHuman(cache)
|
||||
if not status:
|
||||
print('Wrong number, please input again.')
|
||||
else:
|
||||
return num
|
||||
|
||||
def DoQuery():
|
||||
# get config
|
||||
|
||||
print('What are you connecting?')
|
||||
print('r: resistor')
|
||||
print('l: inductor')
|
||||
print('c: capacitor')
|
||||
mode = ValidCommandInput(['c', 'l', 'r'])
|
||||
is_resistor_mode = mode != 'c'
|
||||
|
||||
print('Your target value?')
|
||||
target = ValidNumberInput()
|
||||
|
||||
print('Your tolerance?')
|
||||
target_tolerance = ValidNumberInput()
|
||||
|
||||
print('How to sort result?')
|
||||
print('l: less component')
|
||||
print('a: more accuracy')
|
||||
sort_mode = ValidCommandInput(['l', 'a'])
|
||||
|
||||
# start computing
|
||||
print('Collecting and sorting data...')
|
||||
ResultList.clear()
|
||||
for i in OneComponentList:
|
||||
cache = i.Value - target
|
||||
if abs(cache) < target_tolerance:
|
||||
newobj = ResultStruct()
|
||||
newobj.Subtraction = cache
|
||||
newobj.ComponentCount = 1
|
||||
newobj.CorrespondingClass = i
|
||||
ResultList.append(newobj)
|
||||
for i in TwoComponentList:
|
||||
cache = i.Value - target
|
||||
if abs(cache) < target_tolerance:
|
||||
newobj = ResultStruct()
|
||||
newobj.Subtraction = cache
|
||||
newobj.ComponentCount = 2
|
||||
newobj.CorrespondingClass = i
|
||||
ResultList.append(newobj)
|
||||
for i in ThreeComponentList:
|
||||
cache = i.Value - target
|
||||
if abs(cache) < target_tolerance:
|
||||
newobj = ResultStruct()
|
||||
newobj.Subtraction = cache
|
||||
newobj.ComponentCount = 3
|
||||
newobj.CorrespondingClass = i
|
||||
ResultList.append(newobj)
|
||||
|
||||
if sort_mode == 'l':
|
||||
ResultList.sort(key = lambda x: (x.ComponentCount, abs(x.Subtraction)))
|
||||
else:
|
||||
ResultList.sort(key = lambda x: abs(x.Subtraction))
|
||||
|
||||
# display result
|
||||
if len(ResultList) == 0:
|
||||
print('Sorry, no result!')
|
||||
return
|
||||
|
||||
count = len(ResultList)
|
||||
all_page = int(count / 10)
|
||||
current_page = 0
|
||||
while current_page <= all_page:
|
||||
for i in range(9):
|
||||
picked_index = current_page * 9 + i
|
||||
if (picked_index < count):
|
||||
picked_item = ResultList[picked_index]
|
||||
print("Plan {}\t{}\t{:.2%}".format(picked_index + 1, OutputAsHuman(picked_item.CorrespondingClass.Value), picked_item.Subtraction / target))
|
||||
picked_item.CorrespondingClass.PrintCircuit()
|
||||
|
||||
print('')
|
||||
print("Page {} of {}. p: next page. q: exit".format(current_page + 1, all_page + 1))
|
||||
command = ValidCommandInput(['p', 'q'])
|
||||
if command == 'p':
|
||||
current_page += 1
|
||||
elif command == 'q':
|
||||
break
|
||||
|
||||
def DoHelp():
|
||||
print('LCR Connect Help:')
|
||||
print('')
|
||||
print('query: do a query immediately. following the guider and find the result.')
|
||||
print('help: print this')
|
||||
print('exit: exit this app')
|
||||
|
||||
# ===================================================== program start
|
||||
|
||||
print('LCR Connect')
|
||||
|
||||
# loading file
|
||||
print('Input the component value list file name:')
|
||||
filename = input()
|
||||
if not os.path.isfile(filename + '.cache'):
|
||||
LoadFromFile(filename)
|
||||
else:
|
||||
LoadFromCache(filename)
|
||||
|
||||
# ready for command
|
||||
while True:
|
||||
sys.stdout.write('> ')
|
||||
sys.stdout.flush()
|
||||
cmd = ValidCommandInput(['exit', 'query', 'help'])
|
||||
if cmd == 'exit':
|
||||
break
|
||||
elif cmd == 'query':
|
||||
DoQuery()
|
||||
elif cmd == 'help':
|
||||
DoHelp()
|
||||
Reference in New Issue
Block a user