ngspice画图代码整理

elec/ABCStar_fe_get_fig.py

-#!/usr/bin/env python3
-# TODO: Need to be TOTALLY rewritten
-import ROOT
-import numpy
-
-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]))
-
-    time = numpy.array(time,dtype='float64')
-    volt = numpy.array(volt,dtype='float64')
-
-    return time,volt
-
-def main():
-    file_path = 'output/elec/'
-    #file_name = 'drs4_analog.raw'
-    file_name = 'ABCStar_fe.raw'
-    com_name=file_name.split('.')[0]
-    fig_name=file_path + com_name + '.pdf'
-    time,volt = [],[]
-
-    time,volt = read_file(file_path,file_name)
-    length = len(time)
-
-    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(0,5)
-    f1.GetXaxis().SetLimits(0,30)
-    f1.GetXaxis().CenterTitle()
-    f1.GetXaxis().SetTitleSize(0.05)
-    f1.GetXaxis().SetTitleOffset(0.8)
-
-    f1.GetYaxis().SetTitle('Voltage [V]')
-    f1.GetYaxis().SetLimits(0,-1)
-    f1.GetYaxis().CenterTitle()
-    f1.GetYaxis().SetTitleSize(0.05)
-    f1.GetYaxis().SetTitleOffset(0.93)
-
-    c.cd()
-    f1.Draw('AL')
-    c.SaveAs(fig_name)
-    print("figure  has been saved in " , fig_name)
-
-if __name__ == '__main__':
-    main()
-    

elec/__init__.py

-
+import os
 def main(kwargs):
     label = kwargs['label']
+    os.makedirs('output/elec', exist_ok=True)
 
     if label == 'ngspice_t1':
         import subprocess
         subprocess.run(['ngspice -b output/elec/T1_tmp.cir'], shell=True)
-    elif label == 'ngspice_ABCStar_fe':
-        import subprocess
-        subprocess.run(['ngspice -b param_file/circuit/ABCStar_fe.cir'], shell=True)
-    elif label == 'drs4_get_analog':
+    elif label.startswith('ngspice_'):
         import subprocess
-        subprocess.run(['ngspice -b param_file/circuit/drs4_analog.cir'], shell=True)
-    elif label == 'drs4_get_fig':
-        from . import drs4_get_fig
-        drs4_get_fig.main()
-    elif label == 'ABCStar_fe_get_fig':
-        from . import ABCStar_fe_get_fig
-        ABCStar_fe_get_fig.main()
+        elec_name = label.replace('ngspice_', '')
+        subprocess.run(['ngspice -b param_file/circuit/{}.cir'.format(elec_name)], shell=True)
+    elif label.endswith('_get_fig'):
+        from . import ngspice_get_fig
+        ngspice_get_fig.main(label.replace('_get_fig', ''))
     else:
         from . import readout
         readout.main(label)

elec/drs4_get_fig.py

-#!/usr/bin/env python3
-# TODO: Need to be TOTALLY rewritten
-import ROOT
-import numpy
-
-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]))
-
-    time = numpy.array(time,dtype='float64')
-    volt = numpy.array(volt,dtype='float64')
-
-    return time,volt
-
-def main():
-    file_path = 'output/elec/'
-    file_name = 'drs4_analog.raw'
-    com_name=file_name.split('.')[0]
-    fig_name=file_path + com_name + '.pdf'
-    time,volt = [],[]
-
-    time,volt = read_file(file_path,file_name)
-    length = len(time)
-
-    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(0,5)
-    f1.GetXaxis().CenterTitle()
-    f1.GetXaxis().SetTitleSize(0.05)
-    f1.GetXaxis().SetTitleOffset(0.8)
-
-    f1.GetYaxis().SetTitle('Voltage [V]')
-    f1.GetYaxis().SetLimits(0,-1)
-    f1.GetYaxis().CenterTitle()
-    f1.GetYaxis().SetTitleSize(0.05)
-    f1.GetYaxis().SetTitleOffset(0.93)
-
-    c.cd()
-    f1.Draw('AL')
-    c.SaveAs(fig_name)
-    print("figure  has been saved in " , fig_name)
-
-if __name__ == '__main__':
-    main()
-    

elec/ngspice_get_fig.py

 #!/usr/bin/env python3
 
-import ROOT
 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:
@@ -18,15 +23,14 @@ def read_file(file_path,file_name):
 
     return time,volt
 
-def main():
-    file_path = 'output/elec/ngspice_fig'
-    file_name = sys.argv[1]
-    com_name=file_name.split('.')[0]
-    fig_name=file_path + com_name + '.pdf'
+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,file_name)
+    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)
@@ -37,13 +41,12 @@ def main():
     f1.SetLineWidth(2)
 
     f1.GetXaxis().SetTitle('Time [ns]')
-    f1.GetXaxis().SetLimits(0,20)
+    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().SetLimits(0,-5)
     f1.GetYaxis().CenterTitle()
     f1.GetYaxis().SetTitleSize(0.05)
     f1.GetYaxis().SetTitleOffset(0.7)
@@ -51,7 +54,6 @@ def main():
     c.cd()
     f1.Draw('AL')
     c.SaveAs(fig_name)
-    print("figure  has been saved in " , fig_name)
 
 if __name__ == '__main__':
     main()