AliHLTTPCCAClusterData.h

/*
    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 "AliHLTTPCCAGBHit.h"
#include "AliHLTArray.h"

#include "AliHLTTPCCAParameters.h"

class AliHLTTPCSpacePointData;

class AliHLTTPCCAClusterData
{
  public:

    AliHLTTPCCAClusterData( const AliHLTTPCCAGBHit *hits, int *offset, int numberOfClusters, int nRows8 ):
      fSlice(0), fFirstRow(0), fLastRow(0), fNumberOfClusters(0), fRowOffset(0), fData(0)
    {
      readEvent( hits, offset, numberOfClusters, nRows8 );
    }

    AliHLTTPCCAClusterData(): fSlice(0), fFirstRow(0), fLastRow(0), fNumberOfClusters(0), fRowOffset(0), fData(0) {}

    // void readEvent( const AliHLTArray<AliHLTTPCSpacePointData *> &clusters,
    //     int numberOfClusters, double ClusterZCut );
    void readEvent( const AliHLTTPCCAGBHit *hits, int *offset, int numberOfClusters, int nRows8 );

    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 AliHLTTPCCAClusterData ClusterData;

#endif // CLUSTERDATA_H