interview.py 48.5 KB
Newer Older
Theresa Pollinger's avatar
Theresa Pollinger committed
1
2
#!/usr/bin/env python3

3
# http://cmd2.readthedocs.io
Theresa Pollinger's avatar
Theresa Pollinger committed
4
import cmd2 as cmd
5
# https://github.com/pytransitions/transitions
Theresa Pollinger's avatar
Theresa Pollinger committed
6
7
from transitions import Machine, State
from collections import OrderedDict
8
9
# strings:
# http://mattoc.com/python-yes-no-prompt-cli.html
Theresa Pollinger's avatar
Theresa Pollinger committed
10
11
from distutils.util import strtobool
from pathlib import Path
12
# https://github.com/phfaist/pylatexenc for directly converting Latex commands to unicode
Theresa Pollinger's avatar
Theresa Pollinger committed
13
14
15
16
from pylatexenc.latex2text import LatexNodes2Text
import pyparsing as pp
import re

17
from string_handling import *
Theresa Pollinger's avatar
Theresa Pollinger committed
18
19
20
from exaoutput import ExaOutput
from mmtinterface import *

21

22
23
24
class InterviewError(Exception):
    def __init__(self, err):
        self.error = err
25
        super(InterviewError, self).__init__("Interview error: " + str(self.error))
26

27

28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
###For the part of the simdata whose contents will be cleaned up if there was an error
###to be used in with statements
class CriticalSubdict():
    def __init__(self, subdict):
        self.subdict = subdict
        self.initial_subdict = self.subdict.copy()

    def __enter__(self):
        return self.subdict

    def __exit__(self, type, value, traceback):
        if type is not None:
            #restore the initial state
            self.subdict.clear()
            for key in self.initial_subdict:
                self.subdict[key] = self.initial_subdict[key]
            print(value)
            if isinstance(value, MMTServerError) or isinstance(value, InterviewError):
46
                self.please_repeat(value.args[0])
47
                return True
48
49
50
51
52
53
54
55
56
            else:
                return False
        return True

    def please_repeat(self, moreinfo=None):
        append = ""
        if moreinfo:
            append = "\nDetails: " + moreinfo
        print("I did not catch that. Could you please rephrase?" + append)
Theresa Pollinger's avatar
Theresa Pollinger committed
57

58
59
60
61

# This "main class" is two things: a REPL loop, by subclassing the cmd2 Cmd class
# and a state machine as given by the pytransitions package
class Interview(cmd.Cmd):
Theresa Pollinger's avatar
Theresa Pollinger committed
62
    def __init__(self, *args, **kwargs):
63
        # just act like we were getting the right replies from MMT
64
        self.cheating = True
Theresa Pollinger's avatar
Theresa Pollinger committed
65
66
67
        # initialize legal characters for cmd
        self.legalChars = u'!#$%.:;?@_-<>' + pp.printables + pp.alphas8bit + pp.punc8bit
        # TODO why does "<" not show?
68
69
        # allow all useful unicode characters to be used, and some more
        for i in range(0x20, 0x2E7F):
Theresa Pollinger's avatar
Theresa Pollinger committed
70
71
72
73
74
75
76
            self.legalChars += chr(i)

        # call cmd constructor
        super(Interview, self).__init__(*args, **kwargs)

        # Initialize a state machine
        states = [
77
            # State('greeting'),
Theresa Pollinger's avatar
Theresa Pollinger committed
78
79
80
81
82
            State('dimensions', on_enter=['dimensions_begin']),
            State('domain', on_enter=['domain_begin'], on_exit=['domain_exit']),
            State('unknowns', on_enter=['unknowns_begin'], on_exit=['unknowns_exit']),
            State('parameters', on_enter=['parameters_begin'], on_exit=['parameters_exit']),
            State('pdes', on_enter=['pdes_begin'], on_exit=['pdes_exit']),
83
            State('bcs', on_enter=['bcs_begin'], on_exit=['bcs_exit']),
84
85
            State('props', on_enter=['props_begin'], on_exit=['props_exit']),
            State('sim', on_enter=['sim_begin'], on_exit=['sim_exit']),
Theresa Pollinger's avatar
Theresa Pollinger committed
86
87
88
89
        ]
        states.reverse()
        self.machine = Machine(model=self, states=states, initial=states[-1], after_state_change='update_prompt')
        # this is why we were reverting the states => can always go back
90
91
92
93
94
95
96
97
98
99
100
101
        self.machine.add_ordered_transitions(
            trigger='last_state')  # TODO do something to avoid going back from the first state
        # self.to_dimensions()
        # self.machine.add_transition(trigger='greeting_over', source='greeting', dest='dimensions')
        self.machine.add_transition(trigger='dimensions_parsed', source='dimensions', dest='domain',
                                    before='print_empty_line')
        self.machine.add_transition(trigger='domain_parsed', source='domain', dest='unknowns',
                                    before='print_empty_line')
        self.machine.add_transition(trigger='unknowns_parsed', source='unknowns', dest='parameters',
                                    after='print_empty_line')
        self.machine.add_transition(trigger='parameters_parsed', source='parameters', dest='pdes',
                                    after='print_empty_line')
Theresa Pollinger's avatar
Theresa Pollinger committed
102
103
        self.machine.add_transition(trigger='pdes_parsed', source='pdes', dest='bcs', before='print_empty_line')
        self.machine.add_transition(trigger='bcs_parsed', source='bcs', dest='sim', before='print_empty_line')
104
        #self.machine.add_transition(trigger='props_parsed', source='props', dest='sim', before='print_empty_line')#TODO props
105
        self.machine.add_transition(trigger='sim_finished', source='sim', dest='sim', before='print_empty_line')
Theresa Pollinger's avatar
Theresa Pollinger committed
106
107

        # Initialize cmd member variables
108
        self.myname = 'TheInterview'
Theresa Pollinger's avatar
Theresa Pollinger committed
109
110
        self.username = 'user'
        self.intro = "Hello, " + self.username + "! I am " + self.myname + ", your partial differential equations and simulations expert. " \
111
                                                                           "Let's set up a simulation together.\n" \
112
                                                                           "How many dimensions does your model have?"
Theresa Pollinger's avatar
Theresa Pollinger committed
113

114
        # define what happens when input is received in a certain state
Theresa Pollinger's avatar
Theresa Pollinger committed
115
116
117
118
        self.stateDependentInputHandling = {
            'dimensions': self.dimensions_handle_input,
            'domain': self.domain_handle_input,
            'unknowns': self.unknowns_handle_input,
119
            'parameters': self.parameters_handle_input,
120
121
            'pdes': self.pdes_handle_input,
            'bcs': self.bcs_handle_input,
122
            'props': self.props_handle_input,
123
            'sim': self.sim_handle_input,
Theresa Pollinger's avatar
Theresa Pollinger committed
124
125
126
127
        }

        self.mmtinterface = MMTInterface()

