measindex.h

/***************************************************************************
                          measindex.h  -  description
                             -------------------
    begin                : Mon Apr 2 2007
    copyright            : (C) 2001 by Thies Jochimsen
    email                : jochimse@cns.mpg.de
 ***************************************************************************/

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

#ifndef RECOMEASINDEX_H
#define RECOMEASINDEX_H


#include "controller.h"


template<recoDim dim, class Ignore, class Separate=RecoDim<0> >
class RecoMeasIndex {

 public:

  bool init(const RecoCoord& coord, const RecoController& controller) {
    Log<Reco> odinlog("RecoMeasIndex","init");
    RecoCoord coord4count(coord);
    prep_coord(coord4count);
    coord4count.set_mode(RecoIndex::separate, dim);

    const CoordCountMap* countmap=controller.create_count_map(coord4count);
    if(!countmap) return false;
    init_countmap=(*countmap);  // create deep copy because countmap may be deleted by controller after prep

    return create_measmap(init_countmap);
  }


  bool update(const CoordCountMap& extra) {
    Log<Reco> odinlog("RecoMeasIndex","update");

    if(!extra.size()) return true;

    CoordCountMap countmap_copy(init_countmap);

    ODINLOG(odinlog,normalDebug) << "countmap_copy/extra.size()=" << countmap_copy.size() << "/" << extra.size() << STD_endl;

    for(CoordCountMap::const_iterator it=extra.begin(); it!=extra.end(); ++it) countmap_copy[it->first]+=it->second;

    return create_measmap(countmap_copy);
  }


  ivector get_indices(const RecoCoord& coord) const {
    Log<Reco> odinlog("RecoMeasIndex","get_indices");
    ivector result;
    RecoCoord coord4meas(coord);
    prep_coord(coord4meas);
    coord4meas.set_mode(RecoIndex::ignore,dim);

    MeasMap::const_iterator it=measmap.find(coord4meas);
    if(it==measmap.end()) {
      ODINLOG(odinlog,normalDebug) << "<" << recoDimLabel[dim] << ">: No indices for coordinate " << coord4meas.print() << STD_endl;
    } else {
      result=it->second;
      ODINLOG(odinlog,normalDebug) << "coord4meas/it=" << coord4meas.print() << "/" << it->first.print() << STD_endl;
      ODINLOG(odinlog,normalDebug) << "result=" << result.printbody() << STD_endl;
    }
    return result;
  }


 private:

  static void prep_coord(RecoCoord& coord) {
    Ignore::modify(RecoIndex::ignore, coord);
    Separate::modify(RecoIndex::separate, coord);
  }


  bool create_measmap(const CoordCountMap& countmap) {
    Log<Reco> odinlog("RecoMeasIndex","create_measmap");

    STD_map<RecoCoord, STD_list<int> > indexlist;
    for(CoordCountMap::const_iterator it=countmap.begin(); it!=countmap.end(); ++it) {
      RecoCoord coord4indexlist(it->first);
      coord4indexlist.set_mode(RecoIndex::ignore, dim); // collapse index, indices will be stored in the index vectors of measmap
      indexlist[coord4indexlist].push_back(it->first.index[dim]);
    }

    for(STD_map<RecoCoord, STD_list<int> >::iterator lstit=indexlist.begin(); lstit!=indexlist.end(); ++lstit) {
      STD_list<int>& lstref=lstit->second;
      lstref.sort();
      lstref.unique();
      measmap[lstit->first]=list2vector(lstref);
    }

    for(MeasMap::const_iterator mapit=measmap.begin(); mapit!=measmap.end(); ++mapit) {
      ODINLOG(odinlog,normalDebug) << "measmap[" << mapit->first.print() << "]=" << mapit->second.printbody() << STD_endl;
    }

    return true;
  }


  CoordCountMap init_countmap;

  typedef STD_map<RecoCoord, ivector> MeasMap;
  MeasMap measmap; // will be initialized during init

};


#endif

---