From 2f5a670fdaff8a61fbc2c054954f9bbec0e91c22 Mon Sep 17 00:00:00 2001
From: Sujay Shankar <sujay.shankar@utexas.edu>
Date: Wed, 31 Jan 2024 23:36:57 -0500
Subject: [PATCH] Add parquet read in

---
 src/gollum/V2/phoenix.py         |  24 +++++++++++++-----------
 src/gollum/V2/simple_spectrum.py |   8 +++++---
 test.parquet.gz                  | Bin 0 -> 2481 bytes
 3 files changed, 18 insertions(+), 14 deletions(-)
 create mode 100644 test.parquet.gz

diff --git a/src/gollum/V2/phoenix.py b/src/gollum/V2/phoenix.py
index 70cb748..225ff33 100644
--- a/src/gollum/V2/phoenix.py
+++ b/src/gollum/V2/phoenix.py
@@ -1,9 +1,8 @@
 from gollum.V2.simple_spectrum import SimpleSpectrum
 
-import pandas as pd
-
 from dataclasses import dataclass
 from dotenv import get_key
+from pandas import read_parquet
 
 @dataclass(slots=True)
 class PHOENIXSpectrum(SimpleSpectrum):
@@ -31,17 +30,20 @@ class PHOENIXSpectrum(SimpleSpectrum):
     Z: float
     alpha: float
 
-    def __init__(self, teff: int, logg: float, Z: float, alpha: float, extent: tuple[float, float] = None, **kwargs):
-        self.teff, self.logg, self.Z, self.alpha = teff, logg, Z, alpha
+    def __init__(self, teff: int = None, logg: float = None, Z: float = None, alpha: float = None, extent: tuple[float, float] = None, **kwargs):
+        self.teff, self.logg, self.Z, self.alpha = point =  teff, logg, Z, alpha
         if kwargs:
             super(PHOENIXSpectrum, self).__init__(**kwargs)
             return
-        if not self._env().exists() or not (path := get_key(self._env(), 'PHOENIX_PATH')):
-            self._env().touch()
-            raise RuntimeError('PHOENIX data path not set. Set it with: PHOENIXSpectrum.configure(<path>)')
-        super(PHOENIXSpectrum, self).__init__([10, 20, 30], [40, 50, 60])
 
+        path = get_key(self._env(), 'PHOENIX_PATH') or 'zenodo link'
+        df = read_parquet(path, 'pyarrow', [str(point)], filters=[('index', '>=', extent[0]), ('index', '<=', extent[1])] if extent else None)
+        super(PHOENIXSpectrum, self).__init__(df.index, df[str(point)])
+
+
+import pandas as pd
+pd.DataFrame(index=[1, 2, 3], data={'(6000, 4.5, 0.0, 0.0)': [4, 5, 6]}).to_parquet('test.parquet.gz', compression='gzip')
+
+x = PHOENIXSpectrum(6000, 4.5, 0.0, 0.0)
+print(x)
 
-x = PHOENIXSpectrum(1, 2, 3, 7, (4, 5))
-y = PHOENIXSpectrum(1, 2, 3, 7, wavelength=[1, 2, 3], flux=[4, 5, 6])
-print(x + y)
diff --git a/src/gollum/V2/simple_spectrum.py b/src/gollum/V2/simple_spectrum.py
index f7be245..d89b4a7 100644
--- a/src/gollum/V2/simple_spectrum.py
+++ b/src/gollum/V2/simple_spectrum.py
@@ -151,14 +151,14 @@ def blackbody(spectral_axis: ArrayLike, T: float):
     @staticmethod
     def _env():
         '''
-        [ACCESS] The path to gollum's environment file.
+        [ACCESS] The path to `gollum`'s environment file.
         '''
         return Path(getfile(SimpleSpectrum)).parent / '.env'
     
     @classmethod
-    def configure(cls, data_path: str):
+    def _configure(cls, data_path: str):
         '''
-        [SETUP] Configure gollum's environment for a model grid.
+        [SETUP] Configure `gollum`'s environment for a model grid.
 
         Parameters
         ----------
@@ -166,3 +166,5 @@ def configure(cls, data_path: str):
             The path to the data directory.
         '''
         set_key(cls._env(), cls.__name__.replace('Spectrum', '_PATH'), str(Path(data_path).expanduser().resolve()))
