ngspice_get_fig.py

#!/usr/bin/env python3

import sys
import os

import numpy
import ROOT

from util.output import output

# TODO: Need to be TOTALLY rewritten
def read_file(file_path,file_name):
    with open(file_path + '/' + file_name) as f:
        lines = f.readlines()
        time,volt = [],[]

        for line in lines:
            time.append(float(line.split()[0])*1e9)
            volt.append(float(line.split()[1])*1e3)

    time = numpy.array(time,dtype='float64')
    volt = numpy.array(volt,dtype='float64')

    return time,volt

def main(elec_name):
    file_path = output(__file__)
    fig_name = os.path.join(file_path, elec_name+'.pdf')
    time,volt = [],[]

    time,volt = read_file(file_path, elec_name+'.raw')
    length = len(time)
    t_min, t_max = time[0], time[-1]

    ROOT.gROOT.SetBatch()    
    c = ROOT.TCanvas('c','c',700,600)
    f1 = ROOT.TGraph(length,time,volt)
    f1.SetTitle(' ')

    f1.SetLineColor(2)
    f1.SetLineWidth(2)

    f1.GetXaxis().SetTitle('Time [ns]')
    f1.GetXaxis().SetLimits(t_min, t_max)
    f1.GetXaxis().CenterTitle()
    f1.GetXaxis().SetTitleSize(0.05)
    f1.GetXaxis().SetTitleOffset(0.8)

    f1.GetYaxis().SetTitle('Voltage [mV]')
    f1.GetYaxis().CenterTitle()
    f1.GetYaxis().SetTitleSize(0.05)
    f1.GetYaxis().SetTitleOffset(0.7)

    c.cd()
    f1.Draw('AL')
    c.SaveAs(fig_name)

if __name__ == '__main__':
    main()