aims_project_windows 0.3.1

data/lib/aims_project/app_controller.rb

@@ -0,0 +1,245 @@
+
+module AimsProject
+class AppController < Wx::App
+  
+    include Wx
+  
+    ID_NEW = 102
+    ID_SAVE_IMAGE = 103  
+    ID_MOVE_CLIP_PLANE = 104
+    ID_SAVE_AS = 105
+
+    ID_INSPECTOR = 201
+
+    ID_DELETE_ATOM = 301
+    
+    @frame = nil
+    @menubar = nil
+    @inspector = nil
+    @statusbar = nil
+
+    # The project
+    attr_accessor :project
+    
+    # Used to synchronize directory in open/save dialogs
+    attr_accessor :working_dir
+    
+    # The root frame
+    attr_accessor :frame
+    
+   # Build the application
+   def on_init
+     
+        self.app_name = "AimsViewer"
+        # Create the frame, toolbar and menubar and define event handlers
+        size = [1000,700]
+        @frame = Frame.new(nil, -1, "AimsViewer", DEFAULT_POSITION, size)
+        @statusbar = @frame.create_status_bar
+
+        # This timer will cause the main thread to pass every 2 ms so that other threads
+        # can get work done.
+        timer = Wx::Timer.new(self, Wx::ID_ANY)
+        evt_timer(timer.id) {Thread.pass}
+        timer.start(2)
+
+        # Initialize the selection
+        @selection = {}
+        
+        # Initialize the inspector
+        @inspector = Inspector.new(self, @frame)
+        
+        # Create the geometry notebook page
+        @geomWindow = GeometryWindow.new(self, @frame)
+        frameSizer = VBoxSizer.new
+        # frameSizer.add_item(@geomWindow, :proportion => 1, :flag => EXPAND)
+        #         @frame.set_sizer(frameSizer)
+        @frame.set_menu_bar(menubar)
+
+        # Display
+        @frame.show        
+   end
+   
+   # Process a menu event
+   def process_menu_event(event)
+     case event.id
+     when ID_INSPECTOR
+       show_inspector
+     when ID_DELETE_ATOM
+       delete_atom
+     when Wx::ID_OPEN
+       open_file
+     when Wx::ID_SAVE
+       save_geometry
+     when ID_NEW
+       new_geometry_file
+     when ID_SAVE_AS
+       save_geometry_as
+     when ID_SAVE_IMAGE
+       save_image
+     when Wx::ID_EXIT
+        exit(0)
+     else
+       event.skip
+     end
+     
+   end
+
+   
+   # Return the menubar.  If it is undefined, then define it and attach the event handler
+   def menubar
+     unless @menubar
+       fileMenu = Menu.new
+       fileMenu.append(ID_NEW, "New Geometry ...")
+       fileMenu.append(Wx::ID_OPEN, "Open ...\tCTRL+o")
+       fileMenu.append(Wx::ID_SAVE, "Save Geometry\tCTRL+s")
+       fileMenu.append(ID_SAVE_AS, "Save Geometry As...")
+       fileMenu.append(ID_SAVE_IMAGE, "Export Image ...")
+       fileMenu.append(Wx::ID_EXIT, "Exit")
+       
+       editMenu = Menu.new
+       editMenu.append(ID_DELETE_ATOM, "Delete Atom\tCTRL+d")
+       
+       toolsMenu = Menu.new       
+       # toolsMenu.append(ID_ROTATE, "rotate", "Rotate", Wx::ITEM_CHECK)
+       # toolsMenu.append(ID_ZOOM, "zoom", "Zoom", Wx::ITEM_CHECK)
+       # toolsMenu.append(ID_PAN, "pan", "Pan", Wx::ITEM_CHECK)
+       # toolsMenu.append(ID_MOVE_CLIP_PLANE, "move cilp plane", "Move", Wx::ITEM_CHECK)
+       
+       viewMenu = Menu.new
+       viewMenu.append(ID_INSPECTOR, "inspector\tCTRL+i")
+       
+       
+       @menubar = MenuBar.new
+       @menubar.append(fileMenu, "File")
+       @menubar.append(editMenu, "Edit")
+       @menubar.append(viewMenu, "Views")
+       @menubar.append(toolsMenu, "Tools")
+       
+       evt_menu @menubar, :process_menu_event
+     end
+     @menubar
+   end
+   
+   def set_status(string)
+     @frame.set_status_text(string)
+   end
+   
+   # Get the inspector
+   def inspector
+     if @inspector.nil?
+       @inspector = Inspector.new(self, @frame)
+     end
+     @inspector
+   end
+   
+   # Show the inspector
+   def show_inspector
+     @geomWindow.show_inspector
+     @inspector.show(true)
+   end
+   
+   # Hide the inspector
+   def hide_inspector
+     @inspector.hide
+   end
+   
+   # Create a new geometry file
+   def new_geometry_file(file = nil)
+     fd = TextEntryDialog.new(@frame, :message => "New Geometry", :caption => "Specify name of geometry:")
+      if Wx::ID_OK == fd.show_modal
+        begin
+          geom_name = fd.get_value
+          geometry = GeometryFile.new("")
+          geometry = geometry.save_as(File.new(File.join(GEOMETRY_DIR, geom_name), "w"))
+          @geomWindow.add_geometry(geometry)
+        rescue Exception => e
+          error_dialog(e)
+        end
+      end
+   end
+   
+   # Display a file dialog and attempt to open and display the file
+   def open_geometry_file(file = nil)
+     begin
+       unless file
+         fd = FileDialog.new(@frame, :message => "Open", :style => FD_OPEN, :default_dir => @working_dir)
+         if ID_OK == fd.show_modal
+           file = fd.get_path
+           @working_dir = fd.get_directory
+         else
+           return
+         end
+       end
+       puts "Opening #{file}"
+       @geomWindow.open_geometry_file(file)
+       @frame.set_title(file)
+         
+       
+     rescue Exception => dang
+       error_dialog(dang)
+     end
+   end
+   alias_method :open_file, :open_geometry_file
+   
+   def save_geometry
+     geometry = @geomWindow.geometry
+     begin
+       if geometry.file
+         geometry.save
+       else
+         save_geometry_as
+       end
+     rescue Exception => e
+       error_dialog(e)
+     end
+   end
+   
+   # Save the geometry
+   def save_geometry_as
+     
+       geometry = @geomWindow.geometry
+       
+       fd = FileDialog.new(@frame, :message => "Save Geometry", :style => FD_SAVE, :default_dir => @working_dir)
+       if Wx::ID_OK == fd.show_modal
+         begin
+           @working_dir = fd.get_directory
+           geom_name = fd.get_path
+           new_geom  = geometry.save_as(geom_name)
+           @geomWindow.show_geometry(new_geom)
+         rescue Exception => e
+           error_dialog(e)
+         end
+       end 
+
+   end
+
+   # Display an error dialog for the exception
+   def error_dialog(exception)
+     message = if exception.is_a? String
+       exception
+     elsif exception.is_a? Exception
+       exception.message + "\n\n" + exception.backtrace[0..2].join("\n")
+     end
+     puts message
+     MessageDialog.new(@frame, message, "Error", Wx::ICON_ERROR).show_modal
+   end
+   
+   def save_image
+
+     begin
+         image = @geomWindow.image
+         fd = FileDialog.new(@frame, :message => "Save Image", :style => FD_SAVE, :default_dir => @working_dir)
+         if Wx::ID_OK == fd.show_modal
+           @working_dir = fd.get_directory
+            puts "Writing #{fd.get_path}"
+            image.mirror(false).save_file(fd.get_path)
+          end
+       
+      rescue Exception => e
+        error_dialog(e)
+      end
+   	end
+
+end
+end
+

