source: PSPA/Interface_Web/trunk/pspaWT/src/GWt_LigneFaisceau.cc @ 83

#include "GWt_LigneFaisceau.h"
//#include "driftDefinition.h"
#include "GWt_drift.h"
#include "GWt_initialBeam.h"
#include "GWt_cell.h"
#include "GWt_soleno.h"
#include <iostream>
#include "nomdElements.h"


#include <Wt/WComboBox>
#include <Wt/WVBoxLayout>


GWt_LigneFaisceau::GWt_LigneFaisceau(PspaApplication* ps):
  pspa_(ps)
{

  beamLayout_ = new WHBoxLayout();
  decorationStyle().setBackgroundColor (WColor("orange"));
  

  setLayout(beamLayout_);

  acceptDrops("icons/beam.jpg");
  acceptDrops("icons/bend.jpg");
  acceptDrops("icons/cell.jpg");
  acceptDrops("icons/drift.jpg");
  acceptDrops("icons/final.jpg");
  acceptDrops("icons/lens.jpg");



  //  dtmanage_ = new dataManager();
}


void GWt_LigneFaisceau::dropEvent(WDropEvent evt)
{
  //  pspa_->addConsoleMessage(WString(" GWt_LigneFaisceau::dropEvent")+evt.mimeType().c_str());


  WPushButton* dropped = NULL;



  if ( strstr(evt.mimeType().c_str() , "drift") ) 
    {
      pspa_->addConsoleMessage(" DRIFT");
      abstractElement* newDrift = pspa_->getDataManager()->addDrift();
      GWt_drift* gdrift = new GWt_drift(pspa_,newDrift, evt.mimeType());
      dropped = gdrift->getButton();
    }
  else if ( strstr(evt.mimeType().c_str() , "beam") ) 
    {
      pspa_->addConsoleMessage(" INITIAL_BEAM");
      abstractElement* newInitialBeam = pspa_->getDataManager()->addInitialBeam();
      GWt_initialBeam* gwInitialBeam = new GWt_initialBeam(pspa_,newInitialBeam, evt.mimeType());

      dropped = gwInitialBeam->getButton();
    }
  else if ( strstr(evt.mimeType().c_str() , "cell") ) 
    {
      pspa_->addConsoleMessage(" CELL");
      abstractElement* newCell = pspa_->getDataManager()->addCell();
      GWt_cell* gwCell = new GWt_cell(pspa_,newCell, evt.mimeType());
      dropped = gwCell->getButton();
    }
  else if ( strstr(evt.mimeType().c_str() , "lens") ) 
    {
      pspa_->addConsoleMessage("SOLENOID");
      abstractElement* newSoleno = pspa_->getDataManager()->addSoleno();
      GWt_soleno* gwSoleno = new GWt_soleno(pspa_,newSoleno, evt.mimeType());
      dropped = gwSoleno->getButton();
    }
  else
    {
      pspa_->addConsoleMessage(" RIEN trouve");
    }

  beamLayout_->addWidget(dropped);
}


void GWt_LigneFaisceau::restoreElementCollection()
{
  int nbElem = pspa_->getDataManager()->beamLineSize();
  unsigned int k;
  beamLayout_->clear();
  for ( k=0; k < (unsigned)nbElem; k++) 
    {     
      abstractElement* ptr = pspa_->getDataManager()->getCollection()->getElementPointer(k);
      nomdElement nom = ptr->getName();
      switch (nom)
        {
        case initialBeam :
          {
            GWt_initialBeam* gwInitialBeam = new GWt_initialBeam(pspa_,ptr, string("icons/beam.jpg"));
            beamLayout_->addWidget(gwInitialBeam->getButton());
            break;
          }
        case drift :
          {
            GWt_drift* gdrift = new GWt_drift(pspa_,ptr, string("icons/drift.jpg"));      
            beamLayout_->addWidget(gdrift->getButton());
            break;
          }
        case cell :
          {
              GWt_cell* gwCell = new GWt_cell(pspa_,ptr, string("icons/cell.jpg"));
              beamLayout_->addWidget(gwCell->getButton());
              break;
          }

        case soleno :
          {
              GWt_soleno* gwSoleno = new GWt_soleno(pspa_,ptr, string("icons/lens.jpg"));
              beamLayout_->addWidget(gwSoleno->getButton());
              break;
          }
        default :
          {
            cerr << "  GWt_LigneFaisceau element type " << nom << " inconnu " << endl;
            break;
          }
     }

    }

}