Skip to content

Commit

Permalink
Fix linting.
Browse files Browse the repository at this point in the history 
Signed-off-by: Benjamin Perseghetti <[email protected]>
  • Loading branch information
@bperseghetti
bperseghetti committed May 10, 2024
1 parent 9f715e2 commit 5fda4fa
Show file tree
Hide file tree
Showing 2 changed files with 77 additions and 42 deletions.
114 changes: 74 additions & 40 deletions src/plugins/camera_tracking/CameraTracking.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,7 @@ class CameraTrackingPrivate
msgs::Boolean &_res);

/// \brief Callback for a track message
/// \param[in] _msg Message consists of the target to track, type of tracking, offset and pgain.
/// \param[in] _msg Message is of type CamerTrack.
public: void OnTrackSub(const msgs::CameraTrack &_msg);

/// \brief Callback for a follow request
Expand Down Expand Up @@ -214,7 +214,7 @@ void CameraTrackingPrivate::Initialize()

// track
this->trackTopic = "/gui/track";
this->node.Subscribe(this->trackTopic,
this->node.Subscribe(this->trackTopic,
&CameraTrackingPrivate::OnTrackSub, this);
gzmsg << "Tracking topic on ["
<< this->trackTopic << "]" << std::endl;
Expand Down Expand Up @@ -280,11 +280,11 @@ void CameraTrackingPrivate::OnTrackSub(const msgs::CameraTrack &_msg)
}
if (!_msg.follow_target().empty())
{
this->selectedFollowTarget = _msg.follow_target();
this->selectedFollowTarget = _msg.follow_target();
}
if (!_msg.track_target().empty())
{
this->selectedTrackTarget = _msg.track_target();
this->selectedTrackTarget = _msg.track_target();
}
if (_msg.follow_target().empty() && _msg.track_target().empty()
&& _msg.track_mode() != gz::msgs::CameraTrack::USE_LAST)
Expand Down Expand Up @@ -428,7 +428,8 @@ void CameraTrackingPrivate::OnRender()
// reset track mode if target node got removed
if (!this->selectedFollowTarget.empty())
{
rendering::NodePtr targetFollow = this->scene->NodeByName(this->selectedFollowTarget);
rendering::NodePtr targetFollow = this->scene->NodeByName(
this->selectedFollowTarget);
if (!targetFollow && !this->selectedTargetWait)
{
this->camera->SetFollowTarget(nullptr);
Expand All @@ -437,7 +438,8 @@ void CameraTrackingPrivate::OnRender()
}
if (!this->selectedTrackTarget.empty())
{
rendering::NodePtr targetTrack = this->scene->NodeByName(this->selectedTrackTarget);
rendering::NodePtr targetTrack = this->scene->NodeByName(
this->selectedTrackTarget);
if (!targetTrack && !this->selectedTargetWait)
{
this->camera->SetTrackTarget(nullptr);
Expand All @@ -449,18 +451,22 @@ void CameraTrackingPrivate::OnRender()
return;
rendering::NodePtr selectedFollowTargetTmp = this->camera->FollowTarget();
rendering::NodePtr selectedTrackTargetTmp = this->camera->TrackTarget();
if (!this->selectedTrackTarget.empty() || !this->selectedFollowTarget.empty())
if (!this->selectedTrackTarget.empty() ||
!this->selectedFollowTarget.empty())
{
rendering::NodePtr targetFollow = this->scene->NodeByName(this->selectedFollowTarget);
rendering::NodePtr targetTrack = this->scene->NodeByName(this->selectedTrackTarget);
rendering::NodePtr targetFollow = this->scene->NodeByName(
this->selectedFollowTarget);
rendering::NodePtr targetTrack = this->scene->NodeByName(
this->selectedTrackTarget);
if (targetFollow || targetTrack)
{
if (this->trackMode == gz::msgs::CameraTrack::FOLLOW_FREE_LOOK ||
this->trackMode == gz::msgs::CameraTrack::FOLLOW ||
this->trackMode == gz::msgs::CameraTrack::FOLLOW_LOOK_AT )
{
if (!selectedFollowTargetTmp || targetFollow != selectedFollowTargetTmp
|| this->newTrack)
if (!selectedFollowTargetTmp ||
targetFollow != selectedFollowTargetTmp ||
this->newTrack)
{
this->trackWorldFrame = false;
this->camera->SetFollowTarget(targetFollow,
Expand Down Expand Up @@ -489,8 +495,9 @@ void CameraTrackingPrivate::OnRender()
}
if (this->trackMode == gz::msgs::CameraTrack::TRACK)
{
if (!selectedTrackTargetTmp || targetTrack != selectedTrackTargetTmp
|| this->newTrack)
if (!selectedTrackTargetTmp ||
targetTrack != selectedTrackTargetTmp ||
this->newTrack)
{
this->trackWorldFrame = true;
this->camera->SetFollowTarget(nullptr);
Expand Down Expand Up @@ -547,50 +554,75 @@ CameraTracking::CameraTracking()
if (this->dataPtr->trackMode == gz::msgs::CameraTrack::TRACK)
{
this->dataPtr->trackMsg.set_track_mode(gz::msgs::CameraTrack::TRACK);
this->dataPtr->trackMsg.set_track_target(this->dataPtr->selectedTrackTarget);
this->dataPtr->trackMsg.mutable_track_offset()->set_x(this->dataPtr->trackOffset.X());
this->dataPtr->trackMsg.mutable_track_offset()->set_y(this->dataPtr->trackOffset.Y());
this->dataPtr->trackMsg.mutable_track_offset()->set_z(this->dataPtr->trackOffset.Z());
this->dataPtr->trackMsg.set_track_pgain(this->dataPtr->trackPGain);
this->dataPtr->trackMsg.set_track_target(
this->dataPtr->selectedTrackTarget);
this->dataPtr->trackMsg.mutable_track_offset()->set_x(
this->dataPtr->trackOffset.X());
this->dataPtr->trackMsg.mutable_track_offset()->set_y(
this->dataPtr->trackOffset.Y());
this->dataPtr->trackMsg.mutable_track_offset()->set_z(
this->dataPtr->trackOffset.Z());
this->dataPtr->trackMsg.set_track_pgain(
this->dataPtr->trackPGain);
this->dataPtr->trackMsg.clear_follow_target();
this->dataPtr->trackMsg.clear_follow_offset();
this->dataPtr->trackMsg.clear_follow_pgain();
}
else if (this->dataPtr->trackMode == gz::msgs::CameraTrack::FOLLOW)
{
this->dataPtr->trackMsg.set_track_mode(gz::msgs::CameraTrack::FOLLOW);
this->dataPtr->trackMsg.set_follow_target(this->dataPtr->selectedFollowTarget);
this->dataPtr->trackMsg.mutable_follow_offset()->set_x(this->dataPtr->followOffset.X());
this->dataPtr->trackMsg.mutable_follow_offset()->set_y(this->dataPtr->followOffset.Y());
this->dataPtr->trackMsg.mutable_follow_offset()->set_z(this->dataPtr->followOffset.Z());
this->dataPtr->trackMsg.set_follow_target(
this->dataPtr->selectedFollowTarget);
this->dataPtr->trackMsg.mutable_follow_offset()->set_x(
this->dataPtr->followOffset.X());
this->dataPtr->trackMsg.mutable_follow_offset()->set_y(
this->dataPtr->followOffset.Y());
this->dataPtr->trackMsg.mutable_follow_offset()->set_z(
this->dataPtr->followOffset.Z());
this->dataPtr->trackMsg.set_follow_pgain(this->dataPtr->followPGain);
this->dataPtr->trackMsg.clear_track_target();
this->dataPtr->trackMsg.clear_track_offset();
this->dataPtr->trackMsg.clear_track_pgain();
}
else if (this->dataPtr->trackMode == gz::msgs::CameraTrack::FOLLOW_FREE_LOOK)
else if (this->dataPtr->trackMode ==
gz::msgs::CameraTrack::FOLLOW_FREE_LOOK)
{
this->dataPtr->trackMsg.set_track_mode(gz::msgs::CameraTrack::FOLLOW_FREE_LOOK);
this->dataPtr->trackMsg.set_follow_target(this->dataPtr->selectedFollowTarget);
this->dataPtr->trackMsg.mutable_follow_offset()->set_x(this->dataPtr->followOffset.X());
this->dataPtr->trackMsg.mutable_follow_offset()->set_y(this->dataPtr->followOffset.Y());
this->dataPtr->trackMsg.mutable_follow_offset()->set_z(this->dataPtr->followOffset.Z());
this->dataPtr->trackMsg.set_track_mode(
gz::msgs::CameraTrack::FOLLOW_FREE_LOOK);
this->dataPtr->trackMsg.set_follow_target(
this->dataPtr->selectedFollowTarget);
this->dataPtr->trackMsg.mutable_follow_offset()->set_x(
this->dataPtr->followOffset.X());
this->dataPtr->trackMsg.mutable_follow_offset()->set_y(
this->dataPtr->followOffset.Y());
this->dataPtr->trackMsg.mutable_follow_offset()->set_z(
this->dataPtr->followOffset.Z());
this->dataPtr->trackMsg.set_follow_pgain(this->dataPtr->followPGain);
this->dataPtr->trackMsg.clear_track_target();
this->dataPtr->trackMsg.clear_track_offset();
this->dataPtr->trackMsg.clear_track_pgain();
}
else if (this->dataPtr->trackMode == gz::msgs::CameraTrack::FOLLOW_LOOK_AT)
else if (this->dataPtr->trackMode ==
gz::msgs::CameraTrack::FOLLOW_LOOK_AT)
{
this->dataPtr->trackMsg.set_track_mode(gz::msgs::CameraTrack::FOLLOW_LOOK_AT);
this->dataPtr->trackMsg.set_follow_target(this->dataPtr->selectedFollowTarget);
this->dataPtr->trackMsg.set_track_target(this->dataPtr->selectedTrackTarget);
this->dataPtr->trackMsg.mutable_follow_offset()->set_x(this->dataPtr->followOffset.X());
this->dataPtr->trackMsg.mutable_follow_offset()->set_y(this->dataPtr->followOffset.Y());
this->dataPtr->trackMsg.mutable_follow_offset()->set_z(this->dataPtr->followOffset.Z());
this->dataPtr->trackMsg.mutable_track_offset()->set_x(this->dataPtr->trackOffset.X());
this->dataPtr->trackMsg.mutable_track_offset()->set_y(this->dataPtr->trackOffset.Y());
this->dataPtr->trackMsg.mutable_track_offset()->set_z(this->dataPtr->trackOffset.Z());
this->dataPtr->trackMsg.set_track_mode(
gz::msgs::CameraTrack::FOLLOW_LOOK_AT);
this->dataPtr->trackMsg.set_follow_target(
this->dataPtr->selectedFollowTarget);
this->dataPtr->trackMsg.set_track_target(
this->dataPtr->selectedTrackTarget);
this->dataPtr->trackMsg.mutable_follow_offset()->set_x(
this->dataPtr->followOffset.X());
this->dataPtr->trackMsg.mutable_follow_offset()->set_y(
this->dataPtr->followOffset.Y());
this->dataPtr->trackMsg.mutable_follow_offset()->set_z(
this->dataPtr->followOffset.Z());
this->dataPtr->trackMsg.mutable_track_offset()->set_x(
this->dataPtr->trackOffset.X());
this->dataPtr->trackMsg.mutable_track_offset()->set_y(
this->dataPtr->trackOffset.Y());
this->dataPtr->trackMsg.mutable_track_offset()->set_z(
this->dataPtr->trackOffset.Z());
this->dataPtr->trackMsg.set_follow_pgain(this->dataPtr->followPGain);
this->dataPtr->trackMsg.set_track_pgain(this->dataPtr->trackPGain);
}
Expand Down Expand Up @@ -641,7 +673,8 @@ void CameraTracking::LoadConfig(const tinyxml2::XMLElement *_pluginElem)
}
if (auto followPGainElem = _pluginElem->FirstChildElement("follow_pgain"))
{
this->dataPtr->followPGain = std::stod(std::string(followPGainElem->GetText()));
this->dataPtr->followPGain = std::stod(
std::string(followPGainElem->GetText()));
gzmsg << "CameraTracking: Loaded follow pgain from sdf ["
<< this->dataPtr->followPGain << "]" << std::endl;
this->dataPtr->newTrack = true;
Expand All @@ -657,7 +690,8 @@ void CameraTrackingPrivate::HandleKeyRelease(events::KeyReleaseOnScene *_e)
if (_e->Key().Key() == Qt::Key_Escape)
{
this->trackMode = gz::msgs::CameraTrack::NONE;
if (!this->selectedFollowTarget.empty() || !this->selectedTrackTarget.empty() )
if (!this->selectedFollowTarget.empty() ||
!this->selectedTrackTarget.empty())
{
this->selectedFollowTarget = std::string();
this->selectedTrackTarget = std::string();
Expand Down
5 changes: 3 additions & 2 deletions src/plugins/camera_tracking_config/CameraTrackingConfig.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,8 @@ void CameraTrackingConfig::LoadConfig(const tinyxml2::XMLElement *)
// Track target pose service
this->dataPtr->cameraTrackingTopic = "/gui/track";
this->dataPtr->trackingPub =
this->dataPtr->node.Advertise<msgs::CameraTrack>(this->dataPtr->cameraTrackingTopic);
this->dataPtr->node.Advertise<msgs::CameraTrack>(
this->dataPtr->cameraTrackingTopic);
gzmsg << "CameraTrackingConfig: Tracking topic publisher advertised on ["
<< this->dataPtr->cameraTrackingTopic << "]" << std::endl;

Expand All @@ -112,7 +113,7 @@ bool CameraTrackingConfig::eventFilter(QObject *_obj, QEvent *_event)
/////////////////////////////////////////////////
void CameraTrackingConfig::SetTracking(
double _tx, double _ty, double _tz, double _tp,
double _fx,double _fy, double _fz, double _fp)
double _fx, double _fy, double _fz, double _fp)
{
if (!this->dataPtr->newTrackingUpdate)
{
Expand Down

0 comments on commit 5fda4fa

Please sign in to comment.