data/lib/aims_project/atom.rb

@@ -0,0 +1,95 @@
+#
+#  copyright 2012, Joshua Shapiro
+#  joshua.shapiro@gmail.com
+#
+#    This file is part of AimsProjectManager.
+#
+#    AimsProjectManager is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU General Public License as published by
+#    the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    AimsProjectManager is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU General Public License for more details.
+#
+#
+module Aims
+    
+    # Adds display behavior to atoms.
+    class Atom
+    
+	  @@black = nil
+	  @@white = nil
+      @@blue = nil
+      @@red = nil
+      @@green = nil
+      @@dark = nil
+      @@light = nil
+	  @@yellow = nil
+    @@orange = nil
+    @@graypurple = nil
+  
+      def material
+	    unless @@yellow
+			@@yellow = AimsProject::Material.new(1,1,0,1)
+		end
+		
+	    unless @@black
+		  @@lback = AimsProject::Material.new(0,0,0,1)
+		end
+        
+		unless @@white
+		  @@lback = AimsProject::Material.new(1,1,1,1)
+		end
+        
+		unless @@blue
+          @@blue = AimsProject::Material.new(0,0,1,1)
+        end
+    
+        unless @@red
+          @@red = AimsProject::Material.new(1,0.4,0.4,1)
+        end
+    
+        unless @@green
+          @@green = AimsProject::Material.new(0,1,0,1)
+        end
+    
+        unless @@orange
+          @@orange = AimsProject::Material.new(1, 0.5, 0, 1)
+        end
+        
+        unless @@graypurple
+          @@graypurple = AimsProject::Material.new(0.45, 0.55, 0.55, 1)
+        end
+        
+        unless @@dark
+          @@dark =  AimsProject::Material.new(0.2,0.2,0.2,1)
+        end
+    
+        unless @@light
+          @@light = AimsProject::Material.new(1,1,1,1)
+        end
+    
+        case self.species
+        when /Ga/
+          @@dark
+        when /As/
+          @@light
+        when /In/
+          @@graypurple
+        when /P/
+          @@orange
+		when /Si/
+		  @@yellow
+		when /C/
+		  @@red
+		else 
+		  @@blue
+        end
+      end
+    end
+
+
+end