128
        # for ladder-like views
Theresa Pollinger's avatar
Theresa Pollinger committed
129
        self.viewfrom = OrderedDict([
130
131
132
133
134
            ('domain', "mDomain"),
            ('unknowns', "mUnknown"),
            ('parameters', "mParameter"),
            ('pdes', "mPDE"),
            ('bcs', "mBCsRequired"),
135
136
            ('props', "mEllipticLinearDirichletBoundaryValueProblem"),
            ('sim', "mSolvability"),
Theresa Pollinger's avatar
Theresa Pollinger committed
137
138
139
        ])
        # to include all the necessary theories every time
        self.bgthys = OrderedDict([
140
            ('domain', ["mInterval", "http://mathhub.info/MitM/smglom/arithmetics?realarith"]),
141
142
143
            # new: RealArithmetics
            ('unknowns', ["http://mathhub.info/MitM/Foundation?Strings", "ephdomain",
                          "http://mathhub.info/MitM/smglom/calculus?higherderivative"]),
144
145
            ('parameters', ["http://mathhub.info/MitM/smglom/arithmetics?realarith", "ephdomain",
                            "http://mathhub.info/MitM/Foundation?Math"]),
146
            ('pdes', ["mDifferentialOperators"]),#+params, unknowns,
147
            ('bcs',
148
149
             ["ephdomain", "mLinearity",
              "http://mathhub.info/MitM/smglom/arithmetics?realarith"]),#+params, unknowns, pdes, bctypes
150
            ('props',
151
152
             ["mLinearity",
              "http://mathhub.info/MitM/Foundation?Strings"]), #+bcs, pde
153
154
            ('sim',
             ["http://mathhub.info/MitM/Foundation?Strings"]), #+props
Theresa Pollinger's avatar
Theresa Pollinger committed
155
156
        ])

157
        # the things we'd like to find out
Theresa Pollinger's avatar
Theresa Pollinger committed
158
        self.simdata = {
159
160
161
162
163
164
165
166
            "num_dimensions": None,
            "domain": {
                "name": None,
                "theoryname": None,
                # "viewname" : None,
                "axes": OrderedDict(),
                "from": None,
                "to": None,
Theresa Pollinger's avatar
Theresa Pollinger committed
167
            },
168
            "unknowns": OrderedDict(),
169
            "parameters": OrderedDict(),
170
            "pdes": {
171
                #               "theoryname": None,
172
                "pdes": [],
Theresa Pollinger's avatar
Theresa Pollinger committed
173
            },
174
175
176
177
            "bcs": {
                "theoryname": None,
                "bcs": None,
            },
178
179
180
181
182
183
            "props": {

            },
            "sim" : {
                "type": None,
            },
Theresa Pollinger's avatar
Theresa Pollinger committed
184
185
186
187
188
189
        }

        axes = OrderedDict([
            ("x_1", "[0;1]"),
        ])
        self.examplesimdata = {
190
191
192
193
194
            "num_dimensions": 1,
            "domain": {
                "name": "Ω",
                "theoryname": "Omega",
                "axes": axes,  # names and intervals
195
196
                "from": 0.0,
                "to": 1.0,
197
198
199
200
201
            },
            "unknowns": {  # names and theorynames #TODO OrderedDict
                "u": {
                    "theoryname": "u",
                    "string": "u : Ω → ℝ",
Theresa Pollinger's avatar
Theresa Pollinger committed
202
                },
203
204
205
206
207
            },
            "parameters": {  # names and theorynames
                "μ": {
                    "theoryname": "mu",
                    "string": "μ : ℝ = 1",
Theresa Pollinger's avatar
Theresa Pollinger committed
208
                },
209
210
211
                "f": {
                    "theoryname": "f",
                    "string": "f : Ω → ℝ = [x] x ⋅ x",
Theresa Pollinger's avatar
Theresa Pollinger committed
212
213
                },
            },
214
215
            "pdes": {
                "pdes": [
216
                    {
217
                        "theoryname": "pde1",
218
219
220
221
222
223
224
                        "string": "μ ∆u = f(x)",  # TODO use function arithmetic
                        'lhsstring': 'μ Δu ',
                        'rhsstring': 'f(x)',
                        'op': 'Δ',
                        'lhsparsestring': ' [ anyu : Ω → ℝ ] Δ anyu ',
                        'rhsparsestring': ' [ x : Ω ]  f(x)',
                        # this is more of a wish list... cf https://github.com/UniFormal/MMT/issues/295
225
226
227
                        "expanded": "μ d²/dx_1² u = f(x_1)",
                        "order_in_unknown": {
                            "u": 2,
Theresa Pollinger's avatar
Theresa Pollinger committed
228
                        },
229
                    },
Theresa Pollinger's avatar
Theresa Pollinger committed
230
                ],
231
            },
232
233
234
235
236
237
238
            "bcs": {
                "theoryname": "ephbcs",
                "bcs": [
                    {
                        "name": "bc1",
                        "type": "Dirichlet",
                        "string": "u (0) = x_1**2",
239
                        "on": "0",
240
241
242
243
244
                    },
                    {
                        "name": "bc2",
                        "type": "Dirichlet",
                        "string": "u (1) = x_1**2",
245
                        "on": "1",
246
247
248
                    },
                ],
            },
249
250
251
252
253
254
255
256
257
258
259
260
261
            "props": {
                "theoryname": "ephboundaryvalueproblem",
                "ops": [
                    {
                        "name": "op1",
                        "linear": True, #or false or unknown
                        "props": ["elliptic"]
                    }
                ]
            },
            "sim":{
                "type": "FD",
            },
Theresa Pollinger's avatar
Theresa Pollinger committed
262
263
        }

264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
        self.testsimdata = {
            'num_dimensions': 1,
            'domain': {'name': 'Ω', 'theoryname': 'ephdomain', 'axes': OrderedDict([('x_1', '[0.0;1.0]')]),
                        'from': 0.0, 'to': 1.0, 'boundary_name': 'Ω', 'viewname': 'ephdomainASmDomain'},
            'unknowns': OrderedDict([('u', {'theoryname': 'u', 'string': 'u : Ω → ℝ', 'type': 'Ω → ℝ', 'codomain': 'ℝ',
                                             'viewname': 'uASmUnknown'})]),
            'parameters': {
                'f': {'theoryname': 'f', 'string': 'f = x', 'parsestring': 'f = [ x : Ω] x', 'type': '{ : Ω } Ω',
                      'viewname': 'fASmParameter'}},
            'pdes': {'pdes': [
                {'theoryname': 'ephpde1', 'string': 'Δu = 0.0', 'lhsstring': 'Δu', 'rhsstring': '0.0',
                 'viewname': 'ephpde1ASmPDE', 'op': 'Δ', 'lhsparsestring': ' [ anyu : Ω → ℝ ] Δ anyu ',
                 'rhsparsestring': ' [ x : Ω ]  0.0'}]},
            'bcs': {'theoryname': 'ephbcs', 'bcs': [
                {'name': 'bc0', 'string': 'u = f', 'lhsstring': 'u ', 'rhsstring': ' f', 'type': ('Dirichlet',),
                 'on': ('x',), 'measure': (2,)}], 'bctypes': {'theoryname': 'uBCTypes'},
                                                                 'viewname': 'ephbcsASmBCsRequired',
                                                                 'measure_given': 2},
            'props': {},
            'sim': {},
        }
