galaxy_class.hpp

Go to the documentation of this file.

 
#ifndef GALAXY_CLASS_H
#define GALAXY_CLASS_H
 
#include <math.h>
#include <string>
#include "astro.hpp"
#include "cosmo.hpp"
#include "exceptions.hpp"
 
class UnionFind{ 
 
private:
  
    UnionFind* parent;
 
public:
 
    UnionFind(): parent(nullptr){}
 
    UnionFind * find(){
        if (parent == nullptr)
            parent = this;
        if (parent != this)
            parent = parent->find();
        return parent;
    }
 
    void join(UnionFind* u){
        UnionFind* a = find();
        UnionFind* b = u->find();
        if(a != b)
            a->parent = b;
    }
 
    bool is_singleton(){
        return parent == nullptr;
    }
 
};
 
class Galaxy { 
 
public:
 
    UnionFind uf;
 
    int num;
 
    int bin;
    std::vector<int> bins;
 
    Point P;
 
    double z;
 
    double z_err;
 
    double da;
 
    double v;
 
    unsigned long id;
 
    std::vector<bool> in_cluster;
 
    Galaxy(int num_val, unsigned long id_val, double ra_val, double dec_val,
            double z_val) {
        if (z_val < 0)
            throw BadArgumentException("Galaxy", "z_val", ">= 0.0");
        num = num_val;
        id = id_val;
        P.P[0] = ra_val;
        P.P[1] = dec_val;
        z = z_val;
        v = z / (1 + z);
        da = 0;
        bin = 0;
    };
 
    Galaxy(int num_val,  unsigned long id_val, double ra_val, double dec_val,
            double z_val, double z_err_val) {
        if (z_val < 0)
            throw BadArgumentException("Galaxy", "z_val", ">= 0.0");
        if (z_err_val < 0)
            throw BadArgumentException("Galaxy", "z_err_val", ">= 0.0");
        num = num_val;
        id = id_val;
        P.P[0] = ra_val;
        P.P[1] = dec_val;
        z = z_val;
        z_err = z_err_val;
        da = 0;
        bin = 0;
    };
 
    void assign_dist (double, double, double, double);
 
    void assign_bin (double, double);
 
    void assign_bins (double, double, double);
 
    void set_cluster_status (int);
 
    friend bool operator== (const Galaxy &gal1, const Galaxy &gal2);
    friend bool operator< (const Galaxy &gal1, const Galaxy &gal2);
 
private:
 
    Astro astro;
 
    Cosmo cosmo;
 
};
 
inline bool operator== (const Galaxy &gal1, const Galaxy &gal2) {
    return gal1.id == gal2.id;
}
 
inline bool operator< (const Galaxy& gal1, const Galaxy& gal2) {
    return gal1.id < gal2.id;
}
 
#endif // GALAXY_CLASS_H