[21303] | 1 | import numpy as np
|
---|
[12038] | 2 | from fielddisplay import fielddisplay
|
---|
[17497] | 3 | from checkfield import checkfield
|
---|
| 4 | from WriteData import WriteData
|
---|
[12038] | 5 |
|
---|
[24213] | 6 |
|
---|
[12958] | 7 | class steadystate(object):
|
---|
[24213] | 8 | """
|
---|
| 9 | STEADYSTATE class definition
|
---|
[13023] | 10 |
|
---|
[24213] | 11 | Usage:
|
---|
| 12 | steadystate = steadystate()
|
---|
| 13 | """
|
---|
[13023] | 14 |
|
---|
[24213] | 15 | def __init__(self): # {{{
|
---|
| 16 | self.reltol = 0
|
---|
| 17 | self.maxiter = 0
|
---|
| 18 | self.requested_outputs = []
|
---|
[12123] | 19 |
|
---|
[24213] | 20 | #set defaults
|
---|
| 21 | self.setdefaultparameters()
|
---|
[12123] | 22 |
|
---|
[27458] | 23 | # }}}
|
---|
[24213] | 24 | def __repr__(self): # {{{
|
---|
| 25 | string = ' steadystate solution parameters:'
|
---|
| 26 | string = "%s\n%s" % (string, fielddisplay(self, 'reltol', 'relative tolerance criterion'))
|
---|
| 27 | string = "%s\n%s" % (string, fielddisplay(self, 'maxiter', 'maximum number of iterations'))
|
---|
| 28 | string = "%s\n%s" % (string, fielddisplay(self, 'requested_outputs', 'additional requested outputs'))
|
---|
| 29 | return string
|
---|
[27458] | 30 | # }}}
|
---|
[16470] | 31 |
|
---|
[24213] | 32 | def defaultoutputs(self, md): # {{{
|
---|
| 33 | return md.stressbalance.defaultoutputs(md) + md.thermal.defaultoutputs(md)
|
---|
[16470] | 34 |
|
---|
[27458] | 35 | # }}}
|
---|
[24213] | 36 | def setdefaultparameters(self): # {{{
|
---|
| 37 | #maximum of steady state iterations
|
---|
| 38 | self.maxiter = 100
|
---|
| 39 | #Relative tolerance for the steadystate convertgence
|
---|
| 40 | self.reltol = 0.01
|
---|
| 41 | #default output
|
---|
| 42 | self.requested_outputs = ['default']
|
---|
| 43 | return self
|
---|
[27458] | 44 | # }}}
|
---|
[12123] | 45 |
|
---|
[24213] | 46 | def checkconsistency(self, md, solution, analyses): # {{{
|
---|
| 47 | #Early return
|
---|
| 48 | if not solution == 'SteadystateSolution':
|
---|
| 49 | return md
|
---|
[12123] | 50 |
|
---|
[24213] | 51 | if not md.timestepping.time_step == 0:
|
---|
| 52 | md.checkmessage("for a steadystate computation, timestepping.time_step must be zero.")
|
---|
[13023] | 53 |
|
---|
[24213] | 54 | if np.isnan(md.stressbalance.reltol):
|
---|
| 55 | md.checkmessage("for a steadystate computation, stressbalance.reltol (relative convergence criterion) must be defined!")
|
---|
[13023] | 56 |
|
---|
[24213] | 57 | md = checkfield(md, 'fieldname', 'steadystate.requested_outputs', 'stringrow', 1)
|
---|
[13023] | 58 |
|
---|
[24213] | 59 | return md
|
---|
| 60 | # }}}
|
---|
[13023] | 61 |
|
---|
[24213] | 62 | def marshall(self, prefix, md, fid): # {{{
|
---|
| 63 | WriteData(fid, prefix, 'object', self, 'fieldname', 'reltol', 'format', 'Double')
|
---|
| 64 | WriteData(fid, prefix, 'object', self, 'fieldname', 'maxiter', 'format', 'Integer')
|
---|
[16385] | 65 |
|
---|
[24213] | 66 | #process requested outputs
|
---|
| 67 | outputs = self.requested_outputs
|
---|
| 68 | indices = [i for i, x in enumerate(outputs) if x == 'default']
|
---|
| 69 | if len(indices) > 0:
|
---|
| 70 | outputscopy = outputs[0:max(0, indices[0] - 1)] + self.defaultoutputs(md) + outputs[indices[0] + 1:]
|
---|
| 71 | outputs = outputscopy
|
---|
| 72 | WriteData(fid, prefix, 'data', outputs, 'name', 'md.steadystate.requested_outputs', 'format', 'StringArray')
|
---|
| 73 | # }}}
|
---|