arrow.h

/*
    OpenMBV - Open Multi Body Viewer.
    Copyright (C) 2009 Markus Friedrich
 
  This library is free software; you can redistribute it and/or 
  modify it under the terms of the GNU Lesser General Public 
  License as published by the Free Software Foundation; either 
  version 2.1 of the License, or (at your option) any later version. 
   
  This library 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 
  Lesser General Public License for more details. 
   
  You should have received a copy of the GNU Lesser General Public 
  License along with this library; if not, write to the Free Software 
  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
*/
 
#ifndef _OPENMBV_ARROW_H_
#define _OPENMBV_ARROW_H_
 
#include <openmbvcppinterface/dynamiccoloredbody.h>
#include <hdf5serie/vectorserie.h>
#include <stdexcept>
 
namespace OpenMBV {
 
  class Arrow : public DynamicColoredBody {
    friend class ObjectFactory;
    public:
      enum Type {
        line,
        fromHead,
        toHead,
        bothHeads,
        fromDoubleHead,
        toDoubleHead,
        bothDoubleHeads
      };
      enum ReferencePoint {
        toPoint,
        fromPoint,
        midPoint
      };
    protected:
      std::string pathStr;
      void createHDF5File() override;
      void openHDF5File() override;
      H5::VectorSerie<double>* data{nullptr};
      double headDiameter{0.5};
      double headLength{0.75};
      double diameter{0.25};
      double scaleLength{1};
      Type type{toHead};
      ReferencePoint referencePoint{toPoint};
 
      Arrow();
      ~Arrow() override;
    public:
      void setPath(bool p) { pathStr=(p)?"true":"false"; }
 
      bool getPath() { return pathStr=="true"?true:false; }
      
      template<typename T>
      void append(const T& row) {
        if(data==nullptr) throw std::runtime_error("can not append data to an environment object");
        if(row.size()!=8) throw std::runtime_error("the dimension does not match");
        if(!std::isnan(dynamicColor))
        {
          std::vector<double> tmprow(8);
          std::copy(&row[0], &row[8], tmprow.begin());
          tmprow[7]=dynamicColor;
          data->append(tmprow);
        }
        else
          data->append(row);
      }
 
      int getRows() override { return data?data->getRows():0; }
      std::vector<double> getRow(int i) override { return data?data->getRow(i):std::vector<double>(8); }
 
      void setArrowHead(double diameter, double length) {
        headDiameter=diameter;
        headLength=length;
      }
 
      void setHeadDiameter(double diameter) {
        headDiameter=diameter;
      }
 
      double getHeadDiameter() { return headDiameter; }
 
      void setHeadLength(double length) {
        headLength=length;
      }
 
      double getHeadLength() { return headLength; }
 
      void setDiameter(double diameter_) {
        diameter=diameter_;
      }
 
      double getDiameter() { return diameter; }
      
      void setType(Type type_) {
        type=type_;
      }
      
      Type getType() { return type; }
      
      void setReferencePoint(ReferencePoint referencePoint_) {
        referencePoint=referencePoint_;
      }
      
      ReferencePoint getReferencePoint() { return referencePoint; }
 
      void setScaleLength(double scale) {
        scaleLength=scale;
      }
 
      double getScaleLength() { return scaleLength; }
 
      void initializeUsingXML(xercesc::DOMElement *element) override;
 
      xercesc::DOMElement *writeXMLFile(xercesc::DOMNode *parent) override;
  };
 
}
 
#endif