wip

Readme.md

@@ -1,21 +1,27 @@
 EPICS simulator
 ---------------
+```
 This provides very dummy images and motor(to be done) records
 
 cd /home/zamofing_t/Documents/prj/SwissFEL/epics_ioc_modules/ESB_MX/EpicsSim/iocBoot/iocSwissMxSim
 ./st.cmd
+```
 
 simulate camera (with EPICS simulator)
 --------------------------------------
+```
 cd /home/zamofing_t/Documents/prj/SwissFEL/epics_ioc_modules/ESB_MX/python/SwissMX
 EPICS_CA_ADDR_LIST=localhost
 ./simCam.py 
+```
 
 test camera display
 -------------------
+```
 cd /home/zamofing_t/Documents/prj/SwissFEL/epics_ioc_modules/ESB_MX/python/SwissMX
 EPICS_CA_ADDR_LIST=localhost
 ./camera.py -u -b SwissMxSim
+```
 
 
 https://git.psi.ch/SwissMX/swissmx_cristallina/-/wikis/Instructions%20on%20how%20to%20use%20software

camera.py

@@ -27,16 +27,16 @@ logger = logging.getLogger(__name__)
 global imv
 
 
-class camera_server(object):
+class camera(object):
 
   def __init__(self, basename="ESB-MX-CAM"):
     self._pv=pv=dict()
     os.environ['EPICS_CA_ADDR_LIST'] = 'localhost'
     if basename[-1]!=':': basename+=':'
-    self._transformations = []
     pv['pic']=epics.PV(basename+"FPICTURE", auto_monitor=True, callback=self.new_frame_callback)
     for k,v in (('w','WIDTH'),('h','HEIGHT'),):
       pv[k]=epics.PV(basename+v)
+    self._sz=(int(pv['w'].value), int(pv['h'].value))
     return
 
   def new_frame_callback(self, **kwargs):
@@ -71,36 +71,34 @@ class camera_server(object):
     print('new_frame_callback')
     try:
       pv=self._pv
-      w, h = int(pv['w'].value), int(pv['h'].value)
+      sz=self._sz
-      pic = pv['pic'].get(count=w * h, as_numpy=True).reshape((h, w))
+      pic = pv['pic'].get(count=sz[0]*sz[1], as_numpy=True).reshape(sz[::-1])
+    except AttributeError as e:
+      logger.warning("failed to fetch image")
+    else:
       if pic.dtype==np.int16:
         pic.dtype=np.uint16
+      try:
+        trf=self._transformation
+      except AttributeError as e:
+        pass
+      else:
+        if trf[1,0]==0:
+         pic=pic[::trf[0,0],::trf[1,1]]
+        else:
+         pic=pic[::trf[0,1],::trf[1,0]].T
+
       self.pic=pic
       print(pic[-1][-1])
-      imv.setImage(pic)
+      imv.setImage(pic, autoRange=False, autoLevels=False)
-    except:
-        logger.warning("failed to fetch image")
 
   def get_image(self):
     pv=self._pv
     try:
       pic = self.pic
-    except:
+    except AttributeError:
       self.new_frame_callback()
       pic = self.pic
-    for t in self._transformations:
-        if Transforms.ROTATE_90 == t:
-            pic = np.rot90(pic, 1)
-        elif Transforms.ROTATE_180 == t:
-            pic = np.rot90(pic, 2)
-        elif Transforms.ROTATE_270 == t:
-            pic = np.rot90(pic, 3)
-        elif Transforms.FLIP_LR == t:
-            pic = np.fliplr(pic)
-        elif Transforms.FLIP_UD == t:
-            pic = np.flipud(pic)
-        elif Transforms.TRANSPOSE == t:
-            pic = pic.transpose()
     return pic
 
 
@@ -118,7 +116,7 @@ if __name__ == "__main__":
 
   print('STARTING')
   os.environ['EPICS_CA_ADDR_LIST']='localhost'
