import numpy as np
import cosmoglobe as cg
import healpy as hp

import matplotlib.pyplot as plt
import astropy.units as u
from astropy.coordinates import SkyCoord

ra=51.65593750000001*u.deg
dec=30.257886111111112*u.deg

ra = 274.37475*u.deg
dec=-4.660916666666667*u.deg
c = SkyCoord(ra=ra, dec=dec)
c_gal = c.galactic
l, b = c_gal.l.deg, c_gal.b.deg
print(l,b)

ps_rad = 5

# Find all scans that come within 5 degrees of this point.

# Loop all scans in /mn/stornext/d23/cmbco/globe/akari/tod/w140/filelist_akari_w140.txt


from cosmoglobe.tod_tools import TODLoader

# Takes the directory where .h5 files are located and the dataset in question as
# arguments

data = np.loadtxt('/mn/stornext/d23/cmbco/globe/akari/tod/w140/filelist_akari_w140.txt',
        skiprows=1, dtype=str)
print(data)

comm_tod = TODLoader("/mn/stornext/d23/cmbco/akari/tamakim/h5_test/WIDE-L/", "")
NSIDE = 2048
from tqdm import tqdm
for i in tqdm(range(200,len(data))):
    scan_id = data[i][0].zfill(6)
    file = data[i][1].split('/')[-1].split(".h5")[0]
    comm_tod.init_file(file, '')
    pix = comm_tod.load_field(f'{scan_id}/AKARI_WIDE-L_23/pix')
    lonlat = hp.pix2ang(NSIDE, pix, lonlat=True)
    dist = hp.rotator.angdist(np.array([l,b]), lonlat, lonlat=True)*180/np.pi
    try:
        if np.nanmin(dist) < ps_rad:
            print(scan_id,np.nanmin(dist))
            fig, axes = plt.subplots(sharex=True, nrows=3)
            axes[0].plot(lonlat[0])
            axes[1].plot(lonlat[1])
            axes[2].plot(dist)
            plt.savefig(f'dists_{scan_id}png')
    except RuntimeWarning:
        print(f"What is wrong with scan {scan_id}?")