XT_MJG_SpotsResized3.py

#Resize Spots

# Written by Matthew J. Gastinger
# June 2020 - Imaris 9.6.0


#    <CustomTools>
#      <Menu>
#       <Submenu name="Spots Functions">
#        <Item name="Resize Spots" icon="Python3">
#          <Command>Python3XT::XT_MJG_SpotsResized3(%i)</Command>
#        </Item>
#       </Submenu>
#      </Menu>
#      <SurpassTab>
#        <SurpassComponent name="bpSpots">
#          <Item name="Resize Spots" icon="Python3">
#            <Command>Python3XT::XT_MJG_SpotsResized3(%i)</Command>
#          </Item>
#        </SurpassComponent>
#      </SurpassTab>
#    </CustomTools>

#Description:
#Resize all Spots to defined diameter (default if not spot are selected)
#If a only a selected group of spots are selected, only those will be resized.
#New Spots object will be created

#Python libraries - no special libraries are required for this XTension

import ImarisLib
import time
import random

# GUI imports
import tkinter as tk
from tkinter import ttk
from tkinter.ttk import *
from tkinter import *
from tkinter import messagebox
from tkinter import simpledialog
#aImarisId=0
def XT_MJG_SpotsResized3(aImarisId):
    # Create an ImarisLib object
    vImarisLib = ImarisLib.ImarisLib()
    # Get an imaris object with id aImarisId
    vImarisApplication = vImarisLib.GetApplication(aImarisId)
    # Get the factory
    vFactory = vImarisApplication.GetFactory()
    # Get the currently loaded dataset
    vImage = vImarisApplication.GetDataSet()
    # Get the Surpass scene
    vSurpassScene = vImarisApplication.GetSurpassScene()

    ############################################################################
    ############################################################################

    #testing surface masking
    vExtendMin = (vImage.GetExtendMinX(),vImage.GetExtendMinY(),vImage.GetExtendMinZ())
    vExtendMax = (vImage.GetExtendMaxX(),vImage.GetExtendMaxY(),vImage.GetExtendMaxZ())
    vImageSize = (vImage.GetSizeX(),vImage.GetSizeY(),vImage.GetSizeZ())

    # get all spots in selected Surpass object
    vSpots = vFactory.ToSpots(vImarisApplication.GetSurpassSelection())
    vSpotsRadius = vSpots.GetRadii()
    vSpotsPositionXYZ = vSpots.GetPositionsXYZ()
    vSpotsTime = vSpots.GetIndicesT()
    vAllSpotsIds = vSpots.GetIds()
    vNumberOfSpots = len(vSpotsTime)
    vSelectedSpotsIds = vSpots.GetSelectedIds()
    vNewSpots = vImarisApplication.GetFactory().CreateSpots()
    #remove the error of Selecvted SPot ID getting track id as well
    vSelectedSpotsIds=[item for item in vSelectedSpotsIds if item <1000000000 ]

    root = tk.Tk()
    root.eval('tk::PlaceWindow . center')
    root.withdraw()
    root.attributes("-topmost", True)

    ##################################################################
    #Set input in center on screen
    # Gets the requested values of the height and widht.
    windowWidth = root.winfo_reqwidth()
    windowHeight = root.winfo_reqheight()
    # Gets both half the screen width/height and window width/height
    positionRight = int(root.winfo_screenwidth()/2 - windowWidth/2)
    positionDown = int(root.winfo_screenheight()/2 - windowHeight/2)
    # Positions the window in the center of the page.
    root.geometry("+{}+{}".format(positionRight, positionDown))
    ##################################################################

    #Set new Spot size
    if vSpotsRadius!=[]:
        vNewSpotDiameter = simpledialog.askfloat('Set new Spot Size',
                                'NEW Spot Diameter: ', initialvalue=(round(float(vSpotsRadius[0]),2))*2)
        root.destroy()
    else:
        messagebox.showwarning("Resize Spot Fail",
                               "Please generate some Spots!")
        root.destroy()

    if len(vSelectedSpotsIds)==0:#checks if list is empty
        vNewSpotsRadius = [vNewSpotDiameter/2]*vNumberOfSpots
        vNewSpots.Set(vSpotsPositionXYZ, vSpotsTime, vNewSpotsRadius)
        vNewSpots.SetTrackEdges(vSpots.GetTrackEdges())
        vNewSpots.SetName(str(vSpots.GetName()) + ' Resized All')

    else:
        for s in range (len(vSelectedSpotsIds)):
            vCurrentSpotId = vSelectedSpotsIds[s]
            vCurrentIndex=vAllSpotsIds.index(vCurrentSpotId)#Find specific index
            vSpotsRadius[vCurrentIndex]=vNewSpotDiameter/2
        vNewSpots.Set(vSpotsPositionXYZ, vSpotsTime, vSpotsRadius)
        vNewSpots.SetName(str(vSpots.GetName()) + ' Resized Selected')


    #Set Track edges if present
    vNewSpots.SetTrackEdges(vSpots.GetTrackEdges())
    #Create the a new Spots with new size including tracks if they are there
    vRGBA = vSpots.GetColorRGBA()
    vNewSpots.SetColorRGBA(vRGBA)

    vSpots.SetVisible(0)
    vImarisApplication.GetSurpassScene().AddChild(vNewSpots, -1)