LogRecord.hh

Go to the documentation of this file.

/*
 * Copyright 2012 Nate Koenig
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
*/
/* Desc: A class to log data
 * Author: Nate Koenig
 * Date: 1 Jun 2010
 */

#ifndef _LOGRECORD_HH_
#define _LOGRECORD_HH_

#include <fstream>
#include <string>
#include <map>
#include <boost/thread.hpp>
#include <boost/archive/iterators/base64_from_binary.hpp>
#include <boost/archive/iterators/insert_linebreaks.hpp>
#include <boost/archive/iterators/transform_width.hpp>
#include <boost/archive/iterators/ostream_iterator.hpp>
#include <boost/filesystem.hpp>

#include "common/Event.hh"
#include "common/SingletonT.hh"

#define GZ_LOG_VERSION "1.0"

namespace gazebo
{
  namespace common
  {

    class LogRecord : public SingletonT<LogRecord>
    {
      private: LogRecord();

      private: virtual ~LogRecord();

      public: bool Init(const std::string &_subdir);

      public: void Add(const std::string &_name, const std::string &_filename,
                    boost::function<bool (std::ostringstream &)> _logCallback);

      public: bool Remove(const std::string &_name);

      public: void Stop();

      public: void Start(const std::string &_encoding="bz2");

      public: const std::string &GetEncoding() const;

      private: void Update();

      private: void Run();

      // private: void WriteHeader();

      private: class Log
      {
        public: Log(LogRecord *_parent, const std::string &_filename,
                    boost::function<bool (std::ostringstream &)> _logCB);

        public: virtual ~Log();

        public: void Write();

        public: void Update();

        public: void ClearBuffer();

        public: LogRecord *parent;
        public: boost::function<bool (std::ostringstream &)> logCB;
        public: std::string buffer;
        public: std::ofstream logFile;
        public: std::string filename;
      };

      private: typedef std::map<std::string, Log*> Log_M;

      private: Log_M logs;

      private: Log_M::iterator updateIter;

      private: Log_M::iterator logsEnd;

      private: event::ConnectionPtr updateConnection;

      private: bool stop;

      private: boost::thread *writeThread;

      private: boost::mutex writeMutex;

      private: boost::mutex controlMutex;

      private: boost::condition_variable dataAvailableCondition;

      private: boost::filesystem::path logPath;

      private: std::string encoding;

      private: bool initialized;

      private: friend class SingletonT<LogRecord>;
    };
  }
}
#endif