Пример #1
0
def tensor_solver(b4_eqns,
                  aft_eqns,
                  e_knowns=[],
                  levels=-1,
                  num_sols=1,
                  verbose=True,
                  solution_file=None,
                  logic_files=None):
    """Updater calling tensor solvers."""
    if verbose or DEBUG:
        print "tensor_solver:"
        print "  b4_eqns:", pprint.pformat(b4_eqns, 4, 80)
        print "  aft_eqns:", pprint.pformat(aft_eqns, 4, 80)
        print "  e_knowns:", pprint.pformat(e_knowns, 4, 80)
    knowns = set(flatten([eqn.atoms()
                          for eqn in b4_eqns])).union(set(e_knowns))
    knowns.union(CONSTANTS)
    eqns = aft_eqns + b4_eqns
    if verbose or DEBUG or True:
        print "=" * 80
        print "Calling Generator with following:"
        print "*" * 80
        print "Knowns:", pprint.pformat(knowns, 4, 80)
        print "-" * 80
        unknown = set(flatten([eqn.atoms() for eqn in aft_eqns])) - knowns
        print "Unknowns:", pprint.pformat(unknown, 4, 80)
        print "-" * 80
        print "eqns:", pprint.pformat(eqns, 4, 80)
        print "=" * 80
    multiple_sols = True
    sub_all = True
    sol_dicts = all_back_sub(eqns, knowns, levels, multiple_sols, sub_all)
    #sol_dicts = map(sol_cse, sol_dicts)
    if solution_file:
        fp = open(solution_file, 'w')
        fp.write("%"*80 + \
                 "\n%% This file was automatically generated by Ignition\n" + \
                 "%" * 80 + "\n")
        for n, dict_ord in enumerate(sol_dicts):
            fp.write("Algorithm %d\n\n" % (n + 1))
            fp.write(update_dict_to_latex(*dict_ord))
            fp.write("\n\n")
    if logic_files:
        print "Writing out logic files to %s_{0--%d}.out" % \
            (logic_files, len(sol_dicts))
        for n, dict_ord in enumerate(sol_dicts):
            fp = open("%s_%d.out" % (logic_files, n), 'w')
            backward_sub(eqns, knowns, dict_ord[1], multiple_sols, sub_all, fp)
            fp.close()
    sol_dicts = sol_dicts[:num_sols]
    return sol_dicts
Пример #2
0
 def _is_nonlinear (self):
     """Simple check for non-linear fluxes"""
     for field in self.conserved.fields():
         for term in flatten(self.A.tolist()):
             if field in term:
                 return True
     return False
Пример #3
0
def tensor_solver (b4_eqns, aft_eqns, e_knowns=[], levels= -1, num_sols=1,
                   verbose=True, solution_file=None, logic_files=None,
                   allow_recompute=False):
    """Updater calling tensor solvers."""
    if verbose or DEBUG:
        print "tensor_solver:"
        print "  b4_eqns:", pprint.pformat(b4_eqns, 4, 80)
        print "  aft_eqns:", pprint.pformat(aft_eqns, 4, 80)
        print "  e_knowns:", pprint.pformat(e_knowns, 4, 80)
    knowns = set(flatten([eqn.atoms() for eqn in b4_eqns])).union(set(e_knowns))
    knowns.union(CONSTANTS)
    eqns = aft_eqns + b4_eqns
    if verbose or DEBUG or True:
        print "=" * 80
        print "Calling Generator with following:"
        print "*" * 80
        print "Knowns:", pprint.pformat(knowns, 4, 80)
        print "-" * 80
        unknown = set(flatten([eqn.atoms() for eqn in aft_eqns])) - knowns
        print "Unknowns:", pprint.pformat(unknown, 4, 80)
        print "-" * 80
        print "eqns:", pprint.pformat(eqns, 4, 80)
        print "=" * 80
    multiple_sols = True
    sub_all = True
    sol_dicts = all_back_sub(eqns, knowns, levels, multiple_sols, sub_all, allow_recompute)
    #sol_dicts = map(sol_cse, sol_dicts)
    if solution_file:
        fp = open(solution_file, 'w')
        fp.write("%"*80 + \
                 "\n%% This file was automatically generated by Ignition\n" + \
                 "%" * 80 + "\n")
        for n, dict_ord in enumerate(sol_dicts):
            fp.write("Algorithm %d\n\n" % (n + 1))
            fp.write(update_dict_to_latex(*dict_ord))
            fp.write("\n\n")
    if logic_files:
        print "Writing out logic files to %s_{0--%d}.out" % \
            (logic_files, len(sol_dicts))
        for n, dict_ord in enumerate(sol_dicts):
            fp = open("%s_%d.out" % (logic_files, n), 'w')
            backward_sub(eqns, knowns, dict_ord[1], multiple_sols, sub_all, fp)
            fp.close()
    sol_dicts = sol_dicts[:num_sols]
    return sol_dicts
Пример #4
0
def AK_KJ_Rule(A, K, J):
    [A] = A
    [K_l, k_m, K_r] = K
    [[J_tl, _, _], [Tj_ml, _, _], [_, j_bm, J_br]] = J
    op = (A * K_l - K_l * J_tl - k_m * Tj_ml)
    if type(op) is matrix:
        op = op.tolist()
    if type(op) is list:
        op = flatten(op)
    return op, []
Пример #5
0
def AK_KJ_Rule (A, K, J):
    [A] = A
    [K_l, k_m, K_r] = K
    [[J_tl, _, _],
     [Tj_ml, _, _],
     [_, j_bm, J_br]] = J
    op = (A * K_l - K_l * J_tl - k_m * Tj_ml)
    if type(op) is matrix:
        op = op.tolist()
    if type(op) is list:
        op = flatten(op)
    return op, []
Пример #6
0
 def used_constant_fields(self):
     """Constant Fields used inside jacobian evaluation"""
     A = self.A
     atoms = set(flatten([x.atoms() for x in flatten(A.tolist())]))
     return filter(lambda a: isinstance(a, ConstantField), atoms)