lib/maracasVisuLib/src/CutModule/interface/CutModelMainPanel.cxx

   1 /*=========================================================================
   2
   3 Program:   wxMaracas
   4 Module:    $RCSfile: CutModelMainPanel.cxx,v $
   5 Language:  C++
   6 Date:      $Date: 2010/02/24 13:56:06 $
   7 Version:   $Revision: 1.6 $
   8
   9 Copyright: (c) 2002, 2003
  10 License:
  11
  12 This software is distributed WITHOUT ANY WARRANTY; without even
  13 the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  14 PURPOSE.  See the above copyright notice for more information.
  15
  16 =========================================================================*/
  17
  18
  19
  20 // EOF - wxMaracasMPR.cxx
  21
  22 #include "CutModelMainPanel.h"
  23 #include "HistogramDialog.h"
  24 #include "creaSystem.h"
  25
  26 #include <wx/colordlg.h>
  27 #include <wx/bmpbuttn.h>
  28
  29 #include <Add.xpm>
  30 #include <Undo.xpm>
  31 #include <Redo.xpm>
  32 #include <OkAll.xpm>
  33
  34
  35
  36 CutModelMainPanel* CutModelMainPanel::instance=NULL;
  37
  38 CutModelMainPanel::CutModelMainPanel( wxWindow* parent, std::string path)
  39 : wxPanel(parent, -1, wxDefaultPosition, wxDefaultSize){
  40
  41         cutmanager=NULL;
  42         _isCheck=false;
  43         _isFirstTime=true;
  44         initialize(path);
  45
  46 }
  47 CutModelMainPanel::~CutModelMainPanel( ){
  48         viewpanels.clear();
  49         delete cutmanager;
  50 }
  51
  52 void CutModelMainPanel::initialize(std::string path){
  53         cutmanager = new CutModelManager(path);
  54         _panelid = 0;
  55         _wxauimanager = new wxAuiManager(this);
  56
  57         wxAuiPaneInfo paneinfo;
  58
  59         //RaC 5-02-2010 Add tabs
  60         _notebook = new wxAuiNotebook(this,wxID_ANY,wxDefaultPosition,wxDefaultSize,wxAUI_NB_TOP |wxAUI_NB_TAB_MOVE | wxAUI_NB_SCROLL_BUTTONS);
  61
  62         _notebook->AddPage(new ToolBarCutModel(_notebook),"Cutter",true);
  63         //_wxauimanager->AddPane(new ToolBarCutModel(this),paneinfo.ToolbarPane().Top());
  64         _wxauimanager->AddPane(_notebook,paneinfo.Center());
  65         addPolygonCutterTab();
  66         _notebook->Update();
  67
  68         _wxauimanager->Update();
  69 }
  70
  71 CutModelMainPanel* CutModelMainPanel::getInstance(wxWindow* parent, std::string path){
  72         if(instance==NULL){
  73                 if(parent == NULL){
  74                         parent = new wxFrame();
  75                 }
  76                 instance = new CutModelMainPanel(parent, path);
  77         }
  78         return instance;
  79 }
  80
  81 /*CutModelMainPanel* CutModelMainPanel::getInstance(){
  82 return instance;
  83 }*/
  84
  85
  86 void CutModelMainPanel::setImageData(vtkImageData* img){
  87         try{
  88                 checkInvariant();
  89                 cutmanager->setImageData(img);
  90         }catch(CutModelException e){
  91                 showErrorDialog(e.getCause());
  92         }
  93 }
  94
  95 void CutModelMainPanel::setInteractor(vtkRenderWindowInteractor* interactor){
  96         try{
  97                 checkInvariant();
  98                 cutmanager->setInteractor(interactor);
  99         }catch(CutModelException e){
 100                 showErrorDialog(e.getCause());
 101         }
 102 }
 103
 104 void CutModelMainPanel::setRenderer(vtkRenderer* renderer){
 105         try{
 106                 checkInvariant();
 107                 cutmanager->setRenderer(renderer);
 108         }catch(CutModelException e){
 109                 showErrorDialog(e.getCause());
 110         }
 111 }
 112
 113 void CutModelMainPanel::checkInvariant()throw (CutModelException){
 114         if(cutmanager==NULL){
 115                 throw CutModelException("The manager is not initialize");
 116         }
 117 }
 118
 119 void CutModelMainPanel::showErrorDialog(std::string str){
 120         wxMessageDialog* diag = new wxMessageDialog(this, wxString(str.c_str(),wxConvUTF8 ), wxString(str.c_str(),wxConvUTF8 ), wxICON_ERROR);
 121         diag->ShowModal();
 122 }
 123
 124 void CutModelMainPanel::RemoveActor(int id){
 125         try{
 126                 checkInvariant();
 127
 128                 if(!this->IsBeingDeleted()){
 129                         for(int i = 0; i < viewpanels.size()-1;i++){
 130                                 CutModelView* view = viewpanels[i];
 131                                 if(view->getId()==id){
 132                                         for(int j = i; j < viewpanels.size()-1;j++){
 133                                                 viewpanels[j]=viewpanels[j+1];
 134                                         }
 135                                         i = viewpanels.size();
 136                                 }
 137                         }
 138                         viewpanels.pop_back();
 139                         cutmanager->RemoveActor(id);
 140                 }
 141
 142         }catch(CutModelException e){
 143                 showErrorDialog(e.getCause());
 144         }
 145 }
 146
 147 //RaC
 148 void CutModelMainPanel::onCheckChanged(){
 149
 150         if(_isFirstTime==true){
 151                 _isFirstTime=false;
 152                 cutmanager->InitializePolygonInteractorStyle();
 153         }
 154
 155         if(_isCheck==false)
 156         {
 157                 _isCheck=true;
 158                 //Draw Polygon
 159
 160                 cutmanager->ParallelProjectionOn();
 161                 btnExecutePolygonCut->Enable(false);
 162                 _notebook->GetPage(0)->Enable(false);
 163                 _radioinsideout->Enable(false);
 164                 lblMessagePolygon->SetLabel("Drawing polygon...");
 165         }
 166         else
 167         {
 168                 _isCheck=false;
 169                 //Finish Drawing
 170                 cout<<"Finish Drawing"<<endl;
 171                 cutmanager->ParallelProjectionOff();
 172                 btnExecutePolygonCut->Enable(true);
 173                 _notebook->GetPage(0)->Enable(true);
 174                 _radioinsideout->Enable(true);
 175                 lblMessagePolygon->SetLabel("Contour saved! Click on Execute Cut");
 176         }
 177
 178         cutmanager->UpdatePolygon(_isCheck);
 179
 180 }
 181
 182
 183 //RaC
 184 void CutModelMainPanel::onExecuteCutPolygon(){
 185         //Cuts Polygon
 186         cutmanager->ExecuteCutPolygon(_radioinsideout->GetSelection());
 187         lblMessagePolygon->SetLabel("No contour drawed");
 188
 189 }
 190
 191
 192 void CutModelMainPanel::onAddCutModel(){
 193         try{
 194                 checkInvariant();
 195
 196                 int id = addNewViewPanel();
 197                 cutmanager->onAddCutModel(id, getModelView(id));
 198
 199                 _panelid++;
 200
 201                 ShowCurrentPanel(id);
 202
 203         }catch(CutModelException e){
 204                 showErrorDialog(e.getCause());
 205         }
 206 }
 207 void CutModelMainPanel::ShowCurrentPanel(int id){
 208         for(int i = 0; i < viewpanels.size();i++){
 209                 if(viewpanels[i]->getId()==id){
 210                         _wxauimanager->GetPane(viewpanels[i]).Show(true);
 211                         cutmanager->RefreshActor(id);
 212                 }else{
 213                         _wxauimanager->GetPane(viewpanels[i]).Show(false);
 214                 }
 215         }
 216         _wxauimanager->Update();
 217 }
 218 int CutModelMainPanel::addNewViewPanel()throw( CutModelException){
 219
 220         CutModelView* viewpanel = new CutModelView(this,cutmanager->getImageRange());
 221
 222         wxAuiPaneInfo paneinfo0;
 223         _wxauimanager->AddPane(viewpanel, paneinfo0.DefaultPane().DestroyOnClose().Centre());
 224
 225         viewpanel->setId(_panelid);
 226         viewpanels.push_back(viewpanel);
 227
 228         return viewpanel->getId();
 229
 230
 231 }
 232 CutModelView* CutModelMainPanel::getModelView(int id)throw( CutModelException){
 233
 234         CutModelView* current = NULL;
 235         for(int i = 0; i < viewpanels.size();i++){
 236                 if(viewpanels[i]->getId()==id){
 237                         current = viewpanels[i];
 238                 }
 239         }
 240         if(current == NULL){
 241                 std::string s = "Id not found";
 242                 throw CutModelException(s);
 243         }
 244         return current;
 245 }
 246 void CutModelMainPanel::onUndo(){
 247         try{
 248                 checkInvariant();
 249                 int result = cutmanager->Undo();
 250
 251         }catch(CutModelException e){
 252                 showErrorDialog(e.getCause());
 253         }
 254
 255 }
 256
 257 void CutModelMainPanel::onRedo(){
 258
 259         try{
 260                 checkInvariant();
 261                 int result = cutmanager->Redo();
 262
 263         }catch(CutModelException e){
 264                 showErrorDialog(e.getCause());
 265         }
 266
 267 }
 268
 269 void CutModelMainPanel::changeOpacity(int id,int opacity){
 270         try{
 271                 checkInvariant();
 272                 cutmanager->changeOpacity(id, opacity);
 273
 274         }catch(CutModelException e){
 275                 showErrorDialog(e.getCause());
 276         }
 277 }
 278
 279 void CutModelMainPanel::ShowViewBox(int id,bool check){
 280         try{
 281                 checkInvariant();
 282                 cutmanager->ShowViewBox(id, check);
 283
 284         }catch(CutModelException e){
 285                 showErrorDialog(e.getCause());
 286         }
 287 }
 288 void CutModelMainPanel::ShowPopUpMenu(int id){
 289         showErrorDialog("test");
 290 }
 291 void CutModelMainPanel::changeColor(int id,double r,double g,double b){
 292
 293         try{
 294                 checkInvariant();
 295                 cutmanager->changeColor(id, r, g, b);
 296
 297         }catch(CutModelException e){
 298                 showErrorDialog(e.getCause());
 299         }
 300
 301 }
 302 void CutModelMainPanel::ChangeShape(int id,int selection){
 303         try{
 304                 checkInvariant();
 305
 306                 cutmanager->ChangeShape(id, selection);
 307
 308         }catch(CutModelException e){
 309                 showErrorDialog(e.getCause());
 310         }
 311 }
 312 void CutModelMainPanel::updateActorDirection(int id){
 313         try{
 314                 checkInvariant();
 315
 316                 cutmanager->updateActorDirection(id);
 317
 318         }catch(CutModelException e){
 319                 showErrorDialog(e.getCause());
 320         }
 321 }
 322
 323 void CutModelMainPanel::ExecuteCut(int id, double* range, bool isinside){
 324         try{
 325                 checkInvariant();
 326
 327                 cutmanager->ExecuteCut(id, range, isinside);
 328         }catch(CutModelException e){
 329                 showErrorDialog(e.getCause());
 330         }
 331 }
 332
 333 void CutModelMainPanel::ExecuteAll(){
 334         for(int i = 0; i < viewpanels.size(); i++){
 335                 viewpanels[i]->ExecuteCut();
 336         }
 337
 338 }
 339 vtkImageData* CutModelMainPanel::GetResultImage(){
 340         try{
 341                 checkInvariant();
 342                 return cutmanager->GetResultImage();
 343         }catch(CutModelException e){
 344                 showErrorDialog(e.getCause());
 345         }
 346         return NULL;
 347 }
 348
 349 void CutModelMainPanel::ShowStatistics(int id){
 350
 351         checkInvariant();
 352         HistogramDialog *histo = new HistogramDialog(this, _T("Statistics"));
 353
 354         histo->initializeHistogram(cutmanager->GetResultImage());
 355
 356         histo->ShowModal();
 357
 358         delete histo;
 359 }
 360
 361 void CutModelMainPanel::SaveCutModelData(std::string filename){
 362         cutmanager->SaveCutModelData(filename);
 363 }
 364
 365 void CutModelMainPanel::LoadCutModelData(std::string filename){
 366         cutmanager->LoadCutModelData(filename);
 367 }
 368
 369 void CutModelMainPanel::SetType(int type){
 370         _type = type;
 371 }
 372
 373 int CutModelMainPanel::GetType(){
 374         return _type;
 375 }
 376
 377 void CutModelMainPanel::addPolygonCutterTab(){
 378         wxPanel *panel = new wxPanel(_notebook);
 379         panel->SetAutoLayout(true);
 380         wxBoxSizer *sizer = new wxBoxSizer(wxVERTICAL);
 381         panel->SetSizer(sizer);
 382         sizer->AddSpacer(20);
 383
 384         wxCheckBox *item = new wxCheckBox(panel,10,"Draw Polygon",wxDefaultPosition,wxDefaultSize,0,wxDefaultValidator,"");
 385         item->SetValue(false);
 386         Connect(item->GetId(), wxEVT_COMMAND_CHECKBOX_CLICKED,(wxObjectEventFunction)(&PolygonCutterEventHandlerCutModel::onCheckChanged));
 387         sizer->Add(item, 0, wxALIGN_LEFT);
 388
 389         lblMessagePolygon = new wxStaticText(panel,30,"No contour drawed",wxDefaultPosition,wxDefaultSize,0,"");
 390         sizer->Add(lblMessagePolygon, 0, wxALIGN_LEFT);
 391
 392         wxString choices0[2];
 393         choices0[0] = _T("inside");
 394         choices0[1] = _T("outside");
 395         _radioinsideout = new  wxRadioBox(panel,-1,_T(""), wxDefaultPosition, wxDefaultSize,2,choices0);
 396         sizer->Add(_radioinsideout, wxALIGN_LEFT);
 397
 398         btnExecutePolygonCut = new wxButton(panel,20,"Execute Cut",wxDefaultPosition,wxDefaultSize,0,wxDefaultValidator,"");
 399         Connect(btnExecutePolygonCut->GetId(), wxEVT_COMMAND_BUTTON_CLICKED,(wxObjectEventFunction)(&PolygonCutterEventHandlerCutModel::onExecuteCutPolygon));
 400         sizer->Add(btnExecutePolygonCut, 0, wxALIGN_LEFT);
 401
 402
 403         _notebook->AddPage(panel, "Polygon Cutter", false);
 404
 405 }
 406
 407
 408 void PolygonCutterEventHandlerCutModel::onCheckChanged(wxCommandEvent& event){
 409         CutModelMainPanel::getInstance()->onCheckChanged();
 410 }
 411
 412 void PolygonCutterEventHandlerCutModel::onExecuteCutPolygon(wxCommandEvent& event){
 413         CutModelMainPanel::getInstance()->onExecuteCutPolygon();
 414 }
 415
 416
 417
 418 /**
 419 **
 420 **/
 421 ToolBarCutModel::ToolBarCutModel(wxWindow * parent)
 422 : wxToolBar(parent, -1, wxDefaultPosition, wxDefaultSize)
 423 {
 424         wxBitmap bitmap10(Add_xpm);
 425         this->AddTool(10, wxString(_T("Add")),bitmap10);
 426
 427         this->AddSeparator();
 428
 429         wxBitmap bitmap20(Undo_xpm);
 430         this->AddTool(20, wxString(_T("Undo")),bitmap20);
 431
 432         wxBitmap bitmap30(Redo_xpm);
 433         this->AddTool(30, wxString(_T("Redo")),bitmap30);
 434
 435         this->AddSeparator();
 436
 437         wxBitmap bitmap40(OkAll_xpm);
 438         this->AddTool(40, wxString(_T("Ok All")),bitmap40);
 439
 440         this->Realize();
 441
 442         _evthand = new ToolBarEventHandlerCutModel();
 443         this->SetEventHandler(_evthand);
 444
 445 }
 446
 447 ToolBarCutModel::~ToolBarCutModel(void){
 448 }
 449
 450 ToolBarEventHandlerCutModel::ToolBarEventHandlerCutModel()
 451 : wxEvtHandler(){
 452 }
 453 ToolBarEventHandlerCutModel::~ToolBarEventHandlerCutModel(){
 454 }
 455
 456 void ToolBarEventHandlerCutModel::onAdd(wxCommandEvent& event){
 457         CutModelMainPanel::getInstance()->onAddCutModel();
 458 }
 459
 460 void ToolBarEventHandlerCutModel::onUndo(wxCommandEvent& event){
 461         CutModelMainPanel::getInstance()->onUndo();
 462 }
 463
 464 void ToolBarEventHandlerCutModel::onRedo(wxCommandEvent& event){
 465         CutModelMainPanel::getInstance()->onRedo();
 466 }
 467
 468 void ToolBarEventHandlerCutModel::onExecuteAll(wxCommandEvent& event){
 469         CutModelMainPanel::getInstance()->ExecuteAll();
 470 }
 471
 472
 473 BEGIN_EVENT_TABLE(ToolBarEventHandlerCutModel, wxEvtHandler)
 474         EVT_MENU(10, ToolBarEventHandlerCutModel::onAdd)
 475         EVT_MENU(20, ToolBarEventHandlerCutModel::onUndo)
 476         EVT_MENU(30, ToolBarEventHandlerCutModel::onRedo)
 477         EVT_MENU(40, ToolBarEventHandlerCutModel::onExecuteAll)
 478 END_EVENT_TABLE()
 479