import numpy
from .ButcherTableaux import ButcherTableau
wso432A = numpy.asarray(
[[0.01900072890, 0, 0, 0],
[0.40434605601, 0.38435717512, 0, 0],
[0.06487908412, -0.16389640295, 0.51545231222, 0],
[0.02343549374, -0.41207877888, 0.96661161281, 0.42203167233]])
wso432b = numpy.asarray(
[0.02343549374, -0.41207877888, 0.96661161281, 0.42203167233])
wso432c = numpy.asarray(
[0.01900072890, 0.78870323114, 0.41643499339, 1])
wso433A = numpy.array(
[[0.13756543551, 0, 0, 0],
[0.56695122794, 0.23483888782, 0, 0],
[-1.08354072813, 2.96618223864, 0.44915521951, 0],
[0.59761291500, -0.43420997584, -0.05305815322, 0.88965521406]])
wso433b = numpy.array(
[0.59761291500, -0.43420997584, -0.05305815322, 0.88965521406])
wso433c = numpy.array(
[0.13756543551, 0.80179011576, 2.33179673002, 1])
wso643A = numpy.array(
[[0.079672377876931, 0, 0, 0, 0, 0],
[0.328355391763968, 0.136009256546967, 0, 0, 0, 0],
[-0.650772774016417, 1.742859063495349, 0.256472952467792,
0, 0, 0],
[-0.714580550967259, 1.793745752775934, -0.078254785672497,
0.311753794172585, 0, 0],
[-1.120092779092918, 1.983452339867353, 3.117393885836001,
-3.761930177913743, 0.770646024799205, 0],
[0.214823667785537, 0.536367363903245, 0.154488125726409,
-0.217748592703941, 0.072226422925896, 0.239843012362853]])
wso643b = numpy.array(
[0.214823667785537, 0.536367363903245, 0.154488125726409,
-0.217748592703941, 0.072226422925896, 0.239843012362853])
wso643c = numpy.array(
[0.079672377876931, 0.464364648310935, 1.348559241946724,
1.312664210308764, 0.989469293495897, 1])
wso744A = numpy.array(
[[1.290066345260422e-01, 0, 0, 0, 0, 0, 0],
[3.315354455306989e-01, 1.177478680001996e-01,
0, 0, 0, 0, 0],
[-8.009819642882672e-02, -2.408450965101765e-03, 9.242630648045402e-02, 0, 0, 0, 0],
[-1.730636616639455e+00, 1.513225984674677e+00, 1.221258626309848e+00, 2.266279031096887e-01, 0, 0, 0],
[1.475353790517696e-01, 3.618481772236499e-01, -5.603544220240282e-01, 2.455453653222619e+00, 5.742190161395324e-01, 0, 0],
[2.099717815888321e-01, 7.120237463672882e-01, -2.012023940726332e-02, -1.913828539529156e-02, -5.556044541810300e-03, 3.707277349712966e-01, 0],
[2.387938238483883e-01, 4.762495400483653e-01, 1.233935151213300e-02, 6.011995982693821e-02, 6.553618225489034e-05, -1.270730910442124e-01, 3.395048796261326e-01]])
wso744b = numpy.asarray(
[2.387938238483883e-01, 4.762495400483653e-01, 1.233935151213300e-02,
6.011995982693821e-02, 6.553618225489034e-05, -1.270730910442124e-01,
3.395048796261326e-01])
wso744c = numpy.asarray(
[1.290066345260422e-01, 4.492833135308985e-01, 9.919659086525534e-03,
1.230475897454758e+00, 2.978701803613543e+00, 1.247908693583052e+00,
1.000000000000000e+00])
wso1254A = numpy.array(
[[2.345371908646273e-01,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[6.874344413888787e-01, 5.515270980695153e-02,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[-1.183552669539587e-01, 5.463563002913454e-03, 1.458584459918280e-01,
0, 0, 0, 0, 0, 0, 0, 0, 0],
[-1.832235204042292e-01, 5.269029412008775e-02, 8.203685085133529e-01,
4.812118949092085e-02, 0, 0, 0, 0, 0, 0, 0, 0],
[9.941572060659400e-02, 4.977904930055774e-03, 5.414758174284321e-02,
-1.666571741820749e-03, 8.078975617332473e-02, 0, 0, 0, 0, 0, 0, 0],
[-9.896614721582678e-01, 2.860682690577833e+00, -1.236119341063179e+00,
2.130219523351530e+00, -1.260655031676537e+00, 2.457717913099987e-01,
0, 0, 0, 0, 0, 0],
[-5.656238413439102e-02, 1.661985685769353e-01, 6.464600922362508e-01,
6.608854962269927e-01, 3.736054198873429e-01, 6.294456964407685e-01,
5.702752607818027e-01, 0, 0, 0, 0, 0],
[8.048962104724392e-01, -6.232034990249100e-02, 5.737234603323347e-01,
-9.613723511489970e-02, 5.524106361737929e-01, 5.961002486833255e-01,
1.978411600659203e-01, 3.156238724024008e-01, 0, 0, 0, 0],
[-1.606381759216300e-01, 6.833397073337708e-01, 4.734578665308685e-01,
8.037708984872738e-01, -1.094498069459834e-02, 6.151263362711297e-01,
3.908946848682723e-01, 8.966103265353116e-02, 2.973255537857041e-02,
0, 0, 0],
[7.074283235644631e-01, 4.392037300952482e-01, -3.623592480237268e-02,
7.189990308645932e-04, 5.820968279166545e-01, 3.302003177175218e-01,
-2.394564021215881e-01, -7.540283547997615e-03, 1.702137469523672e-01,
6.268780138721711e-01, 0, 0],
[1.361197981133694e-01, -7.486549901902831e-01, 1.893908350024949e+00,
3.940485196730028e-01, 6.240233526545023e-02, 7.511983862200027e-01,
-5.283465265730526e-01, -1.661625677872943e+00, 9.998723833190827e-01,
1.377776742457387e+00, 8.905676409277480e-01, 0],
[-7.433675378768276e-01, 1.490594423766965e-01, -2.042884056742363e-02,
8.565329438087443e-04, 1.357261590983184e+00, 2.067512027776675e-03,
9.836884265759428e-02, -1.357936974507222e-02, -5.428992174996300e-02,
-3.803299038293005e-02, -9.150525836295019e-03, 2.712352651694511e-01]])
wso1254b = numpy.array(
[-7.433675378768276e-01, 1.490594423766965e-01, -2.042884056742363e-02,
8.565329438087443e-04, 1.357261590983184e+00, 2.067512027776675e-03,
9.836884265759428e-02, -1.357936974507222e-02, -5.428992174996300e-02,
-3.803299038293005e-02, -9.150525836295019e-03, 2.712352651694511e-01])
wso1254c = numpy.array(
[2.345371908646273e-01, 7.425871511958302e-01, 3.296674204078279e-02,
7.379564717201322e-01, 2.376643917109970e-01, 1.750238160341377e+00,
2.990308150015702e+00, 2.882138003112822e+00, 2.914399924907188e+00,
2.573507348677332e+00, 3.567266961364713e+00, 1.000000000000000e+00])
wso1255A = numpy.array(
[[4.113473525867655e-02, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[1.603459327727949e-01, 6.663913326722831e-02,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[-3.424389044264752e-01, 8.658006324816373e-01, 9.893519116923277e-02,
0, 0, 0, 0, 0, 0, 0, 0, 0],
[9.437182028870806e+00, -1.088783359642350e+01, 2.644025436733866e+00,
1.846155800500574e-01, 0, 0, 0, 0, 0, 0, 0, 0],
[-3.425409029430815e-01, 5.172239272544332e-01, 9.163589909678043e-01,
5.225142808845742e-02, 1.165485436026433e-01, 0, 0, 0, 0, 0, 0, 0],
[-2.094441177460360e+00, 2.577655753533404e+00, 5.704405293326313e-01,
1.213637180023516e-01, -4.752289775376601e-01, 5.285605969257756e-01,
0, 0, 0, 0, 0, 0],
[3.391631788320480e-01, -2.797427027028997e-01, 1.039483063369094e+00,
5.978770926212172e-02, -2.132900327070380e-01, 8.344318363436753e-02,
2.410106515779412e-01, 0, 0, 0, 0, 0],
[5.904282488642163e+00, 3.171195765985073e+00, -1.236822836316587e+01,
-4.989519066913001e-01, 2.160529620826442e+00, 1.916104322021480e+00,
1.988059486291180e+00, 2.232092386922440e-01, 0, 0, 0, 0],
[4.616443509508975e-01, -1.933433560549238e-01, -1.212541486279519e-01,
6.662362039716674e-02, 4.254912950625259e-01, 7.856131647013712e-01,
8.369551389357689e-01, 1.604780447895926e-01, 3.616125951766939e-01,
0, 0, 0],
[-7.087669749878204e-01, 6.466527094491541e-01, 4.758821526542215e-01,
-2.570518451375722e-01, 1.123185062554392e+00, 5.546921612875290e-01,
3.192424333237050e-01, 3.612077612576969e-01, 5.866779836068974e-01,
2.353799736246102e-01, 0, 0],
[4.264162484855930e-01, 1.322816663477840e+00, 4.245673729758231e-01,
-2.530402764527700e+00, -7.822016897497742e-02, 1.054463080605071e+00,
4.645590541391895e-01, 1.145097379521439e+00, 4.301337846893282e-01,
1.499513057076809e+00, 1.447942640822165e-02, 0],
[1.207394392845339e-02, 5.187080074649261e-01, 1.121304244847239e-01,
-4.959806334780896e-03, -1.345031364651444e+00, 3.398828703760807e-01,
8.159251531671077e-01, -2.640104266439604e-03, 1.439060901763520e-02,
-6.556567796749947e-03, 6.548135446843367e-04, 5.454220210658036e-01]])
wso1255c = numpy.array(
[4.113473525867655e-02, 2.269850660400232e-01, 6.222969192243949e-01,
1.377989449231234e+00, 1.259841986970257e+00, 1.228350442796143e+00,
1.269855051265635e+00, 2.496200652601413e+00, 2.783820705331141e+00,
3.337101417632813e+00, 4.173423133876636e+00, 1.000000000000000e+00])
wso1255b = numpy.array(
[1.207394392845339e-02, 5.187080074649261e-01, 1.121304244847239e-01,
-4.959806334780896e-03, -1.345031364651444e+00, 3.398828703760807e-01,
8.159251531671077e-01, -2.640104266439604e-03, 1.439060901763520e-02,
-6.556567796749947e-03, 6.548135446843367e-04, 5.454220210658036e-01])
wsodict = {
(4, 3, 2): (wso432A, wso432b, wso432c),
(4, 3, 3): (wso433A, wso433b, wso433c),
(6, 4, 3): (wso643A, wso643b, wso643c),
(7, 4, 4): (wso744A, wso744b, wso744c),
(12, 5, 4): (wso1254A, wso1254b, wso1254c),
(12, 5, 5): (wso1255A, wso1255b, wso1255c)
}
[docs]
class WSODIRK(ButcherTableau):
def __init__(self, ns, order, ws_order):
try:
A, b, c = wsodict[ns, order, ws_order]
except KeyError:
raise NotImplementedError("No WSO DIRK method for that combination of stages, order and weak stage order")
self.ws_order = ws_order
super(WSODIRK, self).__init__(A, b, None, c, order, None, None)
