Next/Previous button, and checkbox / bool like button? #942

NikDrummond · 2023-10-04T17:04:22Z

NikDrummond
Oct 4, 2023

I am currently trying to create a plot where I can cycle through a set of meshs iteratively, showing each one at a time with a previous/next button.

As this is for screening a number of meshs, the idea would be to be able to also have a button each time which can be pressed to say, for example, this mesh is good (or bad) then move to the next one, in order to screen the quality of a large number of objects by eye.

I've looked through the button examples, but haven't been able to figure this out!

Is this something that would be possible?

Thank you in advance!

Answered by marcomusy

Oct 5, 2023

Hi, I made some modifications to Button class, try

pip install -U git+https://github.com/marcomusy/vedo.git

then

from vedo import *


def scroll_left(obj, ename):
    global index
    i = (index - 1) % len(meshes)
    txt.text(meshes[i].name).c("k")
    plt.remove(meshes[index]).add(meshes[i])
    plt.reset_camera()
    index = i

def scroll_right(obj, ename):
    global index
    i = (index + 1) % len(meshes)
    txt.text(meshes[i].name).c("k")
    plt.remove(meshes[index]).add(meshes[i])
    plt.reset_camera()
    index = i

def flag(obj, ename):
    global index
    txt.text("Flag Button Pressed!").c("r")
    plt.reset_camera()


