Load and edit a RAT SOC TTree in python

edit_soc.py

import ROOT
import rat
import sys

file_ = ROOT.TFile(sys.argv[1])
tree = file_.Get("T")
soc = ROOT.RAT.DS.SOC()
run_tree = file.Get("runT")
tree.SetBranchAddress("soc", soc)
for iEntry in range(0, tree.GetEntries()):
    tree.GetEntry(iEntry)
    print soc.GetRunID()
    # Edit...


new_file = ROOT.TFile(sys.argv[2])
run_tree.Write()
new_tree = ROOT.TTree("T", "SOC Tree")
new_tree.Branch("soc", soc.ClassName(), soc, 32000, 99 )
soc.Fill()
new_tree.Write()