def main():
parser = OptionParser()
parser.add_option("-f",
"--file",
dest='networkName',
help="open file with networkName",
metavar="FILE")
parser.add_option("-s",
"--save",
dest="save",
help="save solution data, 0 = False, 1 = True")
parser.add_option(
"-n",
"--dataNumber",
dest='dataNumber',
help=
"number of the solution data (last number in filename), default = 1, max 999",
metavar="FILE")
parser.add_option(
'-v',
'--vizBool',
dest='vizBool',
help=
"choose visualisation mode, 0: no visualisation, 1: 2d and 3d, 2: 2d plots, 3: 3d visualisation default = 1"
)
(options, args) = parser.parse_args()
networkName = 'oldNetwork1'
if options.networkName != None:
networkName = options.networkName
else:
print " no networkName passed, visualisation could not be intialized"
print " use -f networkName to define an file you want to visualize"
exit()
filename = str(networkName + '.xml')
dataNumber = '001'
if options.dataNumber != None:
dataNumber = options.dataNumber
if len(dataNumber) == 1:
dataNumber = '00' + dataNumber
if len(dataNumber) == 2:
dataNumber = '0' + dataNumber
save = False
if options.save != None:
if options.save == '0':
save = False
elif options.save == '1':
save = True
vizBool = '1'
vizBool2d = True
vizBool3d = True
print options.vizBool
if options.vizBool != None:
if options.vizBool == '0':
vizBool2d = False
vizBool3d = False
elif options.vizBool == '1':
vizBool2d = True
vizBool3d = True
elif options.vizBool == '2':
vizBool2d = True
vizBool3d = False
elif options.vizBool == '3':
vizBool2d = False
vizBool3d = True
# load network from the path!
Lstart = time.clock()
vascularNetwork = loadNetworkFromXML(filename=filename)
if vascularNetwork == None: exit()
# update the connections of the network, as they prob weren't saved correct
vascularNetwork.evaluateConnections()
Lend = time.clock()
print " needed %1.3f sec to load" % (Lend - Lstart)
vascularNetwork.initialize()
#create visualisation
if vizBool3d == True:
visualisation = Visualisation(vascularNetwork=vascularNetwork)
#print '====================================='
#print '_________ Simulation Context _________'
#print '%-25s %-20s' % ('Numerical scheme',networkSolver['NumScheme'])
#print '%-25s %4d' % ('Equation system',int(networkSolver['EqSystem']))
#print '%-25s %4d' % ('Characteristic system',int(networkSolver['CharSystem']))
#print '________ Visualisation Context _______'
Cstart = time.clock()
flowSolver = FlowSolver(vascularNetwork)
P, Q, A = flowSolver.solve()
print '\n____________ Solver time _____________'
minutes = int((time.clock() - Cstart) / 60.)
secs = (time.clock() - Cstart) - float(minutes) * 60.
print '%d %s %f %s' % (minutes, 'min', secs, 'sec')
if save == True:
#save solution data in c pickle
#standart solution path
networkPath = str(cur + "/NetworkFiles/")
networkDirectory = filename.split('.')[0]
solutionPath = str(networkPath + networkDirectory + '/')
# data to save
solutionDataSave = [{
'Pressure': P,
'Flow': Q,
'Area': A,
'WaveSpeed': Q,
'Name': str('simulation_' + dataNumber)
}]
#create file with filename in soultionPath
endName = "_SolutionData_" + dataNumber + ".pickle"
FILE = open(str(solutionPath + vascularNetwork.name + endName), "w")
# store pickle
cPickle.dump(solutionDataSave, FILE, protocol=2)
FILE.close()
if vizBool2d == True:
string = ' '.join([
'python', cur + '/Visualisation/class2dVisualisationSimple.py',
'-f', vascularNetwork.name, '-n 1', '-i 0'
])
subprocess.call(string, shell=True)
if vizBool3d == True:
# create solution data
solutionData = [{
'Pressure': P,
'Flow': Q,
'Area': A,
'Name': str('simulation ' + dataNumber)
}]
#set the solution data
visualisation.setSolutionData(solutionData)
visualisation.factor = 10
# set the coefficient for the velocity profile
visualisation.powerLawCoefficient = 2
visualisation.visualize()
# clear memory
visualisation.__del__()
del (visualisation)
