def f(a=r_pi, b=r_pi, m1=r_mass, w1=r_width, m2=r_mass, w2=r_width):
ar1.value, ai1.value = np.cos(a), np.sin(a)
ar2.value, ai2.value = np.cos(b), np.sin(b)
mass1.value, width1.value = m1, w1
mass2.value, width2.value = m2, w2
dplt = gf.DalitzPlotter(prod, dp)
arr = dplt.make2D()
arr = np.ma.array(arr, mask=arr == 0)
ax.clear()
v = ax.imshow(arr, extent=ext, origin='lower')
def f(a, b, m1, w1, m2, w2):
signal = makeSignalPdf(m12, m13, eventNumber)
#prodpdf=ProdPdf("prodpdf",[signal])
#mass1.value,width1.value=m1,w1
#mass2.value,width2.value=m2,w2
ar1.value = np.cos(a)
ai1.value = np.sin(a)
ar2.value = np.cos(b)
ai2.value = np.sin(b)
mass1.value = m1
width1.value = w1
mass2.value = m2
width2.value = w2
#ar1.value,ai1.value= np.cos(a),np.sin(a)
#ar2.value,ai2.value=np.cos(b),np.sin(b)
#mass1.value,width1.value=m1,w1
#mass2.value,width2.value=m2,w2
#ar1.setValue(np.cos(a))
#ai1.setValue(np.sin(a))
#ar2.setValue(np.cos(b))
#ai2.setValue(np.sin(b))
#mass1.setValue(m1)
#width1.setValue(w1)
#mass2.setValue(m2)
#width2.setValue(w2)
#ar1._goofit.Indexable(np.cos(a))
#ai1._goofit.Indexable(np.sin(a))
#ar2._goofit.Indexable(np.cos(b))
#ai2._goofit.Indexable(np.sin(b))
#mass1._goofit.Indexable(m1)
#width1._goofit.Indexable(w1)
#mass2._goofit.Indexable(m2)
#width2._goofit.Indexable(w2)
#signal =makeSignalPdf(m12,m13,eventNumber)
prodpdf = ProdPdf("prodpdf", [signal])
dplt = gf.DalitzPlotter(prodpdf, signal)
arr = dplt.make2D()
arr = np.ma.array(arr, mask=arr == 0)
ax.clear()
ax.imshow(arr, origin='lower')
ax1.plot(a, '_')
plt.show()
def main():
a = cv2.getTrackbarPos('a', 'Image')
b = cv2.getTrackbarPos('b', 'Image')
m1 = cv2.getTrackbarPos('m1', 'Image')
w1 = cv2.getTrackbarPos('w1', 'Image')
m2 = cv2.getTrackbarPos('m2', 'Image')
w2 = cv2.getTrackbarPos('w2', 'Image')
# cv2.imshow('Image', image)
# cv2.waitKey(0)
ar1.value = np.cos(a * 0.001)
ai1.value = np.sin(a * 0.001)
ar2.value = np.cos(b * 0.001)
ai2.value = np.sin(b * 0.001)
mass1.value = m1 * 0.005
width1.value = w1 * 0.001
mass2.value = m2 * 0.005
width2.value = w2 * 0.001
signal = makeSignalPdf(m12, m13, eventNumber)
prodpdf = ProdPdf("prodpdf", [signal])
dplt = gf.DalitzPlotter(prodpdf, signal)
arr = dplt.make2D()
extent = dplt.getExtent()
arr = np.ma.array(arr, mask=arr == 0)
# ax.clear()
# cv2.imshow('Image', arr)
# cv2.waitKey(0)
# plt.imshow(arr,extent=extent,origin='lower')
# plt.colorbar(cax=None,ax=None,shrink=0.5)
# cv2.imshow('Image', arr)
# cv2.waitKey(0)
image[:] = [b, b, b]
# plt.savefig('RBW_pdf_plot.png')
#plt.show()
cv2.imshow('Image', arr)
cv2.waitKey(0)
def f(r1, i1, m1, w1, r2, i2, m2, w2):
ar1.value = r1
ai1.value = i1
ar2.value = r2
ai2.value = i2
mass1.value = m1
width1.value = w1
mass2.value = m2
width2.value = w2
signal = makeSignalPdf(m12, m13, eventNumber)
prodpdf = ProdPdf("prodpdf", [signal])
dplt = gf.DalitzPlotter(prodpdf, signal)
arr = dplt.make2D()
extent = dplt.getExtent()
arr = np.ma.array(arr, mask=arr == 0)
ax.clear()
plt.imshow(arr, extent=extent, origin='lower')
plt.colorbar(cax=None, ax=None, shrink=0.5)
plt.savefig('RBW_pdf_plot.png')
plt.show()
# -*- coding: utf-8 -*-
from goofit import *
import matplotlib.pyplot as plt
import numpy as np
import goofit as gf
import matplotlib
matplotlib.use('Agg')
from dalitz import makeSignalPdf
fig, ax = plt.subplots()
m12 = Observable("m12", 0, 3.)
m13 = Observable("m13", 0, 3.)
m12.numbins = 240
m13.numbins = 240
eventNumber = EventNumber("eventNumber")
signal = makeSignalPdf(m12, m13, eventNumber)
prodpdf = ProdPdf("prodpdf", [signal])
dplt = gf.DalitzPlotter(prodpdf, signal)
arr = dplt.make2D()
extent = dplt.getExtent()
arr = np.ma.array(arr, mask=arr == 0) #background
ax.clear()
plt.imshow(arr, extent=extent, origin='lower')
plt.colorbar(cax=None, ax=None, shrink=0.5)
plt.savefig("dalitz_pdf_eval.png")
plt.show()