Changeset 821

Timestamp:

09/12/07 15:16:15 (12 months ago)

Author:

mankoff

Message:

Initial support for vectors. UI elements and functionality show up if variables exist named U,V, and W

Files:

• misc/clouds/xyztn.pro (modified) (11 diffs)

misc/clouds/xyztn.pro

@@ -76 +76 @@
   xyztnSampleOrthoPlane,vdata,2,z,sState.oSlices[2], $
                         sState.oImages[2],ctab,sState.iAlpha
-  
   ;; Restore the data
   sState.oVols[sState.iImgVol]->SetProperty,DATA0=vdata,/NO_COPY
+
+
+  ;; Vectors?
+  IF sState.vec_show THEN BEGIN
+     sState.oVols[0]->GetProperty,DATA0=bMag,RGB_TABLE0=pal,/NO_COPY
+
+     ;; x,y,z
+     u = (*sState.pdata).u & v = (*sState.pdata).v & w = (*sState.pdata).w
+     xyztnMkVectors,u,v,w,fVerts,iConn, $
+                    SCALE=scale, RANDOM=sState.bRandom, NVECTORS=nvectors,$
+                    STEPSIZE=stepsize,BMAG=bMag,Vc=vc,X=x
+     sState.oVectors[0]->SetProperty,DATA=fVerts,POLYLINES=iConn
+     xyztnMkVectors,u,v,w,fVerts,iConn, $
+                    SCALE=scale, RANDOM=sState.bRandom, NVECTORS=nvectors,$
+                    STEPSIZE=stepsize,BMAG=bMag,Vc=vc,Y=y
+     sState.oVectors[1]->SetProperty,DATA=fVerts,POLYLINES=iConn
+     xyztnMkVectors,u,v,w,fVerts,iConn, $
+                    SCALE=scale, RANDOM=sState.bRandom, NVECTORS=nvectors,$
+                    STEPSIZE=stepsize,BMAG=bMag,Vc=vc,Z=z
+     sState.oVectors[2]->SetProperty,DATA=fVerts,POLYLINES=iConn
+     ;;; x,y,z, end
+     
+     sState.oVols[0]->SetProperty,DATA0=bMag,/NO_COPY
+  ENDIF
+
+
 END
 
