def put_th1(self, cursor, name):
copy_cursor = copy(cursor)
cursor.skip(self._format_cntvers.size)
vers = 8
if len(self.fields["_fBuffer"]) != 0:
raise NotImplementedError
buff = (
self.put_tnamed(cursor, name, self.fTitle) +
self.put_tattline(cursor) + self.put_tattfill(cursor) +
self.put_tattmarker(cursor) +
cursor.put_fields(self._format_th1_1, self.fields["_fNcells"]) +
self.put_taxis(cursor, self.fXaxis) +
self.put_taxis(cursor, self.fYaxis) +
self.put_taxis(cursor, self.fZaxis) + cursor.put_fields(
self._format_th1_2, self.fields["_fBarOffset"],
self.fields["_fBarWidth"], self.fields["_fEntries"],
self.fields["_fTsumw"], self.fields["_fTsumw2"],
self.fields["_fTsumwx"], self.fields["_fTsumwx2"],
self.fields["_fMaximum"], self.fields["_fMinimum"],
self.fields["_fNormFactor"]) +
self.put_tarray(cursor, self.fields["_fContour"]) +
self.put_tarray(cursor, self.fields["_fSumw2"]) +
cursor.put_string(self.fields["_fOption"]) +
self.put_tlist(cursor, self.fields["_fFunctions"]) +
cursor.put_fields(
self._format_th1_3,
self.fields["_fBufferSize"],
0, # FIXME: empty fBuffer
self.fields["_fBinStatErrOpt"],
self.fields["_fStatOverflows"]))
length = len(buff) + self._format_cntvers.size
cnt = numpy.int64(length - 4) | uproot.const.kByteCountMask
return copy_cursor.put_fields(self._format_cntvers, cnt, vers) + buff
def _put_th3d(self, cursor, name):
copy_cursor = copy(cursor)
cursor.skip(self._format_cntvers.size)
vers = 4
buff = self._put_th3(cursor, name) + self._put_tarray(cursor, self._valuesarray)
length = len(buff) + self._format_cntvers.size
cnt = numpy.int64(length - 4) | uproot.const.kByteCountMask
return copy_cursor.put_fields(self._format_cntvers, cnt, vers) + buff
def _put_tobjstring(self, cursor, value, bit=0):
copy_cursor = copy(cursor)
cursor.skip(self._format_tobjstring.size)
vers = 1
buff = cursor.put_string(value)
length = len(buff) + self._format_tobjstring.size
cnt = numpy.int64(length - 4) | uproot.const.kByteCountMask
return copy_cursor.put_fields(self._format_tobjstring, cnt, vers, 1, bit, numpy.uint32(0x03000000)) + buff
def put_th2(self, cursor, name):
copy_cursor = copy(cursor)
cursor.skip(self._format_cntvers.size)
vers = 5
buff = (self.put_th1(cursor, name) + cursor.put_fields(
self._format_th2_1, self.fields["_fScalefactor"],
self.fields["_fTsumwy"], self.fields["_fTsumwy2"],
self.fields["_fTsumwxy"]))
length = len(buff) + self._format_cntvers.size
cnt = numpy.int64(length - 4) | uproot.const.kByteCountMask
return copy_cursor.put_fields(self._format_cntvers, cnt, vers) + buff
def write(self, cursor, sink, isjagged=False):
if isjagged:
raise Exception("isjagged flag should be False")
self.cursor = uproot.write.sink.cursor.Cursor(cursor.index)
self.sink = sink
self.update()
cursor.skip(self._format1.size)
cursor.write_string(sink, self.fClassName)
cursor.write_string(sink, self.fName)
cursor.write_string(sink, self.fTitle)
def put_tattaxis(self, cursor, axis):
copy_cursor = copy(cursor)
cursor.skip(self._format_cntvers.size)
vers = 4
buff = (cursor.put_fields(self._format_tattaxis, axis["_fNdivisions"],
axis["_fAxisColor"], axis["_fLabelColor"],
axis["_fLabelFont"], axis["_fLabelOffset"],
axis["_fLabelSize"], axis["_fTickLength"],
axis["_fTitleOffset"], axis["_fTitleSize"],
axis["_fTitleColor"], axis["_fTitleFont"]))
length = len(buff) + self._format_cntvers.size
cnt = numpy.int64(length - 4) | uproot.const.kByteCountMask
return copy_cursor.put_fields(self._format_cntvers, cnt, vers) + buff
def _put_tlist(self, cursor, values):
copy_cursor = copy(cursor)
cursor.skip(self._format_cntvers.size)
vers = 5
buff = b""
givenbytes = (self._put_tobject(cursor) +
cursor.put_string(b"") +
cursor.put_fields(self._format_tlist1, len(values)))
for value in values:
buff += self.util.put_objany(cursor, (value, "TObjString"), self.keycursor)
buff += cursor.put_fields(self._format_tlist2, 0)
buff += b"" # cursor.bytes(source, n)
givenbytes += buff
length = len(givenbytes) + self._format_cntvers.size
cnt = numpy.int64(length - 4) | uproot.const.kByteCountMask
return copy_cursor.put_fields(self._format_cntvers, cnt, vers) + givenbytes
def put_taxis(self, cursor, axis):
copy_cursor = copy(cursor)
cursor.skip(self._format_cntvers.size)
vers = 10
buff = (
self.put_tnamed(cursor, axis["_fName"], axis["_fTitle"]) +
self.put_tattaxis(cursor, axis) +
cursor.put_fields(self._format_taxis_1, axis["_fNbins"],
axis["_fXmin"], axis["_fXmax"]) +
self.put_tarray(cursor, axis["_fXbins"]) + cursor.put_fields(
self._format_taxis_2, axis["_fFirst"], axis["_fLast"],
axis["_fBits2"], axis["_fTimeDisplay"]) +
cursor.put_string(axis["_fTimeFormat"]) + self.util.put_objany(
cursor, (axis["_fLabels"], "THashList"), self.keycursor) +
self.util.put_objany(cursor,
(axis["_fModLabs"], "TList"), self.keycursor))
length = len(buff) + self._format_cntvers.size
cnt = numpy.int64(length - 4) | uproot.const.kByteCountMask
return copy_cursor.put_fields(self._format_cntvers, cnt, vers) + buff
def write(self, cursor):
copy_cursor = copy(cursor)
cursor.skip(self._format_cntvers.size)
vers = 13
buff = (
self.put_tnamed(cursor,
self.name,
self.nametitle,
hexbytes=numpy.uint32(0x03400000)) +
self.put_tattfill(cursor) + cursor.put_fields(
self._format_tbranch1, self.fields["_fCompress"],
self.fields["_fBasketSize"], self.fields["_fEntryOffsetLen"],
self.fields["_fWriteBasket"], self.fields["_fEntryNumber"]) +
self.put_rootiofeatures(cursor) + cursor.put_fields(
self._format_tbranch2, self.fields["_fOffset"],
self.fields["_fMaxBaskets"], self.fields["_fSplitLevel"],
self.fields["_fEntries"], self.fields["_fFirstEntry"],
self.fields["_fTotBytes"], self.fields["_fZipBytes"]) +
self.put_tobjarray(
cursor, self.fields["_fBranches"], classname="TBranch") +
self.put_tobjarray(cursor,
self.fields["_fLeaves"][0],
classname=self.fields["_fLeaves"][1]) +
#self.put_tobjarray(cursor, self.fields["fBaskets"]) +
b"@\x00\x00\x15\x00\x03\x00\x01\x00\x00\x00\x00\x03\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"
)
cursor.skip(
len(b"@\x00\x00\x15\x00\x03\x00\x01\x00\x00\x00\x00\x03\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"
))
buff += (b"\x01")
cursor.skip(len(b"\x01"))
buff += (cursor.put_array(self.fields["_fBasketBytes"]) + b"\x01")
cursor.skip(len(b"\x01"))
buff += (cursor.put_array(self.fields["_fBasketEntry"]) + b"\x01")
cursor.skip(len(b"\x01"))
buff += (cursor.put_array(self.fields["_fBasketSeek"]) +
cursor.put_string(self.fields["_fFileName"]))
length = len(buff) + self._format_cntvers.size
cnt = numpy.int64(length - 4) | uproot.const.kByteCountMask
return copy_cursor.put_fields(self._format_cntvers, cnt, vers) + buff
def write(self, cursor, sink, isjagged=False):
self.cursor = uproot.write.sink.cursor.Cursor(cursor.index)
self.sink = sink
self.update()
cursor.skip(self._format1.size)
cursor.write_string(sink, self.fClassName)
cursor.write_string(sink, self.fName)
cursor.write_string(sink, self.fTitle)
basketversion = 3
if isjagged:
if self.old_fLast == 0:
raise Exception("isjagged flag should be False")
cursor.write_fields(sink, self._format_basketkey, basketversion,
self.fBufferSize, self.fNevBufSize,
self.fNevBuf, self.old_fLast)
else:
cursor.write_fields(sink, self._format_basketkey, basketversion,
self.fBufferSize, self.fNevBufSize,
self.fNevBuf, self.fLast)
self.old_fLast = self.fLast
cursor.write_data(sink, b"\x00")
