Skip to content
GitLab
Explore
Sign in
Register
Primary navigation
Search or go to…
Project
E
eigenpy
Manage
Activity
Members
Labels
Plan
Issues
Issue boards
Milestones
Wiki
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Snippets
Build
Pipelines
Jobs
Pipeline schedules
Artifacts
Deploy
Releases
Container Registry
Model registry
Operate
Environments
Monitor
Incidents
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
Stack Of Tasks
eigenpy
Commits
4f8980b6
Verified
Commit
4f8980b6
authored
5 years ago
by
Justin Carpentier
Browse files
Options
Downloads
Patches
Plain Diff
core: externalize defaut implementation for EigenFromPy
parent
5e7ae5b1
No related branches found
No related tags found
No related merge requests found
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
include/eigenpy/eigen-from-python.hpp
+115
-105
115 additions, 105 deletions
include/eigenpy/eigen-from-python.hpp
with
115 additions
and
105 deletions
include/eigenpy/eigen-from-python.hpp
+
115
−
105
View file @
4f8980b6
...
@@ -89,31 +89,56 @@ namespace eigenpy
...
@@ -89,31 +89,56 @@ namespace eigenpy
typedef
typename
MatType
::
Scalar
Scalar
;
typedef
typename
MatType
::
Scalar
Scalar
;
/// \brief Determine if pyObj can be converted into a MatType object
/// \brief Determine if pyObj can be converted into a MatType object
static
void
*
convertible
(
PyArrayObject
*
pyArray
)
static
void
*
convertible
(
PyArrayObject
*
pyArray
);
/// \brief Allocate memory and copy pyObj in the new storage
static
void
construct
(
PyObject
*
pyObj
,
bp
::
converter
::
rvalue_from_python_stage1_data
*
memory
);
static
void
registration
();
};
template
<
typename
MatType
>
void
*
EigenFromPy
<
MatType
>::
convertible
(
PyArrayObject
*
pyArray
)
{
if
(
!
PyArray_Check
(
pyArray
))
return
0
;
if
(
!
np_type_is_convertible_into_scalar
<
Scalar
>
(
EIGENPY_GET_PY_ARRAY_TYPE
(
pyArray
)))
return
0
;
if
(
MatType
::
IsVectorAtCompileTime
)
{
{
if
(
!
PyArray_Check
(
pyArray
))
const
Eigen
::
DenseIndex
size_at_compile_time
return
0
;
=
MatType
::
IsRowMajor
?
MatType
::
ColsAtCompileTime
:
MatType
::
RowsAtCompileTime
;
if
(
!
np_type_is_convertible_into_scalar
<
Scalar
>
(
EIGENPY_GET_PY_ARRAY_TYPE
(
pyArray
)))
switch
(
PyArray_NDIM
(
pyArray
))
return
0
;
if
(
MatType
::
IsVectorAtCompileTime
)
{
{
const
Eigen
::
DenseIndex
size_at_compile_time
case
0
:
=
MatType
::
IsRowMajor
return
0
;
?
MatType
::
ColsAtCompileTime
case
1
:
:
MatType
::
RowsAtCompileTime
;
switch
(
PyArray_NDIM
(
pyArray
))
{
{
case
0
:
if
(
size_at_compile_time
!=
Eigen
::
Dynamic
)
return
0
;
{
case
1
:
// check that the sizes at compile time matche
if
(
PyArray_DIMS
(
pyArray
)[
0
]
==
size_at_compile_time
)
return
pyArray
;
else
return
0
;
}
else
// This is a dynamic MatType
return
pyArray
;
}
case
2
:
{
// Special care of scalar matrix of dimension 1x1.
if
(
PyArray_DIMS
(
pyArray
)[
0
]
==
1
&&
PyArray_DIMS
(
pyArray
)[
1
]
==
1
)
{
{
if
(
size_at_compile_time
!=
Eigen
::
Dynamic
)
if
(
size_at_compile_time
!=
Eigen
::
Dynamic
)
{
{
// check that the sizes at compile time matche
if
(
size_at_compile_time
==
1
)
if
(
PyArray_DIMS
(
pyArray
)[
0
]
==
size_at_compile_time
)
return
pyArray
;
return
pyArray
;
else
else
return
0
;
return
0
;
...
@@ -121,108 +146,93 @@ namespace eigenpy
...
@@ -121,108 +146,93 @@ namespace eigenpy
else
// This is a dynamic MatType
else
// This is a dynamic MatType
return
pyArray
;
return
pyArray
;
}
}
case
2
:
if
(
PyArray_DIMS
(
pyArray
)[
0
]
>
1
&&
PyArray_DIMS
(
pyArray
)[
1
]
>
1
)
{
return
0
;
}
if
(((
PyArray_DIMS
(
pyArray
)[
0
]
==
1
)
&&
(
MatType
::
ColsAtCompileTime
==
1
))
||
((
PyArray_DIMS
(
pyArray
)[
1
]
==
1
)
&&
(
MatType
::
RowsAtCompileTime
==
1
)))
{
{
// Special care of scalar matrix of dimension 1x1.
return
0
;
if
(
PyArray_DIMS
(
pyArray
)[
0
]
==
1
&&
PyArray_DIMS
(
pyArray
)[
1
]
==
1
)
}
{
if
(
size_at_compile_time
!=
Eigen
::
Dynamic
)
if
(
size_at_compile_time
!=
Eigen
::
Dynamic
)
{
{
// This is a fixe size vector
if
(
size_at_compile_time
==
1
)
const
Eigen
::
DenseIndex
pyArray_size
return
pyArray
;
=
PyArray_DIMS
(
pyArray
)[
0
]
>
PyArray_DIMS
(
pyArray
)[
1
]
else
?
PyArray_DIMS
(
pyArray
)[
0
]
return
0
;
:
PyArray_DIMS
(
pyArray
)[
1
];
}
if
(
size_at_compile_time
!=
pyArray_size
)
else
// This is a dynamic MatType
return
pyArray
;
}
if
(
PyArray_DIMS
(
pyArray
)[
0
]
>
1
&&
PyArray_DIMS
(
pyArray
)[
1
]
>
1
)
{
return
0
;
}
if
(((
PyArray_DIMS
(
pyArray
)[
0
]
==
1
)
&&
(
MatType
::
ColsAtCompileTime
==
1
))
||
((
PyArray_DIMS
(
pyArray
)[
1
]
==
1
)
&&
(
MatType
::
RowsAtCompileTime
==
1
)))
{
return
0
;
return
0
;
}
if
(
size_at_compile_time
!=
Eigen
::
Dynamic
)
{
// This is a fixe size vector
const
Eigen
::
DenseIndex
pyArray_size
=
PyArray_DIMS
(
pyArray
)[
0
]
>
PyArray_DIMS
(
pyArray
)[
1
]
?
PyArray_DIMS
(
pyArray
)[
0
]
:
PyArray_DIMS
(
pyArray
)[
1
];
if
(
size_at_compile_time
!=
pyArray_size
)
return
0
;
}
break
;
}
}
default
:
break
;
return
0
;
}
}
default
:
return
0
;
}
}
else
// this is a matrix
}
else
// this is a matrix
{
if
(
PyArray_NDIM
(
pyArray
)
==
1
)
// We can always convert a vector into a matrix
{
{
if
(
PyArray_NDIM
(
pyArray
)
==
1
)
// We can always convert a vector into a matrix
return
pyArray
;
{
return
pyArray
;
}
if
(
PyArray_NDIM
(
pyArray
)
!=
2
)
{
return
0
;
}
if
(
PyArray_NDIM
(
pyArray
)
==
2
)
{
const
int
R
=
(
int
)
PyArray_DIMS
(
pyArray
)[
0
];
const
int
C
=
(
int
)
PyArray_DIMS
(
pyArray
)[
1
];
if
(
(
MatType
::
RowsAtCompileTime
!=
R
)
&&
(
MatType
::
RowsAtCompileTime
!=
Eigen
::
Dynamic
)
)
return
0
;
if
(
(
MatType
::
ColsAtCompileTime
!=
C
)
&&
(
MatType
::
ColsAtCompileTime
!=
Eigen
::
Dynamic
)
)
return
0
;
}
}
}
#ifdef NPY_1_8_API_VERSION
if
(
PyArray_NDIM
(
pyArray
)
!=
2
)
if
(
!
(
PyArray_FLAGS
(
pyArray
)))
#else
if
(
!
(
PyArray_FLAGS
(
pyArray
)
&
NPY_ALIGNED
))
#endif
{
{
return
0
;
return
0
;
}
}
return
pyArray
;
if
(
PyArray_NDIM
(
pyArray
)
==
2
)
}
{
const
int
R
=
(
int
)
PyArray_DIMS
(
pyArray
)[
0
];
/// \brief Allocate memory and copy pyObj in the new storage
const
int
C
=
(
int
)
PyArray_DIMS
(
pyArray
)[
1
];
static
void
construct
(
PyObject
*
pyObj
,
bp
::
converter
::
rvalue_from_python_stage1_data
*
memory
)
if
(
(
MatType
::
RowsAtCompileTime
!=
R
)
{
&&
(
MatType
::
RowsAtCompileTime
!=
Eigen
::
Dynamic
)
)
PyArrayObject
*
pyArray
=
reinterpret_cast
<
PyArrayObject
*>
(
pyObj
);
return
0
;
assert
((
PyArray_DIMS
(
pyArray
)[
0
]
<
INT_MAX
)
&&
(
PyArray_DIMS
(
pyArray
)[
1
]
<
INT_MAX
));
if
(
(
MatType
::
ColsAtCompileTime
!=
C
)
&&
(
MatType
::
ColsAtCompileTime
!=
Eigen
::
Dynamic
)
)
void
*
storage
=
reinterpret_cast
<
bp
::
converter
::
rvalue_from_python_storage
<
MatType
>*>
return
0
;
(
reinterpret_cast
<
void
*>
(
memory
))
->
storage
.
bytes
;
}
EigenAllocator
<
MatType
>::
allocate
(
pyArray
,
storage
);
memory
->
convertible
=
storage
;
}
}
static
void
registration
()
#ifdef NPY_1_8_API_VERSION
if
(
!
(
PyArray_FLAGS
(
pyArray
)))
#else
if
(
!
(
PyArray_FLAGS
(
pyArray
)
&
NPY_ALIGNED
))
#endif
{
{
bp
::
converter
::
registry
::
push_back
return
0
;
(
reinterpret_cast
<
void
*
(
*
)(
_object
*
)
>
(
&
EigenFromPy
::
convertible
),
&
EigenFromPy
::
construct
,
bp
::
type_id
<
MatType
>
());
}
}
};
return
pyArray
;
}
template
<
typename
MatType
>
void
EigenFromPy
<
MatType
>::
construct
(
PyObject
*
pyObj
,
bp
::
converter
::
rvalue_from_python_stage1_data
*
memory
)
{
PyArrayObject
*
pyArray
=
reinterpret_cast
<
PyArrayObject
*>
(
pyObj
);
assert
((
PyArray_DIMS
(
pyArray
)[
0
]
<
INT_MAX
)
&&
(
PyArray_DIMS
(
pyArray
)[
1
]
<
INT_MAX
));
void
*
storage
=
reinterpret_cast
<
bp
::
converter
::
rvalue_from_python_storage
<
MatType
>*>
(
reinterpret_cast
<
void
*>
(
memory
))
->
storage
.
bytes
;
EigenAllocator
<
MatType
>::
allocate
(
pyArray
,
storage
);
memory
->
convertible
=
storage
;
}
template
<
typename
MatType
>
void
EigenFromPy
<
MatType
>::
registration
()
{
bp
::
converter
::
registry
::
push_back
(
reinterpret_cast
<
void
*
(
*
)(
_object
*
)
>
(
&
EigenFromPy
::
convertible
),
&
EigenFromPy
::
construct
,
bp
::
type_id
<
MatType
>
());
}
template
<
typename
MatType
>
template
<
typename
MatType
>
struct
EigenFromPyConverter
struct
EigenFromPyConverter
...
...
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment