PndFTSCAClusterData.h

Go to the documentation of this file.

/*
    Copyright (C) 2009 Matthias Kretz <kretz@kde.org>

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

    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
    Library General Public License for more details.

    You should have received a copy of the GNU Library General Public License
    along with this library; see the file COPYING.LIB.  If not, write to
    the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
    Boston, MA 02110-1301, USA.

*/

#ifndef CLUSTERDATA_H
#define CLUSTERDATA_H

#include <vector>
#include "PndFTSCAGBHit.h"
#include "PndFTSArray.h"

#include "PndFTSCAParameters.h"

class PndFTSSpacePointData;

class PndFTSCAClusterData {
 public:
  PndFTSCAClusterData(const PndFTSCAGBHit *hits, int *offset, int numberOfClusters) { readEvent(hits, offset, numberOfClusters); }

  PndFTSCAClusterData() {}

  void readEvent(const PndFTSArray<PndFTSSpacePointData *> &clusters, int numberOfClusters, double ClusterZCut);
  void readEvent(const PndFTSCAGBHit *hits, int *offset, int numberOfClusters);

  void Merge(int index1, int index2);

  // void Split( int index, /* TODO: need some parameters how to split */ );

  // TODO: some access to history of merges and splits

  int Slice() const { return fSlice; }

  int FirstRow() const { return fFirstRow; }

  int LastRow() const { return fLastRow; }

  int NumberOfClusters() const { return fData.size(); }

  int NumberOfClusters(unsigned int rowIndex) const { return rowIndex < fNumberOfClusters.size() ? fNumberOfClusters[rowIndex] : 0; }

  int RowOffset(unsigned int rowIndex) const { return rowIndex < fRowOffset.size() ? fRowOffset[rowIndex] : fData.size(); }

  float X(int index) const
  {
    assert(index < static_cast<int>(fData.size()));
    return fData[index].fX;
  }

  float Y(int index) const
  {
    assert(index < static_cast<int>(fData.size()));
    return fData[index].fY;
  }

  float Z(int index) const
  {
    assert(index < static_cast<int>(fData.size()));
    return fData[index].fZ;
  }

  int Id(int index) const
  {
    assert(index < static_cast<int>(fData.size()));
    return fData[index].fId;
  }

  int RowNumber(int index) const
  {
    assert(index < static_cast<int>(fData.size()));
    return fData[index].fRow;
  }

 private:
  struct Data {
    float fX;
    float fY;
    float fZ;
    int fId;
    int fRow;
  };

  int fSlice;                         // the slice index this data belongs to
  int fFirstRow;                      // see FirstRow()
  int fLastRow;                       // see LastRow()
  std::vector<int> fNumberOfClusters; // list of NumberOfClusters per row for NumberOfClusters(int)
  std::vector<int> fRowOffset;        // see RowOffset()
  std::vector<Data> fData;            // list of data of clusters
};

typedef PndFTSCAClusterData ClusterData;

#endif // CLUSTERDATA_H