TraverseSegments.h

Go to the documentation of this file.

#ifndef TRAVERSE_SEGMENTS_H_
#define TRAVERSE_SEGMENTS_H_

#ifdef SWIG
#include "Python.h"
#endif
#include "Track.h"
#include "Track3D.h"
#include "Geometry.h"
#include "TrackGenerator3D.h"


class TraverseSegments {

private:

  /* Functions defining how to loop over Tracks */
  void loopOverTracks2D(MOCKernel* kernel);
  void loopOverTracksExplicit(MOCKernel* kernel);
  void loopOverTracksByTrackOTF(MOCKernel* kernel);
  void loopOverTracksByStackOTF(MOCKernel* kernel);

  /* Functions defining how to traverse segments */
  void traceSegmentsExplicit(Track* track, MOCKernel* kernel);
  void traceSegmentsOTF(Track* flattened_track, Point* start,
                        double theta, MOCKernel* kernel);
  void traceStackOTF(Track* flattened_track, int polar_index,
                     MOCKernel* kernel);

  void traceStackTwoWay(Track* flattened_track, int polar_index,
                        TransportKernel* kernel);


  int findMeshIndex(double* values, int size, double val, int sign);

protected:

  TrackGenerator* _track_generator;
  TrackGenerator3D* _track_generator_3D;

  double* _global_z_mesh;

  int _mesh_size;

  segmentationType _segment_formation;

  TraverseSegments(TrackGenerator* track_generator);
  virtual ~TraverseSegments();

  /* Functions defining how to loop over and operate on Tracks */
  void loopOverTracks(MOCKernel* kernel);
  virtual void onTrack(Track* track, segment* segments) = 0;

  //FIXME Rework function calls to make this private
  void loopOverTracksByStackTwoWay(TransportKernel* kernel);

  /* Returns a kernel of the requested type */
  template <class KernelType>
  MOCKernel* getKernel() {

    /* Check for segmentation kernel in explicit methods */
    if ((typeid(KernelType) != typeid(SegmentationKernel)) ||
        ((_segment_formation != EXPLICIT_2D) &&
        (_segment_formation != EXPLICIT_3D))) {

      /* Allocate kernel */
      MOCKernel* kernel = new KernelType(_track_generator);
      return kernel;
    }
    else
      return NULL;
  }

public:
  virtual void execute() = 0;

};

#endif