Statistics
| Branch: | Revision:

root / scout / scoutsim / src / sim_frame.h @ 0e0ef125

History | View | Annotate | Download (5.91 KB)

1 c492be62 Alex Zirbel
/**
2
 * The code in this package was developed using the structure of Willow
3
 * Garage's turtlesim package.  It was modified by the CMU Robotics Club
4
 * to be used as a simulator for the Colony Scout robot.
5 266ae7f2 Alex Zirbel
 *
6 c492be62 Alex Zirbel
 * All redistribution of this code is limited to the terms of Willow Garage's
7
 * licensing terms, as well as under permission from the CMU Robotics Club.
8
 * 
9
 * Copyright (c) 2011 Colony Project
10
 * 
11
 * Permission is hereby granted, free of charge, to any person
12
 * obtaining a copy of this software and associated documentation
13
 * files (the "Software"), to deal in the Software without
14
 * restriction, including without limitation the rights to use,
15
 * copy, modify, merge, publish, distribute, sublicense, and/or sell
16
 * copies of the Software, and to permit persons to whom the
17
 * Software is furnished to do so, subject to the following
18
 * conditions:
19
 * 
20
 * The above copyright notice and this permission notice shall be
21
 * included in all copies or substantial portions of the Software.
22
 * 
23
 * Copyright (c) 2009, Willow Garage, Inc.
24 266ae7f2 Alex Zirbel
 * All rights reserved.
25 c492be62 Alex Zirbel
 * 
26
 * Redistribution and use in source and binary forms, with or without
27
 * modification, are permitted provided that the following conditions are met:
28
 * 
29
 *    Redistributions of source code must retain the above copyright
30
 *       notice, this list of conditions and the following disclaimer.
31
 *    Redistributions in binary form must reproduce the above copyright
32
 *       notice, this list of conditions and the following disclaimer in the
33
 *       documentation and/or other materials provided with the distribution.
34
 *    Neither the name of the Willow Garage, Inc. nor the names of its
35
 *       contributors may be used to endorse or promote products derived from
36
 *       this software without specific prior written permission.
37
 * 
38
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
39
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
40
 * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
41
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
42
 * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
43
 * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
44
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
45
 * OTHER DEALINGS IN THE SOFTWARE.
46 266ae7f2 Alex Zirbel
 */
