Track.h

Go to the documentation of this file.

#ifndef TRACK_H_
#define TRACK_H_

#ifdef __cplusplus
#ifdef SWIG
#include "Python.h"
#endif
#include "Point.h"
#include "Material.h"
#include "boundary_type.h"
#include <vector>
#include <algorithm>
#endif

struct segment {

  double _length;

  Material* _material;

  int _region_id;

  int _track_idx;

  int _cmfd_surface_fwd;

  int _cmfd_surface_bwd;

  FP_PRECISION _starting_position[3];

  segment() {
    _track_idx = 0;
    _material = NULL;
    _cmfd_surface_fwd = -1;
    _cmfd_surface_bwd = -1;
    _starting_position[2] = 0.;
  }
};


class Track {

protected:

  int _uid;

  Point _start;

  Point _end;

  double _phi;

  std::vector<segment> _segments;

  int _num_segments;

  boundaryType _bc_fwd;

  boundaryType _bc_bwd;

  /* Indices that are used to locate the track in the various track arrays */
  int _azim_index;
  int _xy_index;
  int _link_index;

  long _track_next_fwd;
  long _track_next_bwd;
  long _track_prdc_fwd;
  long _track_prdc_bwd;
  long _track_refl_fwd;
  long _track_refl_bwd;

  bool _next_fwd_fwd;
  bool _next_bwd_fwd;

  //FIXME
  bool _direction_in_cycle;

  int _surface_in;
  int _surface_out;
  
  int _domain_fwd;
  int _domain_bwd;

public:
  Track();
  virtual ~Track();

  /* Setter methods */
  virtual void setValues(const double start_x, const double start_y,
                         const double end_x, const double end_y,
                         const double phi);
  virtual void setCoords(double x0, double y0,
                         double x1, double y1);
  void setUid(int uid);
  void setPhi(const double phi);
  void setBCFwd(const boundaryType bc_fwd);
  void setBCBwd(const boundaryType bc_bwd);
  void setTrackNextFwd(long track_id);
  void setTrackNextBwd(long track_id);
  void setTrackPrdcFwd(long track_id);
  void setTrackPrdcBwd(long track_id);
  void setTrackReflFwd(long track_id);
  void setTrackReflBwd(long track_id);
  void setNextFwdFwd(bool fwd);
  void setNextBwdFwd(bool fwd);
  void setXYIndex(int index);
  void setAzimIndex(int index);
  void setSurfaceIn(int surface_in);
  void setSurfaceOut(int surface_out);
  void setDomainFwd(int neighbor);
  void setDomainBwd(int neighbor);
  void setLinkIndex(int index);

  /* Getter methods */
  int getUid();
  Point* getEnd();
  Point* getStart();
  double getPhi() const;
  double getLength();
  long getTrackNextFwd();
  long getTrackNextBwd();
  long getTrackPrdcFwd();
  long getTrackPrdcBwd();
  long getTrackReflFwd();
  long getTrackReflBwd();
  bool getNextFwdFwd();
  bool getNextBwdFwd();
  int getXYIndex();
  int getAzimIndex();
  int getLinkIndex();
  boundaryType getBCFwd() const;
  boundaryType getBCBwd() const;
  segment* getSegment(int s);
  segment* getSegments();
  int getNumSegments();
  int getDomainFwd();
  int getDomainBwd();
  int getSurfaceIn();
  int getSurfaceOut();

  /* Worker methods */
  void addSegment(segment* segment);
  void removeSegment(int index);
  void insertSegment(int index, segment* segment);
  void clearSegments();
  void setNumSegments(int num_segments);
  virtual std::string toString();
};


inline int Track::getUid() {
  return _uid;
}


inline segment* Track::getSegment(int segment) {

  /* If Track doesn't contain this segment, exits program */
  if (segment >= (int)_segments.size())
    log_printf(ERROR, "Attempted to retrieve segment s = %d but Track only "
               "has %d segments", segment, _segments.size());

  return &_segments[segment];
}



inline segment* Track::getSegments() {
  return &_segments[0];
}


inline int Track::getNumSegments() {
  if (_num_segments == 0)
    return _segments.size();
  else
    return _num_segments;
}

inline void Track::setNumSegments(int num_segments) {
    _num_segments = num_segments;
}


inline Point* Track::getEnd() {
  return &_end;
}


inline Point* Track::getStart() {
  return &_start;
}


inline double Track::getPhi() const {
  return _phi;
}


inline boundaryType Track::getBCFwd() const {
  return _bc_fwd;
}


inline boundaryType Track::getBCBwd() const {
  return _bc_bwd;
}

#endif /* TRACK_H_ */