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 Components {
        vectorForm,
        componentsInWorld,
        componentsInLocal,
      };
      enum ReferencePoint {
        toPoint,
        fromPoint,
        midPoint
      };
    protected:
      std::string pathStr;
      void createHDF5File() override;
      void openHDF5File() override;
      H5::VectorSerie<Float>* data{nullptr};
      double headDiameter{0.5};
      double headLength{0.75};
      double diameter{0.25};
      double scaleLength{1};
      bool createLocalFrame {false};
      Type type{toHead};
      Components components{vectorForm};
      int hdf5Size { -1 };
      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(hdf5Size==-1) {
          if( createLocalFrame && row.size()!=11)
            throw std::runtime_error("Need the columns t,x,y,z,dx,dy,dz,c,alpha,beta,gamma, but got "+std::to_string(row.size())+" columns");
          if(!createLocalFrame && row.size()!= 8)
            throw std::runtime_error("Need the columns t,x,y,z,dx,dy,dz,c, but got "+std::to_string(row.size())+" columns");
          hdf5Size = row.size();
        }
        else if(static_cast<int>(row.size())!=hdf5Size)
          throw std::runtime_error("The dimension does not match: need "+std::to_string(hdf5Size)+" but got "+std::to_string(row.size()));
 
        if(!std::isnan(dynamicColor))
        {
          std::vector<Float> tmprow(hdf5Size);
          std::copy(&row[0], &row[hdf5Size], tmprow.begin());
          tmprow[7]=dynamicColor;
          data->append(tmprow);
        }
        else
          data->append(row);
      }
 
      int getRows() override { return data?data->getRows():0; }
      std::vector<Float> getRow(int i) override { return data?data->getRow(i):std::vector<Float>(11); }
 
      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 setComponents(Components components_) {
        components=components_;
      }
      
      Components getComponents() { return components; }
 
      void setReferencePoint(ReferencePoint referencePoint_) {
        referencePoint=referencePoint_;
      }
      
      ReferencePoint getReferencePoint() { return referencePoint; }
 
      void setScaleLength(double scale) {
        scaleLength=scale;
      }
 
      double getScaleLength() { return scaleLength; }
 
      void setCreateLocalFrame(bool l) {
        createLocalFrame=l;
      }
 
      double getCreateLocalFrame() { return createLocalFrame; }
 
      void initializeUsingXML(xercesc::DOMElement *element) override;
 
      xercesc::DOMElement *writeXMLFile(xercesc::DOMNode *parent) override;
  };
 
}
 
#endif