lc.py

import os
from collections import defaultdict
import itertools
import reprlib

def lc_reader(filename):
    lclist=[]
    facet_names = []
    facet_values = []

    i = 0
    with open(filename) as fp:
        for line in fp:
            if i == 0:
                facet_names = line[1:].strip().split()
            if i == 1:
                facet_values = line[1:].strip().split()
            if line.find('#')!=0:
                lclist.append([float(f) for f in line.strip().split()])
            i += 1
    return lclist,{facet_names[i]:facet_values[i] for i in range(len(facet_names))}


class LightCurve:

    def __init__(self, data, metadict):
        self._time = [x[0] for x in data]
        self._amplitude = [x[1] for x in data]
        self._error = [x[2] for x in data]
        self.metadata = metadict
        self.filename = None

    @classmethod
    def from_file(cls, filename):
        lclist, metadict = lc_reader(filename)
        instance = cls(lclist, metadict)
        instance.filename = filename
        return instance

    def __repr__(self):
        class_name = type(self).__name__
        components = reprlib.repr(list(itertools.islice(self.timeseries,0,10)))
        components = components[components.find('['):]
        return '{}({})'.format(class_name, components)

    @property
    def time(self):
        return self._time

    @property
    def amplitude(self):
        return self._amplitude

    @property
    def timeseries(self):
        return zip(self._time,self._amplitude)

    def __getitem__(self,i):
        return (self._time[i],self._amplitude[i])

    def __len__(self):
        return len(self._time)


class LightCurveDB:

    def __init__(self):
        self._collection={}
        self._field_index=defaultdict(list)
        self._tile_index=defaultdict(list)
        self._color_index=defaultdict(list)

    def populate(self, folder):
        for root,dirs,files in os.walk(folder): # DEPTH 1 ONLY
            for file in files:
                if file.find('.mjd')!=-1:
                    the_path = root+"/"+file
                    self._collection[file] = LightCurve.from_file(the_path)

    def index(self):
        for f in self._collection:
            lc, tilestring, seq, color, _ = f.split('.')
            field = int(lc.split('_')[-1])
            tile = int(tilestring)
            self._field_index[field].append(self._collection[f])
            self._tile_index[tile].append(self._collection[f])
            self._color_index[color].append(self._collection[f])

    def retrieve(self,facet,value):
        if facet == "field":
            return self._field_index[value]
        elif facet == "tile":
            return self._tile_index[value]
        elif facet == "color":
            return self._color_index[value]
        else:
            print("Fact not found")
            return None