#!/usr/bin/env python
# coding: utf-8

# # mrt
# Build ROOT Ntuple from other source.
# This program reads the `aptuple.txt' file row by row, then creates
# the Ntuple by adding row by row.
# 
# 
# 
# 
# **Author:** Wim Lavrijsen  
# <i><small>This notebook tutorial was automatically generated with <a href= "https://github.com/root-project/root/blob/master/documentation/doxygen/converttonotebook.py">ROOTBOOK-izer</a> from the macro found in the ROOT repository  on Wednesday, April 17, 2024 at 11:16 AM.</small></i>

# In[1]:


import sys, os
from ROOT import TFile, TNtuple, TROOT


ifn = os.path.join(str(TROOT.GetTutorialDir()), 'pyroot', 'aptuple.txt')
ofn = 'aptuple.root'

print('opening file %s ...' % ifn)
infile = open( ifn, 'r' )
lines  = infile.readlines()
title  = lines[0]
labels = lines[1].split()

print('writing file %s ...' % ofn)
outfile = TFile( ofn, 'RECREATE', 'ROOT file with an NTuple' )
ntuple  = TNtuple( 'ntuple', title, ':'.join( labels ) )

for line in lines[2:]:
    words = line.split()
    row = map( float, words )
    ntuple.Fill(*row)

outfile.Write()

print('done')