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)