Theresa Pollinger's avatar
Theresa Pollinger committed
285
        self.exaout = ExaOutput()
286
        # self.greeting()
Theresa Pollinger's avatar
Theresa Pollinger committed
287
288
        self.update_prompt()

289
290
291
292
        self.prompted = False
        self.if_yes = None
        self.if_no = None

293
    ##### for state dimensions
Theresa Pollinger's avatar
Theresa Pollinger committed
294
    def dimensions_begin(self):
295
        self.poutput("How many dimensions does your model have?")
296
        self.poutput("I am just assuming it's 1, since that is all we can currently handle.")  # TODO
Theresa Pollinger's avatar
Theresa Pollinger committed
297
298
299
300
        self.simdata["num_dimensions"] = 1
        self.dimensions_parsed()

    def dimensions_handle_input(self, userstring):
301
302
303
304
        # reply_diffops = self.mmtinterface.query_for("mDifferentialOperators")
        # self.poutput(reply_diffops.tostring())
        # self.poutput(element_to_string(reply_diffops.getConstant("derivative")))
        # self.poutput(element_to_string(reply_diffops.getConstant("laplace_operator")))
Theresa Pollinger's avatar
Theresa Pollinger committed
305
306
307
        try:
            numdim = int(userstring)
        except ValueError:
308
            self.poutput("Please enter a number.")
Theresa Pollinger's avatar
Theresa Pollinger committed
309
310
311
            return
        if numdim < 1:
            self.obviously_stupid_input()
312
            self.exaout.create_output(self.testsimdata)
Theresa Pollinger's avatar
Theresa Pollinger committed
313
            self.dimensions_begin()
314
        elif numdim == 1:  # or self.numdim == 2:
Theresa Pollinger's avatar
Theresa Pollinger committed
315
316
317
            self.simdata["num_dimensions"] = numdim
            self.dimensions_parsed()
        else:
318
            self.poutput(
319
                "Sorry, cannot handle " + str(numdim) + " dimensions as of now. Please try less than that.")
Theresa Pollinger's avatar
Theresa Pollinger committed
320

321
    ##### for state domain
Theresa Pollinger's avatar
Theresa Pollinger committed
322
    def domain_begin(self):
323
        self.poutput("What is the domain you would like to simulate for?     Ω : type ❘ = [?;?], e.g. Ω = [0.0;1.0]")
324
        self.poutput("By the way, you can always try and use LaTeX-type input.")
Theresa Pollinger's avatar
Theresa Pollinger committed
325
326
327
328
        self.simdata[self.state]["axes"] = OrderedDict()
        self.domain_mmt_preamble()

    def domain_handle_input(self, userstring):
329
        domain_name = get_first_word(userstring)
330
331
332
        # subdict = self.simdata[self.state]
        with CriticalSubdict(self.simdata[self.state]) as subdict:
            parsestring = userstring
333
            mmtreply = self.mmtinterface.mmt_new_decl(domain_name, subdict["theoryname"], parsestring)
334
335
            mmttype = self.mmtinterface.mmt_infer_type(subdict["theoryname"], domain_name)
            if mmttype.inferred_type_to_string() != "type":
336
                raise InterviewError("This seems to not be a type. It should be!")
337
            result = self.mmtinterface.query_for(subdict["theoryname"])  # if not self.cheating else
Theresa Pollinger's avatar
Theresa Pollinger committed
338
            #print(result.tostring())
339
            subdict["name"] = domain_name
Theresa Pollinger's avatar
Theresa Pollinger committed
340
            (fro, to) = mmtreply.getIntervalBoundaries(result, domain_name) #if not self.cheating else (0.0, 1.0)  # todo make work again
341
342
            subdict["axes"]["x_1"] = "[" + str(fro) + ";" + str(to) + "]"
            (subdict["from"], subdict["to"]) = (fro, to)
343
344
345
346

            self.poutput("we will just assume that the variable is called x for now.")
            # mmtreply = self.mmtinterface.mmt_new_decl(domain_name, subdict["theoryname"], "x : " + domain_name)
            self.trigger('domain_parsed')
Theresa Pollinger's avatar
Theresa Pollinger committed
347
348
349
350
351

    def domain_exit(self):
        self.domain_mmt_postamble()

    def domain_mmt_preamble(self):
352
353
354
        # set the current MMT theoryname for parsing the input TODO use right dimension
        self.simdata[self.state]["theoryname"] = "ephdomain"
        self.new_theory(self.simdata[self.state]["theoryname"])
355
        # (ok, root) = self.mmtinterface.query_for(self.simdata[self.state]["theoryname"])
Theresa Pollinger's avatar
Theresa Pollinger committed
356
357

    def domain_mmt_postamble(self):
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
        with CriticalSubdict(self.simdata[self.state]) as subdict:
            subdict["boundary_name"] = subdict["name"] #todo
            if not self.cheating:
                self.mmtinterface.mmt_new_decl('mydomainpred', subdict["theoryname"],
                                               "myDomainPred = " + subdict["name"] + ".interval_pred")
                self.mmtinterface.mmt_new_decl('mydomain', subdict["theoryname"],
                                               "myDomain = intervalType " + subdict["name"])
                # and a view to understand our interval as a domain -- view ephDomainAsDomain : ?GeneralDomains → ?ephDomain =
                self.new_view(subdict)
                self.mmtinterface.mmt_new_decl('Vecspace', subdict["viewname"],
                                               "Vecspace = real_lit")  # TODO adjust for higher dimensions
                self.mmtinterface.mmt_new_decl('DomainPred', subdict["viewname"], "DomainPred = " + subdict[
                    "name"] + ".interval_pred")  # the . is unbound, apparently...
            else:
                self.new_view(subdict)
                self.mmtinterface.mmt_new_decl('dom', subdict["viewname"],
                                               "domain = " + subdict["name"])
                self.mmtinterface.mmt_new_decl('boun', subdict["viewname"],
                                               "boundary = " + subdict["boundary_name"])
Theresa Pollinger's avatar
Theresa Pollinger committed
377

378
    ##### for state unknowns
Theresa Pollinger's avatar
Theresa Pollinger committed
379
    def unknowns_begin(self):
380
381
        self.poutput("Which variable(s) are you looking for? / What are the unknowns in your model?  u : " +
                     self.simdata["domain"]["name"] + " → ??,  e.g., u : " + self.simdata["domain"]["name"] + " → ℝ ?")
