]> Creatis software - bbtkGEditor.git/blobdiff - lib/EditorGraphicBBS/bbsKernelEditorGraphic/wxVtkSceneManager.cxx
Implemented deleting of boxes from the scene....
[bbtkGEditor.git] / lib / EditorGraphicBBS / bbsKernelEditorGraphic / wxVtkSceneManager.cxx
index daa6327c157cceea4e7cd481750fa60fa7654e82..c4efe5605d986f87da642f9bbcd499ea122c47e3 100644 (file)
@@ -41,32 +41,32 @@ namespace bbtk
 
 
        //=========================================================================
-       wxVtkSceneManager::wxVtkSceneManager(wxDropTarget *parent, wxVtk3DBaseView *baseView,int idManager)
+       wxVtkSceneManager::wxVtkSceneManager(wxGEditorTabPanel *parent, wxVtk3DBaseView *baseView,int idManager)
        {
+               _parent = parent;
                _numBoxes=0;
                _idManager=idManager;
                _baseView=baseView;
                _idLastController=0;
+               _startDragging=false;
 
                if( _baseView!=NULL )
                {
-                       _baseView->GetWxVTKRenderWindowInteractor()->SetDropTarget(parent);
-                       registerController(this);
+                       _baseView->GetWxVTKRenderWindowInteractor()->SetDropTarget((wxDropTarget*)parent);                      
                        configureBaseView();
                        _worldState=NOTHING_HAPPENS;
+                       registerController(this);
                        
                }
 
-               
        }
 
-
        //=========================================================================
 
        void wxVtkSceneManager::disconnectDrop()
        {
                _baseView->GetWxVTKRenderWindowInteractor()->SetDropTarget(NULL);
-       }
+       }
 
        //=========================================================================
 
@@ -98,7 +98,7 @@ namespace bbtk
 
        //=========================================================================
 
