tkinter_windows.py 文件源码

def submit(self, chosen, chosen2):
        new_bodies, dims, num, mass_r = [], self.dims, self.num.get(), self.findEntries('mass_r')
        if chosen == "Gradient":
            colors = (self.color1ValTrue, self.color2ValTrue)
            if chosen2 == "Density":
                densities = self.findEntries('densities')
                new_bodies = Gradient(dims, num, mass_r, colors).preset('Density', densities)
            else:
                new_bodies = Gradient(dims, num, mass_r, colors).preset('Diffusion')
        else:
            dist_r, density = self.findEntries('dist_r'), self.density.get()
            if chosen2 == "Cluster":
                new_bodies = System(dims, num, mass_r, dist_r, density).preset('Cluster')
            else:
                star_density, star_mass = self.star_density.get(), self.star_mass.get()
                if chosen2 == "Binary":
                    star_mass2 = self.star_mass2.get()
                    new_bodies = System(dims, num, mass_r, dist_r, density).preset('Binary', (star_mass, star_mass2),
                                                                                   star_density)
                else:
                    circular = self.circular.get()
                    new_bodies = System(dims, num, mass_r, dist_r, density).preset('Unary', star_mass, star_density,
                                                                                   circular)
        if not self.bodies or messagebox.askokcancel("Discard Changes", "Are you sure you want to discard changes?"):
            self.bodies[:] = new_bodies
            for window in self.parent.properties_windows:
                window.destroy()
            self.parent.properties_windows = []
            self.parent.name.set("Unnamed Simulation")
            self.parent.filename = ''