ivscreenannotation.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_IVSCREENANNOTATION_H_
#define _OPENMBV_IVSCREENANNOTATION_H_
 
#include "body.h"
#include <hdf5serie/vectorserie.h>
 
namespace OpenMBV {
 
  class IvScreenAnnotation : public Body {
    friend class ObjectFactory;
    public:
      void initializeUsingXML(xercesc::DOMElement *element) override;
      xercesc::DOMElement* writeXMLFile(xercesc::DOMNode *parent) override;
      void setScale1To1(bool scale1To1_);
      bool getScale1To1();
 
      void setScale1To1At(const std::vector<double> &scale1To1Center_);
 
      std::vector<double> getScale1To1At();
 
      void setIvFileName(std::string ivFileName_) { ivContent=""; ivFileName=std::move(ivFileName_); }
      std::string getIvFileName() { return ivFileName; }
 
      void setIvContent(std::string ivContent_) { ivFileName=""; ivContent=std::move(ivContent_); }
      const std::string& getIvContent() { return ivContent; }
 
      void setColumnLabels(const std::vector<std::string> &columnLabels_);
      void addColumnLabel(const std::string &columnLabel_);
      const std::vector<std::string>& getColumnLabels() const;
      void setColumnIntLabels(const std::vector<std::string> &columnIntLabels_);
      void addColumnIntLabel(const std::string &columnIntLabel_);
      const std::vector<std::string>& getColumnIntLabels() const;
      void setColumnStrLabels(const std::vector<std::string> &columnStrLabels_);
      void addColumnStrLabel(const std::string &columnStrLabel_);
      const std::vector<std::string>& getColumnStrLabels() const;
      void setFixedStrSize(int fss) { fixedStrSize = fss; }
      int getFixedStrSize() { return fixedStrSize; }
 
      void createHDF5File() override;
      void openHDF5File() override;
 
      template<typename T>
      void append(const T& row) {
        if(data==nullptr) throw std::runtime_error("IvScreenAnnotation: Cannot append data to an environment object");
        if(row.size()!=1+columnLabels.size()) throw std::runtime_error("IvScreenAnnotation: The dimension does not match (append: "+
                                                    std::to_string(row.size())+", columns: "+std::to_string(1+columnLabels.size())+")");
        data->append(&row[0], row.size());
      }
 
      template<typename T>
      void appendInt(const T& row) {
        if(dataInt==nullptr) throw std::runtime_error("IvScreenAnnotation: Cannot append dataInt to an environment object");
        if(row.size()!=columnIntLabels.size()) throw std::runtime_error("IvScreenAnnotation: The dimension does not match (appendInt: "+
                                                    std::to_string(row.size())+", columns: "+std::to_string(columnIntLabels.size())+")");
        dataInt->append(&row[0], row.size());
      }
 
      template<typename T>
      void appendStr(const T& row) {
        if(dataStr==nullptr) throw std::runtime_error("IvScreenAnnotation: Cannot append dataStr to an environment object");
        if(row.size()!=columnStrLabels.size()) throw std::runtime_error("IvScreenAnnotation: The dimension does not match (append: "+
                                                    std::to_string(row.size())+", columns: "+std::to_string(columnStrLabels.size())+")");
        dataStr->append(&row[0], row.size());
      }
 
      int getRows() override { return data?data->getRows():0; }
      std::vector<Float> getRow(int i) override {
        if(!data)
          return std::vector<Float>(1+columnLabels.size());
 
        auto row = data->getRow(i);
 
        // handle legacy data (no "time" column as first column)
        if(columnLabels.size() == row.size()) {
          std::vector<Float> ret(1+row.size());
          ret[0] = 0; // use 0 for time column
          for(size_t i = 0; i < row.size(); ++i)
            ret[i+1] = row[i];
          return ret;
        }
 
        return row;
      }
 
      std::vector<int> getRowInt(int i) {
        if(!dataInt)
          return std::vector<int>(columnIntLabels.size());
        return  dataInt->getRow(i);
      }
 
      std::vector<std::string> getRowStr(int i) {
        if(!dataStr)
          return std::vector<std::string>(columnStrLabels.size());
        return  dataStr->getRow(i);
      }
    protected:
      IvScreenAnnotation();
      ~IvScreenAnnotation() override = default;
      bool scale1To1 { false };
      std::vector<double> scale1To1Center;
      std::string ivFileName;
      std::string ivContent;
      int fixedStrSize { -1 };
      std::vector<std::string> columnLabels;
      std::vector<std::string> columnIntLabels;
      std::vector<std::string> columnStrLabels;
      H5::VectorSerie<Float>* data{nullptr};
      H5::VectorSerie<int>* dataInt{nullptr};
      H5::VectorSerie<std::string>* dataStr{nullptr};
  };
 
}
 
#endif