source: PSPA/Interface_Web/trunk/pspaWT/sources/userInterface/src/GWt_ligneFaisceau.cc @ 323

#include <iostream>

#include "GWt_elementLigneFaisceau.h"
#include "GWt_ligneFaisceau.h"
#include "GWt_drift.h"
#include "GWt_rfgun.h"
#include "GWt_cell.h"
#include "GWt_soleno.h"
#include "GWt_bend.h"

#include "mixedTools.h"

#include <Wt/WVBoxLayout>


GWt_LigneFaisceau::GWt_LigneFaisceau(PspaApplication* ps) :
WContainerWidget(),
pspa_(ps),
nObjets_(NULL)
{
    
    cout<<"GWt_LigneFaisceau::GWt_LigneFaisceau() %d"<< this << "<---------------"<< endl;
    
    ligneFaisceauLayout_ = new WHBoxLayout();
    //  decorationStyle().setBackgroundColor (WColor(245,245,245));
    setLayout(ligneFaisceauLayout_);
    ligneFaisceauLayout_->setContentsMargins(0,0,0,0);
    ligneFaisceauLayout_->setSpacing    (0);
    
    int nElts= nomdElements::getNumberOfElements();
    nObjets_= new Compteur[nElts];
    
    // set the object name
    setObjectName("ligneFaisceau");
    
    
    clear();
}


void GWt_LigneFaisceau::clear() {
    ligneFaisceauLayout_->clear();

    // add the first drop zone
    decorationStyle().setBorder (WBorder (WBorder::Dotted));
    allDropZone_ = new GWt_dropZoneLigneFaisceau(pspa_);
    WVBoxLayout* fit = new WVBoxLayout();
    allDropZone_->setLayout(fit);
    
    fit->addWidget(new WText("Drag elements here...."),1,Wt::AlignCenter |Wt::AlignMiddle);
    ligneFaisceauLayout_->addWidget(allDropZone_,1);
}


GWt_abstractElement* GWt_LigneFaisceau::addElement(typedElement eType){
    return addElement(eType,NULL);
}



GWt_abstractElement* GWt_LigneFaisceau::addElement(typedElement eType, GWt_dropZoneLigneFaisceau* dropContainer ){

    GWt_abstractElement* GWtAbstractElem = NULL;
    
    cout << " Entree addElement element " << nomdElements::getLabelFromType(eType) << endl;
    
    int nElements= nomdElements ::getNumberOfElements();
    nObjets_= new Compteur[nElements];
    
    // 3 cases :
    // -dropContainer is an element of the ligneFaisceauLayout_
    // -dropContainer is an element of the GWt_elementLigneFaisceau_
    // -dropContainer is NULL, insert in the last element
    
    if(eType == inconnu) {
        cout << "GWt_LigneFaisceau::element type " << eType << " inconnu " << endl;
        return NULL;
    }
    
    // add the element to the datamanager
    
    // first time we insert an element :
    GWt_elementLigneFaisceau* firstElemIsElement = dynamic_cast <GWt_elementLigneFaisceau*> (ligneFaisceauLayout_->itemAt(0)->widget());
    if ((ligneFaisceauLayout_->count() == 1) && (!firstElemIsElement)) {
        int index = indexOf (dropContainer);
        if ((index != -1) || (dropContainer == NULL)){ // found
            // remove global drop zone
            ligneFaisceauLayout_->clear();

            // add first element
            GWt_elementLigneFaisceau* elemLigneFaisceau = new GWt_elementLigneFaisceau(eType, pspa_);
            
            ligneFaisceauLayout_->addWidget(elemLigneFaisceau,1);
            GWtAbstractElem = elemLigneFaisceau->getGWt_AbstractElement();
            // remove arrow from begin and end
            
        } else { // error
            return NULL;
        }
    } else {
        int index = -1;
        for (int a=0; a< ligneFaisceauLayout_->count(); a++) {
            GWt_elementLigneFaisceau* elem = static_cast <GWt_elementLigneFaisceau*> (ligneFaisceauLayout_->itemAt(a)->widget ());
            if (elem) {
                if (elem->getDropZoneLeft() == dropContainer) {
                    index = a;
                }
                if (elem->getDropZoneRight() == dropContainer) {
                    index = a+1;
                }
            }
            
        }
        
        
        //  ligneFaisceauLayout_->addWidget(new GWt_elementLigneFaiseau(ptr),0);
        GWt_elementLigneFaisceau* elemLigneFaisceau = new GWt_elementLigneFaisceau(eType, pspa_);
        if (index != -1) {
//            nObjets_[ptr->getNomdElement().getElementType()].incr();
            ligneFaisceauLayout_->insertWidget(index,elemLigneFaisceau,1);
            GWtAbstractElem = elemLigneFaisceau->getGWt_AbstractElement();
        } else {
            ligneFaisceauLayout_->addWidget(elemLigneFaisceau,1);
            GWtAbstractElem = elemLigneFaisceau->getGWt_AbstractElement();
        }
    }
    manageLineFaisceauLayout();

    return GWtAbstractElem;
}