-  cam = camera_server(basename=args.base_pv)
+  cam = camera(basename=args.base_pv)
   #pv = cam._pv
   #w, h = int(pv['w'].value), int(pv['h'].value)
   #a = np.arange(w*h, dtype=np.uint16)

simCam.py

@@ -10,25 +10,173 @@ import time
 
 import numpy as np
 
-print('STARTING')
+class SimCam:
-os.environ['EPICS_CA_ADDR_LIST'] = 'localhost'
+  def __init__(self):
+    pass
 
-pv=dict()
+  def generate(self,sz=(1500,1000),t=100,mode=0):
-basename='SwissMxSim:'
+    w,h=sz
-#pv['pic'] = epics.PV(basename + "FPICTURE", auto_monitor=True, callback=self.new_frame_callback)
+    self._imgSeq=imgSeq=np.ndarray(shape=(t,h,w),dtype=np.uint16)
-pv['pic'] = epics.PV(basename + "FPICTURE")
+    x = np.linspace(-5, 5, w)
-for k, v in (('w', 'WIDTH'), ('h', 'HEIGHT'),):
+    y = np.linspace(-5, 5, h)
-    pv[k] = epics.PV(basename + v)
+    # full coordinate arrays
+    xx, yy = np.meshgrid(x, y)
 
