Vector.h

Go to the documentation of this file.

#ifndef SRC_VECTOR_H_
#define SRC_VECTOR_H_


#ifdef __cplusplus
#include <math.h>
#include <stdlib.h>
#include <stdio.h>
#include <map>
#include <vector>
#include <string>
#include <iostream>
#include <sstream>
#include <iomanip>
#include "log.h"
#include "pairwise_sum.h"
#endif


class Vector {
 private:
  CMFD_PRECISION* _array;
  int _num_rows;
  int _num_x;
  int _num_y;
  int _num_z;
  int _num_groups;

  omp_lock_t* _cell_locks;

  void setNumX(int num_x);
  void setNumY(int num_y);
  void setNumZ(int num_z);
  void setNumGroups(int num_groups);

 public:
  Vector(omp_lock_t* cell_locks, int num_x = 1, int num_y = 1, int num_z = 1,
         int num_groups = 1);
  virtual ~Vector();

  /* Worker functions */
  void incrementValue(int cell, int group, CMFD_PRECISION val);
  void incrementValues(int cell, int group_start, int group_end,
                       CMFD_PRECISION* vals);
  void clear();
  void scaleByValue(CMFD_PRECISION val);
  void printString();
  void copyTo(Vector* vector);

  /* Getter functions */
  CMFD_PRECISION getValue(int cell, int group);
  CMFD_PRECISION* getArray();
  int getNumX();
  int getNumY();
  int getNumZ();
  int getNumGroups();
  int getNumRows();
  double getSum();
  omp_lock_t* getCellLocks();

  /* Setter functions */
  void setValue(int cell, int group, CMFD_PRECISION val);
  void setValues(int cell, int group_start, int group_end,
                 CMFD_PRECISION* vals);
  void setAll(CMFD_PRECISION val);
};

#endif  // SRC_VECTOR_H_