def return_th1(self, cursor, name):
cnt = numpy.int64(self.length_th1(name) -
4) | uproot.const.kByteCountMask
vers = 8
if len(self.fields["_fBuffer"]) != 0:
raise NotImplementedError
return (cursor.return_fields(self._format_cntvers, cnt, vers) +
self.return_tnamed(cursor, name, self.fTitle) +
self.return_tattline(cursor) + self.return_tattfill(cursor) +
self.return_tattmarker(cursor) + cursor.return_fields(
self._format_th1_1, self.fields["_fNcells"]) +
self.return_taxis(cursor, self.fXaxis) +
self.return_taxis(cursor, self.fYaxis) +
self.return_taxis(cursor, self.fZaxis) + cursor.return_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.return_tarray(cursor, self.fields["_fContour"]) +
self.return_tarray(cursor, self.fields["_fSumw2"]) +
cursor.return_string(self.fields["_fOption"]) +
self.return_tlist(cursor, self.fields["_fFunctions"]) +
cursor.return_fields(
self._format_th1_3,
self.fields["_fBufferSize"],
0, # FIXME: empty fBuffer
self.fields["_fBinStatErrOpt"],
self.fields["_fStatOverflows"]))
def return_tattmarker(self, cursor):
cnt = numpy.int64(self.length_tattmarker() -
4) | uproot.const.kByteCountMask
vers = 2
return (cursor.return_fields(self._format_cntvers, cnt, vers) +
cursor.return_fields(
self._format_tattmarker, self.fields["_fMarkerColor"],
self.fields["_fMarkerStyle"], self.fields["_fMarkerSize"]))
def return_tattline(self, cursor):
cnt = numpy.int64(self.length_tattline() -
4) | uproot.const.kByteCountMask
vers = 2
return (cursor.return_fields(self._format_cntvers, cnt, vers) +
cursor.return_fields(
self._format_tattline, self.fields["_fLineColor"],
self.fields["_fLineStyle"], self.fields["_fLineWidth"]))
def return_tattfill(self, cursor):
cnt = numpy.int64(self.length_tattfill() -
4) | uproot.const.kByteCountMask
vers = 2
return (cursor.return_fields(self._format_cntvers, cnt, vers) +
cursor.return_fields(self._format_tattfill,
self.fields["_fFillColor"],
self.fields["_fFillStyle"]))
def return_th2(self, cursor, name):
cnt = numpy.int64(self.length_th2(name) -
4) | uproot.const.kByteCountMask
vers = 4
return (cursor.return_fields(self._format_cntvers, cnt, vers) +
self.return_th1(cursor, name) + cursor.return_fields(
self._format_th2_1, self.fields["_fScalefactor"],
self.fields["_fTsumwy"], self.fields["_fTsumwy2"],
self.fields["_fTsumwxy"]))
def return_tlist(self, cursor, values):
cnt = numpy.int64(self.length_tlist(values) -
4) | uproot.const.kByteCountMask
vers = 5
for value in values:
raise NotImplementedError
return (cursor.return_fields(self._format_cntvers, cnt, vers) +
self.return_tobject(cursor) + cursor.return_string(b"") +
cursor.return_fields(self._format_tlist, len(values)))
def return_th3(self, cursor, name):
cnt = numpy.int64(self.length_th3(name) -
4) | uproot.const.kByteCountMask
vers = 5
return (cursor.return_fields(self._format_cntvers, cnt, vers) +
self.return_th1(cursor, name) +
self.return_tatt3d(cursor) + cursor.return_fields(
self._format_th3_1, self.fields["_fTsumwy"],
self.fields["_fTsumwy2"], self.fields["_fTsumwxy"],
self.fields["_fTsumwz"], self.fields["_fTsumwz2"],
self.fields["_fTsumwxz"], self.fields["_fTsumwyz"]))
def return_tattaxis(self, cursor, axis):
cnt = numpy.int64(self.length_tattaxis() -
4) | uproot.const.kByteCountMask
vers = 4
return (
cursor.return_fields(self._format_cntvers, cnt, vers) +
cursor.return_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"]))
def write(self, context, cursor, name, compression, key, keycursor):
givenbytes = 0
cnt = numpy.int64(self.length(name) - 4) | uproot.const.kByteCountMask
if "TH1" in self.fClassName.decode("utf-8"):
vers = 2
givenbytes = cursor.return_fields(self._format_cntvers, cnt,
vers) + self.return_th1(
cursor, name)
givenbytes += self.return_tarray(cursor, self.valuesarray)
elif "TH2" in self.fClassName.decode("utf-8"):
vers = 3
givenbytes = cursor.return_fields(self._format_cntvers, cnt,
vers) + self.return_th2(
cursor, name)
givenbytes += self.return_tarray(cursor, self.valuesarray)
elif "TH3" in self.fClassName.decode("utf-8"):
vers = 3
givenbytes = cursor.return_fields(self._format_cntvers, cnt,
vers) + self.return_th3(
cursor, name)
givenbytes += self.return_tarray(cursor, self.valuesarray)
elif "TProfile" == self.fClassName.decode("utf-8"):
vers = 6
givenbytes = (
cursor.return_fields(self._format_cntvers, cnt, vers) +
self.return_th1d(cursor, name) +
self.return_tarray(cursor, self.fields["_fBinEntries"]) +
cursor.return_fields(
self._format_tprofile, self.fields["_fErrorMode"],
self.fields["_fYmin"], self.fields["_fYmax"],
self.fields["_fTsumwy"], self.fields["_fTsumwy2"]) +
self.return_tarray(cursor, self.fields["_fBinSumw2"]))
elif "TProfile2D" == self.fClassName.decode("utf-8"):
vers = 7
givenbytes = (
cursor.return_fields(self._format_cntvers, cnt, vers) +
self.return_th2d(cursor, name) +
self.return_tarray(cursor, self.fields["_fBinEntries"]) +
cursor.return_fields(
self._format_tprofile, self.fields["_fErrorMode"],
self.fields["_fZmin"], self.fields["_fZmax"],
self.fields["_fTsumwz"], self.fields["_fTsumwz2"]) +
self.return_tarray(cursor, self.fields["_fBinSumw2"]))
elif "TProfile3D" == self.fClassName.decode("utf-8"):
vers = 7
givenbytes = (
cursor.return_fields(self._format_cntvers, cnt, vers) +
self.return_th3d(cursor, name) +
self.return_tarray(cursor, self.fields["_fBinEntries"]) +
cursor.return_fields(
self._format_tprofile, self.fields["_fErrorMode"],
self.fields["_fTmin"], self.fields["_fTmax"],
self.fields["_fTsumwt"], self.fields["_fTsumwt2"]) +
self.return_tarray(cursor, self.fields["_fBinSumw2"]))
uproot.write.compress.write(context, cursor, givenbytes, compression,
key, keycursor)
def return_taxis(self, cursor, axis):
cnt = numpy.int64(self.length_taxis(axis) -
4) | uproot.const.kByteCountMask
vers = 10
if axis["_fFirst"] != 0 or axis["_fLast"] != 0 or axis[
"_fBits2"] != 0 or axis[
"_fTimeDisplay"] or axis["_fTimeFormat"] != b"" or axis[
"_fLabels"] or axis["_fModLabs"]:
raise NotImplementedError
return (cursor.return_fields(self._format_cntvers, cnt, vers) +
self.return_tnamed(cursor, axis["_fName"], axis["_fTitle"]) +
self.return_tattaxis(cursor, axis) +
cursor.return_fields(self._format_taxis_1, axis["_fNbins"],
axis["_fXmin"], axis["_fXmax"]) +
self.return_tarray(cursor, axis["_fXbins"]) + (b"\x00" * 20))
def return_th3d(self, cursor, name):
cnt = numpy.int64(self.length_th3d(name) -
4) | uproot.const.kByteCountMask
vers = 3
return (cursor.return_fields(self._format_cntvers, cnt, vers) +
self.return_th3(cursor, name) +
self.return_tarray(cursor, self.valuesarray))
def return_tnamed(self, cursor, name, title):
cnt = numpy.int64(self.length_tnamed(name, title) -
4) | uproot.const.kByteCountMask
vers = 1
return (cursor.return_fields(self._format_cntvers, cnt, vers) +
self.return_tobject(cursor) + cursor.return_string(name) +
cursor.return_string(title))
def write(self, context, cursor, name, compression, key, keycursor):
givenbytes = 0
cnt = numpy.int64(self.length(name) - 4) | uproot.const.kByteCountMask
if "TH1" in self.fClassName.decode("utf-8"):
vers = 2
givenbytes = cursor.return_fields(self._format_cntvers, cnt,
vers) + self.return_th1(
cursor, name)
elif "TH2" in self.fClassName.decode("utf-8"):
vers = 3
givenbytes = cursor.return_fields(self._format_cntvers, cnt,
vers) + self.return_th2(
cursor, name)
elif "TH3" in self.fClassName.decode("utf-8"):
vers = 3
givenbytes = cursor.return_fields(self._format_cntvers, cnt,
vers) + self.return_th3(
cursor, name)
givenbytes += self.return_tarray(cursor, self.valuesarray)
uproot.write.compress.write(context, cursor, givenbytes, compression,
key, keycursor)
def write(self, context, cursor, name, compression, key, keycursor):
cnt = numpy.int64(self.length(name) - 4) | uproot.const.kByteCountMask
vers = 1
givenbytes = cursor.return_fields(self._format, cnt, vers, 1, 0, uproot.const.kNotDeleted) + cursor.return_string(self.value)
uproot.write.compress.write(context, cursor, givenbytes, compression, key, keycursor)
def return_tobject(self, cursor):
return cursor.return_fields(self._format_tobject1, 1, 0,
uproot.const.kNotDeleted)
def return_tarray(self, cursor, values):
return cursor.return_fields(self._format_tarray,
values.size) + cursor.return_array(values)
def return_tatt3d(self, cursor):
cnt = numpy.int64(self.length_tatt3d() -
4) | uproot.const.kByteCountMask
vers = 1
return cursor.return_fields(self._format_cntvers, cnt, vers)
