moved to log space sampling for DM and pulse width

This commit is contained in:
Sakimori 2025-07-30 13:02:59 -04:00
parent c4d88037f3
commit d9dad7cd68
No known key found for this signature in database

View file

@ -24,10 +24,10 @@ logFmt = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
logging.basicConfig(level=logging.INFO, format=logFmt) logging.basicConfig(level=logging.INFO, format=logFmt)
#Set values for random sampling #Set values for random sampling
MIN_DM = 300 LOG_MIN_DM = 2.5
MAX_DM = 10000 LOG_MAX_DM = 3.5
MIN_WIDTH = 0.001 #1 ms LOG_MIN_WIDTH = 0
MAX_WIDTH = 1 #1 s LOG_MAX_WIDTH = 3 #in ms
rng = np.random.default_rng() rng = np.random.default_rng()
#This function from https://josephwkania.github.io/will/examples/inject_pulse.html#Show-how-we-can-inject-a-pulse-into-a-GREENBURST-filterbank. #This function from https://josephwkania.github.io/will/examples/inject_pulse.html#Show-how-we-can-inject-a-pulse-into-a-GREENBURST-filterbank.
@ -92,7 +92,11 @@ def randomDMandWidth():
""" """
Helper function that returns (DM, pulseWidth) tuple with bounds set at start of script. Helper function that returns (DM, pulseWidth) tuple with bounds set at start of script.
""" """
return (rng.uniform(low=MIN_DM, high=MAX_DM), rng.uniform(low=MIN_WIDTH, high=MAX_WIDTH)) log_dm = rng.uniform(low=LOG_MIN_DM, high=LOG_MAX_DM)
log_pw = rng.uniform(low=LOG_MIN_WIDTH, high=LOG_MAX_WIDTH)
dm = 10**log_dm
pw = 10**(log_pw - 3) #-3 to convert ms to s
return (dm, pw)
def addBurst(values): def addBurst(values):