Theresa Pollinger's avatar
Theresa Pollinger committed
382
383
384
        self.simdata["unknowns"] = OrderedDict()

    def unknowns_handle_input(self, userstring):
385
        unknown_name = get_first_word(userstring)
Theresa Pollinger's avatar
Theresa Pollinger committed
386
        # replace interval with domain
387
388
389
        parsestring = (
            userstring.replace(self.simdata["domain"]["name"],
                               "pred myDomainPred") if not self.cheating else userstring)
390

391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
        with CriticalSubdict(self.simdata[self.state]) as usubdict:
            # create mmt theory with includes
            once = self.new_theory(unknown_name)
            # self.include_in(unknown_name, self.simdata["domain"]["theoryname"])

            # and one to "throw away" to infer the type
            self.new_theory(unknown_name + "_to_go_to_trash")
            test = self.mmtinterface.mmt_new_decl(unknown_name, unknown_name + "_to_go_to_trash",
                                                  parsestring)

            type = self.get_inferred_type(unknown_name + "_to_go_to_trash", unknown_name)
            usubdict[unknown_name] = {
                "theoryname": unknown_name,
                "string": parsestring,
                "type": type,
406
                "codomain": type.replace(self.simdata["domain"]["name"] + " →", "", 1).strip(),
407
408
409
410
411
412
413
414
415
416
417
418
419
            }
            with CriticalSubdict(self.simdata["unknowns"][unknown_name]) as subdict:
                if self.mmtinterface.query_for(unknown_name + "_to_go_to_trash").hasDefinition(unknown_name):
                    raise InterviewError("Unknowns cannot be defined!")
                if not type_is_function_from(subdict["type"], self.simdata["domain"]["name"]):
                    raise InterviewError("Unknown should be a function on " + self.simdata["domain"]["name"] + "!")

                # add unknown's type as constant
                twice = self.mmtinterface.mmt_new_decl(unknown_name, subdict["theoryname"],
                                                       "myUnkType = " + subdict["type"])
                twice = (self.mmtinterface.mmt_new_decl('diffable', subdict["theoryname"],
                                                        "anyuwillbediffable : {u : myUnkType} ⊦ twodiff u ") if not self.cheating else twice)
                self.new_view(subdict)
420
421
                self.mmtinterface.mmt_new_decl("codomain", subdict["viewname"], "ucodomain = " + subdict["codomain"])
                self.mmtinterface.mmt_new_decl("unktype", subdict["viewname"], "unknowntype = myUnkType")
422
                self.poutput("Ok, " + userstring)
423
424
                #if self.please_prompt("Are these all the unknowns?"): #TODO
                self.trigger('unknowns_parsed')
Theresa Pollinger's avatar
Theresa Pollinger committed
425
426
427

    def unknowns_exit(self):
        for unknown in self.simdata["unknowns"]:
428
            self.poutput(self.simdata["unknowns"][unknown]["string"])
Theresa Pollinger's avatar
Theresa Pollinger committed
429

430
    ##### for state parameters
Theresa Pollinger's avatar
Theresa Pollinger committed
431
    def parameters_begin(self):
432
433
        self.poutput(
            "Would you like to name additional parameters like constants or functions (that are independent of your unknowns)?  c : ℝ = ? or f : Ω → ℝ = ?")  # ℝ
434
        self.simdata["parameters"] = OrderedDict()
Theresa Pollinger's avatar
Theresa Pollinger committed
435
436

    def parameters_handle_input(self, userstring):
437
        # self.poutput ("parameterinput "+ userstring)
438
        if means_no(userstring):
Theresa Pollinger's avatar
Theresa Pollinger committed
439
440
441
            self.trigger('parameters_parsed')
            return

442
        parameter_name = get_first_word(userstring)
443
444
445
446
447
448
449
450
451
452
        self.simdata["parameters"][parameter_name] = {}
        with CriticalSubdict(self.simdata["parameters"][parameter_name]) as subdict:
            # create mmt theory
            self.new_theory(parameter_name)
            # we might need the other parameters created so far, so use them
            for otherparamentry in get_recursively(self.simdata["parameters"], "theoryname"):
                self.include_in(parameter_name, otherparamentry)

            # sanitize userstring - check if this works for all cases
            parsestring = add_ods(userstring)
453
454
            if parsestring.startswith(parameter_name + "(") or parsestring.startswith(parameter_name + " ("):#todo make smarter for more dimensions
               parsestring = remove_apply_brackets(parsestring)
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
            parsestring = functionize(parsestring, self.simdata["domain"]["name"])
            # self.poutput(parsestring)
            reply_pconstant = self.mmtinterface.mmt_new_decl("param", parameter_name, parsestring)
            reply_pconstant = self.mmtinterface.query_for(parameter_name)
            subdict["theoryname"] = parameter_name
            subdict["string"] = userstring
            subdict["parsestring"] = parsestring
            subdict["type"] = self.get_inferred_type(parameter_name, parameter_name)

            # if not reply_pconstant.hasDefinition(parameter_name) and not self.cheating:
            #    InterviewError("Please define this parameter.")

            # create view
            self.new_view(subdict)
            self.mmtinterface.mmt_new_decl("ptype", subdict["viewname"],
                                                                         "ptype = " + subdict["type"])
            self.mmtinterface.mmt_new_decl("param", subdict["viewname"],
                                                                         "param = " + parameter_name)
            self.poutput("Ok, " + parsestring)
474
            self.please_prompt("Are these all the parameters?", lambda: self.trigger('parameters_parsed'))
Theresa Pollinger's avatar
Theresa Pollinger committed
475
476

    def parameters_exit(self):
477
        # print(str(self.simdata["parameters"]))
Theresa Pollinger's avatar
Theresa Pollinger committed
478
        for parameter in self.simdata["parameters"]:
479
            self.poutput(self.simdata["parameters"][parameter]["string"])
Theresa Pollinger's avatar
Theresa Pollinger committed
480

481
    ##### for state pdes
Theresa Pollinger's avatar
Theresa Pollinger committed
482
    def pdes_begin(self):
483
        self.poutput(
484
            "Let's talk about your partial differential equation(s). What do they look like? Δu = 0.0, or laplace_operator Ω ℝ u = f ?")
Theresa Pollinger's avatar
Theresa Pollinger committed
485
486
487
        self.simdata["pdes"]["pdes"] = []

    def pdes_handle_input(self, userstring):
488
        self.simdata["pdes"]["pdes"].append({})
489
        with CriticalSubdict(self.simdata["pdes"]["pdes"][-1]) as subdict:
490
            subdict["theoryname"] = "ephpde" + str(len(self.simdata["pdes"]["pdes"]))
491
492
493
494
495
496
            self.new_theory(subdict["theoryname"])

            # TODO use symbolic computation to order into LHS and RHS
            parts = re.split("=", userstring)

            if len(parts) is not 2:
497
                raise InterviewError("This does not look like an equation.")
