mirror of
https://github.com/pybind/pybind11.git
synced 2024-11-26 07:02:11 +00:00
25 lines
871 B
C++
25 lines
871 B
C++
|
/*
|
||
|
|
example/example16.cpp -- automatic upcasting for polymorphic types
|
||
|
|
|
||
|
|
Copyright (c) 2015 Wenzel Jakob <wenzel@inf.ethz.ch>
|
||
|
|
|
||
|
|
All rights reserved. Use of this source code is governed by a
|
||
|
|
BSD-style license that can be found in the LICENSE file.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "example.h"
|
||
|
|
|
||
|
|
struct BaseClass { virtual ~BaseClass() {} };
|
||
|
|
struct DerivedClass1 : BaseClass { };
|
||
|
|
struct DerivedClass2 : BaseClass { };
|
||
|
|
|
||
|
|
void init_ex16(py::module &m) {
|
||
|
|
py::class_<BaseClass>(m, "BaseClass").def(py::init<>());
|
||
|
|
py::class_<DerivedClass1>(m, "DerivedClass1").def(py::init<>());
|
||
|
|
py::class_<DerivedClass2>(m, "DerivedClass2").def(py::init<>());
|
||
|
|
|
||
|
|
m.def("return_class_1", []() -> BaseClass* { return new DerivedClass1(); });
|
||
|
|
m.def("return_class_2", []() -> BaseClass* { return new DerivedClass2(); });
|
||
|
|
m.def("return_none", []() -> BaseClass* { return nullptr; });
|
||
|
|
}
|