worldview_insert_tool.cxx

   1 //  $Id: worldview_insert_tool.cxx,v 1.13 2003/07/28 09:23:46 grumbel Exp $
   2 //
   3 //  Construo - A wire-frame construction gamee
   4 //  Copyright (C) 2002 Ingo Ruhnke <grumbel@gmx.de>
   5 //
   6 //  This program is free software; you can redistribute it and/or
   7 //  modify it under the terms of the GNU General Public License
   8 //  as published by the Free Software Foundation; either version 2
   9 //  of the License, or (at your option) any later version.
  10 //
  11 //  This program is distributed in the hope that it will be useful,
  12 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 //  GNU General Public License for more details.
  15 //
  16 //  You should have received a copy of the GNU General Public License
  17 //  along with this program; if not, write to the Free Software
  18 //  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  19
  20 #include <float.h>
  21 #include "colors.hxx"
  22 #include "particle_factory.hxx"
  23 #include "input_context.hxx"
  24 #include "graphic_context.hxx"
  25 #include "world.hxx"
  26 #include "controller.hxx"
  27 #include "worldview_component.hxx"
  28 #include "worldview_insert_tool.hxx"
  29 #include "world_gui_manager.hxx"
  30
  31 WorldViewInsertTool::WorldViewInsertTool ()
  32 {
  33   current_particle = 0;
  34   particle_mass = 0.1f;
  35 }
  36
  37 WorldViewInsertTool::~WorldViewInsertTool ()
  38 {
  39 }
  40
  41 void
  42 WorldViewInsertTool::draw_background (ZoomGraphicContext* gc)
  43 {
  44   float x = WorldViewComponent::instance()->get_gc()->screen_to_world_x (input_context->get_mouse_x ());
  45   float y = WorldViewComponent::instance()->get_gc()->screen_to_world_y (input_context->get_mouse_y ());
  46
  47   World& world = *Controller::instance()->get_world();
  48
  49   Particle* selected_particle = world.get_particle (x, y);
  50   if (selected_particle)
  51     {
  52       selected_particle->draw_highlight(gc);
  53     }
  54 }
  55
  56 void
  57 WorldViewInsertTool::draw_foreground (ZoomGraphicContext* gc)
  58 {
  59   World& world = *Controller::instance()->get_world();
  60
  61   Vector2d click_pos = WorldViewComponent::instance()->get_gc()->screen_to_world (input_context->get_mouse_pos ());
  62
  63   Vector2d new_particle_pos;
  64
  65   if (WorldViewComponent::instance()->uses_grid())
  66     new_particle_pos = Vector2d(Math::round_to(click_pos.x, 10),
  67                                 Math::round_to(click_pos.y, 10));
  68   else
  69     new_particle_pos = Vector2d(click_pos.x, click_pos.y);
  70
  71   Spring* selected_spring = world.get_spring (click_pos.x, click_pos.y);
  72
  73   if (selected_spring)
  74     {
  75       selected_spring->draw_highlight (gc);
  76     }
  77
  78   if (current_particle)
  79     {
  80       gc->GraphicContext::draw_line (current_particle->pos, new_particle_pos,
  81                                      Colors::new_spring, 2);
  82     }
  83
  84   float x = WorldViewComponent::instance()->get_gc()->screen_to_world_x(input_context->get_mouse_x ());
  85   float y = WorldViewComponent::instance()->get_gc()->screen_to_world_y(input_context->get_mouse_y ());
  86
  87   Particle* selected_particle = world.get_particle (x, y);
  88   if (selected_particle)
  89     {
  90       selected_particle->draw_infos (gc);
  91     }
  92   else
  93     {
  94       gc->draw_fill_circle(new_particle_pos.x,
  95                            new_particle_pos.y,
  96                            3.0f,
  97                            Colors::highlight);
  98     }
  99 }
 100
 101 void
 102 WorldViewInsertTool::on_primary_button_press (int screen_x, int screen_y)
 103 {
 104   World& world = *Controller::instance()->get_world ();
 105   float x = WorldViewComponent::instance()->get_gc()->screen_to_world_x (screen_x);
 106   float y = WorldViewComponent::instance()->get_gc()->screen_to_world_y (screen_y);
 107
 108   if (current_particle)
 109     {
 110       // We are going to place a second particle and convert the last an
 111       // the new one with a spring
 112       Particle* new_current_particle = world.get_particle (x, y);
 113       if (new_current_particle != current_particle)
 114         {
 115           if (new_current_particle) // connect to particles
 116             {
 117               world.add_spring (current_particle, new_current_particle);
 118             }
 119           else // add a new particle and connect it with the current one
 120             {
 121               Vector2d new_particle_pos;
 122               if (WorldViewComponent::instance()->uses_grid())
 123                 new_particle_pos = Vector2d(Math::round_to(x, 10),
 124                                             Math::round_to(y, 10));
 125               else
 126                 new_particle_pos = Vector2d(x, y);
 127
 128               new_current_particle = world.get_particle_mgr()->add_particle(new_particle_pos,
 129                                                                             Vector2d(),
 130                                                                             particle_mass);
 131               world.add_spring (current_particle, new_current_particle);
 132             }
 133           // Lower the spring links count, since we have increased it
 134           // at insertion and now connected it to a real spring, so
 135           // its no longer needed
 136           current_particle->spring_links -= 1;
 137
 138           current_particle = 0;
 139         }
 140       WorldGUIManager::instance()->ungrab_mouse (WorldViewComponent::instance());
 141     }
 142   else
 143     {
 144       // We are going to create a new particle and making it the
 145       // current one, so that the next click would result in a new
 146       // spring
 147       current_particle = world.get_particle (x, y);
 148
 149       if (current_particle)
 150         {
 151           // We have clicked on a particle and made it  the current
 152         }
 153       else
 154         {
 155           Vector2d new_particle_pos;
 156
 157           if (WorldViewComponent::instance()->uses_grid())
 158             new_particle_pos = Vector2d(Math::round_to(x, 10),
 159                                         Math::round_to(y, 10));
 160           else
 161             new_particle_pos = Vector2d(x, y);
 162
 163           Particle* p = world.get_particle_mgr()->add_particle(new_particle_pos,
 164                                                                Vector2d(),
 165                                                                particle_mass);
 166           current_particle = p;
 167           // Increase the spring count so that the particle isn't cleaned up
 168           current_particle->spring_links += 1;
 169           WorldGUIManager::instance()->grab_mouse (WorldViewComponent::instance());
 170         }
 171     }
 172 }
 173
 174 void
 175 WorldViewInsertTool::on_primary_button_release (int x, int y)
 176 {
 177 }
 178
 179 void
 180 WorldViewInsertTool::on_secondary_button_press (int screen_x, int screen_y)
 181 {
 182   on_delete_press (screen_x, screen_y);
 183 }
 184
 185 void
 186 WorldViewInsertTool::on_secondary_button_release (int screen_x, int screen_y)
 187 {
 188 }
 189
 190 void
 191 WorldViewInsertTool::on_delete_press (int screen_x, int screen_y)
 192 {
 193   World& world = *Controller::instance()->get_world ();
 194
 195   float x = WorldViewComponent::instance()->get_gc()->screen_to_world_x (screen_x);
 196   float y = WorldViewComponent::instance()->get_gc()->screen_to_world_y (screen_y);
 197
 198   if (current_particle)
 199     { // We are currently creating a new spring, abort that
 200       current_particle->spring_links -= 1;
 201       current_particle = 0;
 202       WorldGUIManager::instance()->ungrab_mouse (WorldViewComponent::instance());
 203     }
 204   else
 205     {
 206       Spring*   spring   = world.get_spring (x, y);
 207       Particle* particle = world.get_particle (x, y);
 208
 209       if (particle)
 210         {
 211           Controller::instance()->push_undo();
 212           world.remove_particle (particle);
 213         }
 214       else if (spring)
 215         {
 216           Controller::instance()->push_undo();
 217           world.remove_spring(spring);
 218         }
 219     }
 220 }
 221
 222 void
 223 WorldViewInsertTool::on_fix_press (int screen_x, int screen_y)
 224 {
 225   float x = WorldViewComponent::instance()->get_gc()->screen_to_world_x (screen_x);
 226   float y = WorldViewComponent::instance()->get_gc()->screen_to_world_y (screen_y);
 227
 228   Particle* particle = Controller::instance()->get_world ()->get_particle (x, y);
 229   if (particle)
 230     {
 231       particle->set_fixed (!particle->get_fixed ());
 232     }
 233 }
 234
 235 /* EOF */