@@ -244 +269 @@
 
 
+
+
+
+PRO xyztnMkVectors, u, v, w, fVerts, iConn, X=x, Y=y, Z=z, SCALE=scale,$
+                    RANDOM=random, NVECTORS=nvectors, STEPSIZE=stepsize,$
+                    bMAG=bMag, VC=vc
+  
+  ;; Ensure volumes match in number of elements.
+  nSamples = N_ELEMENTS(u)
+  IF (nSamples NE N_ELEMENTS(v) OR nSamples NE N_ELEMENTS(w)) THEN BEGIN
+     MESSAGE,'Number of elements in u, v, and w must match.'
+     RETURN
+  ENDIF
+  sz = SIZE(u)
+
+  ;; Handle keywords, set defaults.
+  IF (N_ELEMENTS(scale) EQ 0) THEN scale = 1.0
+  IF (N_ELEMENTS(random) EQ 0) THEN random=0
+  IF (N_ELEMENTS(nvectors) EQ 0) THEN nvectors = 100
+  IF (N_ELEMENTS(stepsize) eq 0) then stepsize = 1
+  IF (N_ELEMENTS(bMag) EQ 0) THEN bMag = BYTSCL(SQRT(u^2+v^2+w^2))
+
+  ;; Get plane information.
+  doPlane = 0
+  IF (N_ELEMENTS(x) GT 0) THEN BEGIN
+     IF (random EQ 0) THEN BEGIN
+        nRow = sz[2] / stepsize
+        nCol = sz[3] / stepsize
+     ENDIF
+     doPlane = 1
+  ENDIF
+  IF (N_ELEMENTS(y) GT 0) THEN BEGIN
+     IF (doPlane) THEN MESSAGE,'X, Y, and Z keywords are mutually exclusive'
+     IF (random EQ 0) THEN BEGIN
+        nRow = sz[3] / stepsize
+        nCol = sz[1] / stepsize
+     ENDIF
+     doPlane = 2
+  ENDIF
+  IF (N_ELEMENTS(z) GT 0) THEN BEGIN
+     IF (doPlane) THEN MESSAGE,'X, Y, and Z keywords are mutually exclusive'
+     IF (random EQ 0) THEN BEGIN
+        nRow = sz[2] / stepsize
+        nCol = sz[1] / stepsize
+     ENDIF
+     doPlane = 3
+  ENDIF
+  IF (doPlane EQ 0) THEN MESSAGE, 'Must specify a plane.'
+  
+  ;; Grab max, min values in vector volumes.
+  maxU = MAX(u, MIN=minU)
+  maxV = MAX(v, MIN=minV)
+  maxW = MAX(w, MIN=minW)
+  
+  ;; Compute the magnitude.
+  mag = SQRT((maxU-minU)^2 + (maxV-minV)^2 + (maxW-minW)^2)
+  fNorm = scale / mag
+  
+  nV = nRow*nCol
+  
+  fVerts = FLTARR(3, 2*nV)
+  iConn = LONARR(3*nV)
+  vc = BYTARR(2*nV)
+  
+  CASE doPlane OF
+     1: BEGIN                   ; X=x
+        x0 = REPLICATE(x,nV)
+        y0 = REFORM((REPLICATE(1,nCol) # (LINDGEN(nRow)*stepsize)),nV)
+        z0 = REFORM(((LINDGEN(nCol)*stepsize) # REPLICATE(1,nRow)),nV)
+     END
+     2: BEGIN                   ; Y=y
+        y0 = REPLICATE(y,nRow*nCol)
+        z0 = REFORM((REPLICATE(1,nCol) # (LINDGEN(nRow)*stepsize)),nV)
+        x0 = REFORM(((LINDGEN(nCol)*stepsize) # REPLICATE(1,nRow)),nV)
+     END
+     3: BEGIN                   ; Z=z
+        z0 = REPLICATE(z,nRow*nCol)
+        y0 = REFORM((REPLICATE(1,nCol) # (LINDGEN(nRow)*stepsize)),nV)
+        x0 = REFORM(((LINDGEN(nCol)*stepsize) # REPLICATE(1,nRow)),nV)
+     END
+  ENDCASE
+  
+  inds = LINDGEN(nV)
+  v0 = transpose([[x0],[y0],[z0]])
+  v1 = transpose([[x0+u[x0,y0,z0]*fNorm], $
+                  [y0+v[x0,y0,z0]*fNorm], $
+                  [z0+w[x0,y0,z0]*fNorm]])
+  fVerts[*,inds*2] = v0[*,inds]
+  fVerts[*,inds*2+1] = v1[*,inds]
+  iConn[inds*3] = 2
+  iConn[inds*3+1] = inds*2
+  iConn[inds*3+2] = inds*2+1
+  
+  vc[inds*2] = bMag[x0,y0,z0]
+  vc[inds*2+1] = bMag[x0,y0,z0]
+END
+
+
+
+
 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 ;; Cleanup
@@ -339 +464 @@
      END
 
+     ;; Image variable changed
+     'VEC_SHOW': BEGIN
+        sState.oVectors[0]->SetProperty, HID=sState.vec_show
+        sState.oVectors[1]->SetProperty, HID=sState.vec_show
+        sState.oVectors[2]->SetProperty, HID=sState.vec_show
+        sState.vec_show = 1-sState.vec_show
+        bUpdate = 4
+        bRedraw = 1
+     END
+
      ;; Show/Hide Contour Planes
      'CONT_SHOW': BEGIN
@@ -413 +548 @@
          WIDGET_CONTROL, sState.wXtext, SET_VALUE= $
                          STRtrim(string(slider_value, FORMAT='(F10.3)'), 2)
-        IF (sState.bShow) THEN BEGIN
            bUpdate = 1
            bRedraw = 1
-        ENDIF
      END
      ;; Y plane location
@@ -426 +559 @@
          WIDGET_CONTROL, sState.wYtext, SET_VALUE= $
                          STRtrim(string(slider_value, FORMAT='(F10.3)'), 2)
-        IF (sState.bShow) THEN BEGIN
            bUpdate = 2
            bRedraw = 1
-        ENDIF
      END
      ;; Z plane location
@@ -440 +571 @@
          WIDGET_CONTROL, sState.wZtext, SET_VALUE= $
                          STRtrim(string(slider_value, FORMAT='(F10.3)'), 2)
-        IF (sState.bShow) THEN BEGIN
            bUpdate = 3
            bRedraw = 1
-        ENDIF
     END
 
@@ -578 +707 @@
      END
 
+     ELSE: MESSAGE, "UI Event with no handler", /CONT
+
   ENDCASE
   
@@ -584 +715 @@
      ;; start by hiding everything
      hid = (1-sState.bShow)
-     sState.oXPolyline->SetProperty,HIDE=hid+sState.bImage
-     sState.oYPolyline->SetProperty,HIDE=hid+sState.bImage
-     sState.oZPolyline->SetProperty,HIDE=hid+sState.bImage
      sState.oSlices[0]->SetProperty,HIDE=hid+(1-sState.bImage)
      sState.oSlices[1]->SetProperty,HIDE=hid+(1-sState.bImage)
@@ -690 +818 @@
      XIASTREC(data,'W')THEN BEGIN
      wVecBase = WIDGET_BASE(wFrameBase,/NONEXCLUSIVE)
-     wIVolSel = WIDGET_BUTTON(wVecBase,VALUE="U/V/W Vectors",UVAL='IMG_VOLSEL')
+     wIVolSel = WIDGET_BUTTON(wVecBase,VALUE="U/V/W Vectors",UVAL='VEC_SHOW')
   ENDIF
   
@@ -1069 +1197 @@
             oImages: [oXImage,oYImage,oZImage], $
             oContours: [oXContour, oYContour, oZContour], $
+            oVectors: [oXPolyline, oYPolyline, oZPolyline ], $
+            vec_show: 0B,          $
             iIsoVol: 0,            $
             fIsoLevel: 128,        $
@@ -1074 +1204 @@
             bIsoShow: 0,           $
             bVolShow: 0,           $
-            oXPolyline: oXPolyline,$
-            oYPolyline: oYPolyline,$
-            oZPolyline: oZPolyline,$
             ColorTable: colorTable,$ ; Color table to restore at exit
             debug: keyword_set(debug), $