# load some meshes
m1 = Mesh(dataurl + "bunny.obj").c("…

View full answer

marcomusy · 2023-10-04T20:03:45Z

marcomusy
Oct 4, 2023
Maintainer

Hi, you can try something like this:

from vedo import *


def func(evt):
    global index
    if evt.actor and evt.actor.name == "ok_button":
        if evt.actor not in goods: 
            i = (index+1) % len(meshes)
            printc("GOOD! adding", meshes[i-1].name, c='g')
            goods.append(meshes[i-1])
            txt.text("that was good!").c('g')
            plt.remove(meshes[i-1]).add(meshes[i])
            plt.reset_camera()
            index += 1
    
    if evt.actor and evt.actor.name == "ko_button":
        if evt.actor not in bads:
            i = (index+1) % len(meshes)
            printc("BAD!  adding", meshes[i-1].name, c='r')
            bads.append(meshes[i-1])
            txt.text("that was bad!").c('r')
            plt.remove(meshes[i-1]).add(meshes[i])
            plt.reset_camera()
            index += 1

# load some meshes
m1 = Mesh(dataurl+'bunny.obj').normalize().c('green5')
m2 = Mesh(dataurl+'apple.ply').normalize().c('red5')
m3 = Mesh(dataurl+'beethoven.ply').normalize().c('blue5')

m1.name = "a bunny"
m2.name = "an apple"
m3.name = "mr. beethoven"

index = 0  # init global index
meshes = [m1, m2, m3]

goods, bads = [], []

plt = Plotter()

bu = plt.add_button(
    func,
    pos=(0.2, 0.05),  # x,y fraction from bottom left corner
    states=["Good"],  # text for each state
    c=["w"],          # font color for each state
    bc=["dg"],        # background color for each state
    font="Courier",   # font type
    size=40,          # font size
    bold=True,        # bold font
    name="ok_button", # button name
)

bu = plt.add_button(
    func,
    pos=(0.8, 0.05),  # x,y fraction from bottom left corner
    states=["Bad"],   # text for each state
    c=["w"],          # font color for each state
    bc=["dr"],        # background color for each state
    font="Courier",   # font type
    size=40,          # font size
    bold=True,        # bold font
    name="ko_button", # button name
)

txt = Text2D(meshes[0].name, font="Courier", s=1.5, c='k')
plt += txt
plt.show(meshes[0]).close()

print("List of good ones", goods)

Note that I just found a small bug.. (the callback function is called twice for some reason) so I add a tag to remind myself to look into it..

5 replies

NikDrummond Oct 5, 2023
Author

Hi, when I try to run this i get the following error:

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
Cell In[1], line 42
     38 goods, bads = [], []
     40 plt = Plotter()
---> 42 bu = plt.add_button(
     43     func,
     44     pos=(0.2, 0.05),  # x,y fraction from bottom left corner
     45     states=["Good"],  # text for each state
     46     c=["w"],          # font color for each state
     47     bc=["dg"],        # background color for each state
     48     font="Courier",   # font type
     49     size=40,          # font size
     50     bold=True,        # bold font
     51     name="ok_button", # button name
     52 )
     54 bu = plt.add_button(
     55     func,
     56     pos=(0.8, 0.05),  # x,y fraction from bottom left corner
   (...)
     63     name="ko_button", # button name
     64 )
     66 txt = Text2D(meshes[0].name, font="Courier", s=1.5, c='k')

TypeError: Plotter.add_button() got an unexpected keyword argument 'name'

vedo is version 2023.4.6

marcomusy Oct 5, 2023
Maintainer

Try

pip install -U git+https://github.com/marcomusy/vedo.git

NikDrummond Oct 5, 2023
Author

Ah thanks, got this working - Just a note though, installing the dev branch from github didn't work with python 3.10, so I made a new environment with 3.9 and it works fine.

NikDrummond Oct 5, 2023
Author

Hi -

So I modified the code you sent to just cycle back and fourth through meshes like so:

def func(evt):
    # adds 'index' to global namespace
    global index
    # if 'next' pressed 
    if evt.actor and evt.actor.name == "next_button":
        # not 100% on this
        i = (index+1) % len(meshes)
        # mesh name
        txt.text(meshes[i].name).c('k')
        # remove the current mesh, add the next one
        plt.remove(meshes[index]).add(meshes[i])
        # reset the camera
        plt.reset_camera()
        # increase the index 
        index = i
    # if previous button pressed
    if evt.actor and evt.actor.name == "prev_button":
        i = (index-1) % len(meshes)
        # mesh name
        txt.text(meshes[i].name).c('k')
        plt.remove(meshes[index]).add(meshes[i])
        plt.reset_camera()
        index = i

# load some meshes
m1 = vd.Mesh(vd.dataurl+'bunny.obj').normalize().c('green5')
m2 = vd.Mesh(vd.dataurl+'apple.ply').normalize().c('red5')
m3 = vd.Mesh(vd.dataurl+'beethoven.ply').normalize().c('blue5')

m1.name = "a bunny"
m2.name = "an apple"
m3.name = "mr. beethoven"

index = 0  # init global index
meshes = [m1, m2, m3]

plt = vd.Plotter()

bu = plt.add_button(
    func,
    pos=(0.8, 0.05),  # x,y fraction from bottom left corner
    states=[">"],  # text for each state
    c=["w"],          # font color for each state
    bc=[[127,127,127]],        # background color for each state
    font="Courier",   # font type
    size=40,          # font size
    bold=True,        # bold font
    name="next_button", # button name
)

bu = plt.add_button(
    func,
    pos=(0.2, 0.05),  # x,y fraction from bottom left corner
    states=["<"],   # text for each state
    c=["w"],          # font color for each state
    bc=[[127,127,127]],        # background color for each state
    font="Courier",   # font type
    size=40,          # font size
    bold=True,        # bold font
    name="prev_button", # button name
)

txt = vd.Text2D(meshes[index].name, pos = 'top-center',font="Courier", s=1.5, c='k')
plt += txt
plt.show(meshes[0]).close()

But then I wanted to add a third button. In this example I just add one which changes the text to "Flag button pressed" -

# function to check event
def func(evt):
    # adds 'index' to global namespace
    global index

    # if 'next' pressed 
    if evt.actor and evt.actor.name == "next_button":
        # not 100% on this
        i = (index+1) % len(meshes)
        # mesh name
        txt.text(meshes[i].name).c('k')
        # remove the current mesh, add the next one
        plt.remove(meshes[index]).add(meshes[i])
        # reset the camera
        plt.reset_camera()
        # increase the index 
        index = i
    # if previous button pressed
    if evt.actor and evt.actor.name == "prev_button":
        i = (index-1) % len(meshes)
        txt.text(meshes[i].name).c('k')
        plt.remove(meshes[index]).add(meshes[i])
        plt.reset_camera()
        index = i

    # if flag button pressed    
    if evt.actor and evt.actor.name == "flag":

        txt.text('Flag Button Pressed!').c('k')
        plt.reset_camera()

# load some meshes
m1 = vd.Mesh(vd.dataurl+'bunny.obj').normalize().c('green5')
m2 = vd.Mesh(vd.dataurl+'apple.ply').normalize().c('red5')
m3 = vd.Mesh(vd.dataurl+'beethoven.ply').normalize().c('blue5')

m1.name = "a bunny"
m2.name = "an apple"
m3.name = "mr. beethoven"

index = 0  # init global index
meshes = [m1, m2, m3]

plt = vd.Plotter()

bu = plt.add_button(
    func,
    pos=(0.8, 0.05),  # x,y fraction from bottom left corner
    states=[">"],  # text for each state
    c=["w"],          # font color for each state
    bc=[[127,127,127]],        # background color for each state
    font="Courier",   # font type
    size=40,          # font size
    bold=True,        # bold font
    name="next_button", # button name
)

bu = plt.add_button(
    func,
    pos=(0.2, 0.05),  # x,y fraction from bottom left corner
    states=["<"],   # text for each state
    c=["w"],          # font color for each state
    bc=[[127,127,127]],        # background color for each state
    font="Courier",   # font type
    size=40,          # font size
    bold=True,        # bold font
    name="prev_button", # button name
)

bu = plt.add_button(
    func,
    pos = (0.5,0.05),
    states = ['Flag'],
    c = ["w"],
    bc = ["r"],
    font = "Courier",
    size = 40,
    bold = True,
    name = 'flag',  
)


txt = vd.Text2D(meshes[0].name, font="Courier",  pos = 'top-center', s=1.5, c='k')
plt += txt
plt.show(meshes[0]).close()

And I notice It doesn't seem to work with more than 2 buttons? Is there a reason for this? If I comment out the code for any one button above, the remaining two will work, but with 2 it stops.

Thanks again!

NikDrummond Oct 5, 2023
Author

Oh - sorry to spam!

I am not sure why this is the case, but having multiple functions called by button presses fixes this?

def scroll(evt):
    # adds 'index' to global namespace
    global index

    # if 'next' pressed 
    if evt.actor and evt.actor.name == "next_button":
        # not 100% on this
        i = (index+1) % len(meshes)
        # mesh name
        txt.text(meshes[i].name).c('k')
        # remove the current mesh, add the next one
        plt.remove(meshes[index]).add(meshes[i])
        # reset the camera
        plt.reset_camera()
        # increase the index 
        index = i
    # if previous button pressed
    if evt.actor and evt.actor.name == "prev_button":
        i = (index-1) % len(meshes)
        txt.text(meshes[i].name).c('k')
        plt.remove(meshes[index]).add(meshes[i])
        plt.reset_camera()
        index = i

def flag(evt):
    # if flag button pressed    
    if evt.actor and evt.actor.name == "flag":
        txt.text('Flag Button Pressed!').c('k')
        plt.reset_camera()

# load some meshes
m1 = vd.Mesh(vd.dataurl+'bunny.obj').normalize().c('green5')
m2 = vd.Mesh(vd.dataurl+'apple.ply').normalize().c('red5')
m3 = vd.Mesh(vd.dataurl+'beethoven.ply').normalize().c('blue5')

m1.name = "a bunny"
m2.name = "an apple"
m3.name = "mr. beethoven"

index = 0  # init global index
meshes = [m1, m2, m3]

plt = vd.Plotter()

bu = plt.add_button(
    scroll,
    pos=(0.8, 0.05),  # x,y fraction from bottom left corner
    states=[">"],  # text for each state
    c=["w"],          # font color for each state
    bc=[[127,127,127]],        # background color for each state
    font="Courier",   # font type
    size=40,          # font size
    bold=True,        # bold font
    name="next_button", # button name
)

bu = plt.add_button(
    scroll,
    pos=(0.2, 0.05),  # x,y fraction from bottom left corner
    states=["<"],   # text for each state
    c=["w"],          # font color for each state
    bc=[[127,127,127]],        # background color for each state
    font="Courier",   # font type
    size=40,          # font size
    bold=True,        # bold font
    name="prev_button", # button name
)

bu = plt.add_button(
    flag,
    pos = (0.5,0.05),
    states = ['Flag'],
    c = ["w"],
    bc = ["r"],
    font = "Courier",
    size = 40,
    bold = True,
    name = 'flag',  
)


txt = vd.Text2D(meshes[0].name, font="Courier",  pos = 'top-center', s=1.5, c='k')
plt += txt
plt.show(meshes[0]).close()

why is this the case?

marcomusy · 2023-10-05T19:10:34Z

marcomusy
Oct 5, 2023
Maintainer

Hi, I made some modifications to Button class, try

pip install -U git+https://github.com/marcomusy/vedo.git

then

from vedo import *


def scroll_left(obj, ename):
    global index
    i = (index - 1) % len(meshes)
    txt.text(meshes[i].name).c("k")
    plt.remove(meshes[index]).add(meshes[i])
    plt.reset_camera()
    index = i

def scroll_right(obj, ename):
    global index
    i = (index + 1) % len(meshes)
    txt.text(meshes[i].name).c("k")
    plt.remove(meshes[index]).add(meshes[i])
    plt.reset_camera()
    index = i

def flag(obj, ename):
    global index
    txt.text("Flag Button Pressed!").c("r")
    plt.reset_camera()


# load some meshes
m1 = Mesh(dataurl + "bunny.obj").c("green5")
m2 = Mesh(dataurl + "apple.ply").c("red5")
m3 = Mesh(dataurl + "beethoven.ply").c("blue5")
m1.name = "a bunny"
m2.name = "an apple"
m3.name = "mr. beethoven"

meshes = [m1, m2, m3]
txt = Text2D(meshes[0].name, font="Courier", pos="top-center", s=1.5)

plt = Plotter()

bu = plt.add_button(
    scroll_right,
    pos=(0.8, 0.06),  # x,y fraction from bottom left corner
    states=[">"],     # text for each state
    c=["w"],          # font color for each state
    bc=["k5"],        # background color for each state
    size=40,          # font size
)
bu = plt.add_button(
    scroll_left,
    pos=(0.2, 0.06),  # x,y fraction from bottom left corner
    states=["<"],     # text for each state
    c=["w"],          # font color for each state
    bc=["k5"],        # background color for each state
    size=40,          # font size
)
bu = plt.add_button(
    flag,
    pos=(0.5, 0.06),
    states=["Flag"],
    c=["w"],
    bc=["r"],
    size=40,
)

index = 0  # init global index
plt += txt
plt.show(meshes[0]).close()

1 reply

NikDrummond Oct 6, 2023
Author

Thanks for this! Works a charm!

Just a little more info - I'm building tools to manipulate/view/quantify neuron morphology from EM reconstructions like these: https://flywire.ai/

Vedo is brilliant for handling the various meshes and tree graphs involved, with a bunch of other bits and pieces for analysis!

Thanks again for the help :)

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Next/Previous button, and checkbox / bool like button? #942

