Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F78367169
py_model.cc
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Tue, Aug 20, 07:10
Size
4 KB
Mime Type
text/x-c
Expires
Thu, Aug 22, 07:10 (1 d, 21 h)
Engine
blob
Format
Raw Data
Handle
20029343
Attached To
rAKA akantu
py_model.cc
View Options
/* -------------------------------------------------------------------------- */
#include "py_aka_array.hh"
/* -------------------------------------------------------------------------- */
#include <model.hh>
#include <non_linear_solver.hh>
#include <sparse_matrix_aij.hh>
/* -------------------------------------------------------------------------- */
#include <pybind11/operators.h>
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
/* -------------------------------------------------------------------------- */
namespace
py
=
pybind11
;
/* -------------------------------------------------------------------------- */
namespace
akantu
{
/* -------------------------------------------------------------------------- */
void
register_model
(
py
::
module
&
mod
)
{
py
::
class_
<
SparseMatrix
>
(
mod
,
"SparseMatrix"
)
.
def
(
"getMatrixType"
,
&
SparseMatrix
::
getMatrixType
)
.
def
(
"size"
,
&
SparseMatrix
::
size
)
.
def
(
"clear"
,
&
SparseMatrix
::
clear
)
.
def
(
"saveProfile"
,
&
SparseMatrix
::
saveProfile
)
.
def
(
"saveMatrix"
,
&
SparseMatrix
::
saveMatrix
)
.
def
(
"add"
,
[](
SparseMatrix
&
self
,
UInt
i
,
UInt
j
)
{
self
.
add
(
i
,
j
);
},
"Add entry in the profile"
)
.
def
(
"add"
,
[](
SparseMatrix
&
self
,
UInt
i
,
UInt
j
,
Real
value
)
{
self
.
add
(
i
,
j
,
value
);
},
"Add the value to the matrix"
)
.
def
(
"__call__"
,
[](
const
SparseMatrix
&
self
,
UInt
i
,
UInt
j
)
{
return
self
(
i
,
j
);
});
py
::
class_
<
SparseMatrixAIJ
,
SparseMatrix
>
(
mod
,
"SparseMatrixAIJ"
)
.
def
(
"getIRN"
,
&
SparseMatrixAIJ
::
getIRN
)
.
def
(
"getJCN"
,
&
SparseMatrixAIJ
::
getJCN
)
.
def
(
"getA"
,
&
SparseMatrixAIJ
::
getA
);
py
::
class_
<
SolverVector
>
(
mod
,
"SolverVector"
);
py
::
class_
<
DOFManager
>
(
mod
,
"DOFManager"
)
.
def
(
"getMatrix"
,
&
DOFManager
::
getMatrix
,
py
::
return_value_policy
::
reference
)
.
def
(
"getNewMatrix"
,
[](
DOFManager
&
self
,
const
std
::
string
&
name
,
const
std
::
string
&
matrix_to_copy_id
)
->
decltype
(
auto
)
{
return
self
.
getNewMatrix
(
name
,
matrix_to_copy_id
);
},
py
::
return_value_policy
::
reference
)
.
def
(
"getResidual"
,
[](
DOFManager
&
self
)
->
decltype
(
auto
)
{
return
self
.
getResidual
();
},
py
::
return_value_policy
::
reference
)
.
def
(
"getArrayPerDOFs"
,
&
DOFManager
::
getArrayPerDOFs
)
.
def
(
"assembleToResidual"
,
&
DOFManager
::
assembleToResidual
);
py
::
class_
<
NonLinearSolver
>
(
mod
,
"NonLinearSolver"
)
.
def
(
"set"
,
[](
NonLinearSolver
&
self
,
const
std
::
string
&
id
,
const
Real
&
val
)
{
if
(
id
==
"max_iterations"
)
self
.
set
(
id
,
int
(
val
));
else
self
.
set
(
id
,
val
);
})
.
def
(
"set"
,
[](
NonLinearSolver
&
self
,
const
std
::
string
&
id
,
const
SolveConvergenceCriteria
&
val
)
{
self
.
set
(
id
,
val
);
});
py
::
class_
<
ModelSolver
,
Parsable
>
(
mod
,
"ModelSolver"
,
py
::
multiple_inheritance
())
.
def
(
"getNonLinearSolver"
,
(
NonLinearSolver
&
(
ModelSolver
::*
)(
const
ID
&
))
&
ModelSolver
::
getNonLinearSolver
,
py
::
arg
(
"solver_id"
)
=
""
,
py
::
return_value_policy
::
reference
)
.
def
(
"solveStep"
,
[](
ModelSolver
&
self
)
{
self
.
solveStep
();
})
.
def
(
"solveStep"
,
[](
ModelSolver
&
self
,
const
ID
&
solver_id
)
{
self
.
solveStep
(
solver_id
);
});
py
::
class_
<
Model
,
ModelSolver
>
(
mod
,
"Model"
,
py
::
multiple_inheritance
())
.
def
(
"setBaseName"
,
&
Model
::
setBaseName
)
.
def
(
"getFEEngine"
,
&
Model
::
getFEEngine
,
py
::
arg
(
"name"
)
=
""
,
py
::
return_value_policy
::
reference
)
.
def
(
"getFEEngineBoundary"
,
&
Model
::
getFEEngine
,
py
::
arg
(
"name"
)
=
""
,
py
::
return_value_policy
::
reference
)
.
def
(
"addDumpFieldVector"
,
&
Model
::
addDumpFieldVector
)
.
def
(
"addDumpField"
,
&
Model
::
addDumpField
)
.
def
(
"setBaseNameToDumper"
,
&
Model
::
setBaseNameToDumper
)
.
def
(
"addDumpFieldVectorToDumper"
,
&
Model
::
addDumpFieldVectorToDumper
)
.
def
(
"addDumpFieldToDumper"
,
&
Model
::
addDumpFieldToDumper
)
.
def
(
"dump"
,
&
Model
::
dump
)
.
def
(
"initNewSolver"
,
&
Model
::
initNewSolver
)
.
def
(
"getDOFManager"
,
&
Model
::
getDOFManager
,
py
::
return_value_policy
::
reference
);
}
}
// namespace akantu
Event Timeline
Log In to Comment