src/TraplasObject.cpp

Go to the documentation of this file.

/*

Traplas visualisation.

Copyright (C) 2006 Herbert de Vos & Willem Drost

This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

(For full Licence see ../GPL-Licence.txt)

*/

#ifndef _GLIBCXX_SSTREAM
#include <sstream>
#endif

#ifndef STATECONTROL_H
#include "StateControl.h"
#endif

#ifndef DISPATCHER_H
#include "Dispatcher.h"
#endif

#ifndef _ASSERT_H
#include <assert.h>
#endif

#ifndef GLOBALS_H
#include "Globals.h"
#endif

#ifndef OSGDB_READFILE
#include <osgDB/ReadFile>
#endif

#ifndef OSG_VEC3D
#include <osg/Vec3d>
#endif

#ifndef OSG_QUAT
#include <osg/Quat>
#endif

#ifndef OSG_POSTIONATTITUDETRANSFORM
#include <osg/PositionAttitudeTransform>
#endif

#ifndef OSG_AUTOTRANSFORM
#include <osg/AutoTransform>
#endif

#ifndef OSG_VEC4
#include <osg/Vec4>
#endif

#ifndef OSG_MATRIX
#include <osg/Matrix>
#endif

#ifndef OSG_MATRIXTRANSFORM
#include <osg/MatrixTransform>
#endif

#ifndef OSG_GEOMETRY
#include <osg/Geometry>
#endif

#ifndef OSG_STATESET
#include <osg/StateSet>
#endif

#ifndef OSGTEXT_TEXT
#include <osgText/Text>
#endif

#ifndef OSGPRODUCER_VIEWER
#include <osgProducer/Viewer>
#endif

/*TraplasObject*/
TraplasObject::TraplasObject(){}

TraplasObject::~TraplasObject(){}

void TraplasObject::initLabel(){
        //      Create the label
        osgText::Text* objectText = new osgText::Text();
        objectText->setCharacterSize(FONTSIZE_LABELS);
        objectText->setAxisAlignment(osgText::Text::SCREEN);
        objectText->setDrawMode(osgText::Text::TEXT | osgText::Text::ALIGNMENT | osgText::Text::CENTER_TOP);
        objectText->setAlignment(osgText::Text::CENTER_TOP);
        objectText->setPosition(osg::Vec3d(0,-6,0)); 
        objectText->setColor(osg::Vec4d(1.0f, 0.0f, 0.0f, 1.0f) );
        label = new osg::Geode();
        label->addDrawable(objectText);

  osg::StateSet* txtStateSet = new osg::StateSet();
  txtStateSet->setMode(GL_LIGHTING,osg::StateAttribute::OFF); //disable lighting on text labels.
  label->setStateSet(txtStateSet);


        //      Add the label to the osg-tree.
        osg::AutoTransform* at = dynamic_cast<osg::AutoTransform*>(model->getParent(0));
        assert(at != NULL);
        at->addChild(label);

        return;
}

osg::Vec3d TraplasObject::getPosition(){
        //      Do some checks

        osg::PositionAttitudeTransform* pat = dynamic_cast<osg::PositionAttitudeTransform*>(model->getParent(0)->getParent(0));
        assert (pat != NULL);

        return pat->getPosition();
}

void TraplasObject::setDescription(string descript){
        description = descript;
        model->setName(descript);
}

bool TraplasObject::select(){
        selected = !selected;
                
        //Ok, we're going to change the color of the textlabel of the (de)selected object:
        osg::AutoTransform* tmp_at = dynamic_cast<osg::AutoTransform*>(getModel()->getParent(0));
        assert(tmp_at != NULL);
        osg::Geode* tmpg = dynamic_cast<osg::Geode*>(tmp_at->getChild(1));
        assert(tmpg != NULL);
        osgText::Text* tmpt = dynamic_cast<osgText::Text*>(tmpg->getDrawable(0)); 
        assert(tmpt != NULL);
        if(selected)tmpt->setColor(osg::Vec4d(0.0f, 1.0f, 0.0f, 1.0f) ); 
        else tmpt->setColor(osg::Vec4d(1.0f, 0.0f, 0.0f, 1.0f) );


        return selected;
}

bool TraplasObject::setLabel( shared_ptr<string> desc ){
        osgText::Text* t = dynamic_cast<osgText::Text*>(label->getDrawable(0));
        if(t == NULL) return false; 
        t->setText(*desc);      
        return true;
}

string TraplasObject::toString(){ return ""; }

/*Cargo*/
Cargo::Cargo(int carg_id, int type_carg, osg::Node* carg_model){
        id = carg_id, type = type_carg, model = carg_model;
        ostringstream s;
        s << id;
        setDescription( s.str() );

        selected = false;

        carg_model->setUserData( (TraplasObject*)this); //sets a reference from the osg::Node to the TraplasObject.

}

string Cargo::toString(){
        ostringstream cs;
        cs << description;
        return cs.str();
}

/*Transport*/

Transport::Transport(int transport_id, int type_transport, double spd, osg::Node* trans_model){
        destination = ""; //convention for: No Destination. 
        carg = -1;
        c_desc = "";
        id = transport_id, type = type_transport, speed = spd, model = trans_model;

        ostringstream s;
        s << id;
        setDescription( s.str() );

        selected = false;

        trans_model->setUserData( (TraplasObject*)this); //sets a reference from the osg::Node to the TraplasObject.

}

bool Transport::setSpeed(double s_spd){
        speed = s_spd;
        return true;
}

bool Transport::setDestination(string s_dest){
        destination = s_dest;
        return true;
}

bool Transport::setCargo(int s_carg, string desc){
        carg = s_carg;
        c_desc = desc;
        //cargoId is transferred from Infrastructure to Transport. However, StateControl maintains control of the actual pointers to the objects.
        return true;
}

int Transport::clearCargo(){
        int cleared = carg;
        c_desc = "";
        carg = -1;
        return cleared;
}

string Transport::toString(){
        ostringstream st;
        st << description << "\nSpeed: " << speed; 
  if(destination != ""){
                st << "\nDestination: " << destination ;
        }
        if(carg != -1)
         st << "\nCargo: " << c_desc;
        return st.str();
}

/*Infrastructure*/

Infrastructure::Infrastructure(int infrastructure_id, int type_infrastructure, double spd, int tile_type, unsigned int cap, double dist, osg::Node* infra_model){
        id = infrastructure_id, type= type_infrastructure, speed = spd, model = infra_model;
        capacity = cap, distance = dist;

        ostringstream s;
  s << id;
  setDescription( s.str() );

        selected = false;

        infra_model->setUserData( (TraplasObject*)this);        //sets a reference from the osg::Node to the TraplasObject.
}

bool Infrastructure::setSpeed(double s_spd){
        speed = s_spd;
        return true;
}

int Infrastructure::removeCarg(int r_carg){
        if(cargoSet.find(r_carg) != cargoSet.end()){
                cargoSet.erase(r_carg);
                return r_carg;
        }
        else return -1;
}

bool Infrastructure::addCarg(int a_carg){
        pair<std::_Rb_tree_const_iterator<int>, bool> thing = cargoSet.insert(a_carg);
        return thing.second;
}

bool Infrastructure::addArc(int to){
        int* arc = NULL;
        arc = new int(to);
        arcs[to]=arc;

        return true;
}

bool Infrastructure::isArc(int end){
        int* arc = arcs[end];
        return (arc != NULL);
}

string Infrastructure::toString(){
        ostringstream is;
        is << description << '\n' << "Speed: " << speed << '\n' << "Capacity" << capacity;
        return is.str();
}

---