+
+SimpleSpectrum._env().touch()
\ No newline at end of file
diff --git a/test.parquet.gz b/test.parquet.gz
new file mode 100644
index 0000000000000000000000000000000000000000..9c0962bce5fdd1ab7cf4d996ea808d635f5913f4
GIT binary patch
literal 2481
zcmcguOK;*<6uwC)Mb#|QYK%aNRAi|rRT&Bd(lDAXu1z3jfDBAx9&JVZf|$8B3>b_P
zMar&Ht8PZ~7iQOG_uX_?%^&Eli*BoS+f~oK<N<}KQ&pNPY+s-E_nmtzp5R)N^Kl7o
zuf+K|^Z`O2ej7k+ZD`6RlDA*`U#<Oq`R61ALJo2FxkoXu+=R=&6$<!gwkQ<vUG0Ks
zdxaE~bAi>9lM_u>Wd)%Zzx{gr9%!R4*0(rcID`C~;SjgIn@A*Lk#sy0izMQ(gWJcP
zFX#{cvKIViJ^E!d@L!ULV}KEH>lDByJOIgEnDt8(-F|)V)!NtJ{rt&l%p2fOS17PH
z+aQX0SqjR3!Ti@dx0v5=w@q6&rfp(O4AM@t+jDq-@<spH0L05l?nu1~arjXNh5RG!
z)UXuJ2NZm?eh1St-Eu`J=nrp&V{f@v(YHxc@Q00{KN<+W^xa1&91HmeimfZ7&`aMr
zKkJXyCu7SV@ed<k@Jl`8zlufpnZ>M~1~WY8ZN)Oc#U2v^79G<ddV5heOVoC}0X5{u
zwyE`Xg9mGSYz@2{MPKb1nnUH5;V8P|C}4ih8#b7mc9;4dALSqH@fR1j#Vl6AubeL$
za-p!h&HhoVX*;{=<w|eFbA?g@E@DvcBKXC;%d|pi|1&4!VrVE%f5`YEraDxH9;IvG
zDZ^pkr2r<zaAfvvDov*2iFk^KVy{!rDO2%eJk4LA&^=tM9o8StrYiJ(;8VJ$-5kca
z3Kw0EF-~Kli}M(FvDjrYL0HhcRJebm-*?5?JKR|mrk5~wG5Eog1%W4Xf_oD#!EM$v
z&zg-gY^&fed?s0B+$zZFMyfJV4`fHR%jTges1nI`6tHbIPCIHrPPOXOPC1|LNF*_m
z#9~iPO_ORpb8SO3Pui984#dlxs`X<UW31PwqmqctmPLqKsC85;@3fwRT_KZH3oeaw
zRZpd(9}t)6#9~sl2*j@u$#lCAFR3<!Ok<~((F!y!Hq}(|d8=MY%=Ag37xI9qb}IGy
zUBfF9)v8Qd1;9mX5{t6D-*gVeVzwhM)P6?o>pLU!rY=$HS9;hji*C0frjt;+Tk7GF
zl*8k)_{?01<qdqaLLNBIpUq<cZWEPWmBmzT+0hQjJO=~SuBIxYL^*O=fDQZ;%^U8f
z*^)?m_Xb>x#ZDu2-Qy|IEqT;96FLo##a_k#d~WDr(lz94!e|JD@<q>!o@=h;9&n7*
zgQQM*YVH&X^kkx$(1QcuxIj$J%6kUI0DU*=c~7&lJFJIxi)eObpr+ESzvZJU@SXM~
zoAv9qMoadz6q%?rrYL!(-u}1(eWhod9eFGq(;b|DjLSzbWP7mL7hz~H7TJ)nEq%oT
z-2xA*xS^v|%3b-xq%eC-hZE+h2~&h&fY@AJ&#!<_sRV`(7RygJ7|ZnbIM;N`IeLz&
z<H}RF3)tMNvo)Z47(|HIrE%H3QW<7Q@aHHL87RZ&W5ao%4F(U`M<{ajk%`PcKjGLG
PLhwUBzKhTk__Ovmr}T2I

literal 0
HcmV?d00001