498
499
500

            # store the info
            subdict["string"] = userstring
501
502
503
            subdict["lhsstring"] = parts[0].strip()
            subdict["rhsstring"] = parts[1].strip()#TODO expand
            subdict["rhsstring_expanded"] = self.try_expand(subdict["rhsstring"])
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522

            # to make the left-hand side a function on x, place " [ variablename : domainname ] " in front
            if parts[0].find("x") > -1:
                parts[0] = " [ x : " + self.simdata["domain"]["name"] + " ] " + parts[0]
            # right-hand side: infer type, make function if not one yet
            if not type_is_function_from(self.get_inferred_type(subdict["theoryname"], parts[1]),
                                         self.simdata["domain"]["name"]):
                parts[1] = " [ x : " + self.simdata["domain"]["name"] + " ] " + parts[1]

            # in lhs replace all unknown names used by more generic ones and add lambda clause in front
            for unkname in get_recursively(self.simdata["unknowns"], "theoryname"):
                parts[0] = parts[0].replace(unkname, " any" + unkname)
                parts[0] = " [ any" + unkname + " : " + self.simdata["unknowns"][unkname]["type"] + " ] " + parts[0]
                # and include the original ones as theory
                inc = self.include_in(subdict["theoryname"], unkname)
            for parname in get_recursively(self.simdata["parameters"], "theoryname"):
                inc = self.include_in(subdict["theoryname"], parname)

            # send declarations to mmt
523
524
            self.mmtinterface.mmt_new_decl("lhs", subdict["theoryname"], " mylhs = " + parts[0])
            reply_lhsconstant = self.mmtinterface.query_for(subdict["theoryname"])
525

526
527
            self.mmtinterface.mmt_new_decl("rhs", subdict["theoryname"], " myrhs = " + parts[1])
            reply_rhsconstant = self.mmtinterface.query_for(subdict["theoryname"])
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546

            # create view
            self.new_view(subdict)
            ltype = self.get_inferred_type(subdict["theoryname"], "mylhs")
            eqtype = get_last_type(ltype)
            rtype = self.get_inferred_type(subdict["theoryname"], "myrhs")
            self.mmtinterface.mmt_new_decl("eqtype", subdict["viewname"],
                                                                     "eqtype = " + eqtype)
            self.mmtinterface.mmt_new_decl("lhs", subdict["viewname"],
                                                                     "lhs = " + "mylhs")
            self.mmtinterface.mmt_new_decl("rhs", subdict["viewname"],
                                                                     "rhs = " + "myrhs")
            self.mmtinterface.mmt_new_decl("pde", subdict["viewname"],
                                                                     "pde = " + "[u](mylhs u) funcEq myrhs")

            reply = self.mmtinterface.query_for(subdict["theoryname"])

            for unkname in get_recursively(self.simdata["unknowns"], "theoryname"):
                op = subdict["lhsstring"].replace(unkname, "")
547
                op = op.strip()
548
549
550
551
552
553
554

            # store the info
            subdict["op"] = op
            subdict["lhsparsestring"] = parts[0]
            subdict["rhsparsestring"] = parts[1]

            # TODO query number of effective pdes and unknowns from mmt for higher dimensional PDEs
555
            # => can assume each to be ==1 for now
556
            numpdesgiven = len(self.simdata["pdes"]["pdes"])
557
            self.poutput("Ok, " + reply.tostring())
Theresa Pollinger's avatar
Theresa Pollinger committed
558
559
560
561
            if numpdesgiven == len(self.simdata["unknowns"]):
                self.trigger('pdes_parsed')
            elif numpdesgiven > len(self.simdata["unknowns"]):
                self.poutput("now that's too many PDEs. Please go back and add more unknowns.")
562
563
            else:
                self.poutput("More PDEs, please!")
Theresa Pollinger's avatar
Theresa Pollinger committed
564
565

    def pdes_exit(self):
566
        self.poutput("These are all the PDEs needed.")
Theresa Pollinger's avatar
Theresa Pollinger committed
567

568
    ##### for state bcs
569
    def bcs_begin(self):
570
571
        self.poutput("Let's discuss your boundary conditions. "
                     "What do they look like? u(x) = f(x) or u(" + str(self.simdata["domain"]["to"]) + ") = \\alpha ?") #TODO remove square brakcets
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
        bctypetheoryname = self.redefine_bcs()
        with CriticalSubdict(self.simdata["bcs"]) as subdict:
            subdict["theoryname"] = "ephbcs"
            subdict["bcs"] = []
            self.new_theory(subdict["theoryname"])
            # apparently, need to include everything explicitly so that view works
            for unknownentry in get_recursively(self.simdata["unknowns"], "theoryname"):
                self.include_in(subdict["theoryname"], unknownentry)
            for paramentry in get_recursively(self.simdata["parameters"], "theoryname"):
                self.include_in(subdict["theoryname"], paramentry)
            for pdeentry in get_recursively(self.simdata["pdes"], "theoryname"):
                self.include_in(subdict["theoryname"], pdeentry)
            self.include_in(subdict["theoryname"], bctypetheoryname)
            self.new_view(subdict)
            subdict["measure_given"] = 0
587
588

    def bcs_handle_input(self, userstring):
589
        with CriticalSubdict(self.simdata["bcs"]) as subdict:
590
591
592
593
594
595
596
597
598
            currentname = "bc" + str(len(subdict["bcs"]))
            subdict["bcs"].append({"name": currentname})
            # TODO use symbolic computation to order into LHS and RHS
            parts = re.split("=", userstring)

            if len(parts) is not 2:
                raise InterviewError("This does not look like a boundary condition.")
            # store the info
            subdict["bcs"][-1]["string"] = userstring
599
600
601
            subdict["bcs"][-1]["lhsstring"] = parts[0].strip()
            subdict["bcs"][-1]["rhsstring"] = parts[1].strip()#TODO expand
            subdict["bcs"][-1]["rhsstring_expanded"] = self.try_expand(subdict["bcs"][-1]["rhsstring"])
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619

            # to make a function on x, place " [ variablename : boundaryname ] " in front
            if parts[0].find("x") > -1:
                parts[0] = " [ x : " + self.simdata["domain"]['boundary_name'] + " ] " + parts[0]
            if parts[1].find("x") > -1:
                parts[1] = " [ x : " + self.simdata["domain"]['boundary_name'] + " ] " + parts[1]

            # in lhs replace all unknown names used by more generic ones and add lambda clause in front
            for unkname in get_recursively(self.simdata["unknowns"], "theoryname"):
                parts[0] = parts[0].replace(unkname, " any" + unkname)
                parts[0] = " [ any" + unkname + " : " + self.simdata["unknowns"][unkname]["type"] + " ] " + parts[0]

                type = self.get_inferred_type(subdict["theoryname"], parts[0])
                if type_is_function_to(type, self.simdata["unknowns"][unkname]["type"]):
                    # right-hand side: infer type, make function if not one yet
                    rhstype = self.get_inferred_type(subdict["theoryname"], parts[1])
                    if not type_is_function_from(rhstype, self.simdata["domain"]["name"])\
                            and not type_is_function_from(rhstype, self.simdata["domain"]["boundary_name"]):