-pv['w'].put(300)
+    for i in range(t):
-pv['h'].put(400)
+      #imgSeq[i,:,:] = 100*np.sqrt(np.sin(xx+.1*i)**2 + np.sin(yy+.01*i)**2)#+xx*t+yy*t)
-
+      #imgSeq[i,:,:] = 100*np.sqrt(np.sin(xx+.1*i)**2 + np.sin((1+.1*np.sin(.2*i))*yy+.001*i**2)**2)#+xx*t+yy*t)
-w, h = int(pv['w'].value), int(pv['h'].value)
+      #imgSeq[i,:,:] = 100*np.sqrt(np.sin(xx+2*np.sin(i/t*2*np.pi))**2 + np.sin(yy)**2)
+      px=2*np.sin(i/t*2*np.pi)
+      fx=1
+      py=2*np.sin(i/t*2*np.pi)
+      fy=1+.3*np.sin(i/t*2*np.pi*2)
+      imgSeq[i,:,:] = 100*np.sqrt(np.sin(xx*fx+px)**2 + np.sin(yy*fy+py)**2)
+      #np.random.bytes(100)
+    wr=w//4
+    hr=h//4
+    imgSeq[:,0:hr,0:wr]+=np.random.randint(0,100,(t,hr,wr),dtype=np.uint16)
+
+
+
+  def pushEPICS(self,sz=None,t=100,prefix='SwissMxSim:'):
+    if prefix[-1]!=':': prefix+=':'
+    os.environ['EPICS_CA_ADDR_LIST'] = 'localhost'
+    pv = dict()
+    pv['pic'] = epics.PV(prefix + "FPICTURE")
+    for k, v in (('w', 'WIDTH'), ('h', 'HEIGHT'),):
+      pv[k] = epics.PV(prefix + v)
+    if sz:
+      w,h=sz
+      pv['w'].put(w)
+      pv['h'].put(h)
+    else:
+      w, h = int(pv['w'].value), int(pv['h'].value)
+
+    print('generate data...')
+    self.generate((w,h),t)
+    print('done. Now cycle and publish the data to EPICS')
+    imgSeq=self._imgSeq
+    while(True):
+      for i in range(t):
+        a = imgSeq[i,:,:].ravel()
+        pv['pic'].put(a)
+        print(pv['pic'].get())
+        #time.sleep(.01)
+
+if __name__ == "__main__":
+
+  def QtUI(simCam):
+    global img, i, fps, updateTime
+    from pyqtgraph.Qt import QtCore, QtGui
+    import numpy as np
+    import pyqtgraph as pg
+    import pyqtgraph.ptime as ptime
+    import sys
+
+    app = QtGui.QApplication([])
+    ## Create window with GraphicsView widget
+    win = pg.GraphicsLayoutWidget()
+    win.show()  ## show widget alone in its own window
+    win.setWindowTitle('pyqtgraph example: ImageItem')
+    view = win.addViewBox()
+
+    ## lock the aspect ratio so pixels are always square
+    view.setAspectLocked(True)
+
+    ## Create image item
+    img = pg.ImageItem(border='g')
+    view.addItem(img)
+
+    ## Set initial view bounds
+    view.setRange(QtCore.QRectF(0, 0, 600, 600))
+
+    i=0; fps=0; updateTime = ptime.time()
+    def updateData():
+      global img, i, updateTime, fps
+      img.setImage(simCam._imgSeq[i])
+      i = (i + 1) % simCam._imgSeq.shape[0]
+      QtCore.QTimer.singleShot(1, updateData)
+      now = ptime.time()
+      fps2 = 1.0 / (now - updateTime)
+      updateTime = now
+      fps = fps * 0.9 + fps2 * 0.1
+      print("%d %0.1f fps" % (i,fps))
+    updateData()
+
+    ## Start Qt event loop unless running in interactive mode.
+    if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
+      QtGui.QApplication.instance().exec_()
+
+  def QtUI2(simCam):
+    global img, i, fps, updateTime
+    import sys
+    from pyqtgraph.Qt import QtCore, QtGui
+    import pyqtgraph as pg
+    import pyqtgraph.ptime as ptime
+    print(pg.__version__)
+
+    # Interpret image data as row-major instead of col-major
+    pg.setConfigOptions(imageAxisOrder='row-major')
+
+    app = QtGui.QApplication([])
+
+    ## Create window with ImageView widget
+    win = QtGui.QMainWindow()
+    win.resize(800,800)
+    imv = pg.ImageView()
+    win.setCentralWidget(imv)
+    win.show()
+    win.setWindowTitle('pyqtgraph example: ImageView')
+
+    ## Display the data and assign each frame a time value from 1.0 to 3.0
+    i=0; fps=0; updateTime = ptime.time()
+    imv.setImage(simCam._imgSeq[i])
+    def updateData():
+      global img, i, updateTime, fps
+      imv.setImage(simCam._imgSeq[i], autoRange=False, autoLevels=False)
+      i = (i + 1) % simCam._imgSeq.shape[0]
+      QtCore.QTimer.singleShot(1, updateData)
+      now = ptime.time()
+      fps2 = 1.0 / (now - updateTime)
+      updateTime = now
+      fps = fps * 0.9 + fps2 * 0.1
+      print("%d %0.1f fps" % (i,fps))
+    updateData()
+
+    ## Set a custom color map
+    colors = [ (0, 0, 0), (45, 5, 61), (84, 42, 55), (150, 87, 60), (208, 171, 141), (255, 255, 255) ]
+    cmap = pg.ColorMap(pos=np.linspace(0.0, 1.0, 6), color=colors)
+    imv.setColorMap(cmap)
+
+    ## Start Qt event loop unless running in interactive mode.
+    if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
+        QtGui.QApplication.instance().exec_()
+
+
+
+  import argparse
+  parser = argparse.ArgumentParser()
+  parser.add_argument("--ui", "-u", help="qt local speed test", type=int, default=0)
+  parser.add_argument("--prefix","-p",help="EPICS camera-PV prefix",type=str,default="SwissMxSim",)
+  args = parser.parse_args()
+
+  if args.ui:
+    simCam=SimCam()
+    print('generate data...')
+    simCam.generate()
+    print('done')
+    if args.ui==1:
+      QtUI(simCam)
+    else:
+      QtUI2(simCam)
+  else:
+    obj = SimCam()
+    obj.pushEPICS((1500,1000),100,args.prefix)
+# -*- coding: utf-8 -*-
+"""
+Demonstrates very basic use of ImageItem to display image data inside a ViewBox.
+"""
+
+## Add path to library (just for examples; you do not need this)
 
 
-for j in range(100):
-  for i in range(20):
-    a = np.arange(w * h, dtype=np.uint16)//i
-    pv['pic'].put(a)
-    print(pv['pic'].get())
-    time.sleep(.01)