BCDataSet.h

Go to the documentation of this file.

#ifndef __BCDATASET__H
#define __BCDATASET__H

/*
 * Copyright (C) 2008, Daniel Kollar and Kevin Kroeninger.
 * All rights reserved.
 *
 * For the licensing terms see doc/COPYING.
 */

// ---------------------------------------------------------

#include <vector>

// BAT classes
class BCDataPoint;

// ---------------------------------------------------------

typedef std::vector<BCDataPoint*> BCDataVector;

// ---------------------------------------------------------

class BCDataSet
{
   public:

      /* @{ */

      /*
       * Default constructor */
      BCDataSet();

      /*
       * Default destructor */
      virtual ~BCDataSet();

      /* @} */

      /* @{ */

      /*
       * @param The vector of data points. */
//    BCDataVector * GetDataVector();

      /*
       * @return The number of data points. */
      unsigned int GetNDataPoints();

      /*
       * @return number of values per data point (dimension of data). */
      unsigned int GetNValuesPerPoint();

      /*
       * @param index The index of the data point to be returned.
       * @return The data point at the index. */
      BCDataPoint * GetDataPoint(unsigned int index);


      /* @} */

      /* @{ */

      int ReadDataFromFile(const char * filename, const char * treename, const char * branchnames)
         { return this ->  ReadDataFromFileTree(filename, treename, branchnames); };

      int ReadDataFromFile(const char * filename, int nvariables)
         { return this -> ReadDataFromFileTxt(filename, nvariables); };

      int ReadDataFromFileTree(const char * filename, const char * treename, const char * branchnames);

      int ReadDataFromFileTxt(const char * filename, int nvariables);

      void AddDataPoint(BCDataPoint * datapoint);

      void Reset();

      void Dump();

      /* @} */

   private:

      /*
       * A vector containing the data points */
      BCDataVector * fBCDataVector;

};

// ---------------------------------------------------------

#endif

---