void GWt_LigneFaisceau::manageLineFaisceauLayout() {
    // special : hide/setVisible arrow :
    for (int a=0; a< ligneFaisceauLayout_->count(); a++) {
        if (GWt_elementLigneFaisceau* elem = dynamic_cast <GWt_elementLigneFaisceau*> (ligneFaisceauLayout_->itemAt(a)->widget ())) {

            // update labels
            elem->getGWt_AbstractElement()->updateLabelAndToolTipWidget();
            
            //  one element
            if ((a == 0) && (a == (ligneFaisceauLayout_->count()-1))) {
                elem->getDropZoneLeft()->setStyleClass("elementLigneFaiseau_transparent");
                elem->getDropZoneRight()->setStyleClass("elementLigneFaiseau_transparent");

                // first
            } else if (a == 0){
                elem->getDropZoneLeft()->setStyleClass("elementLigneFaiseau_transparent");
                elem->getDropZoneRight()->setStyleClass("elementLigneFaiseau_line");

                // last
            } else if (a == (ligneFaisceauLayout_->count()-1)){
                elem->getDropZoneLeft()->setStyleClass("elementLigneFaiseau_line");
                elem->getDropZoneRight()->setStyleClass("elementLigneFaiseau_transparent");
                
               
                // middle elements
            } else {
                elem->getDropZoneLeft()->setStyleClass("elementLigneFaiseau_line");
                elem->getDropZoneRight()->setStyleClass("elementLigneFaiseau_line");
            }
        }
    }
    
    // insure that there is still the dropzone !
    if (ligneFaisceauLayout_->count() == 0) {
        clear();
    }
    
    pspa_->getExecuteWidget()->updateSelections();
}



void GWt_LigneFaisceau::update() {
    manageLineFaisceauLayout();
}


void GWt_LigneFaisceau::removeElement(GWt_abstractElement* GWt_abstractElement) {
    if (GWt_abstractElement == NULL)
        return;
    
    // remove Widget
    // - GWt_abstractElement is an element of the GWt_elementLigneFaisceau_
    
    int index = -1;
    for (int a=0; a< ligneFaisceauLayout_->count(); a++) {
        GWt_elementLigneFaisceau* elem = static_cast <GWt_elementLigneFaisceau*> (ligneFaisceauLayout_->itemAt(a)->widget ());
        if (elem) {
            if (elem->getGWt_AbstractElement() == GWt_abstractElement) {
                index = a;
            }
        }
    }
    
    if (index != -1) {
        // remove element from the list
//        pspa_->getDataManager()->removeElement(GWt_abstractElement->getElement()->getLabel());
        
        ligneFaisceauLayout_->removeItem(ligneFaisceauLayout_->itemAt(index));
        delete GWt_abstractElement;
        
        // manage widgets if necessary
        manageLineFaisceauLayout();
    }
}


abstractElement* GWt_LigneFaisceau::getAbstractElement(int i){
    int index = 0;

    // first if 1
//    i--;
    
    for (int a=0; a< ligneFaisceauLayout_->count(); a++) {
        if (GWt_elementLigneFaisceau* elem = dynamic_cast <GWt_elementLigneFaisceau*> (ligneFaisceauLayout_->itemAt(a)->widget ())) {
            if (index == i) {
                printf("GWt_LigneFaisceau::getAbstractElement %d %s\n",i,elem->getGWt_AbstractElement()->getAbstractElement()->getLabel().c_str());
                return elem->getGWt_AbstractElement()->getAbstractElement();
            }
            index ++;
        }
    }
    return NULL;
}


int GWt_LigneFaisceau::getBeamLineSize() {
    int count = 0;
    for (int a=0; a< ligneFaisceauLayout_->count(); a++) {
        if (dynamic_cast <GWt_elementLigneFaisceau*> (ligneFaisceauLayout_->itemAt(a)->widget ())) {
            count ++;
        }
    }
    return count;
}


vector <abstractElement*> GWt_LigneFaisceau::getAllAbstractElements(){
    vector <abstractElement*> elems;
    for (int a=0; a< ligneFaisceauLayout_->count(); a++) {
        if (GWt_elementLigneFaisceau* elem = dynamic_cast <GWt_elementLigneFaisceau*> (ligneFaisceauLayout_->itemAt(a)->widget ())) {
            elems.push_back(elem->getGWt_AbstractElement()->getAbstractElement());
        }
    }
    return elems;
}