main.py

import json
import numpy as np
import pca
import process_object as po

from flask import Flask, request, jsonify, render_template, url_for


app = Flask(__name__)

@app.route('/')
def index():
    return render_template('index.html')

@app.route('/supernova')
def supernova():
    return render_template('supernova.html')

@app.route('/new')
def splinenew():
    return render_template('index_new.html')

@app.route('/plotusers', methods=['post'])
def plotusers():
    data = request.get_json()
    position = po.project(data['lc'], data['p'])
    return jsonify(position)

@app.route('/fastpca')
def fastpca():
    return render_template('fastpca.html')

@app.route('/calculatepca', methods=['post'])
def calculatepca():
    global LAST_PCA
    uids = request.get_json()
    idlist, matrix, status = get_data_by_id(uids)
    pca_result = {}
    if status == 'ok':
        pca_result = pca.calculate(idlist, matrix, LAST_PCA)
        LAST_PCA = pca_result

    final_result = [[{} for x in range(len(pca_result[0]))] for x in range(len(pca_result))]
    for i in range(len(pca_result)):
        for j in range(len(pca_result[0])):
            final_result[i][j] = {'count':pca_result[i][j]}

    return jsonify({'status':status, 'data':final_result})

@app.route('/calculate_average_lc', methods=['post'])
def calculate_average_lc():
    global LCDATA, LCPHASE
    uids = request.get_json()
    # TODO: band as parameter
    band = 'V'

    matrix = []
    for i in uids:
        i = str(i)
        if i in LCDATA:
            if band in LCDATA[i]['bands']:
                vec = LCDATA[i][band]['mag']
                if len(vec) > 0: # some id may not have lc data
                    matrix.append(LCDATA[i][band]['mag'])
    mean = np.mean(np.array(matrix), axis=0)
    std = np.std(np.array(matrix), axis=0)
    return jsonify({'mean':mean.tolist(),
                    'std':std.tolist(),
                    'phase':LCPHASE})



def load_lc_data():
    lcdata = {}
    lcphase = [] # Assume all phase for different surveys are the same
    surveys = json.load(open("./static/data_ogle/list.json"))['surveys']
    for s in surveys:
        path = "./static/data_ogle/lightcurves/{}/fit.json".format(s)
        data = json.load(open(path))
        lcphase = data['phase']
        for objid in data['data']:
            lcdata[objid] = data['data'][objid]
    return lcdata, lcphase

def get_data_by_id(ids, band='V'):
    global LCDATA
    ids_exist = []  # some id may not have lc data
    matrix = []
    status = 'ok'

    for i in ids:
        i = str(i)
        if i in LCDATA:
            if band in LCDATA[i]['bands']:
                vec = LCDATA[i][band]['mag']
                if len(vec) > 0:
                    ids_exist.append(i)
                    matrix.append(vec)

    if len(matrix) == 0:
        return ids_exist, matrix, 'no light curve data'

    c_length = len(matrix[0])
    matrix = np.array(matrix)
    if len(matrix) < c_length:
        status = "numrows < numcols"
    else:
        try:
            matrix.shape = (len(matrix), c_length)
        except:
            status = "not all rows have same numcols"

    return ids_exist, matrix, status

LCDATA, LCPHASE = load_lc_data()
LAST_PCA = []


app.run(port=8080, debug=True)