{{title}}

Replies: 2 comments 6 replies

{{title}}

{{editor}}'s edit

{{editor}}'s edit

{{title}}

{{editor}}'s edit

{{editor}}'s edit

{{title}}

{{title}}

{{title}}

{{title}}

{{editor}}'s edit

{{editor}}'s edit

{{title}}

{{title}}

Select a reply

Next/Previous button, and checkbox / bool like button? #942

NikDrummond Oct 4, 2023

Replies: 2 comments · 6 replies

marcomusy Oct 4, 2023 Maintainer

NikDrummond Oct 5, 2023 Author

marcomusy Oct 5, 2023 Maintainer

NikDrummond Oct 5, 2023 Author

NikDrummond Oct 5, 2023 Author

NikDrummond Oct 5, 2023 Author

marcomusy Oct 5, 2023 Maintainer

NikDrummond Oct 6, 2023 Author

NikDrummond
Oct 4, 2023

Replies: 2 comments 6 replies

marcomusy
Oct 4, 2023
Maintainer

NikDrummond Oct 5, 2023
Author

marcomusy Oct 5, 2023
Maintainer

NikDrummond Oct 5, 2023
Author

NikDrummond Oct 5, 2023
Author

NikDrummond Oct 5, 2023
Author

marcomusy
Oct 5, 2023
Maintainer

NikDrummond Oct 6, 2023
Author