data_manager.hpp

Go to the documentation of this file.

#ifndef EXPLICIT_MAPMAKING_DATA_MANAGER_HPP_
#define EXPLICIT_MAPMAKING_DATA_MANAGER_HPP_

#include "ces.hpp"
#include <mpi.h>
#include <vector>
#include <list>
#include <boost/filesystem.hpp>

namespace fs = boost::filesystem;

namespace mapmaking {

class DataManager {
 public:
  DataManager(const std::vector<int>& obspix, const MPI_Comm comm,
              char stokes, int n_poly);
  void SetNCESHitThreshold(int n_ces_hits, int max_n_pixel);
  int n_pixel() const { return obspix_.size(); };
  char stokes() const { return stokes_; };
  std::vector<int> obspix() const { return obspix_; }
  CES& AddCES(int n_sample) { ceses_.emplace_back(n_sample, filters_); return ceses_.back(); }
  std::list<CES>& ceses() { return ceses_; }
  void CompressCESes();
  void RemovePathologicalPixels(double threshold);
  void WhiteNoise();
  void SetObspix(fs::path obspix_filename);
  void SaveObspix(fs::path obspix_filename);
  void ComputeCcCsSsNhits();
  void SaveCcCsSsNhitsWeigths(fs::path filename);

 private:
  std::vector<double> cc_;  // Root only >
  std::vector<double> cs_;
  std::vector<double> ss_;
  std::vector<double> weights_;
  std::vector<int> nhits_;  // Root only <
  std::vector<int> obspix_;
  Filters filters_;
  int my_rank_;
  int n_proc_;
  char stokes_;
  const MPI_Comm comm_;
  std::list<CES> ceses_;
  void MaskPixels(std::vector<int> mask);  // mask[p] = 0 => pixel p is rejected
};
}

#endif  // EXPLICIT_MAPMAKING_DATA_MANAGER_HPP_