ctools

#! /usr/bin/env python 

# ================================================================

# This class stores/plots/prints/manipulates data given as a python

# dictionary of form:

#

# dict = {"parameter1":{"value":[..],"eu_value":[..],"ed_value":[..],

# "unit":[..]}, "parameter2",...}

#

# Author: Rolf Buehler (rolf.buehler@desy.de)

# Date: 22.7.2014

# ================================================================

import astropy.table as apytab

import json

import matplotlib.pyplot as plt

from astropy.io import ascii

class dicdata:

    "Class to store spectra"

    def __init__(self):

        self.data=apytab.Table()

    def adddic(self,dic):

        "Adds dictionary to data"

        for key in dic:

            self.data[key]       = apytab.Column(dic[key]["value"], unit=dic[key]["unit"])

            if len(dic[key]["eu_value"])==len(dic[key]["value"]):

                self.data["eu_"+key] = apytab.Column(dic[key]["eu_value"], unit=dic[key]["unit"])

            if len(dic[key]["ed_value"])==len(dic[key]["value"]):

                self.data["ed_"+key] = apytab.Column(dic[key]["ed_value"], unit=dic[key]["unit"])

    def addjson(self,filename):

        "Adds json file to data"

        json_data=open(filename)

        self.adddic(json.load(json_data))

    def printout(self):

        "Prints table out"

        print self.data

    def draw(self,xvar,yvar,log=True,fmt="o"):

        "Draws spectrum data"

        if log:

            plt.xscale('log')

            plt.yscale('log')

        plt.xlabel(xvar+" ["+str(self.data[xvar].unit)+"]")

        plt.ylabel(yvar+" ["+str(self.data[yvar].unit)+"]")

        plt.errorbar(self.data[xvar],self.data[yvar], xerr=[self.data["ed_"+xvar],self.data["eu_"+xvar]],

                     yerr=[self.data["ed_"+yvar],self.data["eu_"+yvar]],fmt=fmt)

        plt.show()

    def writetoascii(self,filename,asciiformat='ipac'):

        "Writes data to ascii"

        ascii.write(self.data, filename,format=asciiformat)

if __name__ == '__main__':

    print "Testing dicdata class.."

    specdata = dicdata()

    specdata.addjson("spectrum.json")

    specdata.printout()

    specdata.draw("energy","flux")

    specdata.writetoascii("spectrum.ipac")