interface.cpp

Go to the documentation of this file.

# include <math.h>

# include <interface.h>
# include <network.h>
# include <qgl.h>
# include <inn.h>
# include <qtextedit.h>

using namespace qglviewer ;

extern Network * network ;
extern Inn * inn ;

// Constructor must call the base class constructor.
Viewer::Viewer(QWidget *parent,const char *name) : QGLViewer(parent, name) {
        //help();
        startTimer(20);
        setDrawAxis(false) ;
        idfrom_ = -1 ;
        idto_ = -1 ;
}


void Viewer::init() {
        mode_ = NONE ;
        
        glEnable (GL_BLEND);
        glBlendFunc (GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
        
        // Make camera the default manipulated frame.
        setManipulatedFrame( camera()->frame() );
        
        // Set a generic plane constraint
        constraint = new LocalConstraint() ;
        constraint->setTranslationConstraint(AxisPlaneConstraint::PLANE, Vec(0.0,1.0,0.0));
        layerconstraint = new LocalConstraint() ;
        layerconstraint->setTranslationConstraint(AxisPlaneConstraint::FORBIDDEN, Vec(1.0,1.0,1.0));
        layerconstraint->setRotationConstraint(AxisPlaneConstraint::FORBIDDEN, Vec(1.0,1.0,1.0));
        
        // Use "frame(i)->setConstraint(constraint)" to affect this constraint
        
        // Swap the CAMERA and FRAME state keys (NoButton and Control)
        // Save CAMERA binding first. See setMouseStateKey() documentation.
        setMouseStateKey(QGLViewer::FRAME,  Qt::ControlButton);
        
        setSceneRadius(3.0);
        showEntireScene();
        
        float x, y, z ;
        camera()->getPosition( x, y, z ) ;
        camera()->setPosition( x, y+1.3, z ) ;
}

int Viewer::mode(void) {
        return mode_ ;
}

void Viewer::setMode(int mode) {
        mode_ = mode ;
}

void Viewer::resetMode(void) {
        idfrom_ = -1 ;
        idto_ = -1 ;
        mode_ = NONE ;
}


void Viewer::draw() {
        network->setShininess( 32 ) ;
        network->draw(selectedName()) ;
        if(mode_==ADDAXON) {
                QFont font("Arial", 16) ;
                renderText( 10, 50, "Axon mode is on", font ) ;
        }
}

void Viewer::drawText( float x, float y, float z, const QString& s, const QFont& q ) {
        renderText( x, y, z, s, q ) ;
}


void Viewer::drawWithNames() {
        network->setShininess( 4 ) ;
        network->draw(selectedName(), true) ;
}

void Viewer::fastDraw() {
        network->setShininess( 4 ) ;
        network->draw(selectedName()) ;
}

void Viewer::postSelection(int x, int y) {
        if(mode()==ADDAXON  && network->existNeuron(selectedName())) {
                if( idfrom_ < 0 ) idfrom_ = selectedName() ;
                else {
                        idto_ = selectedName() ;
                        if( network->getLayerNumTheNeuronIsOn(idfrom_)<network->getLayerNumTheNeuronIsOn(idto_)
                                && !network->existAxon( idfrom_, idto_ ) ) 
                                network->addAxon( idfrom_, idto_ ) ;
                        else if( network->getLayerNumTheNeuronIsOn(idfrom_)>network->getLayerNumTheNeuronIsOn(idto_)
                                && !network->existAxon( idto_, idfrom_ ) )
                                network->addAxon( idto_, idfrom_ ) ;
                        idfrom_ = -1 ;
                        idto_ = -1 ;
                }
        } else if (selectedName() == -1 && !network->frame(selectedName()) ) {
      // Camera will be the default frame is no object is selected.
      setManipulatedFrame( camera()->frame() );
    } else {
      setManipulatedFrame(network->frame(selectedName()));
      if(network->existNeuron(selectedName()))
        network->frame(selectedName())->setConstraint(constraint) ;
      else if(network->existLayer(selectedName()))
        network->frame(selectedName())->setConstraint(layerconstraint) ;
    }
}

void Viewer::timerEvent(QTimerEvent*) {
    updateGL();
}

QString Viewer::helpString() const {
  QString text("<h2>Intuitive Neural Network</h2>");
  text += "";
  text += "";
  text += "";
  return text;
}

---