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()