예제 #1
0
파일: hbook.py 프로젝트: alexhsamuel/pyhep
    def _make1DHistogram(self, hist, title, rz_id):
        assert hist.dimensions == 1
        axis = hist.axis
        num_bins = axis.number_of_bins
        range_lo, range_hi = map(float, axis.range)

        # Create the HBOOK histogram.
        ext.hbook1(rz_id, title, num_bins, range_lo, range_hi, 0)
        # Activate storage of per-bin errors for this histogam.
        ext.hbarx(rz_id)
        # Construct a C array containing the bin contents as floats.
        bin_contents = [ hist.getBinContent(n)
                         for n in xrange(0, num_bins) ]
        pack_array = array.array("f", bin_contents)
        # Stuff the bin contents into the HBOOK histogram.
        ext.hpak(rz_id, pack_array.buffer_info()[0])
        # Construct a C array containing the bin errors as floats.
        bin_errors = \
            [ max(*hist.getBinError(bin)) for bin in xrange(num_bins) ]
        pack_array = array.array("f", bin_errors)
        # Stuff the bin errors into the HBOOK histogram.
        ext.hpake(rz_id, pack_array.buffer_info()[0])
        # Set the underflow and overflow contents.
        # FIXME: This is hackish, but I can't find any other way in the API
        # to do this.
        ext.hf1(rz_id, range_lo - 1, hist.getBinContent("underflow"))
        ext.hf1(rz_id, range_hi + 1, hist.getBinContent("overflow"))
        # Set the number of entries.
        ext.hfnoent(rz_id, hist.number_of_samples)
예제 #2
0
파일: hbook.py 프로젝트: alexhsamuel/pyhep
    def _make2DHistogram(self, hist, title, rz_id):
        assert hist.dimensions == 2

        x_axis, y_axis = hist.axes
        num_x_bins = x_axis.number_of_bins
        num_y_bins = y_axis.number_of_bins
        x_min, x_max = map(float, x_axis.range)
        y_min, y_max = map(float, y_axis.range)

        # Create the HBOOK histogram.
        ext.hbook2(rz_id, title, num_x_bins, x_min, x_max,
                   num_y_bins, y_min, y_max, 0)
        # Activate storage of per-bin errors for this histogam.
        ext.hbarx(rz_id)

        # Construct a C array containing the bin contents as floats.
        pack_array = array.array("f")
        for y in xrange(0, num_y_bins):
            for x in xrange(0, num_x_bins):
                pack_array.append(hist.getBinContent((x, y)))
        # Fill the bin contents.
        ext.hpak(rz_id, pack_array.buffer_info()[0])
        # Now fill the C array with bin errors.
        for y in xrange(0, num_y_bins):
            for x in xrange(0, num_x_bins):
                pack_array[y * num_x_bins + x] = max(*hist.getBinError((x, y)))
        # Fill the bin errors.
        ext.hpake(rz_id, pack_array.buffer_info()[0])

        # Set the underflow and overflow bins.  This is kind of hackish;
        # using 'hfcxy' here depends on our having called 'hpak' or 'hpake'
        # directly before.
        for x in xrange(0, num_x_bins):
            ext.hfcxy(x + 1, 0, hist.getBinContent((x, "underflow")))
            ext.hfcxy(x + 1, num_y_bins + 1, hist.getBinContent((x, "overflow")))
        for y in xrange(0, num_y_bins):
            ext.hfcxy(0, y + 1, hist.getBinContent(("underflow", y)))
            ext.hfcxy(num_x_bins + 1, y + 1, hist.getBinContent(("overflow", y)))
        ext.hfcxy(0, 0,
                  hist.getBinContent(("underflow", "underflow")))
        ext.hfcxy(num_x_bins + 1, 0,
                  hist.getBinContent(("overflow", "underflow")))
        ext.hfcxy(0, num_y_bins + 1,
                  hist.getBinContent(("underflow", "overflow")))
        ext.hfcxy(num_x_bins + 1, num_y_bins + 1,
                  hist.getBinContent(("overflow", "overflow")))
        # FIXME: Set underflow and overflow errors.  (Is this supported?)
        # Set the number of entries.
        ext.hfnoent(rz_id, hist.number_of_samples)