recorder.h
Go to the documentation of this file.
00001 /*********************************************************************
00002 * Software License Agreement (BSD License)
00003 *
00004 *  Copyright (c) 2008, Willow Garage, Inc.
00005 *  All rights reserved.
00006 *
00007 *  Redistribution and use in source and binary forms, with or without
00008 *  modification, are permitted provided that the following conditions
00009 *  are met:
00010 *
00011 *   * Redistributions of source code must retain the above copyright
00012 *     notice, this list of conditions and the following disclaimer.
00013 *   * Redistributions in binary form must reproduce the above
00014 *     copyright notice, this list of conditions and the following
00015 *     disclaimer in the documentation and/or other materials provided
00016 *     with the distribution.
00017 *   * Neither the name of Willow Garage, Inc. nor the names of its
00018 *     contributors may be used to endorse or promote products derived
00019 *     from this software without specific prior written permission.
00020 *
00021 *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
00022 *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
00023 *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
00024 *  FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
00025 *  COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
00026 *  INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
00027 *  BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
00028 *  LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
00029 *  CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
00030 *  LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
00031 *  ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
00032 *  POSSIBILITY OF SUCH DAMAGE.
00033 ********************************************************************/
00034 
00035 #ifndef ROSBAG_RECORDER_H
00036 #define ROSBAG_RECORDER_H
00037 
00038 #include <sys/stat.h>
00039 #if !defined(_MSC_VER)
00040   #include <termios.h>
00041   #include <unistd.h>
00042 #endif
00043 #include <time.h>
00044 
00045 #include <queue>
00046 #include <string>
00047 #include <vector>
00048 
00049 #include <boost/thread/condition.hpp>
00050 #include <boost/thread/mutex.hpp>
00051 #include <boost/regex.hpp>
00052 
00053 #include <ros/ros.h>
00054 #include <ros/time.h>
00055 
00056 #include <std_msgs/Empty.h>
00057 #include <topic_tools/shape_shifter.h>
00058 
00059 #include "rosbag/bag.h"
00060 #include "rosbag/stream.h"
00061 #include "rosbag/macros.h"
00062 
00063 namespace rosbag {
00064 
00065 class ROSBAG_DECL OutgoingMessage
00066 {
00067 public:
00068     OutgoingMessage(std::string const& _topic, topic_tools::ShapeShifter::ConstPtr _msg, boost::shared_ptr<ros::M_string> _connection_header, ros::Time _time);
00069 
00070     std::string                         topic;
00071     topic_tools::ShapeShifter::ConstPtr msg;
00072     boost::shared_ptr<ros::M_string>    connection_header;
00073     ros::Time                           time;
00074 };
00075 
00076 class ROSBAG_DECL OutgoingQueue
00077 {
00078 public:
00079     OutgoingQueue(std::string const& _filename, std::queue<OutgoingMessage>* _queue, ros::Time _time);
00080 
00081     std::string                  filename;
00082     std::queue<OutgoingMessage>* queue;
00083     ros::Time                    time;
00084 };
00085 
00086 struct ROSBAG_DECL RecorderOptions
00087 {
00088     RecorderOptions();
00089 
00090     bool            trigger;
00091     bool            record_all;
00092     bool            regex;
00093     bool            do_exclude;
00094     bool            quiet;
00095     bool            append_date;
00096     bool            snapshot;
00097     bool            verbose;
00098     CompressionType compression;
00099     std::string     prefix;
00100     std::string     name;
00101     boost::regex    exclude_regex;
00102     uint32_t        buffer_size;
00103     uint32_t        chunk_size;
00104     uint32_t        limit;
00105     bool            split;
00106     uint32_t        max_size;
00107     ros::Duration   max_duration;
00108     std::string     node;
00109     unsigned long long min_space;
00110     std::string min_space_str;
00111 
00112     std::vector<std::string> topics;
00113 };
00114 
00115 class ROSBAG_DECL Recorder
00116 {
00117 public:
00118     Recorder(RecorderOptions const& options);
00119 
00120     void doTrigger();
00121 
00122     bool isSubscribed(std::string const& topic) const;
00123 
00124     boost::shared_ptr<ros::Subscriber> subscribe(std::string const& topic);
00125 
00126     int run();
00127 
00128 private:
00129     void printUsage();
00130 
00131     void updateFilenames();
00132     void startWriting();
00133     void stopWriting();
00134 
00135     bool checkLogging();
00136     bool scheduledCheckDisk();
00137     bool checkDisk();
00138 
00139     void snapshotTrigger(std_msgs::Empty::ConstPtr trigger);
00140     //    void doQueue(topic_tools::ShapeShifter::ConstPtr msg, std::string const& topic, boost::shared_ptr<ros::Subscriber> subscriber, boost::shared_ptr<int> count);
00141     void doQueue(const ros::MessageEvent<topic_tools::ShapeShifter const>& msg_event, std::string const& topic, boost::shared_ptr<ros::Subscriber> subscriber, boost::shared_ptr<int> count);
00142     void doRecord();
00143     bool checkSize();
00144     bool checkDuration(const ros::Time&);
00145     void doRecordSnapshotter();
00146     void doCheckMaster(ros::TimerEvent const& e, ros::NodeHandle& node_handle);
00147 
00148     bool shouldSubscribeToTopic(std::string const& topic, bool from_node = false);
00149 
00150     template<class T>
00151     static std::string timeToStr(T ros_t);
00152 
00153 private:
00154     RecorderOptions               options_;
00155 
00156     Bag                           bag_;
00157 
00158     std::string                   target_filename_;
00159     std::string                   write_filename_;
00160 
00161     std::set<std::string>         currently_recording_;  
00162     int                           num_subscribers_;      
00163 
00164     int                           exit_code_;            
00165 
00166     boost::condition_variable_any queue_condition_;      
00167     boost::mutex                  queue_mutex_;          
00168     std::queue<OutgoingMessage>*  queue_;                
00169     uint64_t                      queue_size_;           
00170     uint64_t                      max_queue_size_;       
00171 
00172     uint64_t                      split_count_;          
00173 
00174     std::queue<OutgoingQueue>     queue_queue_;          
00175 
00176     ros::Time                     last_buffer_warn_;
00177 
00178     ros::Time                     start_time_;
00179 
00180     bool                          writing_enabled_;
00181     boost::mutex                  check_disk_mutex_;
00182     ros::WallTime                 check_disk_next_;
00183     ros::WallTime                 warn_next_;
00184 };
00185 
00186 } // namespace rosbag
00187 
00188 #endif


rosbag
Author(s): Tim Field, Jeremy Leibs, James Bowman
autogenerated on Tue Mar 7 2017 03:45:48