pt_recorder.h

Go to the documentation of this file.

/* =========================================================================== *
|
|  Copyright (c) 1994-2008 by Kobus Barnard (author).
|
|  Personal and educational use of this code is granted, provided that this
|  header is kept intact, and that the authorship is not misrepresented, that
|  its use is acknowledged in publications, and relevant papers are cited.
|
|  For other use contact the author (kobus AT cs DOT arizona DOT edu).
|
|  Please note that the code in this file has not necessarily been adequately
|  tested. Naturally, there is no guarantee of performance, support, or fitness
|  for any particular task. Nonetheless, I am interested in hearing about
|  problems that you encounter.
|
| Authors:
|     Ernesto Brau, Jinyan Guan
|
* =========================================================================== */

/* $Id$ */

#ifndef PT_RECORDER_H
#define PT_RECORDER_H

#include <people_tracking_cpp/pt_scene.h>
#include <people_tracking_cpp/pt_scene_posterior.h>
#include <string>
#include <sstream>

namespace kjb {
namespace pt {

template <class OutputIterator>
class Posterior_detail_recorder
{
public:
    typedef std::string record_type;

    Posterior_detail_recorder(OutputIterator it, const Scene_posterior& post) :
        it_(it), post_(post), increment_(true)
    {}

    Posterior_detail_recorder& replace()
    {
        increment_ = false;
        return *this;
    }

    template <class Step>
    void operator()(const Step& step, const Scene&, double)
    {
        std::stringstream ss;
        write_posterior_details(post_, *step.proposed_model(), 0.0, ss);
        *it_ = ss.str();
        if(increment_) it_++;
    }

private:
    OutputIterator it_;
    const Scene_posterior& post_;
    bool increment_;
};

template <class OutputIterator>
inline
Posterior_detail_recorder<OutputIterator> make_pd_recorder
(
    OutputIterator it,
    const Scene_posterior& post
)
{
    return Posterior_detail_recorder<OutputIterator>(it, post);
}

}} // namespace kjb::pt

#endif /* PT_RECORDER_H */