TrackTraversingAlgorithms.h

Go to the documentation of this file.

#ifndef TRACK_TRAVERSING_ALGORITHMS_H_
#define TRACK_TRAVERSING_ALGORITHMS_H_

#include "TraverseSegments.h"
#include "ExpEvaluator.h"

#ifdef THREED
#define _NUM_COEFFS 6
#endif

class CPUSolver;
class CPULSSolver;


class MaxOpticalLength: public TraverseSegments {
private:
  FP_PRECISION _max_tau;

public:

  MaxOpticalLength(TrackGenerator* track_generator);
  void execute();
  void onTrack(Track* track, segment* segments);
};


class SegmentCounter: public TraverseSegments {
private:
  int _max_num_segments;
  long _total_num_segments;
  bool _count_total_segments;
  bool _total_segments_counted;

public:

  SegmentCounter(TrackGenerator* track_generator);
  void execute();
  void onTrack(Track* track, segment* segments);
  void countTotalNumSegments();
  long getTotalNumSegments();
};


class SegmentSplitter: public TraverseSegments {

public:

  SegmentSplitter(TrackGenerator* track_generator);
  void execute();
  void onTrack(Track* track, segment* segments);
};


class VolumeCalculator: public TraverseSegments {

public:

  VolumeCalculator(TrackGenerator* track_generator);
  void execute();
  void onTrack(Track* track, segment* segments);
};


class CentroidGenerator: public TraverseSegments {

private:

  Point** _centroids;
  FP_PRECISION* _FSR_volumes;
  omp_lock_t* _FSR_locks;
  Quadrature* _quadrature;
  Point** _starting_points;

public:

  CentroidGenerator(TrackGenerator* track_generator);
  virtual ~CentroidGenerator();
  void setCentroids(Point** centroids);
  void execute();
  void onTrack(Track* track, segment* segments);
};

  //TODO Improve
class LinearExpansionGenerator: public TraverseSegments {

private:

  double* _lin_exp_coeffs;
  FP_PRECISION* _FSR_volumes;
  omp_lock_t* _FSR_locks;
  double* _src_constants;
  Quadrature* _quadrature;
  CPULSSolver* _solver;
  int _num_groups;
#ifndef THREED
  int _NUM_COEFFS;
#endif
  int _num_flat;
  Point** _starting_points;
  double** _thread_source_constants;
  ExpEvaluator* _exp_evaluator;
  Progress* _progress;

public:

  LinearExpansionGenerator(CPULSSolver* solver);
  virtual ~LinearExpansionGenerator();
  void execute();
  void onTrack(Track* track, segment* segments);
};


class TransportSweep: public TraverseSegments {

private:

  CPUSolver* _cpu_solver;
  CPULSSolver* _ls_solver;
  Geometry* _geometry;
  int _num_groups;
  FP_PRECISION** _thread_fsr_fluxes;
  FP_PRECISION** _thread_scratch_pads;

public:

  TransportSweep(CPUSolver* cpu_solver);
  virtual ~TransportSweep();
  void execute();
  void onTrack(Track* track, segment* segments);

};


class DumpSegments: public TraverseSegments {

private:

  FILE* _out;

public:

  DumpSegments(TrackGenerator* track_generator);
  void setOutputFile(FILE* out);
  void execute();
  void onTrack(Track* track, segment* segments);
};


class ReadSegments: public TraverseSegments {

private:

  FILE* _in;

public:

  ReadSegments(TrackGenerator* track_generator);
  void setInputFile(FILE* input);
  void execute();
  void onTrack(Track* track, segment* segments);
};


class TransportSweepOTF: public TraverseSegments {

private:

  CPUSolver* _cpu_solver;

public:

  TransportSweepOTF(TrackGenerator* track_generator);
  void setCPUSolver(CPUSolver* cpu_solver);
  void onTrack(Track* track, segment* segments);
  void execute();
};


class RecenterSegments: public TraverseSegments {

private:

  Geometry* _geometry;

public:

  RecenterSegments(TrackGenerator* track_generator);
  void onTrack(Track* track, segment* segments);
  void execute();
};


class PrintSegments: public TraverseSegments {

private:

  FILE* _out;

public:

  PrintSegments(TrackGenerator* track_generator);
  void setOutputFile(FILE* out);
  void execute();
  void onTrack(Track* track, segment* segments);
};

#endif