create a tgraph for the mean amplitude versus time and correlated with temperature

plotVtime.py

#!/usr/bin/env python

from ROOT import *
import os, sys
import numpy as np

def main(*args):

    runFile = os.path.join('/sps/edelweis/kdata/data/current/raw', os.path.basename(args[0]))
    boloName = args[1]
    f = KDataReader(runFile)
    e = f.GetEvent()
    

    ff = TFile(os.path.join('$HOME/analysis/tempVtime/tempPlots', 'tempPlots' + boloName + '_'+ os.path.basename(runFile)), 'recreate')
    gTemperature = TGraph(1)
    gResolution = TGraph(1)
    gCorrelation = TGraph(1)
    
    tPoint = 0
    rPoint = 0
    cPoint = 0
    
    for i in range(f.GetEntries()):
        f.GetEntry(i)
        
        for j in range(e.GetNumBolos()):
            
            b = e.GetBolo(j)
            
            if b.GetDetectorName() == boloName:
                s = b.GetSambaRecord()
                gTemperature.SetPoint(tPoint, s.GetNtpDateSec(), s.GetTemperature())
                tPoint += 1
                
                for n in range(b.GetNumPulseRecords()):
                    p = b.GetPulseRecord(n)
                    if p.GetIsHeatPulse():
                        trace = np.array(p.GetTrace()[:200])
                        mean = trace.mean()
                        gResolution.SetPoint(rPoint, s.GetNtpDateSec(), mean)
                        rPoint += 1
                        gCorrelation.SetPoint(cPoint, mean, s.GetTemperature())
                        cPoint += 1
                
    ff.cd()
    gTemperature.SetName('temp')
    gTemperature.Write()
    gResolution.SetName('meanamp')
    gResolution.Write()
    gCorrelation.SetName('cor')
    gCorrelation.Write()
    ff.Close()
    f.Close()

if __name__ == '__main__':
    main(*sys.argv[1:])