controller.h

gehe zur Dokumentation dieser Datei

#ifndef MICROCONTROLLER_H
#define MICROCONTROLLER_H

#include "seemicro.h"
#include <iostream>
#include <list>
#include <iomanip>
#include "timer.h"

using namespace std;

namespace seemicro
{

int DEBUGLEVEL = 0;

struct MicroController
{
  Micro *ergebnis;

  Micro *kern;

  ImageSource *src;

  videoDevice *v;

  GrayModule* g;
  CannyModule *ca, *canny2;
  CorrelationModule *co;
  SearchRadiusModule *sr;
  SearchAnglesAdam *se;
  FindLEDSModule *sleds;
  HSVRange *range;
#ifdef POLAR
  PolarModule *polar;
  PolarCorrelationModule *polarCorr;
#endif

  list<RecognitionModule*> frameRecognitionList;
  list<OutputModule*> outputList;

  bool keyframe;

  Timer timer;
  long timeForFrame, timeForKeyFrame, timeForFrameOut, timeForKeyFrameOut;

  void modulesProcessFrame(void)
  {
    Timer t;
#ifdef DEBUG
    if(DEBUGLEVEL>=3) cerr << " ProcessFrame: ";
#endif
    myrect *ROI, ROI2;

    if(se) ROI=&se->rect;
    else if(co && co->micro.radiusValid && co->micro.mitteValid)
    {
      mypoint p = mypoint(int(co->micro.radius), int(co->micro.radius));
      ROI2 = myrect(co->micro.mitte - p, co->micro.mitte+p);
      ROI2.crop(co->src.img->width, co->src.img->height);
      ROI = &ROI2;
    }
    else ROI=NULL;

    list<RecognitionModule*>::iterator i;
    for(i = frameRecognitionList.begin();
        i != frameRecognitionList.end(); i++)
    {
#ifdef DEBUG
      if(DEBUGLEVEL>=3) cerr << "+";
#endif
      t.start();
      (*i)->processFrame(ROI);
      t.stop();
      (*i)->timeForFrame = t.get();
    }
  }

  void modulesProcessKeyFrame(void)
  {
    Timer t;
#ifdef DEBUG
    if(DEBUGLEVEL>=3) cerr << " key: ";
#endif
    list<RecognitionModule*>::iterator i;
    for(i = frameRecognitionList.begin();
        i != frameRecognitionList.end(); i++)
    {
#ifdef DEBUG
    if(DEBUGLEVEL>=3) cerr << "+";
#endif
      t.start();
      (*i)->processKeyFrame();
      t.stop();
      (*i)->timeForKeyFrame = t.get();
    }
  }

  void callOutputModules(bool keyframe)
  {
#ifdef DEBUG
    if(DEBUGLEVEL>=3) cerr << " out: ";
#endif
    list<OutputModule*>::iterator i;
    for(i = outputList.begin();
        i != outputList.end(); i++)
    {
#ifdef DEBUG
    if(DEBUGLEVEL>=3) cerr << "#";
#endif
      (*i)->output(keyframe);
    }
#ifdef DEBUG
    if(DEBUGLEVEL>=3) cerr << ".\n";
#endif
  }

  void reEvalIfParamChanged(void)
  {
#ifdef DEBUG
    if(DEBUGLEVEL>=3) cerr << "Reeval: ";
#endif
    bool cascade = false;
    list<RecognitionModule*>::iterator i;
    for(i = frameRecognitionList.begin();
        i != frameRecognitionList.end(); i++)
    {
#ifdef DEBUG
      if(DEBUGLEVEL>=3) cerr << (cascade ? "+" : "-");
#endif
      if((!cascade) && ((*i)->paramChanged)) cascade = true;
      if(!cascade) continue;
      (*i)->processKeyFrame();
    }
#ifdef DEBUG
    if(DEBUGLEVEL>=3) cerr << ".";
#endif

    if(cascade) callOutputModules(true);
  }