620
                        parts[1] = " [ x : " + self.simdata["domain"]["boundary_name"] + " ] " + parts[1]
621
622
623
624
                    self.add_list_of_declarations(subdict["viewname"], [
                        "firstBC = myDirichletBCfun " + parts[1],
                        "secondBC = myDirichletBCfun " + parts[1],
                    ])
625
626
627
                    subdict["bcs"][-1]["type"] = "Dirichlet",
                    subdict["bcs"][-1]["on"] = "x",
                    subdict["bcs"][-1]["measure"] = 2,
628
629
                    subdict["measure_given"] = 2
                elif type_is_function_to(type, self.simdata["unknowns"][unkname]["codomain"]):
630
631
632
633
                    #at_x = re.split('[\(\)]', subdict["bcs"][-1]["lhsstring"])[-1] #TODO
                    at_x = subdict["bcs"][-1]["lhsstring"].split('(', 1)[1].split(')')[0].strip()
                    if not at_x is self.simdata["domain"]["from"] or at_x is self.simdata["domain"]["to"]:
                        raise InterviewError(at_x + " is not on the boundary!")
634
635
                    if len(subdict["bcs"]) == 1:
                        self.mmtinterface.mmt_new_decl("bc1", subdict["viewname"],
636
                                                       "firstBC = solutionat " + at_x + " is " + parts[1])
637
638
                    elif len(subdict["bcs"]) == 2:
                        self.mmtinterface.mmt_new_decl("bc2", subdict["viewname"],
639
                                                       "secondBC = solutionat " + at_x + " is " + parts[1]) #TODO store at and type
640
641
642
                    else:
                        raise InterviewError("too many boundary conditions saved")
                    subdict["measure_given"] += 1
643
644
645
                    subdict["bcs"][-1]["type"] = "Dirichlet",
                    subdict["bcs"][-1]["on"] = at_x,
                    subdict["bcs"][-1]["measure"] = 1,
646
647
648
649
650
651
652
653
654
655
656
657

            #try:
            #    type = self.get_inferred_type(subdict["theoryname"], "[u : Ω → ℝ] u(0.0)")
            #    type = self.get_inferred_type(subdict["theoryname"], "[u : Ω → ℝ] u")
            #except MMTServerError as error:
            #    self.poutput(error.args[0])

            self.poutput("Ok ")
            if subdict["measure_given"] == len(self.simdata["unknowns"])*2: #TODO times order
                self.trigger('bcs_parsed')
            elif subdict["measure_given"] > len(self.simdata["unknowns"]):
                raise InterviewError("now that's too many boundary conditions. ignoring last input.")
658
659
660
661

    def bcs_exit(self):
        self.poutput("These are all the boundary conditions needed.")

662
663
    def redefine_bcs(self):
        for unknown in get_recursively(self.simdata["unknowns"], "theoryname"):
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
            with CriticalSubdict(self.simdata["bcs"]) as subdict:
                subdict["bctypes"] = {}
                bctypetheoryname = unknown + "BCTypes"
                subdict["bctypes"]["theoryname"] = bctypetheoryname
                self.new_theory(bctypetheoryname)
                self.include_in(bctypetheoryname, unknown)
                self.include_in(bctypetheoryname, "mDifferentialOperators")
                self.add_list_of_declarations(bctypetheoryname,
                        [
                            "myDirichletBC: {where: " + self.simdata["domain"]["boundary_name"] + ", rhs: " +
                                self.simdata["unknowns"][unknown]["codomain"] + "}(" + self.simdata["domain"]["name"] + " → " +
                                self.simdata["unknowns"][unknown]["codomain"] + ") → prop "
                                " ❘ = [where, rhs][u] u where ≐ rhs ❘  # solutionat 1 is 2 ",
                            "myDirichletBCfun : {rhs: " + self.simdata["domain"]["boundary_name"] + " → " +
                                self.simdata["unknowns"][unknown]["codomain"] + " }(" + self.simdata["domain"]["name"] + " → " +
                                self.simdata["unknowns"][unknown]["codomain"] + ") → prop ❘ = [rhs] [u] ∀[x:" + self.simdata["domain"]["boundary_name"] + " ] u x ≐ rhs x "
                                "❘ # solutionatboundaryis 1",
                        ]
                )
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
                err = """when trying to include include ?mDomain = ?ephdomainASmDomain and all others in view
                info.kwarc.mmt.api.InvalidObject of level 2
                invalid object (ill-formed morphism: expected http://mathhub.info/MitM/smglom/calculus?mDomain -> (http://cds.omdoc.org/urtheories?ModExp?complextheory [] ), found http://mathhub.info/MitM/smglom/calculus?mDomain -> http://mathhub.info/MitM/smglom/calculus?ephdomain): http://mathhub.info/MitM/smglom/calculus?ephdomainASmDomain
                info.kwarc.mmt.api.checking.MMTStructureChecker.checkMorphism(MMTStructureChecker.scala:535)
                info.kwarc.mmt.api.checking.MMTStructureChecker.checkRealization(MMTStructureChecker.scala:545)
                info.kwarc.mmt.api.checking.MMTStructureChecker.check(MMTStructureChecker.scala:187)
                info.kwarc.mmt.api.checking.MMTStructureChecker.applyElementBegin(MMTStructureChecker.scala:57)
                info.kwarc.mmt.interviews.InterviewServer$$anon$1.onElement(InterviewServer.scala:94)
                info.kwarc.mmt.api.parser.KeywordBasedParser.seCont(StructureParser.scala:96)
                info.kwarc.mmt.api.parser.KeywordBasedParser.addDeclaration$1(StructureParser.scala:481)
                info.kwarc.mmt.api.parser.KeywordBasedParser.readInModuleAux(StructureParser.scala:525)
                info.kwarc.mmt.api.parser.KeywordBasedParser.readInModule(StructureParser.scala:461)
                info.kwarc.mmt.interviews.InterviewServer.parseDecl(InterviewServer.scala:102)
                info.kwarc.mmt.interviews.InterviewServer.apply(InterviewServer.scala:50)
                info.kwarc.mmt.api.web.Server.resolveExtension(Server.scala:95)
                info.kwarc.mmt.api.web.Server.resolve(Server.scala:76)
                info.kwarc.mmt.api.web.Server.handleRequest(Server.scala:53)
                info.kwarc.mmt.api.web.TiscafServerImplementation$RequestHandler$$anon$1.act(TiscafServerImplementation.scala:49)
                tiscaf.HSimpleLet.aact(HLet.scala:166)
                tiscaf.HSimpleLet.aact$(HLet.scala:165)
                info.kwarc.mmt.api.web.TiscafServerImplementation$RequestHandler$$anon$1.aact(TiscafServerImplementation.scala:47)
                tiscaf.HAcceptor.talk(HAcceptor.scala:246)
                tiscaf.HSimplePeer.doTalkItself$1(HPeer.scala:101)
                tiscaf.HSimplePeer.$anonfun$readChannel$1(HPeer.scala:114)
                scala.runtime.java8.JFunction0$mcV$sp.apply(JFunction0$mcV$sp.java:12)
                scala.concurrent.Future$.$anonfun$apply$1(Future.scala:653)
                scala.util.Success.$anonfun$map$1(Try.scala:251)
                scala.util.Success.map(Try.scala:209)
                scala.concurrent.Future.$anonfun$map$1(Future.scala:287)
                scala.concurrent.impl.Promise.liftedTree1$1(Promise.scala:29)
                scala.concurrent.impl.Promise.$anonfun$transform$1(Promise.scala:29)
                scala.concurrent.impl.CallbackRunnable.run(Promise.scala:60)
                tiscaf.sync.SyncQuExecutionContext$$anon$1.run(SyncQuExecutionContext.scala:69)
                """#TODO
