source: trunk/GSASIIElemGUI.py @ 529

"""ElementGUI: class defn. for element GUIs
   Copyright: 2008, Robert B. Von Dreele & Brian H. Toby (Argonne National Laboratory)
"""
########### SVN repository information ###################
# $Date: 2012-01-24 14:31:27 -0600 (Tue, 24 Jan 2012) $
# $Author: vondreele & toby $
# $Revision: 456 $
# $URL: https://subversion.xor.aps.anl.gov/pyGSAS/trunk/GSASIIElemGUI.py $
# $Id: GSASIIElemGUI.py 456 2012-01-24 20:31:27Z toby $
########### SVN repository information ###################
import wx
import os
import wx.lib.colourselect as wscs
class PickElement(wx.Dialog):
    "Makes periodic table widget for picking element - caller maintains element list"
    Elem=None
    def _init_ctrls(self, prnt,oneOnly):
        wx.Dialog.__init__(self, id=-1, name='PickElement',
              parent=prnt, pos=wx.DefaultPosition, 
              style=wx.DEFAULT_DIALOG_STYLE, title='Pick Element')
        import ElementTable as ET
        self.butWid = 55
        if 'nt' in os.name:
            self.butWid = 40
        self.SetClientSize(wx.Size(50+18*self.butWid, 250))
        
        i=0
        for E in ET.ElTable:
            if oneOnly:
                color=E[4]
            else:
                color=E[6]
            PickElement.ElButton(self,name=E[0],
               pos=wx.Point(E[1]*self.butWid+25,E[2]*24+24),tip=E[3],color=color,oneOnly=oneOnly)
            i+=1

    def __init__(self, parent,oneOnly=False):
        self._init_ctrls(parent,oneOnly)
        
    def ElButton(self, name, pos, tip, color, oneOnly):
        Black = wx.Colour(0,0,0)
        if oneOnly:
            El = wscs.ColourSelect(label=name[0], parent=self,colour=color,
                pos=pos, size=wx.Size(self.butWid,23), style=wx.RAISED_BORDER)
            El.Bind(wx.EVT_BUTTON, self.OnElButton)
        else:
            El = wx.ComboBox(choices=name, parent=self, pos=pos, size=wx.Size(self.butWid,23),
                style=wx.CB_READONLY, value=name[0])
            El.Bind(wx.EVT_COMBOBOX,self.OnElButton)
        
        El.SetBackgroundColour(color)
        El.SetToolTipString(tip)

    def OnElButton(self, event):
        El = event.GetEventObject().GetValue()
        self.Elem = El
        self.EndModal(wx.ID_OK)        
        
class DeleteElement(wx.Dialog):
    "Delete element from selected set widget"
    def _init_ctrls(self, parent,choice):
        l = len(choice)-1
        wx.Dialog.__init__(self, id=-1, name='Delete', parent=parent, 
              pos=wx.DefaultPosition, size=wx.Size(max(128,64+l*24), 87),
              style=wx.DEFAULT_DIALOG_STYLE, title='Delete Element')
        self.Show(True)
        self.SetAutoLayout(True)
        self.SetHelpText('Select element to delete')
        self.SetWindowVariant(wx.WINDOW_VARIANT_SMALL)

        i = 0
        Elem = []
        for Elem in choice:
            self.ElButton(id=-1,name=Elem,pos=wx.Point(16+i*24, 16))
            i+=1
              
    def __init__(self, parent,choice):
        DeleteElement.El = ' '
        self._init_ctrls(parent,choice)

    def ElButton(self, id, name, pos):
        White = wx.Colour(255, 255, 255)
        El = wscs.ColourSelect(label=name, parent=self, colour = White,
            pos=pos, size=wx.Size(24, 23), style=wx.RAISED_BORDER)
        El.Bind(wx.EVT_BUTTON, self.OnDeleteButton)
    
    def OnDeleteButton(self, event):
        DeleteElement.El=event.GetEventObject().GetLabel()
        self.EndModal(wx.ID_OK)
        
    def GetDeleteElement(self):
        return DeleteElement.El