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/03/05 17:54:39 $
   7 Version:   $Revision: 1.7 $
   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),_T("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(_T("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(_T("Contour saved! Click on Execute Cut"));
 176         }
 177
 178         cutmanager->UpdatePolygon(_isCheck);
 179
 180 }
 181
 182
 183 //RaC
 184 void CutModelMainPanel::onExecuteCutPolygon()
 185 {
 186         //Cuts Polygon
 187         cutmanager->ExecuteCutPolygon(_radioinsideout->GetSelection());
 188         lblMessagePolygon->SetLabel(_T("No contour drawed"));
 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
 208 void CutModelMainPanel::ShowCurrentPanel(int id)
 209 {
 210         int i;
 211         for(i = 0; i < (int) viewpanels.size();i++){
 212                 if(viewpanels[i]->getId()==id){
 213                         _wxauimanager->GetPane(viewpanels[i]).Show(true);
 214                         cutmanager->RefreshActor(id);
 215                 }else{
 216                         _wxauimanager->GetPane(viewpanels[i]).Show(false);
 217                 }
 218         }
 219         _wxauimanager->Update();
 220 }
 221 int CutModelMainPanel::addNewViewPanel()throw( CutModelException){
 222
 223         CutModelView* viewpanel = new CutModelView(this,cutmanager->getImageRange());
 224
 225         wxAuiPaneInfo paneinfo0;
 226         _wxauimanager->AddPane(viewpanel, paneinfo0.DefaultPane().DestroyOnClose().Centre());
 227
 228         viewpanel->setId(_panelid);
 229         viewpanels.push_back(viewpanel);
 230
 231         return viewpanel->getId();
 232
 233
 234 }
 235 CutModelView* CutModelMainPanel::getModelView(int id)throw( CutModelException)
 236 {
 237         CutModelView* current = NULL;
 238         int i;
 239         for(i = 0; i < (int)viewpanels.size();i++)
 240         {
 241                 if(viewpanels[i]->getId()==id)
 242                 {
 243                         current = viewpanels[i];
 244                 }
 245         }
 246         if(current == NULL)
 247         {
 248                 std::string s = "Id not found";
 249                 throw CutModelException(s);
 250         }
 251         return current;
 252 }
 253
 254 void CutModelMainPanel::onUndo()
 255         {
 256         try{
 257                 checkInvariant();
 258                 int result = cutmanager->Undo();
 259
 260         }catch(CutModelException e){
 261                 showErrorDialog(e.getCause());
 262         }
 263
 264 }
 265
 266 void CutModelMainPanel::onRedo(){
 267
 268         try{
 269                 checkInvariant();
 270                 int result = cutmanager->Redo();
 271
 272         }catch(CutModelException e){
 273                 showErrorDialog(e.getCause());
 274         }
 275
 276 }
 277
 278 void CutModelMainPanel::changeOpacity(int id,int opacity){
 279         try{
 280                 checkInvariant();
 281                 cutmanager->changeOpacity(id, opacity);
 282
 283         }catch(CutModelException e){
 284                 showErrorDialog(e.getCause());
 285         }
 286 }
 287
 288 void CutModelMainPanel::ShowViewBox(int id,bool check){
 289         try{
 290                 checkInvariant();
 291                 cutmanager->ShowViewBox(id, check);
 292
 293         }catch(CutModelException e){
 294                 showErrorDialog(e.getCause());
 295         }
 296 }
 297 void CutModelMainPanel::ShowPopUpMenu(int id){
 298         showErrorDialog("test");
 299 }
 300 void CutModelMainPanel::changeColor(int id,double r,double g,double b){
 301
 302         try{
 303                 checkInvariant();
 304                 cutmanager->changeColor(id, r, g, b);
 305
 306         }catch(CutModelException e){
 307                 showErrorDialog(e.getCause());
 308         }
 309
 310 }
 311 void CutModelMainPanel::ChangeShape(int id,int selection){
 312         try{
 313                 checkInvariant();
 314
 315                 cutmanager->ChangeShape(id, selection);
 316
 317         }catch(CutModelException e){
 318                 showErrorDialog(e.getCause());
 319         }
 320 }
 321 void CutModelMainPanel::updateActorDirection(int id){
 322         try{
 323                 checkInvariant();
 324
 325                 cutmanager->updateActorDirection(id);
 326
 327         }catch(CutModelException e){
 328                 showErrorDialog(e.getCause());
 329         }
 330 }
 331
 332 void CutModelMainPanel::ExecuteCut(int id, double* range, bool isinside){
 333         try{
 334                 checkInvariant();
 335
 336                 cutmanager->ExecuteCut(id, range, isinside);
 337         }catch(CutModelException e){
 338                 showErrorDialog(e.getCause());
 339         }
 340 }
 341
 342 void CutModelMainPanel::ExecuteAll(){
 343         int i;
 344         for(i = 0; i < (int)viewpanels.size(); i++){
 345                 viewpanels[i]->ExecuteCut();
 346         }
 347
 348 }
 349 vtkImageData* CutModelMainPanel::GetResultImage(){
 350         try{
 351                 checkInvariant();
 352                 return cutmanager->GetResultImage();
 353         }catch(CutModelException e){
 354                 showErrorDialog(e.getCause());
 355         }
 356         return NULL;
 357 }
 358
 359 void CutModelMainPanel::ShowStatistics(int id){
 360
 361         checkInvariant();
 362         HistogramDialog *histo = new HistogramDialog(this, _T("Statistics"));
 363
 364         histo->initializeHistogram(cutmanager->GetResultImage());
 365
 366         histo->ShowModal();
 367
 368         delete histo;
 369 }
 370
 371 void CutModelMainPanel::SaveCutModelData(std::string filename){
 372         cutmanager->SaveCutModelData(filename);
 373 }
 374
 375 void CutModelMainPanel::LoadCutModelData(std::string filename){
 376         cutmanager->LoadCutModelData(filename);
 377 }
 378
 379 void CutModelMainPanel::SetType(int type){
 380         _type = type;
 381 }
 382
 383 int CutModelMainPanel::GetType(){
 384         return _type;
 385 }
 386
 387 void CutModelMainPanel::addPolygonCutterTab(){
 388         wxPanel *panel = new wxPanel(_notebook);
 389         panel->SetAutoLayout(true);
 390         wxBoxSizer *sizer = new wxBoxSizer(wxVERTICAL);
 391         panel->SetSizer(sizer);
 392         sizer->AddSpacer(20);
 393
 394         wxCheckBox *item = new wxCheckBox(panel,10,_T("Draw Polygon"),wxDefaultPosition,wxDefaultSize,0,wxDefaultValidator,_T("") );
 395         item->SetValue(false);
 396         Connect(item->GetId(), wxEVT_COMMAND_CHECKBOX_CLICKED,(wxObjectEventFunction)(&PolygonCutterEventHandlerCutModel::onCheckChanged));
 397         sizer->Add(item, 0, wxALIGN_LEFT);
 398
 399         lblMessagePolygon = new wxStaticText(panel,30,_T("No contour drawed"),wxDefaultPosition,wxDefaultSize,0,_T("") );
 400         sizer->Add(lblMessagePolygon, 0, wxALIGN_LEFT);
 401
 402         wxString choices0[2];
 403         choices0[0] = _T("inside");
 404         choices0[1] = _T("outside");
 405         _radioinsideout = new  wxRadioBox(panel,-1,_T(""), wxDefaultPosition, wxDefaultSize,2,choices0);
 406         sizer->Add(_radioinsideout, wxALIGN_LEFT);
 407
 408         btnExecutePolygonCut = new wxButton(panel,20,_T("Execute Cut"),wxDefaultPosition,wxDefaultSize,0,wxDefaultValidator,_T("") );
 409         Connect(btnExecutePolygonCut->GetId(), wxEVT_COMMAND_BUTTON_CLICKED,(wxObjectEventFunction)(&PolygonCutterEventHandlerCutModel::onExecuteCutPolygon));
 410         sizer->Add(btnExecutePolygonCut, 0, wxALIGN_LEFT);
 411
 412
 413         _notebook->AddPage(panel, _T("Polygon Cutter"), false);
 414
 415 }
 416
 417
 418 void PolygonCutterEventHandlerCutModel::onCheckChanged(wxCommandEvent& event){
 419         CutModelMainPanel::getInstance()->onCheckChanged();
 420 }
 421
 422 void PolygonCutterEventHandlerCutModel::onExecuteCutPolygon(wxCommandEvent& event){
 423         CutModelMainPanel::getInstance()->onExecuteCutPolygon();
 424 }
 425
 426
 427
 428 /**
 429 **
 430 **/
 431 ToolBarCutModel::ToolBarCutModel(wxWindow * parent)
 432 : wxToolBar(parent, -1, wxDefaultPosition, wxDefaultSize)
 433 {
 434         wxBitmap bitmap10(Add_xpm);
 435         this->AddTool(10, wxString(_T("Add")),bitmap10);
 436
 437         this->AddSeparator();
 438
 439         wxBitmap bitmap20(Undo_xpm);
 440         this->AddTool(20, wxString(_T("Undo")),bitmap20);
 441
 442         wxBitmap bitmap30(Redo_xpm);
 443         this->AddTool(30, wxString(_T("Redo")),bitmap30);
 444
 445         this->AddSeparator();
 446
 447         wxBitmap bitmap40(OkAll_xpm);
 448         this->AddTool(40, wxString(_T("Ok All")),bitmap40);
 449
 450         this->Realize();
 451
 452         _evthand = new ToolBarEventHandlerCutModel();
 453         this->SetEventHandler(_evthand);
 454
 455 }
 456
 457 ToolBarCutModel::~ToolBarCutModel(void){
 458 }
 459
 460 ToolBarEventHandlerCutModel::ToolBarEventHandlerCutModel()
 461 : wxEvtHandler(){
 462 }
 463 ToolBarEventHandlerCutModel::~ToolBarEventHandlerCutModel(){
 464 }
 465
 466 void ToolBarEventHandlerCutModel::onAdd(wxCommandEvent& event){
 467         CutModelMainPanel::getInstance()->onAddCutModel();
 468 }
 469
 470 void ToolBarEventHandlerCutModel::onUndo(wxCommandEvent& event){
 471         CutModelMainPanel::getInstance()->onUndo();
 472 }
 473
 474 void ToolBarEventHandlerCutModel::onRedo(wxCommandEvent& event){
 475         CutModelMainPanel::getInstance()->onRedo();
 476 }
 477
 478 void ToolBarEventHandlerCutModel::onExecuteAll(wxCommandEvent& event){
 479         CutModelMainPanel::getInstance()->ExecuteAll();
 480 }
 481
 482
 483 BEGIN_EVENT_TABLE(ToolBarEventHandlerCutModel, wxEvtHandler)
 484         EVT_MENU(10, ToolBarEventHandlerCutModel::onAdd)
 485         EVT_MENU(20, ToolBarEventHandlerCutModel::onUndo)
 486         EVT_MENU(30, ToolBarEventHandlerCutModel::onRedo)
 487         EVT_MENU(40, ToolBarEventHandlerCutModel::onExecuteAll)
 488 END_EVENT_TABLE()
 489