717
718
719
720
721
722
723
724
                #viewname = bctypetheoryname + "ASmBCTypes"
                #subdict["bctypes"]["viewname"] = viewname
                #self.mmtinterface.mmt_new_view(viewname, bctypetheoryname, "mBCTypes")
                #self.add_list_of_declarations(viewname,
                #                              ["DirichletBC = myDirichletBC ",
                #                               #" = myDirichletBCfun"
                #                               ])
                return bctypetheoryname  # Todo adapt for more than 1
725

726
727
728
729
730
731
732
733
734
735
    ##### for state props
    def props_begin(self):
        with CriticalSubdict(self.simdata["props"]) as subdict:
            # TODO try to find out things about the solvability ourselves
            subdict["theoryname"] = "ephBoundaryValueProblem"
            self.new_theory(subdict["theoryname"])
            #self.new_view(subdict)
            for pde in self.simdata["pdes"]["pdes"]:
                self.poutput("Do you know something about the operator " + pde["op"] + "? "
                             "Is it e.g. linear, or not elliptic ? ")
736

737
    def props_handle_input(self, userstring):
738
        if means_no(userstring):
739
            self.trigger("props_parsed")
740
741
            return

742
743
744
745
746
747
748
749
750
751
752
753
        with CriticalSubdict(self.simdata["props"]) as subdict:
            #            "props": {
            #    "theoryname": "ephboundaryvalueproblem",
            #    "ops": [
            #        {
            #            "name": "op1",
            #            "linear": True, #or false or unknown
            #            "props": ["elliptic"]
            #        }
            #    ]
            #},
            #parts = re.split(" ", userstring) #TODO can add arbitrary proofs of undefined terms??
754
755
756
757
            parsestring = userstring.replace("not", "¬")
            for property in ["linear", "elliptic"]:
                if parsestring.find(property) > -1:
                    self.add_list_of_declarations(subdict["theoryname"], [
758
                        "user_" + property + " : ⊦ " + parsestring + " mylhs"
759
760
761
                    ])
            self.poutput("OK!")
            self.poutput("do you know anything else?")
762

763
764
765
766
767
    def props_exit(self):
        return

    ##### for state sim
    def sim_begin(self):
768
769
        self.please_prompt("Would you like to try and solve the PDE using the Finite Difference Method in ExaStencils?",
                           self.sim_ok_fd)
770
771

    def sim_handle_input(self, userstring):
772
773
        self.please_prompt("Would you like to try and solve the PDE using the Finite Difference Method in ExaStencils?",
                           self.sim_ok_fd)
774

775
    def sim_exit(self):
776
777
        # generate output
        self.exaout.create_output(self.simdata)
778
        self.poutput("Generated ExaStencils input.")
779
        #TODO generate and run simulation
Theresa Pollinger's avatar
Theresa Pollinger committed
780

781
782
783
784
    def sim_ok_fd(self):
        self.simdata["sim"]["type"] = "FiniteDifferences"
        self.sim_exit()

785
    #### functions for user interaction
Theresa Pollinger's avatar
Theresa Pollinger committed
786
    def please_prompt(self, query):
787
        self.poutput(query + " [y/n]? ")
Theresa Pollinger's avatar
Theresa Pollinger committed
788
789
790
791
792
793
794
795
796
797
798
799
        val = input()
        if val == "":
            return True
        try:
            ret = strtobool(val)
        except ValueError:
            # or use as input to callback an input processing fcn..?
            self.poutput("Please answer with Y/n")
            return self.please_prompt(query)
        return ret

    def obviously_stupid_input(self):
800
        self.poutput("Trying to be funny, huh?")
Theresa Pollinger's avatar
Theresa Pollinger committed
801

802
    # mmt input helper functions
Theresa Pollinger's avatar
Theresa Pollinger committed
803
    def include_in(self, in_which_theory, what):
804
        return self.mmtinterface.mmt_new_decl("inc", in_which_theory, "include " + assert_question_mark(what))
Theresa Pollinger's avatar
Theresa Pollinger committed
805

806
807
808
809
    def add_list_of_declarations(self, in_which_theory, declaration_list):
        for declaration in declaration_list:
            self.mmtinterface.mmt_new_decl("inc", in_which_theory, declaration)

Theresa Pollinger's avatar
Theresa Pollinger committed
810
    def include_bgthys(self, in_which_theory):
811
        ok = True
Theresa Pollinger's avatar
Theresa Pollinger committed
812
        for bgthy in self.bgthys[self.state]:
813
814
            ok = ok and self.include_in(in_which_theory, bgthy)
        return ok
Theresa Pollinger's avatar
Theresa Pollinger committed
815
816

    def new_theory(self, thyname):
817
818
819
820
821
        try:
            self.mmtinterface.mmt_new_theory(thyname)
            return self.include_bgthys(thyname)
        except MMTServerError as error:
            self.poutput(error.args[0])
822
            # self.poutput(error.with_traceback())
823
            raise
824
        # (ok, root) = self.mmtinterface.query_for(self.simdata[self.state]["theoryname"])
Theresa Pollinger's avatar
Theresa Pollinger committed
825
826

    def new_view(self, dictentry):
827
        dictentry["viewname"] = self.construct_current_view_name(dictentry)
828
        # self.poutput("new view: "+dictentry["viewname"])
829
        ok = self.mmtinterface.mmt_new_view(dictentry["viewname"], self.viewfrom[self.state], dictentry["theoryname"])
830
        # recursively look for all views already done and try to include them
831
        for viewstring in get_recursively(self.simdata, "viewname"):
832
            if (dictentry["viewname"] != viewstring) and ok:
833
                try:
834
835
                    ok = self.include_in(dictentry["viewname"],
                                         "?" + re.split('AS', viewstring)[-1] + " = " + "?" + viewstring)