  void handleFrame()
  {
#ifdef DEBUG
    if(DEBUGLEVEL>=3)
      cerr << "->MicroController::handleFrame\n";
#endif
    assert(ergebnis);
    keyframe = !v || // kein videomodus
      !ergebnis->mitteValid || // mittelpunkt noch nicht bekannt
      !(v->framesCaptured % v->keyFrameRate);

    timer.start();

    if((DEBUGLEVEL>=2) && (v!=0))
    {
      cout << "\rframe: " << setw(2) <<
        (v->framesCaptured % v->keyFrameRate);
      cout << (keyframe ? " keyframe" : "          ") << flush;
    }

    if(keyframe) modulesProcessKeyFrame();
    else modulesProcessFrame();

    timer.stop();
    if(keyframe) timeForKeyFrame = timer.get();
    else timeForFrame = timer.get();

    timer.start();
    callOutputModules(keyframe);
    timer.stop();
    if(keyframe) timeForKeyFrameOut = timer.get();
    else timeForFrameOut = timer.get();

#ifdef DEBUG
    if(DEBUGLEVEL>=3) cerr << "<-MicroController::handleFrame\n";
#endif
  }

  MicroController()
  {
    range = NULL;
    sleds = NULL;
    g = NULL;
    se = NULL;
    sr = NULL;
    co = NULL;
    ca = NULL;
    kern = NULL;
  }

  MicroController(ImageSource *quelle, Micro *Amicro) : ergebnis(Amicro), src(quelle)
  {
    range = NULL;
    sleds = NULL;
    g = NULL;
    se = NULL;
    sr = NULL;
    co = NULL;
    ca = NULL;
    kern = NULL;

    // clone ergebnis to create kern of exactly same type
    kern = ergebnis->clone();

    // klappt immer. liefert NULL, wenn cast nicht möglich
    v = dynamic_cast<videoDevice*>(src);

    if(DEBUGLEVEL>=2) cerr << "Creating Recognition Modules...";
    src->registerMe(frameRecognitionList);

    if(src->img->nChannels>1)
    {
      g = new GrayModule(*src);
      g->registerMe(frameRecognitionList);
      ca = new CannyModule(*g);
#ifdef POLAR
      polar = new PolarModule(*g, *ergebnis, 200, 200);
      polar->registerMe(frameRecognitionList);
      canny2 = new CannyModule(*polar);
      canny2->registerMe(frameRecognitionList);
      polarCorr = new PolarCorrelationModule(*ergebnis, *canny2);
      polarCorr->registerMe(frameRecognitionList);
#endif
    }
    else
    {
      ca = new CannyModule(*src);
    }
    ca->registerMe(frameRecognitionList);


    co = new CorrelationModule(*ergebnis, *ca, kern);
    co->registerMe(frameRecognitionList);

    sr = new SearchRadiusModule(*co);
    sr->registerMe(frameRecognitionList);

    range = new HSVRange();
    range->registerMe(frameRecognitionList);
    // Dieses Modul verhält sich verschieden,
    // je nachdem ob es auf Adam oder Eva schaut
    sleds = new FindLEDSModule(*src, *ergebnis, *co, *range);
    sleds->registerMe(frameRecognitionList);

    // Only if we try to recognize a micro.Adam
    MicroAdam *adam = dynamic_cast<MicroAdam*>(ergebnis);
    if(adam)
    {
      se = new SearchAnglesAdam(*adam, *co, *ca);
      se->registerMe(frameRecognitionList);
    }

  }

  ~MicroController()
  {
    if(DEBUGLEVEL>=2) cerr << "sleds ";
    if(sleds)
    {
      delete sleds; sleds = NULL;
    }
    if(DEBUGLEVEL>=2) cerr << "se ";
    if(se)
    {
      delete se; se = NULL;
    }
    if(DEBUGLEVEL>=2) cerr << "sr ";
    if(sr)
    {
      delete sr; sr = NULL;
    }
    if(DEBUGLEVEL>=2) cerr << "co ";
    if(co)
    {
      delete co; co = NULL;
    }
    if(DEBUGLEVEL>=2) cerr << "ca ";
    if(ca)
    {
      delete ca;ca = NULL;
    }
    if(DEBUGLEVEL>=2) cerr << "g ";
    if(g)
    {
      delete g; g = NULL;
    }
    if(DEBUGLEVEL>=2) cerr << "kern ";
    if(kern)
    {
      delete kern; kern = NULL;
    }

  }

}; // MicroController

} // seemicro

#endif

---