FrameObserver.cpp

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
*/


#include <fstream>
#include <string>
#include "FrameObserver.h"

#include <statistiken.h>

using namespace std;


MatchObs::MatchObs(const char *defFileName) {

    fstream objs(defFileName, ios::in);
    
    while (!objs.fail() && !objs.eof()) {
        string type;
        string fname;
        double threshold;
        
        objs >> type >> fname >> threshold;
        
        if (objs.fail() || objs.eof()) break;

        // Todo: if text: more info needed!
#if 0
        cout << type.c_str() << " "
            << fname.c_str() << " "
            << threshold << endl;
#endif
        
        matchers.push_back(new MatchImage(type.c_str(), 
            fname.c_str(), threshold));
    }

}


ColorObs::ColorObs(const char *fileName) 
  : win(new FrameObsWindow("color")),
    ImageCopy(2) {
    IplImage *pImage = cvLoadImage(fileName, CV_LOAD_IMAGE_UNCHANGED);
    if(pImage!=nullptr && pImage->depth==IPL_DEPTH_8U) {
#if 0
        for(size_t u=0;u<4;++u) {
            cout << pImage->channelSeq[u];
        }
        cout << endl;

        cout << "size= " << pImage->width << "x" << pImage->height 
            << " (" << pImage->width*pImage->height << ")"
            << endl;
#endif

        std::vector<Var<float>> m(pImage->nChannels);

        const unsigned char* p = reinterpret_cast<unsigned char*>(pImage->imageData);

        for(int y=0; y<pImage->height; ++y) {
            const unsigned char* row = p + y*pImage->widthStep;
            for(int x=0; x<pImage->width; ++x) {
                if(row[pImage->alphaChannel-1] < 100) {
                    row += pImage->nChannels;
                } else {
                    for(int c=0;c<pImage->nChannels;++c) {
                        m[c].add(*row++);
                    }
                }
            }
        }
    
#if 1
        for(const auto &x : m) {
            if(x.get_num_cases() > 1) {
                cout << "M=" << x.get_mean() 
                    << " SD=" << sqrt(x.get_var())
                    << " N=" << x.get_num_cases() << endl;
            }
        }
#endif
        cvReleaseImage(&pImage);

        means.resize(m.size());
        sd.resize(m.size());

        for(size_t u=0;u<m.size();++u) {
            means[u] = m[u].get_mean();
            sd[u] = sqrt(m[u].get_var());
        }
    }
}