836
                except MMTServerError as error:
837
838
839
840
841
                    # self.poutput("no backend available that is applicable to " + "http://mathhub.info/MitM/smglom/calculus" + "?" + re.split('AS', dictentry["viewname"])[-1] + "?")
                    # we are expecting errors if we try to include something that is not referenced in the source theory, so ignore them
                    if error.args[0].find(
                            "no backend available that is applicable to " + "http://mathhub.info/MitM/smglom/calculus" + "?" +
                            re.split('AS', dictentry["viewname"])[-1] + "?") < 1:
842
                        raise
Theresa Pollinger's avatar
Theresa Pollinger committed
843
844
845
        return ok

    def construct_current_view_name(self, dictentry):
846
        return self.construct_view_name(dictentry, self.state)
Theresa Pollinger's avatar
Theresa Pollinger committed
847
848

    def construct_view_name(self, dictentry, state):
849
        return dictentry["theoryname"] + "AS" + (self.viewfrom[state])
Theresa Pollinger's avatar
Theresa Pollinger committed
850

851
852
853
    def get_inferred_type(self, in_theory, term):
        return self.mmtinterface.mmt_infer_type(in_theory, term).inferred_type_to_string()

854
855
856
857
858
859
860
861
862
863
    def try_expand(self, term, in_theory=None): #TODO do using mmt definition expansion
        for param in reversed(self.simdata["parameters"]):
            if term.find(param) > -1:
                parts = self.simdata["parameters"][param]["string"].split("=")
                if (len(parts) != 2):
                    raise InterviewError("no definition for " + param + " given")
                paramdef = parts[-1]
                term = term.replace(param, paramdef.strip())
        return term

864
865
    def print_empty_line(self):
        self.poutput("\n")
Theresa Pollinger's avatar
Theresa Pollinger committed
866

867
    ############# input processing if not explain or undo
Theresa Pollinger's avatar
Theresa Pollinger committed
868
869
870
    def default(self, line):
        raw = line.parsed['raw']
        arg = LatexNodes2Text().latex_to_text(raw)
871
        # pythonic switch-case, cf. https://bytebaker.com/2008/11/03/switch-case-statement-in-python/
872
873
874
875
876
877
878

        if not self.prompt_input_handling(arg):
            self.state_input_handling(arg)

    def state_input_handling(self, arg):
        """The standard input handling, depending on which state we are in"""
        # pythonic switch-case, cf. https://bytebaker.com/2008/11/03/switch-case-statement-in-python/
Theresa Pollinger's avatar
Theresa Pollinger committed
879
880
        try:
            self.stateDependentInputHandling[self.state](arg)
881
        except Exception as error:
882
            #self.exaout.create_output(self.simdata)
Theresa Pollinger's avatar
Theresa Pollinger committed
883
884
            raise

885
    def please_prompt(self, query, if_yes, if_no=None):
Theresa Pollinger's avatar
Theresa Pollinger committed
886
        self.poutput(str(query) + " [y/n]? ")
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
        self.prompted = True
        self.if_yes = if_yes
        self.if_no = if_no

    def prompt_input_handling(self, arg):
        """ If we asked for a yes-no answer, execute what was specified in please_prompt.
        return true if the input was handled here, and false if not."""
        if self.prompted:
            if arg == "":
                self.poutput("Yes")
                ret = True
            else:
                try:
                    ret = strtobool(str(arg).strip().lower())
                except ValueError:
                    # or use as input to callback an input processing fcn..?
                    self.poutput("Please answer with Y/n")
                    return True
            self.prompted = False
            if ret:
                self.if_yes()
            elif self.if_no is not None:
                self.if_no()
            else:
                return False
            return True
        return False

Theresa Pollinger's avatar
Theresa Pollinger committed
915
916
917
918
    # called when user types 'explain [expression]'
    def do_explain(self, expression):
        "Explain an expression or the theoretical background to what we are currently looking for"
        if expression:
919
            explanation = "hello, " + expression  # TODO query flexiformal content through mmt
Theresa Pollinger's avatar
Theresa Pollinger committed
920
921
922
923
924
        else:
            explanation = 'hello'
        self.poutput(explanation)

    def help_explain(self):
925
926
927
        self.poutput('\n'.join(['explain [expression]',
                                'explain the expression given or the theory currently used',
                                ]))
Theresa Pollinger's avatar
Theresa Pollinger committed
928
929
930
931
932
933
934

    # called when user types 'undo'
    def do_undo(self, expression):
        "Go back to the last question"
        self.trigger('last_state')

    def help_undo(self):
935
936
937
        self.poutput('\n'.join(['undo',
                                'Go back to the last question',
                                ]))
Theresa Pollinger's avatar
Theresa Pollinger committed
938
939
940
941

    def update_prompt(self):
        self.prompt = "(" + self.state + ")"

942
    # tab completion for empty lines
Theresa Pollinger's avatar
Theresa Pollinger committed
943
944
945
946
947
948
    def completenames(self, text, line, begidx, endidx):
        """Override of cmd2 method which completes command names both for command completion and help."""
        command = text
        if self.case_insensitive:
            command = text.lower()
        if not command:
949
            # define the "default" input for the different states we can be in
Theresa Pollinger's avatar
Theresa Pollinger committed
950
951
952
            self.stateDependentDefaultInput = {
                'dimensions': '1',
                'domain': ['Ω = [ 0 ; 1 ]'],
953
                'unknowns': ['u : Ω → ℝ'],
954
                'parameters': ['f :  ℝ → ℝ = [x: ℝ] x '],  # ['f : Ω → ℝ = [x:Ω] x ⋅ x'],
955
956
957
                'pdes': ['∆u = f(x_1)'],
                'bcs': ['u (0) = 0'],  # ,'u (1) = x_1**2'],
                'sim': ['FD'],
Theresa Pollinger's avatar
Theresa Pollinger committed
958
959
960
961
962
963
964
965
966
967
968
            }
            return self.stateDependentDefaultInput[self.state]
        else:
            # Call super class method.  Need to do it this way for Python 2 and 3 compatibility
            cmd_completion = cmd.Cmd.completenames(self, command)

            # If we are completing the initial command name and get exactly 1 result and are at end of line, add a space
            if begidx == 0 and len(cmd_completion) == 1 and endidx == len(line):
                cmd_completion[0] += ' '
            return cmd_completion

969
970
971
972
    def greeting(self):  # TODO make work in proper order
        self.poutput(
            "Hello, " + self.username + "! I am " + self.myname + ", your partial differential equations and simulations expert. " \
                                                                  "Let's set up a simulation together.\n")
Theresa Pollinger's avatar
Theresa Pollinger committed
973
974
        self.trigger("greeting_over")

975

Theresa Pollinger's avatar
Theresa Pollinger committed
976
977
if __name__ == '__main__':
    Interview().cmdloop()