forked from sammy-tri/drake
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[primitives] Copy shared_pointer_system from Anzu
This matches commit 142fb0b0 (Tue May 24 07:53:55 2022 -0700).
- Loading branch information
1 parent
e0bce06
commit 4b7c5fb
Showing
3 changed files
with
207 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,26 @@ | ||
#include "common/unique_ptr_system.h" | ||
|
||
namespace anzu { | ||
|
||
using drake::systems::LeafSystem; | ||
using drake::systems::SystemTypeTag; | ||
|
||
template <typename T> | ||
SharedPointerSystem<T>::SharedPointerSystem( | ||
std::shared_ptr<void> held, std::type_index held_type) | ||
: LeafSystem<T>(SystemTypeTag<SharedPointerSystem>{}), | ||
held_(std::move(held)), | ||
held_type_(held_type) {} | ||
|
||
template <typename T> | ||
template <typename U> | ||
SharedPointerSystem<T>::SharedPointerSystem(const SharedPointerSystem<U>& other) | ||
: SharedPointerSystem<T>(other.held_, other.held_type_) {} | ||
|
||
template <typename T> | ||
SharedPointerSystem<T>::~SharedPointerSystem() = default; | ||
|
||
} // namespace anzu | ||
|
||
DRAKE_DEFINE_CLASS_TEMPLATE_INSTANTIATIONS_ON_DEFAULT_SCALARS( | ||
class ::anzu::SharedPointerSystem) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,110 @@ | ||
#pragma once | ||
|
||
#include <memory> | ||
#include <typeindex> | ||
#include <utility> | ||
|
||
#include "drake/common/default_scalars.h" | ||
#include "drake/common/drake_throw.h" | ||
#include "drake/systems/framework/diagram_builder.h" | ||
#include "drake/systems/framework/leaf_system.h" | ||
|
||
// TODO(jwnimmer-tri) This was created as workaround for drake#14011, but is | ||
// useful even beyond that specific circumstance. We plan to promote this to | ||
// Drake (and in doing so, fix it to use the correct filename). | ||
|
||
namespace anzu { | ||
/** %SharedPointerSystem holds a single `shared_ptr` that will be released at | ||
System deletion time (i.e., the end of a Diagram lifespan). It has no input, | ||
output, state, nor parameters. This is useful for storing objects that will be | ||
pointed-to by other systems outside of the usual input/output port connections. | ||
Scalar conversion is supported and will simply increment the reference count | ||
for the contained object. Advanced users who need to disable conversion may | ||
access the get_system_scalar_converter() to do so. Note that the contained | ||
object is not scalar-converted, so should not depend on T. | ||
@tparam_default_scalar */ | ||
template <typename T> | ||
class SharedPointerSystem final : public drake::systems::LeafSystem<T> { | ||
public: | ||
DRAKE_NO_COPY_NO_MOVE_NO_ASSIGN(SharedPointerSystem) | ||
|
||
/** Creates a system holding the given value. | ||
The value is allowed to be `nullptr`. | ||
@tparam Held type used to store the given value. | ||
Calls to get<>() must provide the same type for retrieval. */ | ||
template <typename Held> | ||
explicit SharedPointerSystem(std::shared_ptr<Held> value_to_hold) | ||
: SharedPointerSystem(std::move(value_to_hold), | ||
std::type_index(typeid(Held))) {} | ||
|
||
/** Creates a system holding the given value. | ||
This overload accepts a unique_ptr (but still stores it at a shared_ptr). | ||
The value is allowed to be `nullptr`. | ||
@tparam Held type used to store the given value. | ||
Calls to get<>() must provide the same type for retrieval. */ | ||
template <typename Held> | ||
explicit SharedPointerSystem(std::unique_ptr<Held> value_to_hold) | ||
: SharedPointerSystem(std::shared_ptr<Held>(std::move(value_to_hold))) {} | ||
|
||
/** Scalar-converting copy constructor. See @ref system_scalar_conversion. */ | ||
template <typename U> | ||
explicit SharedPointerSystem(const SharedPointerSystem<U>&); | ||
|
||
~SharedPointerSystem() final; | ||
|
||
/** Creates a system holding the given value and adds it to the builder. | ||
The value is allowed to be `nullptr`. Returns an alias to the value. | ||
@tparam Held type used to store the given value. | ||
Calls to get<>() must provide the same type for retrieval. | ||
@pre builder is non-null */ | ||
template <typename Held> | ||
static Held* AddToBuilder( | ||
drake::systems::DiagramBuilder<T>* builder, | ||
std::shared_ptr<Held> value_to_hold) { | ||
DRAKE_THROW_UNLESS(builder != nullptr); | ||
auto holder = std::make_unique<SharedPointerSystem<T>>( | ||
std::move(value_to_hold)); | ||
auto* result = holder->template get<Held>(); | ||
builder->AddSystem(std::move(holder)); | ||
return result; | ||
} | ||
|
||
/** Creates a system holding the given value and adds it to the builder. | ||
This overload accepts a unique_ptr (but still stores it at a shared_ptr). | ||
The value is allowed to be `nullptr`. Returns an alias to the value. | ||
@tparam Held type used to store the given value. | ||
Calls to get<>() must provide the same type for retrieval. | ||
@pre builder is non-null */ | ||
template <typename Held> | ||
static Held* AddToBuilder( | ||
drake::systems::DiagramBuilder<T>* builder, | ||
std::unique_ptr<Held> value_to_hold) { | ||
return SharedPointerSystem::AddToBuilder( | ||
builder, std::shared_ptr<Held>(std::move(value_to_hold))); | ||
} | ||
|
||
/** (Advanced) Retrieves an alias to the stored value. | ||
@tparam Held type used to store the given value, per our constructor. | ||
@throws std::bad_cast if Held doesn't match the type used at construction. */ | ||
template <typename Held> | ||
Held* get() const { | ||
if (std::type_index(typeid(Held)) != held_type_) { | ||
throw std::bad_cast(); | ||
} | ||
return static_cast<Held*>(held_.get()); | ||
} | ||
|
||
private: | ||
template <typename> friend class SharedPointerSystem; | ||
|
||
SharedPointerSystem(std::shared_ptr<void> held, std::type_index held_type); | ||
|
||
const std::shared_ptr<void> held_; | ||
const std::type_index held_type_; | ||
}; | ||
} // namespace anzu | ||
|
||
DRAKE_DECLARE_CLASS_TEMPLATE_INSTANTIATIONS_ON_DEFAULT_SCALARS( | ||
class ::anzu::SharedPointerSystem) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,71 @@ | ||
#include "common/unique_ptr_system.h" | ||
|
||
#include <string> | ||
#include <string_view> | ||
|
||
#include <gtest/gtest.h> | ||
|
||
namespace anzu { | ||
namespace common { | ||
namespace { | ||
|
||
using drake::systems::DiagramBuilder; | ||
using drake::systems::System; | ||
|
||
GTEST_TEST(SharedPointerSystemTest, CtorFromUnique) { | ||
auto held = std::make_unique<std::string>("held"); | ||
auto dut = std::make_unique<SharedPointerSystem<double>>(std::move(held)); | ||
std::string* gotten = dut->get<std::string>(); | ||
ASSERT_NE(gotten, nullptr); | ||
EXPECT_EQ(*gotten, "held"); | ||
} | ||
|
||
GTEST_TEST(SharedPointerSystemTest, CtorFromShared) { | ||
auto held = std::make_shared<std::string>("held"); | ||
auto dut = std::make_unique<SharedPointerSystem<double>>(std::move(held)); | ||
std::string* gotten = dut->get<std::string>(); | ||
ASSERT_NE(gotten, nullptr); | ||
EXPECT_EQ(*gotten, "held"); | ||
} | ||
|
||
GTEST_TEST(SharedPointerSystemTest, BuilderFromUnique) { | ||
auto held = std::make_unique<std::string>("held"); | ||
auto builder = std::make_unique<DiagramBuilder<double>>(); | ||
std::string* gotten = SharedPointerSystem<double>::AddToBuilder( | ||
builder.get(), std::move(held)); | ||
auto diagram = builder->Build(); | ||
ASSERT_NE(gotten, nullptr); | ||
EXPECT_EQ(*gotten, "held"); | ||
} | ||
|
||
GTEST_TEST(SharedPointerSystemTest, BuilderFromShared) { | ||
auto held = std::make_shared<std::string>("held"); | ||
auto builder = std::make_unique<DiagramBuilder<double>>(); | ||
std::string* gotten = SharedPointerSystem<double>::AddToBuilder( | ||
builder.get(), std::move(held)); | ||
auto diagram = builder->Build(); | ||
ASSERT_NE(gotten, nullptr); | ||
EXPECT_EQ(*gotten, "held"); | ||
} | ||
|
||
GTEST_TEST(SharedPointerSystemTest, ScalarConversion) { | ||
auto dut1 = std::make_unique<SharedPointerSystem<double>>( | ||
std::make_unique<std::string>("held")); | ||
auto dut2 = System<double>::ToSymbolic(*dut1); | ||
ASSERT_NE(dut2, nullptr); | ||
dut1.release(); | ||
std::string* gotten = dut2->get<std::string>(); | ||
ASSERT_NE(gotten, nullptr); | ||
EXPECT_EQ(*gotten, "held"); | ||
} | ||
|
||
// Ensure that type-checking guards operate correctly. | ||
GTEST_TEST(SharedPointerSystemTest, BadCast) { | ||
auto held = std::make_shared<std::string>("held"); | ||
auto dut = std::make_unique<SharedPointerSystem<double>>(std::move(held)); | ||
EXPECT_THROW(dut->get<std::string_view>(), std::bad_cast); | ||
} | ||
|
||
} // namespace | ||
} // namespace common | ||
} // namespace anzu |