fof_class.hpp

Go to the documentation of this file.

 
#ifndef FOF_CLASS_H
#define FOF_CLASS_H
 
#include <string>
#include <vector>
#include "astro.hpp"
#include "cluster_class.hpp"
#include "galaxy_class.hpp"
#include "kdtree_class.hpp"
#include "zbin_class.hpp"
 
class FoF { // Class for friends-of-friends functions
 
public:
 
  std::vector<Cluster> list_of_clusters;
 
  FoF(int max_ngal_val) {
    max_ngal = max_ngal_val;
    cluster_count = -1;
    link_r = 0;
    link_z = 0;
  };
 
  void setup (double, double, const std::string &);
 
  void remove (int);
 
  int friends_of_friends (int, const std::vector<Zbin> &, std::vector<Galaxy> &,
               const Kdtree &);
 
private:
 
  Astro astro;
 
  int cluster_count;
 
  int max_ngal;
 
  double link_r;
 
  double link_z;
 
  std::string mode;
 
  bool bin_check (int, const std::vector<int> &);
 
 
  bool friendship (const Zbin &, const Galaxy &, const Galaxy &, double);
 
  void new_cluster (const Zbin &, Galaxy*, Galaxy*);
 
  void add_member (const Zbin &, Galaxy* , Cluster &);
 
  int find_friends (const Zbin &, Galaxy &, double, std::vector<Galaxy> &, const Kdtree &);
 
  int find_friends_of_friends (const Zbin &, Cluster &, double,
                               std::vector<Galaxy> &, const Kdtree &);
 
};
 
#endif // FOF_CLASS_H