summaryrefslogtreecommitdiff
path: root/cad/src/commands/SelectAtoms/SelectAtoms_Command.py
blob: 81849d0c26c665f2fc87d184b8e8fb41957bdc3e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
# Copyright 2004-2008 Nanorex, Inc.  See LICENSE file for details.
"""
SelectAtoms_Command.py

The 'Command' part of the SelectAtoms Mode (SelectAtoms_basicCommand and
SelectAtoms_basicGraphicsMode are the two split classes of the old
selectAtomsMode)  It provides the command object for its GraphicsMode class.
The Command class defines anything related to the 'command half' of the mode --
For example:
- Anything related to its current Property Manager, its settings or state
- The model operations the command does (unless those are so simple
  that the mouse event bindings in the _GM half can do them directly
  and the code is still clean, *and* no command-half subclass needs
  to override them).

@version: $Id$
@copyright: 2004-2008 Nanorex, Inc.  See LICENSE file for details.


TODO:
- Items mentioned in Select_GraphicsMode.py
- Other items listed in Select_Command.py

History:
Ninad & Bruce 2007-12-13: Created new Command and GraphicsMode classes from
                          the old class selectAtomsMode and moved the
                          Command related methods into this class from
                          selectAtomsMode.py

"""
from utilities import debug_flags
from utilities.debug import print_compact_traceback
from utilities.debug import reload_once_per_event

from commands.Select.Select_Command import Select_Command
from commands.SelectAtoms.SelectAtoms_GraphicsMode import SelectAtoms_GraphicsMode

_superclass = Select_Command
class SelectAtoms_Command(Select_Command):
    """
    SelectAtoms_basicCommand
    The 'Command' part of the SelectAtoms Mode (SelectAtoms_basicCommand and
    SelectAtoms_basicGraphicsMode are the two split classes of the old
    selectAtomsMode)  It provides the command object for its GraphicsMode class.
    The Command class defines anything related to the 'command half' of the
    mode --
    For example:
    - Anything related to its current Property Manager, its settings or state
    - The model operations the command does (unless those are so simple
      that the mouse event bindings in the _GM half can do them directly
      and the code is still clean, *and* no command-half subclass needs
      to override them).
    """
    GraphicsMode_class = SelectAtoms_GraphicsMode

    commandName = 'SELECTATOMS'
    featurename = "Select Atoms Mode"
    from utilities.constants import CL_ABSTRACT
    command_level = CL_ABSTRACT #??

    # Don't highlight singlets in SelectAtoms_Command. Fixes bug 1540.mark 060220.
    highlight_singlets = False

    call_makeMenus_for_each_event = True

    def makeMenus(self):
        selatom, selobj = self.graphicsMode.update_selatom_and_selobj(None)

        self.Menu_spec = []

        # Local minimize [now called Adjust Atoms in history/Undo,
        #Adjust <what> here and in selectMode -- mark & bruce 060705]
        # WARNING: This code is duplicated in depositMode.makeMenus().
        #--mark 060314.
        if selatom is not None and \
           not selatom.is_singlet() and \
           self.w.simSetupAction.isEnabled():

            # see comments in depositMode version
            self.Menu_spec.append((
                "Adjust atom %s" % selatom,
                lambda e1 = None, a = selatom: self.localmin(a, 0) ))

            self.Menu_spec.append((
                "Adjust 1 layer",
                lambda e1 = None, a = selatom: self.localmin(a, 1) ))

            self.Menu_spec.append((
                "Adjust 2 layers",
                lambda e1 = None, a = selatom: self.localmin(a, 2) ))

        # selobj-specific menu items. [revised by bruce 060405;
        #for more info see the same code in depositMode]
        if selobj is not None and hasattr(selobj, 'make_selobj_cmenu_items'):
            try:
                selobj.make_selobj_cmenu_items(self.Menu_spec)
            except:
                print_compact_traceback("bug: exception (ignored) in "
                                        "make_selobj_cmenu_items "
                                        "for %r: " % selobj)

        # separator and other mode menu items.
        if self.Menu_spec:
            self.Menu_spec.append(None)

        # Enable/Disable Jig Selection.
        # This is duplicated in depositMode.makeMenus() and
        # SelectChunks_Command.makeMenus().
        if self.o.jigSelectionEnabled:
            self.Menu_spec.extend( [("Enable Jig Selection",
                                     self.graphicsMode.toggleJigSelection,
                                     'checked')])
        else:
            self.Menu_spec.extend( [("Enable Jig Selection",
                                     self.graphicsMode.toggleJigSelection,
                                     'unchecked')])

        self.Menu_spec.extend( [
            # mark 060303. added the following:
            None,
            ("Edit Color Scheme...", self.w.colorSchemeCommand),
        ])

        return # from makeMenus

    # Local minimize [now called Adjust Atoms in history/Undo, Adjust <what>
    #in menu commands -- mark & bruce 060705]
    def localmin(self, atom, nlayers): #bruce 051207 #e might generalize to
                                       #take a list or pair of atoms, other
                                       #options
        if debug_flags.atom_debug:
            print "debug: reloading sim_commandruns on each use, for development"\
                  "[localmin %s, %d]" % (atom, nlayers)
            import simulation.sim_commandruns as sim_commandruns
            reload_once_per_event(sim_commandruns) #bruce 060705 revised this
        if 1:
            # this does not work,
            #I don't know why, should fix sometime: [bruce 060705]
            self.set_cmdname("Adjust Atoms") # for Undo (should we be more
                                             #specific, like the menu text was?
                                             #why didn't that get used?)
        from simulation.sim_commandruns import LocalMinimize_function # should not be toplevel
        LocalMinimize_function( [atom], nlayers )
        return

    def drag_selected_atom(self, a, delta, computeBaggage = False):
        """
        Delegates this to self's GraphicsMode

        @param computeBaggage: If this is true, the baggage and non-baggage of
        the atom to be dragged will be computed in this method before dragging
        the atom. Otherwise  it assumes that the baggage and non-baggage atoms
        are up-to-date and are computed elsewhere , for example in 'atomSetUp'
        See a comment in the method that illustrates an example use.
        @type recompueBaggage: boolean

        @see: SelectAtoms_GraphicsMode.drag_selected_atom()
        """
        self.graphicsMode.drag_selected_atom(a, delta,
                                             computeBaggage = computeBaggage)