#!/usr/bin/env python
# -*- coding: utf-8 -*-
from astropy.io.votable import parse_single_table
from mapper import Mapper
from iterator import Iterator
import sys
import matplotlib.pyplot as plt
       
def main():
    file_to_process = "../../standardized_votables/laurent/gaia_multicol.xml"
    mapper = Mapper("gaiamulticol", file_to_process, "_table1")
    mapper.read_annotation()
    mapper.add_shorcuts()

    print( mapper.__repr__())
    
    iterator = Iterator(file_to_process, mapper)
    tg = []
    mg = []
    tbp = []
    mbp = []
    trp = []
    mrp = []
    
    while iterator.hasNext() :
        obj = iterator.next_object()
        if obj['magnitude.G'] != None:
           tg.append(obj['point.timestamp']) 
           mg.append(obj['magnitude.G'])
        elif  obj['magnitude.BP'] != None:
           tbp.append(obj['point.timestamp']) 
           mbp.append(obj['magnitude.BP'])
        elif  obj['magnitude.RP'] != None:
           trp.append(obj['point.timestamp']) 
           mrp.append(obj['magnitude.RP'])

    """
    Plotting section
    """
    fig, ax1 = plt.subplots()

    plt.plot(tg, mg, "bo", markersize=1)
    plt.plot(tbp, mbp, "ro", markersize=1)
    plt.plot(trp, mrp, "yo", markersize=1)
    plt.xlabel('Time')
    plt.ylabel('Mag')
    
    ax1.text(tg[0], min(mg), "Gaia multi band G BP RP", fontsize=8)
    plt.show()        
if __name__ == "__main__":
    main()