ResultHolder.h

Go to the documentation of this file.

// -*- mode: c++; indent-tabs-mode: nil; c-basic-offset: 4; coding: iso-8859-1; -*-

/*
ACT-CV - Machine Vision for Cognitive Modeling
Copyright (c) 2008 Marc Halbruegge

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/

#ifndef RESULTHOLDER_H_
#define RESULTHOLDER_H_

#include <vector>
#include <map>
#include <garbagecoll.h> 

template<typename _t>
class ResultSource {
public:
    virtual void EnterCriticalSection()=0;
    virtual std::vector<_t>& GetData() = 0;
    virtual void LeaveCriticalSection()=0;
};


template<typename _t>
class ResultData : public ReferenceCount {
public:
    typename std::vector<_t> data;
    typename std::vector<_t>::const_iterator iter;

    ResultData() {};
    ResultData(const std::vector<_t> &d)
        : data(d) {
        iter = data.begin();
    }
};

// small helper, to be deleted
static std::ostream& operator << (std::ostream& o, const Line& l) {
    o << l.end1x << " "
      << l.end1y << " "
      << l.end2x << " "
      << l.end2y;
    return o;
}

// small helper, to be deleted
static std::ostream& operator << (std::ostream& o, const ObjectPosition& pos) {
    o << pos.x << " "
      << pos.y << " "
      << pos.width << " "
      << pos.height << " "
      << pos.name;
    return o;
}

template<typename _t>
class ResultHolder : public ReferenceCount {
protected:    
    ResultSource<_t> *source;
    std::map<int, GCPointer<ResultData<_t> > > results; 
public:
    ResultHolder(ResultSource<_t> *s) : source(s) {}
    
    int FindFirst() {
        int newHandle = static_cast<int>(results.size()+1);
    
        source->EnterCriticalSection();
        results[newHandle] 
            = new ResultData<_t>(source->GetData());
        source->LeaveCriticalSection();
        MHDBGMSG(__FILE__,__LINE__,__PRETTY_FUNCTION__);
        return newHandle;
    }        

    bool FindNext(int handle, _t *result) {
        MHDBGMSG(__FILE__,__LINE__,__PRETTY_FUNCTION__);
        if (results[handle]->iter == results[handle]->data.end()) {
            return false;
        } else {
            *result = *(results[handle]->iter);
            //std::cout << *result << std::endl;
            results[handle]->iter++;
            return true;
        }
    }
    
    void FindFinish(int handle) {
        results.erase(handle);
        MHDBGMSG(__FILE__,__LINE__,__PRETTY_FUNCTION__);
    }

};

#endif /*RESULTHOLDER_H_*/