/src/parcsr_ls/F90_HYPRE_ams.c

https://github.com/hypre-space/hypre · C · 389 lines · 272 code · 46 blank · 71 comment · 0 complexity · 7d814e626303e3d4a51b07d3d52c77f6 MD5 · raw file 

    

  1. /******************************************************************************
    2. 

  2.  * Copyright 1998-2019 Lawrence Livermore National Security, LLC and other
    3. 

  3.  * HYPRE Project Developers. See the top-level COPYRIGHT file for details.
    4. 

  4.  *
    5. 

  5.  * SPDX-License-Identifier: (Apache-2.0 OR MIT)
    6. 

  6.  ******************************************************************************/
    7. 

  7. 

  8. /******************************************************************************
    9. 

  9.  *
    10. 

  10.  * HYPRE_AMS Fortran interface
    11. 

  11.  *
    12. 

  12.  *****************************************************************************/
    13. 

  13. 

  14. #include "_hypre_parcsr_ls.h"
    15. 

  15. #include "fortran.h"
    16. 

  16. 

  17. #ifdef __cplusplus
    18. 

  18. extern "C" {
    19. 

  19. #endif
    20. 

  20.     
    21. 

  21. /*--------------------------------------------------------------------------
    22. 

  22.  * HYPRE_AMSCreate
    23. 

  23.  *--------------------------------------------------------------------------*/
    24. 

  24. 

  25. void
    26. 

  26. hypre_F90_IFACE(hypre_amscreate, HYPRE_AMSCREATE)
    27. 

  27.    ( hypre_F90_Obj *solver,
    28. 

  28.      hypre_F90_Int *ierr)
    29. 

  29. {
    30. 

  30.    *ierr = (hypre_F90_Int)
    31. 

  31.       ( HYPRE_AMSCreate(
    32. 

  32.            hypre_F90_PassObjRef (HYPRE_Solver, solver) ) );
    33. 

  33. }
    34. 

  34. 

  35. /*--------------------------------------------------------------------------
    36. 

  36.  * HYPRE_AMSDestroy
    37. 

  37.  *--------------------------------------------------------------------------*/
    38. 

  38. 

  39. void
    40. 

  40. hypre_F90_IFACE(hypre_amsdestroy, HYPRE_AMSDESTROY)
    41. 

  41.    ( hypre_F90_Obj *solver,
    42. 

  42.      hypre_F90_Int *ierr)
    43. 

  43. {
    44. 

  44.    *ierr = (hypre_F90_Int)
    45. 

  45.       ( HYPRE_AMSDestroy(
    46. 

  46.            hypre_F90_PassObj (HYPRE_Solver, solver) ) );
    47. 

  47. }
    48. 

  48. 

  49. /*--------------------------------------------------------------------------
    50. 

  50.  * HYPRE_AMSSetup
    51. 

  51.  *--------------------------------------------------------------------------*/
    52. 

  52. 

  53. void
    54. 

  54. hypre_F90_IFACE(hypre_amssetup, HYPRE_AMSSETUP)
    55. 

  55.    ( hypre_F90_Obj *solver,
    56. 

  56.      hypre_F90_Obj *A,
    57. 

  57.      hypre_F90_Obj *b,
    58. 

  58.      hypre_F90_Obj *x,
    59. 

  59.      hypre_F90_Int *ierr)
    60. 

  60. {
    61. 

  61.    *ierr = (hypre_F90_Int)
    62. 

  62.       ( HYPRE_AMSSetup(
    63. 

  63.            hypre_F90_PassObj (HYPRE_Solver, solver),
    64. 

  64.            hypre_F90_PassObj (HYPRE_ParCSRMatrix, A),
    65. 

  65.            hypre_F90_PassObj (HYPRE_ParVector, b),
    66. 

  66.            hypre_F90_PassObj (HYPRE_ParVector, x) ) );
    67. 

  67. }
    68. 

  68. 

  69. /*--------------------------------------------------------------------------
    70. 

  70.  * HYPRE_AMSSolve
    71. 

  71.  *--------------------------------------------------------------------------*/
    72. 

  72. 

  73. void
    74. 

  74. hypre_F90_IFACE(hypre_amssolve, HYPRE_AMSSOLVE)
    75. 

  75.    ( hypre_F90_Obj *solver,
    76. 

  76.      hypre_F90_Obj *A,
    77. 

  77.      hypre_F90_Obj *b,
    78. 

  78.      hypre_F90_Obj *x,
    79. 

  79.      hypre_F90_Int *ierr)
    80. 

  80. {
    81. 

  81.    *ierr = (hypre_F90_Int)
    82. 

  82.       ( HYPRE_AMSSolve(
    83. 

  83.            hypre_F90_PassObj (HYPRE_Solver, solver),
    84. 

  84.            hypre_F90_PassObj (HYPRE_ParCSRMatrix, A),
    85. 

  85.            hypre_F90_PassObj (HYPRE_ParVector, b),
    86. 

  86.            hypre_F90_PassObj (HYPRE_ParVector, x) ) );
    87. 

  87. }
    88. 

  88. 

  89. /*--------------------------------------------------------------------------
    90. 

  90.  * HYPRE_AMSSetDimension
    91. 

  91.  *--------------------------------------------------------------------------*/
    92. 

  92. 

  93. void
    94. 

  94. hypre_F90_IFACE(hypre_amssetdimension, HYPRE_AMSSETDIMENSION)
    95. 

  95.    ( hypre_F90_Obj *solver,
    96. 

  96.      hypre_F90_Int *dim,
    97. 

  97.      hypre_F90_Int *ierr)
    98. 

  98. {
    99. 

  99.    *ierr = (hypre_F90_Int)
    100. 

  100.       ( HYPRE_AMSSetDimension(
    101. 

  101.            hypre_F90_PassObj (HYPRE_Solver, solver),
    102. 

  102.            hypre_F90_PassInt (dim) ) );
    103. 

  103. }
    104. 

  104. 

  105. /*--------------------------------------------------------------------------
    106. 

  106.  * HYPRE_AMSSetDiscreteGradient
    107. 

  107.  *--------------------------------------------------------------------------*/
    108. 

  108. 

  109. void
    110. 

  110. hypre_F90_IFACE(hypre_amssetdiscretegradient, HYPRE_AMSSETDISCRETEGRADIENT)
    111. 

  111.    ( hypre_F90_Obj *solver,
    112. 

  112.      hypre_F90_Obj *G,
    113. 

  113.      hypre_F90_Int *ierr)
    114. 

  114. {
    115. 

  115.    *ierr = (hypre_F90_Int)
    116. 

  116.       ( HYPRE_AMSSetDiscreteGradient(
    117. 

  117.            hypre_F90_PassObj (HYPRE_Solver, solver),
    118. 

  118.            hypre_F90_PassObj (HYPRE_ParCSRMatrix, G) ) );
    119. 

  119. }
    120. 

  120. 

  121. /*--------------------------------------------------------------------------
    122. 

  122.  * HYPRE_AMSSetCoordinateVectors
    123. 

  123.  *--------------------------------------------------------------------------*/
    124. 

  124. 

  125. void
    126. 

  126. hypre_F90_IFACE(hypre_amssetcoordinatevectors, HYPRE_AMSSETCOORDINATEVECTORS)
    127. 

  127.    ( hypre_F90_Obj *solver,
    128. 

  128.      hypre_F90_Obj *x,
    129. 

  129.      hypre_F90_Obj *y,
    130. 

  130.      hypre_F90_Obj *z,
    131. 

  131.      hypre_F90_Int *ierr)
    132. 

  132. {
    133. 

  133.    *ierr = (hypre_F90_Int)
    134. 

  134.       ( HYPRE_AMSSetCoordinateVectors(
    135. 

  135.            hypre_F90_PassObj (HYPRE_Solver, solver),
    136. 

  136.            hypre_F90_PassObj (HYPRE_ParVector, x),
    137. 

  137.            hypre_F90_PassObj (HYPRE_ParVector, y),
    138. 

  138.            hypre_F90_PassObj (HYPRE_ParVector, z) ) );
    139. 

  139. }
    140. 

  140. 

  141. /*--------------------------------------------------------------------------
    142. 

  142.  * HYPRE_AMSSetEdgeConstantVectors
    143. 

  143.  *--------------------------------------------------------------------------*/
    144. 

  144. 

  145. void
    146. 

  146. hypre_F90_IFACE(hypre_amssetedgeconstantvectors, HYPRE_AMSSETEDGECONSTANTVECTORS)
    147. 

  147.    ( hypre_F90_Obj *solver,
    148. 

  148.      hypre_F90_Obj *Gx,
    149. 

  149.      hypre_F90_Obj *Gy,
    150. 

  150.      hypre_F90_Obj *Gz,
    151. 

  151.      hypre_F90_Int *ierr)
    152. 

  152. {
    153. 

  153.    *ierr = (hypre_F90_Int)
    154. 

  154.       ( HYPRE_AMSSetEdgeConstantVectors(
    155. 

  155.            hypre_F90_PassObj (HYPRE_Solver, solver),
    156. 

  156.            hypre_F90_PassObj (HYPRE_ParVector, Gx),
    157. 

  157.            hypre_F90_PassObj (HYPRE_ParVector, Gy),
    158. 

  158.            hypre_F90_PassObj (HYPRE_ParVector, Gz) ) );
    159. 

  159. }
    160. 

  160. 

  161. /*--------------------------------------------------------------------------
    162. 

  162.  * HYPRE_AMSSetAlphaPoissonMatrix
    163. 

  163.  *--------------------------------------------------------------------------*/
    164. 

  164. 

  165. void
    166. 

  166. hypre_F90_IFACE(hypre_amssetalphapoissonmatrix, HYPRE_AMSSETALPHAPOISSONMATRIX)
    167. 

  167.    ( hypre_F90_Obj *solver,
    168. 

  168.      hypre_F90_Obj *A_alpha,
    169. 

  169.      hypre_F90_Int *ierr)
    170. 

  170. {
    171. 

  171.    *ierr = (hypre_F90_Int)
    172. 

  172.       ( HYPRE_AMSSetAlphaPoissonMatrix(
    173. 

  173.            hypre_F90_PassObj (HYPRE_Solver, solver),
    174. 

  174.            hypre_F90_PassObj (HYPRE_ParCSRMatrix, A_alpha) ) );
    175. 

  175. }
    176. 

  176. 

  177. /*--------------------------------------------------------------------------
    178. 

  178.  * HYPRE_AMSSetBetaPoissonMatrix
    179. 

  179.  *--------------------------------------------------------------------------*/
    180. 

  180. 

  181. void
    182. 

  182. hypre_F90_IFACE(hypre_amssetbetapoissonmatrix, HYPRE_AMSSETBETAPOISSONMATRIX)
    183. 

  183.    ( hypre_F90_Obj *solver,
    184. 

  184.      hypre_F90_Obj *A_beta,
    185. 

  185.      hypre_F90_Int *ierr)
    186. 

  186. {
    187. 

  187.    *ierr = (hypre_F90_Int)
    188. 

  188.       ( HYPRE_AMSSetBetaPoissonMatrix(
    189. 

  189.            hypre_F90_PassObj (HYPRE_Solver, solver),
    190. 

  190.            hypre_F90_PassObj (HYPRE_ParCSRMatrix, A_beta) ) );
    191. 

  191. }
    192. 

  192. 

  193. /*--------------------------------------------------------------------------
    194. 

  194.  * HYPRE_AMSSetMaxIter
    195. 

  195.  *--------------------------------------------------------------------------*/
    196. 

  196. 

  197. void
    198. 

  198. hypre_F90_IFACE(hypre_amssetmaxiter, HYPRE_AMSSETMAXITER)
    199. 

  199.    ( hypre_F90_Obj *solver,
    200. 

  200.      hypre_F90_Int *maxiter,
    201. 

  201.      hypre_F90_Int *ierr)
    202. 

  202. {
    203. 

  203.    *ierr = (hypre_F90_Int)
    204. 

  204.       ( HYPRE_AMSSetMaxIter(
    205. 

  205.            hypre_F90_PassObj (HYPRE_Solver, solver),
    206. 

  206.            hypre_F90_PassInt (maxiter) ) );
    207. 

  207. }
    208. 

  208. 

  209. /*--------------------------------------------------------------------------
    210. 

  210.  * HYPRE_AMSSetTol
    211. 

  211.  *--------------------------------------------------------------------------*/
    212. 

  212. 

  213. void
    214. 

  214. hypre_F90_IFACE(hypre_amssettol, HYPRE_AMSSETTOL)
    215. 

  215.    ( hypre_F90_Obj *solver,
    216. 

  216.      hypre_F90_Real *tol,
    217. 

  217.      hypre_F90_Int *ierr)
    218. 

  218. {
    219. 

  219.    *ierr = (hypre_F90_Int)
    220. 

  220.       ( HYPRE_AMSSetTol(
    221. 

  221.            hypre_F90_PassObj (HYPRE_Solver, solver),
    222. 

  222.            hypre_F90_PassReal (tol) ) );
    223. 

  223. }
    224. 

  224. 

  225. /*--------------------------------------------------------------------------
    226. 

  226.  * HYPRE_AMSSetCycleType
    227. 

  227.  *--------------------------------------------------------------------------*/
    228. 

  228. 

  229. void
    230. 

  230. hypre_F90_IFACE(hypre_amssetcycletype, HYPRE_AMSSETCYCLETYPE)
    231. 

  231.    ( hypre_F90_Obj *solver,
    232. 

  232.      hypre_F90_Int *cycle_type,
    233. 

  233.      hypre_F90_Int *ierr)
    234. 

  234. {
    235. 

  235.    *ierr = (hypre_F90_Int)
    236. 

  236.       ( HYPRE_AMSSetCycleType(
    237. 

  237.            hypre_F90_PassObj (HYPRE_Solver, solver),
    238. 

  238.            hypre_F90_PassInt (cycle_type) ) );
    239. 

  239. }
    240. 

  240. 

  241. /*--------------------------------------------------------------------------
    242. 

  242.  * HYPRE_AMSSetPrintLevel
    243. 

  243.  *--------------------------------------------------------------------------*/
    244. 

  244. 

  245. void
    246. 

  246. hypre_F90_IFACE(hypre_amssetprintlevel, HYPRE_AMSSETPRINTLEVEL)
    247. 

  247.    ( hypre_F90_Obj *solver,
    248. 

  248.      hypre_F90_Int *print_level,
    249. 

  249.      hypre_F90_Int *ierr)
    250. 

  250. {
    251. 

  251.    *ierr = (hypre_F90_Int)
    252. 

  252.       ( HYPRE_AMSSetPrintLevel(
    253. 

  253.            hypre_F90_PassObj (HYPRE_Solver, solver),
    254. 

  254.            hypre_F90_PassInt (print_level) ) );
    255. 

  255. }
    256. 

  256. 

  257. /*--------------------------------------------------------------------------
    258. 

  258.  * HYPRE_AMSSetSmoothingOptions
    259. 

  259.  *--------------------------------------------------------------------------*/
    260. 

  260. 

  261. void
    262. 

  262. hypre_F90_IFACE(hypre_amssetsmoothingoptions, HYPRE_AMSSETSMOOTHINGOPTIONS)
    263. 

  263.    ( hypre_F90_Obj *solver,
    264. 

  264.      hypre_F90_Int *relax_type,
    265. 

  265.      hypre_F90_Int *relax_times,
    266. 

  266.      hypre_F90_Real *relax_weight,
    267. 

  267.      hypre_F90_Real *omega,
    268. 

  268.      hypre_F90_Int *ierr)
    269. 

  269. {
    270. 

  270.    *ierr = (hypre_F90_Int)
    271. 

  271.       ( HYPRE_AMSSetSmoothingOptions(
    272. 

  272.            hypre_F90_PassObj (HYPRE_Solver, solver),
    273. 

  273.            hypre_F90_PassInt (relax_type),
    274. 

  274.            hypre_F90_PassInt (relax_times),
    275. 

  275.            hypre_F90_PassReal (relax_weight),
    276. 

  276.            hypre_F90_PassReal (omega) ) );
    277. 

  277. }
    278. 

  278. 

  279. /*--------------------------------------------------------------------------
    280. 

  280.  * HYPRE_AMSSetAlphaAMGOptions
    281. 

  281.  *--------------------------------------------------------------------------*/
    282. 

  282. 

  283. void
    284. 

  284. hypre_F90_IFACE(hypre_amssetalphaamgoptions, HYPRE_AMSSETALPHAAMGOPTIONS)
    285. 

  285.    ( hypre_F90_Obj *solver,
    286. 

  286.      hypre_F90_Int *alpha_coarsen_type,
    287. 

  287.      hypre_F90_Int *alpha_agg_levels,
    288. 

  288.      hypre_F90_Int *alpha_relax_type,
    289. 

  289.      hypre_F90_Real *alpha_strength_threshold,
    290. 

  290.      hypre_F90_Int *alpha_interp_type,
    291. 

  291.      hypre_F90_Int *alpha_Pmax,
    292. 

  292.      hypre_F90_Int *ierr)
    293. 

  293. 

  294. {
    295. 

  295.    *ierr = (hypre_F90_Int)
    296. 

  296.       ( HYPRE_AMSSetAlphaAMGOptions(
    297. 

  297.            hypre_F90_PassObj (HYPRE_Solver, solver),
    298. 

  298.            hypre_F90_PassInt (alpha_coarsen_type),
    299. 

  299.            hypre_F90_PassInt (alpha_agg_levels),
    300. 

  300.            hypre_F90_PassInt (alpha_relax_type),
    301. 

  301.            hypre_F90_PassReal (alpha_strength_threshold),
    302. 

  302.            hypre_F90_PassInt (alpha_interp_type),
    303. 

  303.            hypre_F90_PassInt (alpha_Pmax) ) );
    304. 

  304. }
    305. 

  305. 

  306. /*--------------------------------------------------------------------------
    307. 

  307.  * HYPRE_AMSSetBetaAMGOptions
    308. 

  308.  *--------------------------------------------------------------------------*/
    309. 

  309. 

  310. void
    311. 

  311. hypre_F90_IFACE(hypre_amssetbetaamgoptions, HYPRE_AMSSETBETAAMGOPTIONS)
    312. 

  312.    ( hypre_F90_Obj *solver,
    313. 

  313.      hypre_F90_Int *beta_coarsen_type,
    314. 

  314.      hypre_F90_Int *beta_agg_levels,
    315. 

  315.      hypre_F90_Int *beta_relax_type,
    316. 

  316.      hypre_F90_Real *beta_strength_threshold,
    317. 

  317.      hypre_F90_Int *beta_interp_type,
    318. 

  318.      hypre_F90_Int *beta_Pmax,
    319. 

  319.      hypre_F90_Int *ierr)
    320. 

  320. 

  321. {
    322. 

  322.    *ierr = (hypre_F90_Int)
    323. 

  323.       ( HYPRE_AMSSetBetaAMGOptions(
    324. 

  324.            hypre_F90_PassObj (HYPRE_Solver, solver),
    325. 

  325.            hypre_F90_PassInt (beta_coarsen_type),
    326. 

  326.            hypre_F90_PassInt (beta_agg_levels),
    327. 

  327.            hypre_F90_PassInt (beta_relax_type),
    328. 

  328.            hypre_F90_PassReal (beta_strength_threshold),
    329. 

  329.            hypre_F90_PassInt (beta_interp_type),
    330. 

  330.            hypre_F90_PassInt (beta_Pmax) ) );
    331. 

  331. }
    332. 

  332. 

  333. /*--------------------------------------------------------------------------
    334. 

  334.  * HYPRE_AMSGetNumIterations
    335. 

  335.  *--------------------------------------------------------------------------*/
    336. 

  336. 

  337. void
    338. 

  338. hypre_F90_IFACE(hypre_amsgetnumiterations, HYPRE_AMSGETNUMITERATIONS)
    339. 

  339.    ( hypre_F90_Obj *solver,
    340. 

  340.      hypre_F90_Int *num_iterations,
    341. 

  341.      hypre_F90_Int *ierr)
    342. 

  342. {
    343. 

  343.    *ierr = (hypre_F90_Int)
    344. 

  344.       ( HYPRE_AMSGetNumIterations(
    345. 

  345.            hypre_F90_PassObj (HYPRE_Solver, solver),
    346. 

  346.            hypre_F90_PassIntRef (num_iterations) ) );
    347. 

  347. }
    348. 

  348. 

  349. /*--------------------------------------------------------------------------
    350. 

  350.  * HYPRE_AMSGetFinalRelativeResidualNorm
    351. 

  351.  *--------------------------------------------------------------------------*/
    352. 

  352. 

  353. void
    354. 

  354. hypre_F90_IFACE(hypre_amsgetfinalrelativeresidualnorm, HYPRE_AMSGETFINALRELATIVERESIDUALNORM)
    355. 

  355.    ( hypre_F90_Obj *solver,
    356. 

  356.      hypre_F90_Real *rel_resid_norm,
    357. 

  357.      hypre_F90_Int *ierr)
    358. 

  358. {
    359. 

  359.    *ierr = (hypre_F90_Int)
    360. 

  360.       ( HYPRE_AMSGetFinalRelativeResidualNorm(
    361. 

  361.            hypre_F90_PassObj (HYPRE_Solver, solver),
    362. 

  362.            hypre_F90_PassRealRef (rel_resid_norm) ) );
    363. 

  363. }
    364. 

  364. 

  365. /*--------------------------------------------------------------------------
    366. 

  366.  * HYPRE_AMSConstructDiscreteGradient
    367. 

  367.  *--------------------------------------------------------------------------*/
    368. 

  368. 

  369. void
    370. 

  370. hypre_F90_IFACE(hypre_amsconstructdiscretegradient, HYPRE_AMSCONSTRUCTDISCRETEGRADIENT)
    371. 

  371.    ( hypre_F90_Obj *A,
    372. 

  372.      hypre_F90_Obj *x_coord,
    373. 

  373.      hypre_F90_BigIntArray *edge_vertex,
    374. 

  374.      hypre_F90_Int *edge_orientation,
    375. 

  375.      hypre_F90_Obj *G,
    376. 

  376.      hypre_F90_Int *ierr)
    377. 

  377. {
    378. 

  378.    *ierr = (hypre_F90_Int)
    379. 

  379.       ( HYPRE_AMSConstructDiscreteGradient(
    380. 

  380.            hypre_F90_PassObj (HYPRE_ParCSRMatrix, A),
    381. 

  381.            hypre_F90_PassObj (HYPRE_ParVector, x_coord),
    382. 

  382.            hypre_F90_PassBigIntArray (edge_vertex),
    383. 

  383.            hypre_F90_PassInt (edge_orientation),
    384. 

  384.            hypre_F90_PassObjRef (HYPRE_ParCSRMatrix, G) ) );
    385. 

  385. }
    386. 

  386.     
    387. 

  387. #ifdef __cplusplus
    388. 

  388. }
    389. 

  389. #endif
    390.