/usr/include/arkode/arkode.h is in libsundials-dev 2.7.0+dfsg-2build1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 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 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 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 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 | /*---------------------------------------------------------------
* Programmer(s): Daniel R. Reynolds @ SMU
*---------------------------------------------------------------
* LLNS/SMU Copyright Start
* Copyright (c) 2015, Southern Methodist University and
* Lawrence Livermore National Security
*
* This work was performed under the auspices of the U.S. Department
* of Energy by Southern Methodist University and Lawrence Livermore
* National Laboratory under Contract DE-AC52-07NA27344.
* Produced at Southern Methodist University and the Lawrence
* Livermore National Laboratory.
*
* All rights reserved.
* For details, see the LICENSE file.
* LLNS/SMU Copyright End
*---------------------------------------------------------------
* This is the interface file for the main ARKODE integrator.
*---------------------------------------------------------------
* ARKODE is used to solve numerically the ordinary initial value
* problem:
* M(t)*y' = f(t,y),
* y(t0) = y0,
* where t0, y0 in R^N, M(t)*y: R x R^N -> R^N, and
* f: R x R^N -> R^N are given.
*--------------------------------------------------------------*/
#ifndef _ARKODE_H
#define _ARKODE_H
#include <stdio.h>
#include <sundials/sundials_nvector.h>
#ifdef __cplusplus /* wrapper to enable C++ usage */
extern "C" {
#endif
/*===============================================================
ARKODE CONSTANTS
===============================================================*/
/*---------------------------------------------------------------
Enumerations for inputs to ARKodeCreate and ARKode.
-----------------------------------------------------------------
Symbolic constants for the iter parameter to
ARKodeCreate and the input parameter itask to ARKode, are given
below.
itask: The itask input parameter to ARKode indicates the job
of the solver for the next user step. The ARK_NORMAL
itask is to have the solver take internal steps until
it has reached or just passed the user specified tout
parameter. The solver then interpolates in order to
return an approximate value of y(tout). The ARK_ONE_STEP
option tells the solver to just take one internal step
and return the solution at the point reached by that
step. If fixed time steps are desired, then the user
can run in ARK_ONE_STEP mode, and set hmin and hmax to
be the desired time step -- this will effectively
disable all temporal error control, and the user may
need to increase the iteration counts for the nonlinear
and linear solvers to guarantee convergence for larger
step sizes.
---------------------------------------------------------------*/
/* max number of overall stages allowed */
#define ARK_S_MAX 15
#define ARK_A(A,i,j) (A[i*ARK_S_MAX + j])
/* itask */
#define ARK_NORMAL 1
#define ARK_ONE_STEP 2
/* Butcher table accessors -- explicit */
#define HEUN_EULER_2_1_2 0
#define BOGACKI_SHAMPINE_4_2_3 1
#define ARK324L2SA_ERK_4_2_3 2
#define ZONNEVELD_5_3_4 3
#define ARK436L2SA_ERK_6_3_4 4
#define SAYFY_ABURUB_6_3_4 5
#define CASH_KARP_6_4_5 6
#define FEHLBERG_6_4_5 7
#define DORMAND_PRINCE_7_4_5 8
#define ARK548L2SA_ERK_8_4_5 9
#define VERNER_8_5_6 10
#define FEHLBERG_13_7_8 11
#define DEFAULT_ERK_2 HEUN_EULER_2_1_2
#define DEFAULT_ERK_3 BOGACKI_SHAMPINE_4_2_3
#define DEFAULT_ERK_4 ZONNEVELD_5_3_4
#define DEFAULT_ERK_5 CASH_KARP_6_4_5
#define DEFAULT_ERK_6 VERNER_8_5_6
#define DEFAULT_ERK_8 FEHLBERG_13_7_8
#define MIN_ERK_NUM 0
#define MAX_ERK_NUM 11
/* Butcher table accessors -- implicit */
#define SDIRK_2_1_2 12
#define BILLINGTON_3_3_2 13
#define TRBDF2_3_3_2 14
#define KVAERNO_4_2_3 15
#define ARK324L2SA_DIRK_4_2_3 16
#define CASH_5_2_4 17
#define CASH_5_3_4 18
#define SDIRK_5_3_4 19
#define KVAERNO_5_3_4 20
#define ARK436L2SA_DIRK_6_3_4 21
#define KVAERNO_7_4_5 22
#define ARK548L2SA_DIRK_8_4_5 23
#define DEFAULT_DIRK_2 SDIRK_2_1_2
#define DEFAULT_DIRK_3 ARK324L2SA_DIRK_4_2_3
#define DEFAULT_DIRK_4 SDIRK_5_3_4
#define DEFAULT_DIRK_5 ARK548L2SA_DIRK_8_4_5
#define MIN_DIRK_NUM 12
#define MAX_DIRK_NUM 23
/* Butcher table accessors -- ImEx */
#define DEFAULT_ARK_ETABLE_3 ARK324L2SA_ERK_4_2_3
#define DEFAULT_ARK_ETABLE_4 ARK436L2SA_ERK_6_3_4
#define DEFAULT_ARK_ETABLE_5 ARK548L2SA_ERK_8_4_5
#define DEFAULT_ARK_ITABLE_3 ARK324L2SA_DIRK_4_2_3
#define DEFAULT_ARK_ITABLE_4 ARK436L2SA_DIRK_6_3_4
#define DEFAULT_ARK_ITABLE_5 ARK548L2SA_DIRK_8_4_5
/* ARKODE return flags */
#define ARK_SUCCESS 0
#define ARK_TSTOP_RETURN 1
#define ARK_ROOT_RETURN 2
#define ARK_WARNING 99
#define ARK_TOO_MUCH_WORK -1
#define ARK_TOO_MUCH_ACC -2
#define ARK_ERR_FAILURE -3
#define ARK_CONV_FAILURE -4
#define ARK_LINIT_FAIL -5
#define ARK_LSETUP_FAIL -6
#define ARK_LSOLVE_FAIL -7
#define ARK_RHSFUNC_FAIL -8
#define ARK_FIRST_RHSFUNC_ERR -9
#define ARK_REPTD_RHSFUNC_ERR -10
#define ARK_UNREC_RHSFUNC_ERR -11
#define ARK_RTFUNC_FAIL -12
#define ARK_LFREE_FAIL -13
#define ARK_MASSINIT_FAIL -14
#define ARK_MASSSETUP_FAIL -15
#define ARK_MASSSOLVE_FAIL -16
#define ARK_MASSFREE_FAIL -17
#define ARK_MASSMULT_FAIL -18
#define ARK_MEM_FAIL -20
#define ARK_MEM_NULL -21
#define ARK_ILL_INPUT -22
#define ARK_NO_MALLOC -23
#define ARK_BAD_K -24
#define ARK_BAD_T -25
#define ARK_BAD_DKY -26
#define ARK_TOO_CLOSE -27
#define ARK_POSTPROCESS_FAIL -28
/*===============================================================
FUNCTION TYPES
===============================================================*/
/*---------------------------------------------------------------
Type : ARKRhsFn
-----------------------------------------------------------------
The f functions which define the right hand side of the ODE
system M*y' = fe(t,y) + fi(t,y) must have type ARKRhsFn.
f takes as input the independent variable value t, and the
dependent variable vector y. It stores the result of fe(t,y)
or fi(t,y) in the vector ydot. The y and ydot arguments are of
type N_Vector.
(Allocation of memory for ydot is handled within ARKODE)
The user_data parameter is the same as the user_data
parameter set by the user through the ARKodeSetUserData routine.
This user-supplied pointer is passed to the user's fe or fi
function every time it is called.
A ARKRhsFn should return 0 if successful, a negative value if
an unrecoverable error occured, and a positive value if a
recoverable error (e.g. invalid y values) occured.
If an unrecoverable occured, the integration is halted.
If a recoverable error occured, then (in most cases) ARKODE
will try to correct and retry.
---------------------------------------------------------------*/
typedef int (*ARKRhsFn)(realtype t, N_Vector y,
N_Vector ydot, void *user_data);
/*---------------------------------------------------------------
Type : ARKRootFn
-----------------------------------------------------------------
A function g, which defines a set of functions g_i(t,y) whose
roots are sought during the integration, must have type
ARKRootFn. The function g takes as input the independent
variable value t, and the dependent variable vector y. It
stores the nrtfn values g_i(t,y) in the realtype array gout.
(Allocation of memory for gout is handled within ARKODE.)
The user_data parameter is the same as that passed by the user
to the ARKodeSetUserData routine. This user-supplied pointer
is passed to the user's g function every time it is called.
A ARKRootFn should return 0 if successful or a non-zero value
if an error occured (in which case the integration will be
halted).
---------------------------------------------------------------*/
typedef int (*ARKRootFn)(realtype t, N_Vector y,
realtype *gout, void *user_data);
/*---------------------------------------------------------------
Type : ARKEwtFn
-----------------------------------------------------------------
A function e, which sets the error weight vector ewt, must have
type ARKEwtFn. The function e takes as input the current
dependent variable y. It must set the vector of error weights
used in the WRMS norm:
||y||_WRMS = sqrt [ 1/N * sum ( ewt_i * y_i)^2 ]
Typically, the vector ewt has components:
ewt_i = 1 / (reltol * |y_i| + abstol_i)
The user_data parameter is the same as that passed by the user
to the ARKodeSetUserData routine. This user-supplied pointer
is passed to the user's e function every time it is called.
A ARKEwtFn e must return 0 if the error weight vector has been
successfuly set and a non-zero value otherwise.
---------------------------------------------------------------*/
typedef int (*ARKEwtFn)(N_Vector y, N_Vector ewt, void *user_data);
/*---------------------------------------------------------------
Type : ARKRwtFn
-----------------------------------------------------------------
A function r, which sets the residual weight vector rwt, must
have type ARKRwtFn. The function r takes as input the current
dependent variable y. It must set the vector of residual
weights used in the WRMS norm:
||v||_WRMS = sqrt [ 1/N * sum ( rwt_i * v_i)^2 ]
Typically, the vector rwt has components:
rwt_i = 1 / (reltol * |(M*y)_i| + rabstol_i)
The user_data parameter is the same as that passed by the user
to the ARKodeSetUserData routine. This user-supplied pointer
is passed to the user's r function every time it is called.
A ARKRwtFn e must return 0 if the residual weight vector has
been successfuly set and a non-zero value otherwise.
---------------------------------------------------------------*/
typedef int (*ARKRwtFn)(N_Vector y, N_Vector rwt, void *user_data);
/*---------------------------------------------------------------
Type : ARKErrHandlerFn
-----------------------------------------------------------------
A function eh, which handles error messages, must have type
ARKErrHandlerFn. The function eh takes as input the error code,
the name of the module reporting the error, the error message,
and a pointer to user data, the same as that passed to
ARKodeSetUserData.
All error codes are negative, except ARK_WARNING which indicates
a warning (the solver continues).
A ARKErrHandlerFn has no return value.
---------------------------------------------------------------*/
typedef void (*ARKErrHandlerFn)(int error_code, const char *module,
const char *function, char *msg,
void *user_data);
/*---------------------------------------------------------------
Type : ARKAdaptFn
-----------------------------------------------------------------
A function which sets the new time step h, must have type
ARKAdaptFn. The function takes as input the current dependent
variable y, the current time t, the last 3 step sizes h,
the last 3 error estimates, the method order q, the embedding
order p, and a pointer to user data. The function must set the
scalar step size for the upcoming time step. This value will
subsequently be bounded by the user-supplied values for the
minimum and maximum allowed time step, and the time step
satisfying the explicit stability restriction. The user_data
parameter is the same as that passed by the user to the
ARKodeSetUserData routine. This user-supplied pointer is passed
to the function every time it is called.
A ARKAdaptFn must return 0 if the new time step has been
successfuly set and a non-zero value otherwise.
---------------------------------------------------------------*/
typedef int (*ARKAdaptFn)(N_Vector y, realtype t, realtype h1,
realtype h2, realtype h3,
realtype e1, realtype e2,
realtype e3, int q, int p,
realtype *hnew, void *user_data);
/*---------------------------------------------------------------
Type : ARKExpStabFn
-----------------------------------------------------------------
A function which returns the time step satisfying the stability
restriction for the explicit portion of the ODE. The function
takes as input the current dependent variable y, the current
time t, and a pointer to user data. The function must set the
scalar step size satisfying the stability restriction for the
upcoming time step. This value will subsequently be bounded by
the user-supplied values for the minimum and maximum allowed
time step, and the accuracy-based time step. The user_data
parameter is the same as that passed by the user to the
ARKodeSetUserData routine. This user-supplied pointer is passed
to the function every time it is called.
If this function is not supplied (NULL), or if it returns a
negative time step size, then ARKode will assume that there is
no explicit stability restriction on the time step size.
A ARKExpStabFn must return 0 if the step size limit has been
successfuly set and a non-zero value otherwise.
---------------------------------------------------------------*/
typedef int (*ARKExpStabFn)(N_Vector y, realtype t,
realtype *hstab, void *user_data);
/*---------------------------------------------------------------
Type : ARKVecResizeFn
-----------------------------------------------------------------
When calling ARKodeResize, the user may specify a vector resize
function to be used to convert any existing N_Vectors in the
ARKode memory structure to the new problem size. This would
typically be used if there is a user-supplied N_Vector module
that allows dynamic resizing of the vector data structures
without the need to delete/allocate memory on each call.
The default behavior will be to delete the vector memory and
re-clone from the new vector; if this is the desired behavior
then specification of the ARKVecResizeFn is not recommended.
The first argument, 'y', is the vector to be resized.
The second argument, 'ytemplate', is the user-provided vector
with the "new" size, that may be used as a template.
The third argument, 'user_data', is a user-provided data
structure to ARKodeResize, in case additional data is
necessary for the resize operation.
A ARKVecResizeFn should return 0 if successful, and a nonzero
value if an error occurred.
---------------------------------------------------------------*/
typedef int (*ARKVecResizeFn)(N_Vector y, N_Vector ytemplate,
void *user_data);
/*---------------------------------------------------------------
Type : ARKPostProcessStepFn
-----------------------------------------------------------------
A function that is used to process the results of each timestep
solution, in preparation for subsequent steps. A routine of
this type is designed for tasks such as inter-processor
communication, computation of derived quantities, etc..
IF THIS IS USED TO MODIFY ANY OF THE ACTIVE STATE DATA, THEN ALL
THEORETICAL GUARANTEES OF SOLUTION ACCURACY AND STABILITY ARE
LOST.
Inputs:
t current time of ARKode solution
y current ARKode solution N_Vector for processing
user_data the structure passed by the user to the
ARKodeSetUserData routine.
---------------------------------------------------------------*/
typedef int (*ARKPostProcessStepFn)(realtype t, N_Vector y,
void *user_data);
/*===============================================================
USER-CALLABLE ROUTINES
===============================================================*/
/*---------------------------------------------------------------
Function : ARKodeCreate
-----------------------------------------------------------------
ARKodeCreate creates an internal memory block for a problem to
be solved by ARKODE.
If successful, ARKodeCreate returns a pointer to initialized
problem memory. This pointer should be passed to ARKodeInit.
If an initialization error occurs, ARKodeCreate prints an error
message to standard err and returns NULL.
---------------------------------------------------------------*/
SUNDIALS_EXPORT void *ARKodeCreate();
/*---------------------------------------------------------------
Integrator optional input specification functions
-----------------------------------------------------------------
The following functions can be called to set optional inputs
to values other than the defaults given below.
Function | Optional input / [ default value ]
-----------------------------------------------------------------
ARKodeSetDefaults | resets all optional inputs to ARKode
| default values. Does not change
| problem-defining function pointers
| fe and fi or user_data pointer. Also
| leaves alone any data structures or
| options related to root-finding
| (those can be reset using
! ARKodeRootInit).
| [internal]
|
ARKodeSetOptimalParams | sets all adaptivity and solver
| parameters to our 'best guess' values,
| for a given integration method (ERK,
| DIRK, ARK) and a given method order.
| Should only be called after the method
| order and integration method have been
! set.
| [internal]
|
ARKodeSetErrHandlerFn | user-provided ErrHandler function.
| [internal]
|
ARKodeSetErrFile | the file pointer for an error file
| where all ARKODE warning and error
| messages will be written if the
| default internal error handling
| function is used. This parameter can
| be stdout (standard output), stderr
| (standard error), or a file pointer
| (corresponding to a user error file
| opened for writing) returned by fopen.
| If not called, then all messages will
| be written to stderr.
| [stderr]
|
ARKodeSetUserData | a pointer to user data that will be
| passed to the user's f function every
| time f is called.
| [NULL]
|
ARKodeSetDiagnostics | the file pointer for a diagnostics file
| where all ARKODE step adaptivity and solver
| information is written. This parameter can
| be stdout or stderr, though the preferred
| approach is to specify a file pointer
| (corresponding to a user diagnostics file
| opened for writing) returned by fopen. If
| not called, or if called with a NULL file
| pointer, all diagnostics output is disabled.
| NOTE: when run in parallel, only one process
| should set a non-NULL value for this pointer,
| since statistics from all processes would be
| identical.
| [NULL]
|
ARKodeSetOrder | method order to be used by the solver.
| [4]
|
ARKodeSetDenseOrder | polynomial order to be used for dense
| output. Allowed values are between 0
| and min(q,5) (where q is the order of
| the integrator)
| [3]
|
ARKodeSetLinear | specifies that the implicit portion of
| the problem is linear, and to tighten
| the linear solver tolerances while
| taking only one Newton iteration.
| [FALSE]
|
ARKodeSetNonlinear | specifies that the implicit portion of
| the problem is nonlinear. Used to undo
| a previous call to ARKodeSetLinear.
| [TRUE]
|
ARKodeSetFixedPoint | specifies that the implicit portion of
| the problem should use the accelerated
| fixed-point solver.
| [FALSE]
|
ARKodeSetNewton | specifies that the implicit portion of
| the problem should use the modified Newton
| solver. Used to undo a previous call to
| ARKodeSetFixedPoint.
| [TRUE]
|
ARKodeSetExplicit | specifies that implicit portion of
| problem is disabled, and to use an
| explicit RK method.
| [FALSE]
|
ARKodeSetImplicit | specifies that explicit portion of
| problem is disabled, and to use an
| implicit RK method.
| [FALSE]
|
ARKodeSetImEx | specifies that problem has both
| implicit and explicit parts, and to
| use an ARK method.
| [TRUE]
|
ARKodeSetERKTable | specifies to use a customized Butcher
| table for the explicit portion of the
| system. This automatically calls
! ARKodeSetExplicit.
| [determined by ARKODE based on order]
|
ARKodeSetIRKTable | specifies to use a customized Butcher
| table for the implicit portion of the
| system. This automatically calls
! ARKodeSetImplicit.
| [determined by ARKODE based on order]
|
ARKodeSetARKTables | specifies to use customized Butcher
| tables for the IMEX system. This
! automatically calls ARKodeSetImEx.
| [determined by ARKODE based on order]
|
ARKodeSetERKTableNum | specifies to use a built-in Butcher
| table for the explicit portion of the
| system. The integer argument should
| match an existing method in
| ARKodeLoadButcherTable() within the file
| arkode_butcher.c. Error-checking is
| performed to ensure that the table
| exists, and is not implicit. This
! automatically calls ARKodeSetExplicit.
| [determined by ARKODE based on order]
|
ARKodeSetIRKTableNum | specifies to use a built-in Butcher
| table for the implicit portion of the
| system. The integer argument should
| match an existing method in
| ARKodeLoadButcherTable() within the file
| arkode_butcher.c. Error-checking is
| performed to ensure that the table
| exists, and is not explicit. This
! automatically calls ARKodeSetImplicit.
| [determined by ARKODE based on order]
|
ARKodeSetARKTableNum | specifies to use a built-in Butcher
| tables for the ImEx system. The
! integer arguments should match existing
| methods in ARKodeLoadButcherTable()
| within the file arkode_butcher.c.
| Error-checking is performed to ensure
| that the tables exist. Subsequent
| error-checking is automatically performed
| to ensure that the tables' stage times
| and solution coefficients match. This
! automatically calls ARKodeSetImEx.
| [determined by ARKODE based on order]
|
ARKodeSetMaxNumSteps | maximum number of internal steps to be
| taken by the solver in its attempt to
| reach tout.
| [500]
|
ARKodeSetMaxHnilWarns | maximum number of warning messages
| issued by the solver that t+h==t on
| the next internal step. A value of -1
| means no such messages are issued.
| [10]
|
ARKodeSetInitStep | initial step size.
| [estimated by ARKODE]
|
ARKodeSetMinStep | minimum absolute value of step size
| allowed.
| [0.0]
|
ARKodeSetMaxStep | maximum absolute value of step size
| allowed.
| [infinity]
|
ARKodeSetStopTime | the independent variable value past
| which the solution is not to proceed.
| [infinity]
|
ARKodeSetFixedStep | specifies to use a fixed step size
| throughout integration
| [off]
|
ARKodeSetCFLFraction | safety factor to use for explicitly
! stable steps
| [0.5]
|
ARKodeSetSafetyFactor | safety factor to use for error-based
! step adaptivity
| [0.96]
|
ARKodeSetErrorBias | error bias factor to use in error-based
! step adaptivity
| [1.5]
|
ARKodeSetMaxGrowth | maximum growth factor for successive
! time steps (not including the first step).
| [20.0]
|
ARKodeSetMaxFirstGrowth | maximum growth factor for first step.
| [10000.0]
|
ARKodeSetMaxEFailGrowth | maximum growth factor after an error failure.
| [0.3]
|
ARKodeSetSmallNumEFails | maximum number of error failures before
! MaxFailGrowth factor is used.
| [2]
|
ARKodeSetMaxCFailGrowth | maximum growth factor after a convergence failure.
| [0.25]
|
ARKodeSetFixedStepBounds | step growth interval to force retention of
! the same step size
| [1.0 1.5]
|
ARKodeSetAdaptivityMethod | Method to use for time step adaptivity
| [0]
|
ARKodeSetAdaptivityFn | user-provided time step adaptivity
| function.
| [internal]
|
ARKodeSetNonlinCRDown | user-provided nonlinear convergence
| rate constant.
| [0.3]
|
ARKodeSetNonlinRDiv | user-provided nonlinear divergence ratio.
| [2.3]
|
ARKodeSetDeltaGammaMax | user-provided linear setup decision
| constant.
| [0.2]
|
ARKodeSetMaxStepsBetweenLSet | user-provided linear setup decision
| constant.
| [20]
|
ARKodeSetPredictorMethod | Method to use for predicting implicit
| solutions.
| [0]
|
ARKodeSetStabilityFn | user-provided explicit time step
| stability function.
| [internal]
|
ARKodeSetMaxErrTestFails | Maximum number of error test failures
| in attempting one step.
| [7]
|
ARKodeSetMaxNonlinIters | Maximum number of nonlinear solver
| iterations at one stage solution.
| [3]
|
ARKodeSetMaxConvFails | Maximum number of convergence failures
| allowed in attempting one step.
| [10]
|
ARKodeSetNonlinConvCoef | Coefficient in the nonlinear
| convergence test.
| [0.1]
|
-----------------------------------------------------------------
ARKodeSetRootDirection | Specifies the direction of zero
| crossings to be monitored
| [both directions]
|
ARKodeSetNoInactiveRootWarn | disable warning about possible
| g==0 at beginning of integration
-----------------------------------------------------------------
Return flag:
ARK_SUCCESS if successful
ARK_MEM_NULL if the arkode memory is NULL
ARK_ILL_INPUT if an argument has an illegal value
---------------------------------------------------------------*/
SUNDIALS_EXPORT int ARKodeSetDefaults(void *arkode_mem);
SUNDIALS_EXPORT int ARKodeSetOptimalParams(void *arkode_mem);
SUNDIALS_EXPORT int ARKodeSetErrHandlerFn(void *arkode_mem,
ARKErrHandlerFn ehfun,
void *eh_data);
SUNDIALS_EXPORT int ARKodeSetErrFile(void *arkode_mem,
FILE *errfp);
SUNDIALS_EXPORT int ARKodeSetUserData(void *arkode_mem,
void *user_data);
SUNDIALS_EXPORT int ARKodeSetDiagnostics(void *arkode_mem,
FILE *diagfp);
SUNDIALS_EXPORT int ARKodeSetOrder(void *arkode_mem, int maxord);
SUNDIALS_EXPORT int ARKodeSetDenseOrder(void *arkode_mem, int dord);
SUNDIALS_EXPORT int ARKodeSetLinear(void *arkode_mem, int timedepend);
SUNDIALS_EXPORT int ARKodeSetNonlinear(void *arkode_mem);
SUNDIALS_EXPORT int ARKodeSetFixedPoint(void *arkode_mem, long int fp_m);
SUNDIALS_EXPORT int ARKodeSetNewton(void *arkode_mem);
SUNDIALS_EXPORT int ARKodeSetExplicit(void *arkode_mem);
SUNDIALS_EXPORT int ARKodeSetImplicit(void *arkode_mem);
SUNDIALS_EXPORT int ARKodeSetImEx(void *arkode_mem);
SUNDIALS_EXPORT int ARKodeSetERKTable(void *arkode_mem, int s,
int q, int p, realtype *c,
realtype *A, realtype *b,
realtype *bembed);
SUNDIALS_EXPORT int ARKodeSetIRKTable(void *arkode_mem, int s,
int q, int p, realtype *c,
realtype *A, realtype *b,
realtype *bembed);
SUNDIALS_EXPORT int ARKodeSetARKTables(void *arkode_mem, int s,
int q, int p,
realtype *ci, realtype *ce,
realtype *Ai, realtype *Ae,
realtype *bi, realtype *be,
realtype *b2i, realtype *b2e);
SUNDIALS_EXPORT int ARKodeSetERKTableNum(void *arkode_mem, int itable);
SUNDIALS_EXPORT int ARKodeSetIRKTableNum(void *arkode_mem, int itable);
SUNDIALS_EXPORT int ARKodeSetARKTableNum(void *arkode_mem,
int itable, int etable);
SUNDIALS_EXPORT int ARKodeSetMaxNumSteps(void *arkode_mem,
long int mxsteps);
SUNDIALS_EXPORT int ARKodeSetMaxHnilWarns(void *arkode_mem,
int mxhnil);
SUNDIALS_EXPORT int ARKodeSetInitStep(void *arkode_mem,
realtype hin);
SUNDIALS_EXPORT int ARKodeSetMinStep(void *arkode_mem,
realtype hmin);
SUNDIALS_EXPORT int ARKodeSetMaxStep(void *arkode_mem,
realtype hmax);
SUNDIALS_EXPORT int ARKodeSetStopTime(void *arkode_mem,
realtype tstop);
SUNDIALS_EXPORT int ARKodeSetFixedStep(void *arkode_mem,
realtype hfixed);
SUNDIALS_EXPORT int ARKodeSetCFLFraction(void *arkode_mem,
realtype cfl_frac);
SUNDIALS_EXPORT int ARKodeSetSafetyFactor(void *arkode_mem,
realtype safety);
SUNDIALS_EXPORT int ARKodeSetErrorBias(void *arkode_mem,
realtype bias);
SUNDIALS_EXPORT int ARKodeSetMaxGrowth(void *arkode_mem,
realtype mx_growth);
SUNDIALS_EXPORT int ARKodeSetFixedStepBounds(void *arkode_mem,
realtype lb, realtype ub);
SUNDIALS_EXPORT int ARKodeSetAdaptivityMethod(void *arkode_mem,
int imethod,
int idefault, int pq,
realtype *adapt_params);
SUNDIALS_EXPORT int ARKodeSetAdaptivityFn(void *arkode_mem,
ARKAdaptFn hfun,
void *h_data);
SUNDIALS_EXPORT int ARKodeSetMaxFirstGrowth(void *arkode_mem,
realtype etamx1);
SUNDIALS_EXPORT int ARKodeSetMaxEFailGrowth(void *arkode_mem,
realtype etamxf);
SUNDIALS_EXPORT int ARKodeSetSmallNumEFails(void *arkode_mem,
int small_nef);
SUNDIALS_EXPORT int ARKodeSetMaxCFailGrowth(void *arkode_mem,
realtype etacf);
SUNDIALS_EXPORT int ARKodeSetNonlinCRDown(void *arkode_mem,
realtype crdown);
SUNDIALS_EXPORT int ARKodeSetNonlinRDiv(void *arkode_mem,
realtype rdiv);
SUNDIALS_EXPORT int ARKodeSetDeltaGammaMax(void *arkode_mem,
realtype dgmax);
SUNDIALS_EXPORT int ARKodeSetMaxStepsBetweenLSet(void *arkode_mem,
int msbp);
SUNDIALS_EXPORT int ARKodeSetPredictorMethod(void *arkode_mem,
int method);
SUNDIALS_EXPORT int ARKodeSetStabilityFn(void *arkode_mem,
ARKExpStabFn EStab,
void *estab_data);
SUNDIALS_EXPORT int ARKodeSetMaxErrTestFails(void *arkode_mem,
int maxnef);
SUNDIALS_EXPORT int ARKodeSetMaxNonlinIters(void *arkode_mem,
int maxcor);
SUNDIALS_EXPORT int ARKodeSetMaxConvFails(void *arkode_mem,
int maxncf);
SUNDIALS_EXPORT int ARKodeSetNonlinConvCoef(void *arkode_mem,
realtype nlscoef);
SUNDIALS_EXPORT int ARKodeSetRootDirection(void *arkode_mem,
int *rootdir);
SUNDIALS_EXPORT int ARKodeSetNoInactiveRootWarn(void *arkode_mem);
SUNDIALS_EXPORT int ARKodeSetPostprocessStepFn(void *arkode_mem,
ARKPostProcessStepFn ProcessStep);
/*---------------------------------------------------------------
Function : ARKodeInit
-----------------------------------------------------------------
ARKodeInit allocates and initializes memory for a problem to
to be solved by ARKODE.
arkode_mem is pointer to ARKODE memory returned by ARKodeCreate.
fe is the name of the C function defining the explicit
portion of the right-hand side function in
y' = fe(t,y) + fi(t,y).
fi is the name of the C function defining the implicit
portion of the right-hand side function in
y' = fe(t,y) + fi(t,y).
t0 is the initial value of t.
y0 is the initial condition vector y(t0).
Return flag:
ARK_SUCCESS if successful
ARK_MEM_NULL if the arkode memory was NULL
ARK_MEM_FAIL if a memory allocation failed
ARK_ILL_INPUT if an argument has an illegal value.
---------------------------------------------------------------*/
SUNDIALS_EXPORT int ARKodeInit(void *arkode_mem, ARKRhsFn fe,
ARKRhsFn fi, realtype t0, N_Vector y0);
/*---------------------------------------------------------------
Function : ARKodeReInit
-----------------------------------------------------------------
ARKodeReInit re-initializes ARKode for the solution of a problem,
where a prior call to ARKodeInit has been made with the same
problem size N. ARKodeReInit performs the same input checking
and initializations that ARKodeInit does, but it does no memory
allocation, since the basic data structures initialized in
ARKodeInit may be reused.
All of the arguments to ARKodeReInit have names and meanings
identical to those of ARKodeInit.
The return value of ARKodeReInit is equal to ARK_SUCCESS = 0 if
there were no errors; otherwise it is a negative int equal to:
ARK_MEM_NULL indicating arkode_mem was NULL (i.e.,
ARKodeCreate has not been called).
ARK_NO_MALLOC indicating that arkode_mem has not been
allocated (i.e., ARKodeInit has not been
called).
ARK_ILL_INPUT indicating an input argument was illegal
(including an attempt to increase maxord).
In case of an error return, an error message is also printed.
--------------------------------------------------------------*/
SUNDIALS_EXPORT int ARKodeReInit(void *arkode_mem, ARKRhsFn fe,
ARKRhsFn fi, realtype t0, N_Vector y0);
/*---------------------------------------------------------------
Function : ARKodeResize
-----------------------------------------------------------------
ARKodeResize re-initializes ARKODE's memory for a problem with a
changing vector size. It is assumed that the problem dynamics
before and after the vector resize will be comparable, so that
all time-stepping heuristics prior to calling ARKodeResize
remain valid after the call. If instead the dynamics should be
re-calibrated, the ARKode memory structure should be deleted
with a call to ARKodeFree, and re-created with calls to
ARKodeCreate and ARKodeInit.
To aid in the vector-resize operation, the user can supply a
vector resize function, that will take as input an N_Vector with
the previous size, and return as output a corresponding vector
of the new size. If this function (of type ARKVecResizeFn) is
not supplied (i.e. is set to NULL), then all existing N_Vectors
will be destroyed and re-cloned from the input vector.
In the case that the dynamical time scale should be modified
slightly from the previous time scale, an input "hscale" is
allowed, that will re-scale the upcoming time step by the
specified factor. If a value <= 0 is specified, the default of
1.0 will be used.
Other arguments:
arkode_mem Existing ARKode memory data structure.
ynew The newly-sized solution vector, holding
the current dependent variable values.
t0 The current value of the independent
variable.
resize_data User-supplied data structure that will be
passed to the supplied resize function.
The return value of ARKodeResize is equal to ARK_SUCCESS = 0 if
there were no errors; otherwise it is a negative int equal to:
ARK_MEM_NULL indicating arkode_mem was NULL (i.e.,
ARKodeCreate has not been called).
ARK_NO_MALLOC indicating that arkode_mem has not been
allocated (i.e., ARKodeInit has not been
called).
ARK_ILL_INPUT indicating an input argument was illegal
(including an error from the supplied
resize function).
In case of an error return, an error message is also printed.
---------------------------------------------------------------*/
SUNDIALS_EXPORT int ARKodeResize(void *arkode_mem, N_Vector ynew,
realtype hscale, realtype t0,
ARKVecResizeFn resize,
void *resize_data);
/*---------------------------------------------------------------
Functions : ARKodeSStolerances
ARKodeSVtolerances
ARKodeWFtolerances
-----------------------------------------------------------------
These functions specify the integration tolerances. One of them
SHOULD be called before the first call to ARKode; otherwise
default values of reltol=1e-4 and abstol=1e-9 will be used,
which may be entirely incorrect for a specific problem.
ARKodeSStolerances specifies scalar relative and absolute
tolerances.
ARKodeSVtolerances specifies scalar relative tolerance and a
vector absolute tolerance (a potentially different absolute
tolerance for each vector component).
ARKodeWFtolerances specifies a user-provided function (of type
ARKEwtFn) which will be called to set the error weight vector.
The tolerances reltol and abstol define a vector of error
weights, ewt, with components
ewt[i] = 1/(reltol*abs(y[i]) + abstol) (in SS case), or
ewt[i] = 1/(reltol*abs(y[i]) + abstol[i]) (in SV case).
This vector is used in all error and convergence tests, which
use a weighted RMS norm on all error-like vectors v:
WRMSnorm(v) = sqrt( (1/N) sum(i=1..N) (v[i]*ewt[i])^2 ),
where N is the problem dimension.
The return value of these functions is equal to ARK_SUCCESS=0
if there were no errors; otherwise it is a negative int equal
to:
ARK_MEM_NULL indicating arkode_mem was NULL (i.e.,
ARKodeCreate has not been called).
ARK_NO_MALLOC indicating that arkode_mem has not been
allocated (i.e., ARKodeInit has not been
called).
ARK_ILL_INPUT indicating an input argument was illegal
(e.g. a negative tolerance)
In case of an error return, an error message is also printed.
---------------------------------------------------------------*/
SUNDIALS_EXPORT int ARKodeSStolerances(void *arkode_mem,
realtype reltol,
realtype abstol);
SUNDIALS_EXPORT int ARKodeSVtolerances(void *arkode_mem,
realtype reltol,
N_Vector abstol);
SUNDIALS_EXPORT int ARKodeWFtolerances(void *arkode_mem,
ARKEwtFn efun);
/*---------------------------------------------------------------
Functions : ARKodeResStolerance
ARKodeResVtolerance
ARKodeResFtolerance
-----------------------------------------------------------------
These functions specify the absolute residual tolerance.
Specification of the absolute residual tolerance is only
necessary for problems with non-identity mass matrices in which
the units of the solution vector y dramatically differ from the
units of My, where M is the user-supplied mass matrix. If this
occurs, one of these routines SHOULD be called before the first
call to ARKode; otherwise the default value of rabstol=1e-9 will
be used, which may be entirely incorrect for a specific problem.
ARKodeResStolerances specifies a scalar residual tolerance.
ARKodeResVtolerances specifies a vector residual tolerance
(a potentially different absolute residual tolerance for
each vector component).
ARKodeResFtolerances specifies a user-provides function (of
type ARKRwtFn) which will be called to set the residual
weight vector.
The tolerances reltol (defined for both the solution and
residual) and rabstol define a vector of residual weights,
rwt, with components
rwt[i] = 1/(reltol*abs(My[i]) + abstol) (in S case), or
rwt[i] = 1/(reltol*abs(My[i]) + abstol[i]) (in V case).
This vector is used in all solver convergence tests, which
use a weighted RMS norm on all residual-like vectors v:
WRMSnorm(v) = sqrt( (1/N) sum(i=1..N) (v[i]*rwt[i])^2 ),
where N is the problem dimension.
The return value of these functions is equal to ARK_SUCCESS=0
if there were no errors; otherwise it is a negative int equal
to:
ARK_MEM_NULL indicating arkode_mem was NULL (i.e.,
ARKodeCreate has not been called).
ARK_NO_MALLOC indicating that arkode_mem has not been
allocated (i.e., ARKodeInit has not been
called).
ARK_ILL_INPUT indicating an input argument was illegal
(e.g. a negative tolerance)
In case of an error return, an error message is also printed.
---------------------------------------------------------------*/
SUNDIALS_EXPORT int ARKodeResStolerance(void *arkode_mem,
realtype rabstol);
SUNDIALS_EXPORT int ARKodeResVtolerance(void *arkode_mem,
N_Vector rabstol);
SUNDIALS_EXPORT int ARKodeResFtolerance(void *arkode_mem,
ARKRwtFn rfun);
/*---------------------------------------------------------------
Function : ARKodeRootInit
-----------------------------------------------------------------
ARKodeRootInit initializes a rootfinding problem to be solved
during the integration of the ODE system. It must be called
after ARKodeCreate, and before ARKode. The arguments are:
arkode_mem = pointer to ARKODE memory returned by ARKodeCreate.
nrtfn = number of functions g_i, an integer >= 0.
g = name of user-supplied function, of type ARKRootFn,
defining the functions g_i whose roots are sought.
If a new problem is to be solved with a call to ARKodeReInit,
where the new problem has no root functions but the prior one
did, then call ARKodeRootInit with nrtfn = 0.
The return value of ARKodeRootInit is ARK_SUCCESS = 0 if there
were no errors; otherwise it is a negative int equal to:
ARK_MEM_NULL indicating arkode_mem was NULL, or
ARK_MEM_FAIL indicating a memory allocation failed.
(including an attempt to increase maxord).
ARK_ILL_INPUT indicating nrtfn > 0 but g = NULL.
In case of an error return, an error message is also printed.
---------------------------------------------------------------*/
SUNDIALS_EXPORT int ARKodeRootInit(void *arkode_mem,
int nrtfn,
ARKRootFn g);
/*---------------------------------------------------------------
Function : ARKode
-----------------------------------------------------------------
ARKode integrates the ODE over an interval in t.
ARKode may be run in one of two modes (ARK_NORMAL or
ARK_ONE_STEP), as determined by the itask argument:
If itask is ARK_NORMAL, then the solver integrates from its
current internal t value to a point at or beyond tout, then
interpolates to t = tout and returns y(tout) in the user-
allocated vector yout. This interpolation is typically less
accurate than the full time step solutions produced by the
solver, since the interpolating polynomial relies on the
internal stage solutions, that may have reduced accuracy in
comparison with the full time step solutions. If the user
wishes that this returned value have full method accuracy, they
may issue a call to ARKodeSetStopTime before the call to ARKode
to specify a fixed stop time to end the time step and return to
the user. Once the integrator returns at a tstop time, any
future testing for tstop is disabled (and can be reenabled only
though a new call to ARKodeSetStopTime).
If itask is ARK_ONE_STEP, then the solver takes one internal
time step and returns in yout the value of y at the new internal
time. In this case, tout is used only during the first call to
ARKode to determine the direction of integration and the rough
scale of the t variable. As with the ARK_NORMAL mode, a user
may specify a specific stop time for output of this step,
assuming that the requested step is smaller than the step taken
by the method.
The time reached by the solver is placed in (*tret). The
user is responsible for allocating the memory for this value.
arkode_mem is the pointer to ARKODE memory returned by
ARKodeCreate.
tout is the next time at which a computed solution is desired.
yout is the computed solution vector. In ARK_NORMAL mode with no
errors and no roots found, yout=y(tout).
tret is a pointer to a real location. ARKode sets (*tret) to
the time reached by the solver and returns
yout=y(*tret).
itask is ARK_NORMAL or ARK_ONE_STEP, as described above.
Here is a brief description of each return value:
ARK_SUCCESS: ARKode succeeded and no roots were found.
ARK_ROOT_RETURN: ARKode succeeded, and found one or more roots.
If nrtfn > 1, call ARKodeGetRootInfo to see
which g_i were found to have a root at (*tret).
ARK_TSTOP_RETURN: ARKode succeeded and returned at tstop.
ARK_MEM_NULL: The arkode_mem argument was NULL.
ARK_NO_MALLOC: arkode_mem was not allocated.
ARK_ILL_INPUT: One of the inputs to ARKode is illegal. This
includes the situation when a component of the
error weight vectors becomes < 0 during
internal time-stepping. It also includes the
situation where a root of one of the root
functions was found both at t0 and very near t0.
The ILL_INPUT flag will also be returned if the
linear solver routine ARK--- (called by the user
after calling ARKodeCreate) failed to set one of
the linear solver-related fields in arkode_mem or
if the linear solver's init routine failed. In
any case, the user should see the printed
error message for more details.
ARK_TOO_MUCH_WORK: The solver took mxstep internal steps but
could not reach tout prior to the maximum number
of steps (ark_mxstep).
ARK_TOO_MUCH_ACC: The solver could not satisfy the accuracy
demanded by the user for some internal step.
ARK_ERR_FAILURE: Error test failures occurred too many times
(= ark_maxnef) during one internal time step
or occurred with |h| = hmin.
ARK_CONV_FAILURE: Convergence test failures occurred too many
times (= ark_maxncf) during one internal time
step or occurred with |h| = hmin.
ARK_LINIT_FAIL: The linear solver's initialization function
failed.
ARK_LSETUP_FAIL: The linear solver's setup routine failed in an
unrecoverable manner.
ARK_LSOLVE_FAIL: The linear solver's solve routine failed in an
unrecoverable manner.
---------------------------------------------------------------*/
SUNDIALS_EXPORT int ARKode(void *arkode_mem, realtype tout,
N_Vector yout, realtype *tret, int itask);
/*---------------------------------------------------------------
Function : ARKodeGetDky
-----------------------------------------------------------------
ARKodeGetDky computes the kth derivative of the y function at
time t, where tn-hu <= t <= tn, tn denotes the current
internal time reached, and hu is the last internal step size
successfully used by the solver. The user may request
k=0, 1, ..., d, where d = min(5,q), with q the order of accuracy
for the time integration method. The derivative vector is
returned in dky. This vector must be allocated by the caller. It
is only legal to call this function after a successful return
from ARKode.
arkode_mem is the pointer to ARKODE memory returned by
ARKodeCreate.
t is the time at which the kth derivative of y is evaluated.
The legal range for t is [tn-hu,tn] as described above.
k is the order of the derivative of y to be computed. The
legal range for k is [0,min(q,3)] as described above.
dky is the output derivative vector [((d/dy)^k)y](t).
The return value for ARKodeGetDky is one of:
ARK_SUCCESS: ARKodeGetDky succeeded.
ARK_BAD_K: k is not in the range 0, 1, ..., s-1.
ARK_BAD_T: t is not in the interval [tn-hu,tn].
ARK_BAD_DKY: The dky argument was NULL.
ARK_MEM_NULL: The arkode_mem argument was NULL.
---------------------------------------------------------------*/
SUNDIALS_EXPORT int ARKodeGetDky(void *arkode_mem, realtype t,
int k, N_Vector dky);
/*---------------------------------------------------------------
Integrator optional output extraction functions
-----------------------------------------------------------------
The following functions can be called to get optional outputs
and statistics related to the main integrator.
-----------------------------------------------------------------
ARKodeGetWorkSpace returns the ARKODE real and integer workspaces
ARKodeGetNumSteps returns the cumulative number of internal
steps taken by the solver
ARKodeGetNumExpSteps returns the cumulative number of stability
limited steps taken by the solver
ARKodeGetNumAccSteps returns the cumulative number of accuracy
limited steps taken by the solver
ARKodeGetNumStepAttempts returns the total number of steps
attempted by the solver
ARKodeGetNumRhsEvals returns the number of calls to the user's
fe and fi functions
ARKodeGetNumLinSolvSetups returns the number of calls made to
the linear solver's setup routine
ARKodeGetNumMassSolves returns the number of calls made to
the mass matrix solve routine
ARKodeGetNumMassMultiplies returns the number of calls made to
the mass matrix times vector routine
ARKodeGetNumErrTestFails returns the number of local error test
failures that have occured
ARKodeGetActualInitStep returns the actual initial step size
used by ARKODE
ARKodeGetLastStep returns the step size for the last internal
step
ARKodeGetCurrentStep returns the step size to be attempted on
the next internal step
ARKodeGetCurrentTime returns the current internal time reached
by the solver
ARKodeGetCurrentButcherTables returns the explicit and implicit
Butcher tables currently in use
ARKodeGetTolScaleFactor returns a suggested factor by which the
user's tolerances should be scaled when
too much accuracy has been requested for
some internal step
ARKodeGetErrWeights returns the current error weight vector.
The user must allocate space for eweight.
ARKodeGetEstLocalErrors returns the vector of estimated local
errors. The user must allocate space
for ele.
ARKodeGetNumGEvals returns the number of calls to the user's
g function (for rootfinding)
ARKodeGetRootInfo returns the indices for which g_i was found to
have a root. The user must allocate space for
rootsfound. For i = 0 ... nrtfn-1,
rootsfound[i] = 1 if g_i has a root, and = 0
if not.
ARKodeGet* return values:
ARK_SUCCESS if succesful
ARK_MEM_NULL if the arkode memory was NULL
---------------------------------------------------------------*/
SUNDIALS_EXPORT int ARKodeGetWorkSpace(void *arkode_mem,
long int *lenrw,
long int *leniw);
SUNDIALS_EXPORT int ARKodeGetNumSteps(void *arkode_mem,
long int *nsteps);
SUNDIALS_EXPORT int ARKodeGetNumExpSteps(void *arkode_mem,
long int *expsteps);
SUNDIALS_EXPORT int ARKodeGetNumAccSteps(void *arkode_mem,
long int *accsteps);
SUNDIALS_EXPORT int ARKodeGetNumStepAttempts(void *arkode_mem,
long int *step_attempts);
SUNDIALS_EXPORT int ARKodeGetNumRhsEvals(void *arkode_mem,
long int *nfe_evals,
long int *nfi_evals);
SUNDIALS_EXPORT int ARKodeGetNumLinSolvSetups(void *arkode_mem,
long int *nlinsetups);
SUNDIALS_EXPORT int ARKodeGetNumMassSolves(void *arkode_mem,
long int *nMassSolves);
SUNDIALS_EXPORT int ARKodeGetNumMassMultiplies(void *arkode_mem,
long int *nMassMult);
SUNDIALS_EXPORT int ARKodeGetNumErrTestFails(void *arkode_mem,
long int *netfails);
SUNDIALS_EXPORT int ARKodeGetActualInitStep(void *arkode_mem,
realtype *hinused);
SUNDIALS_EXPORT int ARKodeGetLastStep(void *arkode_mem,
realtype *hlast);
SUNDIALS_EXPORT int ARKodeGetCurrentStep(void *arkode_mem,
realtype *hcur);
SUNDIALS_EXPORT int ARKodeGetCurrentTime(void *arkode_mem,
realtype *tcur);
SUNDIALS_EXPORT int ARKodeGetCurrentButcherTables(void *arkode_mem,
int *s, int *q, int *p,
realtype *Ai, realtype *Ae,
realtype *ci, realtype *ce,
realtype *bi, realtype *be,
realtype *b2i, realtype *b2e);
SUNDIALS_EXPORT int ARKodeGetTolScaleFactor(void *arkode_mem,
realtype *tolsfac);
SUNDIALS_EXPORT int ARKodeGetErrWeights(void *arkode_mem,
N_Vector eweight);
SUNDIALS_EXPORT int ARKodeGetEstLocalErrors(void *arkode_mem,
N_Vector ele);
SUNDIALS_EXPORT int ARKodeGetNumGEvals(void *arkode_mem,
long int *ngevals);
SUNDIALS_EXPORT int ARKodeGetRootInfo(void *arkode_mem,
int *rootsfound);
/*---------------------------------------------------------------
As a convenience, the following functions provides the
optional outputs in one group.
---------------------------------------------------------------*/
SUNDIALS_EXPORT int ARKodeGetIntegratorStats(void *arkode_mem,
long int *nsteps,
long int *expsteps,
long int *accsteps,
long int *step_attempts,
long int *nfe_evals,
long int *nfi_evals,
long int *nlinsetups,
long int *netfails,
realtype *hinused,
realtype *hlast,
realtype *hcur,
realtype *tcur);
/*---------------------------------------------------------------
Nonlinear solver optional output extraction functions
-----------------------------------------------------------------
The following functions can be called to get optional outputs
and statistics related to the nonlinear solver.
-----------------------------------------------------------------
ARKodeGetNumNonlinSolvIters returns the number of nonlinear
solver iterations performed.
ARKodeGetNumNonlinSolvConvFails returns the number of nonlinear
convergence failures.
---------------------------------------------------------------*/
SUNDIALS_EXPORT int ARKodeGetNumNonlinSolvIters(void *arkode_mem,
long int *nniters);
SUNDIALS_EXPORT int ARKodeGetNumNonlinSolvConvFails(void *arkode_mem,
long int *nncfails);
/*---------------------------------------------------------------
As a convenience, the following function provides the
nonlinear solver optional outputs in a group.
---------------------------------------------------------------*/
SUNDIALS_EXPORT int ARKodeGetNonlinSolvStats(void *arkode_mem,
long int *nniters,
long int *nncfails);
/*---------------------------------------------------------------
As a convenience, the following function may be used to retrieve
Butcher tables from arkode_butcher.c. The array A must be
declared of type A[ARK_S_MAX][ARK_S_MAX], and the arrays c, b
and b2 should all have length at least ARK_S_MAX.
---------------------------------------------------------------*/
SUNDIALS_EXPORT int ARKodeLoadButcherTable(int imethod, int *s,
int *q, int *p,
realtype *A, realtype *b,
realtype *c, realtype *b2);
/*---------------------------------------------------------------
The following function returns the name of the constant
associated with a ARKODE return flag
---------------------------------------------------------------*/
SUNDIALS_EXPORT char *ARKodeGetReturnFlagName(long int flag);
/*---------------------------------------------------------------
Function : ARKodeWriteParameters
-----------------------------------------------------------------
ARKodeWriteParameters outputs all solver parameters to the
provided file pointer.
---------------------------------------------------------------*/
SUNDIALS_EXPORT int ARKodeWriteParameters(void *arkode_mem, FILE *fp);
/*---------------------------------------------------------------
Function : ARKodeWriteButcher
-----------------------------------------------------------------
ARKodeWriteButcher outputs the Butcher tables to the
provided file pointer.
---------------------------------------------------------------*/
SUNDIALS_EXPORT int ARKodeWriteButcher(void *arkode_mem, FILE *fp);
/*---------------------------------------------------------------
Function : ARKodeFree
-----------------------------------------------------------------
ARKodeFree frees the problem memory arkode_mem allocated by
ARKodeCreate and ARKodeInit. Its only argument is the pointer
arkode_mem returned by ARKodeCreate.
---------------------------------------------------------------*/
SUNDIALS_EXPORT void ARKodeFree(void **arkode_mem);
#ifdef __cplusplus
}
#endif
#endif
|