myociss

from clustering_algorithm.cluster_flashes import load_dat, to_ecef
import numpy as np

center_geo = np.array([40.4463980, -104.6368130, 1000.00]) # COLMA center
lma_center = to_ecef(np.expand_dims(center_geo, axis=0))

n_grid_points, grid_spacing = 18, 0.2559
grid_start_lat = center_geo[0] - (n_grid_points / 2) * grid_spacing
grid_start_lon = center_geo[1] - (n_grid_points / 2) * grid_spacing

myociss

from sklearn.cluster import DBSCAN

# sources[time, x, y, z, power, grid_lat, grid_lon, init_lat, init_lon, init_alt]
def cluster(sources: array_type, xyz_scale: float, t_scale: float, grid_max: int, min_samples: int=10, epsilon: float=1.0, max_duration: float=3.0) -> Tuple[List[array_type], int]:
    min_t = np.min(sources[:,0])
    max_t = np.max(sources[:,0])
    time_start = min_t

    all_flashes: List[array_type]=[]
    total_removed = 0

myociss

def merge_flashes(flashes: List[array_type], t_threshold: float=0.15, xyz_threshold: float=3000.0) -> List[array_type]:
    flashes_sorted = sorted(flashes, key=lambda x: x[0,0])

    flash_merges = np.zeros((len(flashes), 2)) - 1.0

    for j in range(1, len(flashes_sorted)):
        branch_flash = flashes_sorted[j]

        for i in range(j):
            base_flash = flashes_sorted[i]

myociss

from scipy.spatial import ConvexHull

def get_flash_params(flashes: array_type, start_time: datetime.datetime) -> List[Dict]:
    flash_params: List[Dict] = []

    north_pole = np.array([90.0,0.0,0.0])

    for flash in flashes:
        grid_points = np.unique(flash[:,5:7], axis=0).astype('int64')
        n_sources = len(flash)

myociss

from clustering_algorithm.cluster_flashes import load_dat, to_ecef, cluster, get_flash_params, merge_flashes
import numpy as np
import os
from datetime import date, timedelta
import json
from pathlib import Path

all_files = os.listdir(data_dir)

for f_idx, fname in enumerate(all_files):

myociss

import matplotlib.pyplot as plt

n_buckets = int(max_dist // bucket_size)
param_names = ('duration', 'hull_area', 'mean_power', 'n_sources')
param_units = ('(ms)', '(square km)', '(dBW)', '')
stats_buckets = [{param: [] for param in param_names} for i in range(n_buckets)]

for fname in os.listdir(json_dir):
    with open(os.path.join(json_dir, fname), 'r') as f:
        data = json.load(f)

myociss

import cartopy.crs as ccrs
import cartopy.feature as cfeature
import matplotlib.pyplot as plt
import matplotlib.patheffects as pe
import datetime

all_flash_areas = {}
all_flash_hours = {}

for fname in os.listdir(json_dir):