astroimsim-data 0.1.3

Data package for astroimsim ecosystem.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130

use std::fs::File;
use std::io::Write;
use std::time::Instant;
use astroimsim_geometry::grid1d::GRID1D;
use astroimsim_geometry::grid2d::{Location, GRID2D};
use astroimsim_geometry::points::Point;
use rand::distr::{Distribution, Uniform};
use astroimsim_spectra::power_spectrum::{PowerSpectrum, SpectrumUnits};
use astroimsim_spectra::power_spectrum::SpectrumUnits::f_lambda;
use astroimsim_spectra::spectral_response::SpectralResponseCurve;
use egui::accesskit::Invalid::Spelling;
use rand_distr::Poisson;
use crate::point_sources::BandUnits::{AverageElectronFlux, Electrons};
use crate::spatial_effect::SpatialEffect;

#[derive(Debug,Clone)]
pub struct FullSpectrumPointSource {
    pub point: Point,
    pub spectrum: PowerSpectrum,
    pub scale: f64,
}

#[derive(Debug)]
pub struct Bands {
    pub fuv: f64,
    pub nuv: f64,
    pub units: BandUnits,
}

#[derive(Debug)]
pub enum BandUnits{
    AverageElectronFlux,
    Electrons,
}

impl FullSpectrumPointSource {
    pub fn flat_AB(point:Point,ab_mag:f64,grid1d: GRID1D)->FullSpectrumPointSource{
        let mut spectrum = PowerSpectrum::flat_AB(ab_mag,grid1d," ");
            spectrum.convert_to(&SpectrumUnits::f_lambda);
        //println!("spectrum is {:?}",spectrum);
        FullSpectrumPointSource{point,spectrum,scale:1.0}
    }
    pub fn new_from_spectrum(point: Point,spectrum:PowerSpectrum)->FullSpectrumPointSource{
        FullSpectrumPointSource{point,spectrum,scale:1.0}
    }

    pub fn scale(&mut self,scale:f64){
        self.scale *= scale
    }

    pub fn apply_spectral_response_curve(&mut self, curve:&SpectralResponseCurve){
        self.spectrum.convert_to(&f_lambda);
        self.spectrum.apply_spectral_response(curve);
    }
    pub fn to_bands(&self,fuv_path:&SpectralResponseCurve,nuv_path:&SpectralResponseCurve,area:f64)->Bands{
        let mut fuv_spectrum = self.clone();
        let mut nuv_spectrum = self.clone();
        fuv_spectrum.apply_spectral_response_curve(fuv_path);
        nuv_spectrum.apply_spectral_response_curve(nuv_path);
        fuv_spectrum.spectrum.write_to_dat("fuv_spectrum", "fuv spectrum ");
        nuv_spectrum.spectrum.write_to_dat("nuv_spectrum", "nuv spectrum");
        let fuv = fuv_spectrum.spectrum.total_average_photon_flux(area);
        let nuv = nuv_spectrum.spectrum.total_average_photon_flux(area);
        Bands{fuv,nuv,units:AverageElectronFlux}
    }
}
#[derive(Debug)]
pub struct FullSpectrumSourceList {
    pub sources: Vec<FullSpectrumPointSource>,
}
impl FullSpectrumSourceList {
    pub fn new_from(mut sources:Vec<FullSpectrumPointSource> ) -> FullSpectrumSourceList {
        //sources.sort_by(|a:&point_source, b:&point_source| b.bin.cmp(&a.bin));
        FullSpectrumSourceList {
            sources,
        }
    }
    pub fn new_empty(capacity:usize) -> FullSpectrumSourceList {
        FullSpectrumSourceList {
            sources: Vec::with_capacity(capacity)
        }
    }
    pub fn add_source(&mut self, source: FullSpectrumPointSource) -> &mut FullSpectrumSourceList {
        self.sources.push(source);
        self

    }
    pub fn full_spectrum_point_source_field(number_of_point_sources:usize,
                                           min_brightness: f64,
                                           max_brightness: f64,
                                            spectrum:PowerSpectrum,
                                            grid: &GRID2D,
    ) -> FullSpectrumSourceList {
        assert!(max_brightness>=min_brightness,"max brightness must be greater or equal to min brightness");
        let luminosities = Uniform::new(min_brightness,max_brightness).expect("Could not generate random luminosities in the given range");
        let mut rng = rand::rng();
        let sources: Vec<FullSpectrumPointSource> = (0..number_of_point_sources).map(|_x|{
            let scale = luminosities.sample(&mut rng);
            let point = grid.random();
            FullSpectrumPointSource { point,spectrum: spectrum.clone(), scale}
        }).collect();
        FullSpectrumSourceList::new_from(sources)
    }

    pub fn apply_spatial_effect(&mut self,effect:SpatialEffect){
        let start = Instant::now();
        for source in &mut self.sources{
            let scale = match effect.grid.inside_or_outside(&source.point){
                Location::Outside => continue,
                Location::Inside => effect.get_data(&source.point)
            };
            source.scale(scale)
        }
        println!("Applied spatial effect {:?} to {:?} sources in {:?} ms",effect.label,self.sources.len(),start.elapsed().as_millis())
    }
}


impl Bands{
    pub fn poisson(&mut self){
        let mut rng = rand::rng();
        let fuv_poisson = Poisson::new(self.fuv as f64).unwrap();
        let nuv_poisson = Poisson::new(self.nuv as f64).unwrap();
        let fuv_electrons = fuv_poisson.sample(&mut rng);
        let nuv_electrons = nuv_poisson.sample(&mut rng);
        self.units = Electrons;
        self.nuv = nuv_electrons;
        self.fuv = fuv_electrons;
    }
}