-       void wxVtkSceneManager::createGBlackBox(int x, int y, std::string packageName, std::string boxName)
+       void wxVtkSceneManager::createGBlackBox(int x, int y, std::string packageName, std::string boxType)
        {
 
                int windowWidth=_baseView->GetRenWin()->GetSize()[0];
@@ -112,13 +112,13 @@ namespace bbtk
                vtkGObjectView *view = GObjectsMVCFactory::getInstance()->createGObjectView(type);
                GObjectController* controller = GObjectsMVCFactory::getInstance()->createGObjectController(type);
                
-               BlackBoxDescriptor::Pointer descriptor = GObjectsMVCFactory::getInstance()->getBlackBoxDescriptor(packageName, boxName);
+               BlackBoxDescriptor::Pointer descriptor = GObjectsMVCFactory::getInstance()->getBlackBoxDescriptor(packageName, boxType);
                
                //Prepares the initial model
                //The coordinates obtained are the following. Top-Left:x=0,y=0 Bottom-Right:x=width,y=height  
 
                double xx = x;
-               double yy =  windowHeight-y;
+               double yy = windowHeight-y;
                
                //z value is not important yet, because it is only used a parallel projection
                double zz = 900;
@@ -140,7 +140,8 @@ namespace bbtk
                std::string arraystring = stream.str();
 
                model->setBBTKName(arraystring);
-               model->setBBTKType(boxName);
+               model->setBBTKType(boxType);
+               model->setBBTKPackage(packageName);
                                
                model->addObserver(view);
                model->addObserver(this);
@@ -196,19 +197,19 @@ namespace bbtk
 
        void wxVtkSceneManager::createGInputPort(int portType, int posinBox,GBlackBoxModel *blackBox, BlackBoxInputDescriptor *desc)
        {
-               createGPort(portType,posinBox,blackBox);
+               createGPort(portType,desc->GetName(),desc->GetTypeName(),posinBox,blackBox);
        }
 
        //=========================================================================
 
        void wxVtkSceneManager::createGOutputPort(int portType, int posinBox,GBlackBoxModel *blackBox, BlackBoxOutputDescriptor *desc)
        {
-               createGPort(portType,posinBox,blackBox);
+               createGPort(portType,desc->GetName(),desc->GetTypeName(),posinBox,blackBox);
        }
 
        //=========================================================================
 
-       void wxVtkSceneManager::createGPort(int portType, int posInBox,GBlackBoxModel *blackBox)
+       void wxVtkSceneManager::createGPort(int portType,std::string bbtkName, std::string bbtkType, int posInBox,GBlackBoxModel *blackBox)
        {
                int type = GPORT;
 
@@ -220,6 +221,9 @@ namespace bbtk
                model->registerInBox(blackBox,portType, posInBox);
                blackBox->addOutputPort(model);
                
+               model->setBBTKType(bbtkType);
+               model->setBBTKName(bbtkName);
+
                model->addObserver(view);
                model->addObserver(this);
 
@@ -300,6 +304,7 @@ namespace bbtk
                connectorcontroller->setManualContourController(manContourControl);             
                connectorModel->setManualContourModel(manContourModel);
                connectorView->setManualContourView(manContourView);
+               connectorView->setModel(connectorModel);
                connectorcontroller->setModelAndView(connectorModel,connectorView);
 
                int newId = _controllers.size();
@@ -318,6 +323,14 @@ namespace bbtk
 
        //=========================================================================
 
+       void wxVtkSceneManager::unregisterController(InteractorStyleMaracas *param)
+       {
+               vtkInteractorStyleBaseView* baseViewControlManager = (vtkInteractorStyleBaseView*)_baseView->GetInteractorStyleBaseView();
+               baseViewControlManager->RemoveInteractorStyleMaracas( param );
+       }
+
+       //=========================================================================
+
        vtkRenderer* wxVtkSceneManager::getRenderer()
        {
                return _baseView->GetRenderer();
@@ -331,76 +344,421 @@ namespace bbtk
        }
        //=========================================================================
 
-       bool wxVtkSceneManager::OnMouseMove()
-       {
-               return true;
-       }
-
-       //=========================================================================
-       
+               
        void wxVtkSceneManager::update(int idController,int command)
        {
-               
-               if(command == INIT_CREATION_CONTOUR)
-               {       
-                       GObjectController* cont = _controllers[idController];
-                       GPortModel* startOutputPort = (GPortModel*)cont->getModel();
-                       createGConnector(startOutputPort);
-
-                       // The last one is the controller of the connector
-                       for(int i=0;i<_controllers.size()-1;i++)
+               if(command != NO_COMMAND)
+               {
+                       if(command == ADD_TO_SELECTED)
                        {
-                               GObjectController* cont = _controllers[i];
-                               if(cont->getGObjectType() == GPORT )
+                               GObjectController* cont = _controllers[idController];
+
+                               bool foundID=false;
+                               for (int i=0; i<_selectedObjects.size() && foundID==false; i++)
                                {
-                                       GPortModel* port = (GPortModel*)cont->getModel();
-                                       if(port->getPortType()==GINPUTPORT)
+                                       int id = _selectedObjects[i];
+                                       if(id==idController)
                                        {
-                                               cont->SetActive(true);
+                                               foundID = true;
+                                       }
+                               }
+                               if(!foundID)
+                               {
+                                       int id = idController;
+                                       _selectedObjects.push_back(id);
+                               }
+
+                       }
+                       else if(command == INIT_CREATION_CONTOUR)
+                       {       
+                               _worldState = INIT_CREATION_CONTOUR;
+                               GObjectController* cont = _controllers[idController];
+                               GPortModel* startOutputPort = (GPortModel*)cont->getModel();
+                               
+
+                               // The last one is the controller of the connector
+                               for(int i=0;i<_controllers.size();i++)
+                               {
+                                       GObjectController* cont = _controllers[i];
+                                       if(cont->getGObjectType() == GPORT )
+                                       {
+                                               GPortModel* port = (GPortModel*)cont->getModel();
+                                               if(port->getPortType()==GINPUTPORT)
+                                               {
+                                                       cont->SetActive(true);
+                                               }
+                                               else
+                                               {
+                                                       cont->getView()->setState(NOTHING_HAPPENS);
+                                                       cont->getModel()->notifyObservers(_idManager);
+                                                       cont->SetActive(false);
+                                               }
                                        }
                                        else
                                        {
                                                cont->getView()->setState(NOTHING_HAPPENS);
+                                               cont->getModel()->notifyObservers(_idManager);
                                                cont->SetActive(false);
+                                       }                               
+                               }
+
+                               _selectedObjects.clear();
+
+                               createGConnector(startOutputPort);
+
+                       }
+                       else if(command == FIN_CREATION_CONTOUR && _worldState == INIT_CREATION_CONTOUR)
+                       {                               
+                               _worldState = NOTHING_HAPPENS;
+                               int id = _controllers.size()-1;
+                               GObjectController* cont = _controllers[id];                     
+                               GConnectorModel* modelContour = (GConnectorModel*)cont->getModel();
+
+                               GObjectController* finPort = _controllers[idController];
+                               if(finPort->getGObjectType() == GPORT)
+                               {
+                                       GPortModel* modelPort = (GPortModel*)finPort->getModel();
+                                       modelContour->setEndPort(modelPort);
+                               }                       
+
+                               manualConnectorContourController* manCont = ((GConnectorController*)cont)->getManualContourController();                        
+                               manualConnectorContourView* connView = (manualConnectorContourView*)manCont->GetManualViewBaseContour();
+                               connView->Refresh();
+
+                               for(int i=0;i<_controllers.size();i++)
+                               {
+                                       GObjectController* cont = _controllers[i];
+                                       if(cont->getView()!=NULL)
+                                       {
+                                               cont->getView()->setState(NOTHING_HAPPENS);
+                                               cont->getModel()->notifyObservers(_idManager);
                                        }
+                                       cont->SetActive(true);                                                          
                                }
-                               else
+                       }
+               }
+       }
+
+       //=========================================================================
+
+       bool wxVtkSceneManager::OnMouseMove()
+       {
+               int X,Y;
+               wxVTKRenderWindowInteractor *wxVTKiren;
+               wxVTKiren = _vtkInteractorStyleBaseView->GetWxVtk3DBaseView()->GetWxVTKRenderWindowInteractor();
+               wxVTKiren->GetEventPosition(X,Y);
+       
+               
+               if(_worldState == DRAG_OBJECTS)
+               {                       
+                       for (int i=0; i<_selectedObjects.size(); i++)
+                       {
+                               int id = _selectedObjects[i];
+                               GObjectController* cont = _controllers[id];
+                               if(_startDragging)
+                               {
+                                       cont->getView()->setStartDragging(true);
+                               }
+                               cont->getView()->moveObject(X,Y);
+                               cont->getView()->setState(DRAG);
+                               cont->getModel()->notifyObservers(_idManager);
+                       }
+
+                       std::map<int, GObjectController*>::iterator it;
+
+                       for(it = _controllers.begin(); it != _controllers.end(); ++it)
+                       {
+                               GObjectController *desc = it->second;
+                               if(desc->getGObjectType()==GCONNECTOR)
                                {
-                                       cont->getView()->setState(NOTHING_HAPPENS);
-                                       cont->SetActive(false);
-                               }                               
+                                       GConnectorView* vconn = (GConnectorView*)desc->getView();
+                                       vconn->updateStartEndPoints();
+                               }
                        }
-                       _worldState = CREATING_CONTOUR;
+                       
+                       _startDragging=false;
+
                }
-               else if(command == FIN_CREATION_CONTOUR && _worldState == CREATING_CONTOUR)
+               else if(_worldState == NOTHING_HAPPENS || _worldState == INIT_CREATION_CONTOUR)
+               {
+                       std::map<int, GObjectController*>::iterator it;
+
+                       for(it = _controllers.begin(); it != _controllers.end(); ++it)
+                       {
+                               GObjectController *desc = it->second;
+                               int type = desc->getGObjectType();
+                               int state = desc->getView()->getState();
+                               
+                               if(state == HIGHLIGHTED){
+
+                                       updateStatusBar(desc->getStatusText());
+                                       if(type==GBLACKBOX)
+                                       {
+                                               GBlackBoxModel *mod = (GBlackBoxModel*)desc->getModel();
+                                               _parent->displayBlackBoxInfo(mod->getBBTKPackage(),mod->getBBTKType());
+                                       }
+                               }
+                       }
+               }
+               
+               
+
+
+               return true;
+       }
+
+       //=========================================================================
+       
+       bool wxVtkSceneManager::OnLeftButtonDown()
+       {
+               
+               if(_selectedObjects.size()!=0)
+               {
+                       _worldState = DRAG_OBJECTS;
+                       _startDragging = true;
+
+                       for (int i = 0; i < _selectedObjects.size(); i++)
+                       {
+                               int id = _selectedObjects[i];
+                               GObjectController* cont = _controllers[id];
+                               cont->getView()->setState(DRAG);
+                               cont->getModel()->notifyObservers(_idManager);
+                       }
+               }
+               
+               
+               
+               return true;
+       }
+
+       //=========================================================================
+       
+       bool wxVtkSceneManager::OnLeftButtonUp()
+       {
+               if(_worldState == DRAG_OBJECTS)
                {
-                       
                        _worldState = NOTHING_HAPPENS;
-                       int id = _controllers.size()-1;
-                       GObjectController* cont = _controllers[id];                     
-                       GConnectorModel* modelContour = (GConnectorModel*)cont->getModel();
 
-                       GObjectController* finPort = _controllers[idController];
-                       if(finPort->getGObjectType() == GPORT)
+                       for (int i = 0; i < _selectedObjects.size(); i++)
+                       {
+                               int id = _selectedObjects[i];
+                               GObjectController* cont = _controllers[id];
+                               cont->getView()->setState(SELECTED);
+                               cont->getModel()->notifyObservers(_idManager);
+                       }
+               }
+               return true;
+       }
+
+       //=========================================================================
+
+       bool wxVtkSceneManager::OnRightButtonUp()
+       {
+               for (int i = 0; i < _selectedObjects.size(); i++)
+               {
+                       int id = _selectedObjects[i];
+                       GObjectController* cont = _controllers[id];
+                       cont->SetActive(true);
+                       cont->getView()->setState(NOTHING_HAPPENS);
+                       cont->getModel()->notifyObservers(_idManager);
+               }
+
+               _selectedObjects.clear();
+
+               return true;
+       }
+
+       //=========================================================================
+
+       bool wxVtkSceneManager::OnChar()
+       {       
+               char keyCode = _vtkInteractorStyleBaseView->GetInteractor()-> GetKeyCode();
+               
+               // KeyCode 127 : Delete Key
+               // KeyCode 8 : Backspace Key
+               if(keyCode == 8 || keyCode == 127)
+               {
+                       if(_selectedObjects.size()>0)
+                       {
+                               for(int i=0;i<_selectedObjects.size();i++)
+                               {
+                                       int id = _selectedObjects[i];
+                                       removeObject(id);
+                               }
+                               _selectedObjects.clear();
+                       }
+               }
+
+               return true;
+       }
+
+       //=========================================================================
+
+       void wxVtkSceneManager::removeObject(int id)
+       {
+               GObjectController *control = _controllers[id];
+               std::vector<int> controllersToRemove;
+
+               if(control->getGObjectType()==GBLACKBOX)
+               {
+                       GBlackBoxModel *bbmod = (GBlackBoxModel*)control->getModel();
+                       std::vector<GPortModel*>inputs = bbmod->getInputPorts();
+                       
+                       for(int i = 0;i<inputs.size();i++)
+                       {
+                               controllersToRemove.push_back(inputs[i]->getObjectId());
+                       }
+
+                       std::vector<GPortModel*>outputs = bbmod->getOutputPorts();
+
+                       for(int i = 0;i<outputs.size();i++)
                        {
-                               GPortModel* modelPort = (GPortModel*)finPort->getModel();
-                               modelContour->setEndPort(modelPort);
-                       }                       
+                               controllersToRemove.push_back(outputs[i]->getObjectId());
+                       }
+
+                       controllersToRemove.push_back(control->getId());
+               }
+
+               for(int i = 0;i<controllersToRemove.size();i++)
+               {
+                       int id = controllersToRemove[i];
+                       GObjectController *cont = _controllers[id];
+                       cont->removeFromScene();
+                       unregisterController((InteractorStyleMaracas*)cont);
+                       _controllers.erase(id);
+               }
 
-                       manualConnectorContourController* manCont = ((GConnectorController*)cont)->getManualContourController();                        
-                       manualConnectorContourView* connView = (manualConnectorContourView*)manCont->GetManualViewBaseContour();
-                       connView->Refresh();
 
-                       for(int i=0;i<_controllers.size();i++)
+       }
+
+       //=========================================================================
+
+       void wxVtkSceneManager::displayBlackBoxInfo(std::string packageName, std::string boxName)
+       {
+               _parent->displayBlackBoxInfo(packageName,boxName);
+       }
+
+       //=========================================================================
+
+       void wxVtkSceneManager::updateStatusBar(std::string textStatus)
+       {
+               _parent->updateStatusBar(textStatus);
+       }
+
+       //=========================================================================
+
+       std::string wxVtkSceneManager::getDiagramScript()
+       {
+               bool existsExec=false;
+
+               std::vector<std::string> packages;
+               std::vector<int> boxes;
+               std::vector<int> connections;
+               std::vector<int> execBoxes;
+
+               std::map<int, GObjectController*>::iterator it;
+
+               for(it = _controllers.begin(); it != _controllers.end(); ++it)
+               {
+                       GObjectController *desc = it->second;
+                       int type = desc->getGObjectType();
+
+                       if(type==GBLACKBOX)
                        {
-                               GObjectController* cont = _controllers[i];
-                               if(cont->getView()!=NULL)
+                               GBlackBoxModel *mod = (GBlackBoxModel*)desc->getModel();
+                               
+                               std::string pkg = mod->getBBTKPackage();
+                               bool existsPkg = false;
+                               for(int t = 0;t<packages.size() && existsPkg == false;t++)
+                               {
+                                       if(packages[t]==pkg)
+                                       {
+                                               existsPkg=true;
+                                       }
+                               }
+                               if(!existsPkg)
+                               {
+                                       packages.push_back(pkg);
+                               }
+
+
+                               boxes.push_back(it->first);
+                               if(mod->isExecutable())
                                {
-                                       cont->getView()->setState(NOTHING_HAPPENS);
+                                       execBoxes.push_back(it->first);
+                                       existsExec=true;
                                }
-                               cont->SetActive(true);                                                          
                        }
+                       else if(type==GCONNECTOR)
+                       {
+                               connections.push_back(it->first);
+                       }
+               }
+
+               std::string script = "";
+               script+="# BBTK GEditor Script\n";
+               script+="# ----------------------\n";
+               if(existsExec)
+               {
+                       int i;
+                       for(i = 0; i<packages.size();i++)
+                       {
+                               script+="load ";
+                               script+=packages[i];
+                               script+="\n";
+                       }
+
+                       for(i = 0; i<boxes.size();i++)
+                       {
+                               script+="new ";
+                               int id = boxes[i];
+                               GObjectController *control = _controllers[id];
+                               GBlackBoxModel *model = (GBlackBoxModel*)control->getModel();
+
+                               script+=model->getBBTKType();
+                               script+=" ";
+                               script+=model->getBBTKName();
+                               script+="\n";
+                       }
+
+                       for(i = 0; i<connections.size();i++)
+                       {
+                               script+="connect ";
+                               int id = connections[i];
+                               GObjectController *control = _controllers[id];
+                               GConnectorModel *model = (GConnectorModel*)control->getModel();
+
+                               //Start Connection info
+                               GPortModel *start = model->getStartPort();
+                               script+=start->getParentBox()->getBBTKName();
+                               script+=".";
+                               script+=start->getBBTKName();
+
+                               script+=" ";
+
+                               //End Connection info
+                               GPortModel *end = model->getEndPort();
+                               script+=end->getParentBox()->getBBTKName();
+                               script+=".";
+                               script+=end->getBBTKName();
+
+                               script+="\n";
+                       }
+
+                       for(i = 0; i<execBoxes.size();i++)
+                       {
+                               script+="exec ";
+                               int id = execBoxes[i];
+                               GObjectController *control = _controllers[id];
+                               GBlackBoxModel *model = (GBlackBoxModel*)control->getModel();
+
+                               script+=model->getBBTKName();
+                               script+="\n";
+                       }
+
                }
+
+               return script;
        }
 
        //=========================================================================