47
48
#include <wx/wx.h>
49
#include <wx/event.h>
50
#include <wx/timer.h>
51 2eaafff2 Alex
#include <wx/string.h>
52 0e77683c Alex
#include <wx/utils.h>
53 266ae7f2 Alex Zirbel
54 594c3bb9 Alex
#include <sys/stat.h>
55
56 266ae7f2 Alex Zirbel
#include <ros/ros.h>
57
58
#include <std_srvs/Empty.h>
59
#include <scoutsim/Spawn.h>
60
#include <scoutsim/Kill.h>
61 0e77683c Alex
#include <motors/set_motors.h>
62 266ae7f2 Alex Zirbel
#include <map>
63
64
#include "scout.h"
65 0e0ef125 Priya
#include "ghost_scout.h"
66 e3f69e61 Alex
#include "scoutsim_internal.h"
67 1d1281cc Priya
#include "messages/WirelessPacket.h"
68 266ae7f2 Alex Zirbel
69 a8480867 Alex Zirbel
#define SCOUTSIM_NUM_SCOUTS 1
70 2eaafff2 Alex
#define ID_ABOUT 1
71
#define ID_QUIT 2
72
#define ID_CLEAR 3
73
#define ID_MAP 4
74
#define ID_LINES 5
75 7ffad595 Alex
#define ID_WALLS 6
76 0e77683c Alex
#define ID_TELEOP_NONE 7
77
#define ID_TELEOP_PRECISE 8
78
#define ID_TELEOP_FLUID 9
79 68b23184 viki
#define ID_SONAR 10
80 0e77683c Alex
81 c63c9752 Alex
#define TELEOP_PRECISE_SPEED 1.0f
82
#define TELEOP_FLUID_MAX_SPEED 1.5f
83
#define TELEOP_FLUID_INC 0.1f
84 0e77683c Alex
85
// Teleop types
86
#define TELEOP_OFF 0
87
#define TELEOP_PRECISE 1
88
#define TELEOP_FLUID 2
89 266ae7f2 Alex Zirbel
90
namespace scoutsim
91
{
92
    class SimFrame : public wxFrame
93
    {
94
        public:
95 2eaafff2 Alex
            SimFrame(wxWindow* parent, std::string map_name);
96 266ae7f2 Alex Zirbel
            ~SimFrame();
97
98
            std::string spawnScout(const std::string& name,
99
                    float x, float y, float angle);
100
101 2eaafff2 Alex
            void onQuit(wxCommandEvent& event);
102
            void onAbout(wxCommandEvent& event);
103
            void onClear(wxCommandEvent& event);
104
            void showMap(wxCommandEvent& event);
105
            void showLines(wxCommandEvent& event);
106 7ffad595 Alex
            void showWalls(wxCommandEvent& event);
107 0e77683c Alex
            void stopTeleop(wxCommandEvent& event);
108
            void startTeleopPrecise(wxCommandEvent& event);
109
            void startTeleopFluid(wxCommandEvent& event);
110 68b23184 viki
            void showSonar(wxCommandEvent& event);
111 2eaafff2 Alex
112
            DECLARE_EVENT_TABLE()
113
114 266ae7f2 Alex Zirbel
        private:
115
            void onUpdate(wxTimerEvent& evt);
116
            void onPaint(wxPaintEvent& evt);
117
118 594c3bb9 Alex
            bool fileExists(const std::string& filename);
119
120 0e77683c Alex
            void teleop_move_precise();
121
            void teleop_move_fluid();
122
            void teleop();
123
124 266ae7f2 Alex Zirbel
            void updateScouts();
125
            void clear();
126
            bool hasScout(const std::string& name);
127
128
            bool clearCallback(std_srvs::Empty::Request&, std_srvs::Empty::Response&);
129
            bool resetCallback(std_srvs::Empty::Request&, std_srvs::Empty::Response&);
130 4612f7e4 Alex Zirbel
            bool spawnCallback(Spawn::Request&, Spawn::Response&);
131
            bool killCallback(Kill::Request&, Kill::Response&);
132 266ae7f2 Alex Zirbel
133 1d1281cc Priya
            void wirelessCallback(const messages::WirelessPacket::ConstPtr& msg);
134 82f3f746 Priya
135
            ros::Subscriber wireless_send;
136
            ros::Publisher wireless_receive;
137
138 0e77683c Alex
            // Teleop
139 c63c9752 Alex
            float teleop_l_speed;
140
            float teleop_r_speed;
141 0e77683c Alex
            ros::Publisher teleop_pub;
142
            int teleop_type;
143
144 c63c9752 Alex
            float teleop_fluid_speed;
145
            float teleop_fluid_omega;
146 0e77683c Alex
147 144137a1 Alex Zirbel
            ros::NodeHandle nh;
148
            wxTimer* update_timer;
149
            wxBitmap path_bitmap;
150
            wxImage path_image;
151 ade1b7f9 Alex
            wxImage lines_image;
152 7ffad595 Alex
            wxImage walls_image;
153 144137a1 Alex Zirbel
            wxMemoryDC path_dc;
154 0e77683c Alex
            wxMemoryDC sonar_dc;
155 266ae7f2 Alex Zirbel
156 144137a1 Alex Zirbel
            uint64_t frame_count;
157 266ae7f2 Alex Zirbel
158 144137a1 Alex Zirbel
            ros::WallTime last_scout_update;
159 266ae7f2 Alex Zirbel
160 144137a1 Alex Zirbel
            ros::ServiceServer clear_srv;
161
            ros::ServiceServer reset_srv;
162
            ros::ServiceServer spawn_srv;
163
            ros::ServiceServer kill_srv;
164 266ae7f2 Alex Zirbel
165
            typedef std::map<std::string, ScoutPtr> M_Scout;
166 144137a1 Alex Zirbel
            M_Scout scouts;
167 0e0ef125 Priya
            std::vector<GhostScout*> ghost_scouts;
168 144137a1 Alex Zirbel
            uint32_t id_counter;
169 266ae7f2 Alex Zirbel
170 594c3bb9 Alex
            std::string images_path;
171 266ae7f2 Alex Zirbel
172 144137a1 Alex Zirbel
            float width_in_meters;
173
            float height_in_meters;
174 2eaafff2 Alex
175 68b23184 viki
            bool sonar_on;
176
177 ade1b7f9 Alex
            std::string map_base_name;
178
            std::string map_lines_name;
179 7ffad595 Alex
            std::string map_walls_name;
180 2eaafff2 Alex
            std::string display_map_name;
181 266ae7f2 Alex Zirbel
    };
182
183
}