#ifndef PNDSTTGEOMETRYMAP_H
#define PNDSTTGEOMETRYMAP_H

#include <iostream>
#include <map>
#include <vector>
#include "TVector3.h"
#include "TObject.h"
#include "TArrayI.h"

class PndSttTubeParameters;
class PndSttTube; 
class PndGeoSttPar;
class PndSttHit;
class FairGeoNode;


class FairHit;

using std::map;
using std::vector;

class PndSttGeometryMap : public TObject
{
 public:

  PndSttGeometryMap(); 
  PndSttGeometryMap(TClonesArray *tubearray, Int_t geoType);
  ~PndSttGeometryMap();
  PndSttGeometryMap(const PndSttGeometryMap& o) :
    TObject(o),
    fGeoType(o.fGeoType), 
    fVerbose(o.fVerbose),
    fStrawIndex(o.fStrawIndex),
    fStrawNeighbors(o.fStrawNeighbors),
    fSectorOfStraw(o.fSectorOfStraw),
    fLayerOfStraw(o.fLayerOfStraw),
    fAxialStraw(o.fAxialStraw),
    fSectorStart(o.fSectorStart),
    fSectorEnd(o.fSectorEnd),
    fStrawMapInitialized(o.fStrawMapInitialized),
    fNLayers(o.fNLayers),
    fNSectors(o.fNSectors),
    fNTubes(o.fNTubes),
    fNTubes_inner_parallel(o.fNTubes_inner_parallel),
    fNTubes_outer_parallel(o.fNTubes_outer_parallel),
    fNTubes_fillup_parallel(o.fNTubes_fillup_parallel),
    fNTubes_skewed(o.fNTubes_skewed),
    fNLayers_inner_parallel(o.fNLayers_inner_parallel),
    fNLayers_skew(o.fNLayers_skew),
    fNLayers_outer_parallel(o.fNLayers_outer_parallel),
    fNLayers_fillup_parallel(o.fNLayers_fillup_parallel),
    fStartTube(o.fStartTube),
    fEndTube(o.fEndTube),
    fShift(o.fShift),
    fShiftSkew(o.fShiftSkew),
    fTubeArray(o.fTubeArray)
  {};
  PndSttGeometryMap& operator=(const PndSttGeometryMap& o)
  {
    fGeoType=o.fGeoType; 
    fVerbose=o.fVerbose;
    fStrawIndex=o.fStrawIndex;
    fStrawNeighbors=o.fStrawNeighbors;
    fSectorOfStraw=o.fSectorOfStraw;
    fLayerOfStraw=o.fLayerOfStraw;
    fAxialStraw=o.fAxialStraw;
    fSectorStart=o.fSectorStart;
    fSectorEnd=o.fSectorEnd;
    fStrawMapInitialized=o.fStrawMapInitialized;
    fNLayers=o.fNLayers;
    fNSectors=o.fNSectors;
    fNTubes=o.fNTubes;
    fNTubes_inner_parallel=o.fNTubes_inner_parallel;
    fNTubes_outer_parallel=o.fNTubes_outer_parallel;
    fNTubes_fillup_parallel=o.fNTubes_fillup_parallel;
    fNTubes_skewed=o.fNTubes_skewed;
    fNLayers_inner_parallel=o.fNLayers_inner_parallel;
    fNLayers_skew=o.fNLayers_skew;
    fNLayers_outer_parallel=o.fNLayers_outer_parallel;
    fNLayers_fillup_parallel=o.fNLayers_fillup_parallel;
    fStartTube=o.fStartTube;
    fEndTube=o.fEndTube;
    fShift=o.fShift;
    fShiftSkew=o.fShiftSkew;
    fTubeArray=o.fTubeArray;
    return *this;
  };
  // ----------------------------
  void SetGeneralParameters();
  void GenerateStrawMap(Int_t map);
  Bool_t FillGeometryParameters();
  // ----------------------------

  // *** GEO TYPE 1 *** 
  void SetGeneralParametersGeoType1();
  void GenerateStrawMapGeoType1(Int_t map);
  Bool_t FillGeometryParametersGeoType1();

  void GenerateStrawMapTubeIDGeoType1();
  void GenerateStrawMapAngleGeoType1();

  int GetRow(int strawindex) const {return fLayerOfStraw.at(strawindex);}
  int GetSector(int strawindex) const {return fSectorOfStraw.at(strawindex);}
  const vector<int>& GetStrawRow(int sector, int row) const {return (fStrawIndex.find(sector))->second.at(row);}
  const vector<vector<int> >& GetStrawSector(int sector) const {return (fStrawIndex.find(sector))->second;}

  bool IsEdgeStraw(int strawindex) const;
  int IsSectorBorderStraw(int strawindex) const;

  bool IsAxialStraw(int strawindex) const {return fAxialStraw.at(strawindex);}
  bool IsSkewedStraw(int strawindex) const {return !(fAxialStraw.at(strawindex));}

  bool IsAxialRow(int rowindex) const;
  bool IsSkewedRow(int rowindex) const;

  TArrayI FindNeighborings(PndSttTube *tube);
  TArrayI FindNeighborings(int tubeId);

  TArrayI GetNeighboringsByMap(int tubeId);

  void FillStrawNeighborsMap();

  Double_t CalculateStrawPoca(PndSttHit* hit1, PndSttHit* hit2, TVector3& poca); //returns smallest distance between two Stt Straws. poca is set to the point between the two straws.

  double GetAngleBetweenTubes(int tubeID1, int tubeID2) const;

  bool InStraightLine(int tube1, int tube2, int tube3) const;

  // *********************

  void SetVerbose(int ver) { fVerbose = ver; }


 private:
  Int_t fGeoType, fVerbose;


  void GenerateAngles();
  map<int, vector< vector<int> > > fStrawIndex;
  map<int, TArrayI> fStrawNeighbors;
  vector<int> fSectorOfStraw;
  vector<int> fLayerOfStraw;
  vector<bool> fAxialStraw;
  vector<double> fSectorStart;
  vector<double> fSectorEnd;
  bool fStrawMapInitialized;


  // ------------------ SECTORS/LAYERS
  Int_t fNLayers, fNSectors, fNTubes, fNTubes_inner_parallel, fNTubes_outer_parallel, fNTubes_fillup_parallel, fNTubes_skewed, fNLayers_inner_parallel, fNLayers_skew, fNLayers_outer_parallel, fNLayers_fillup_parallel;
  Int_t **fStartTube, **fEndTube, **fShift, **fShiftSkew;

  
  TClonesArray *fTubeArray;

 protected:
  
  
  ClassDef(PndSttGeometryMap,2)
    
    };

#endif