/* This file is part of MAUS: http://micewww.pp.rl.ac.uk:8080/projects/maus
 *
 * MAUS is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * MAUS 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with MAUS.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

/** @class SciFiStraightPRTrack Pattern Recognition straight track class */

#ifndef  _SRC_COMMON_CPP_DATASTRUCTURE_SCIFISTRAIGHTPRTRACK_HH_
#define _SRC_COMMON_CPP_DATASTRUCTURE_SCIFISTRAIGHTPRTRACK_HH_

// C++ headers
#include <vector>

// MAUS headers
#include "src/common_cpp/Utils/VersionNumber.hh"
#include "src/common_cpp/DataStructure/SciFiSpacePoint.hh"
#include "src/common_cpp/DataStructure/SciFiBasePRTrack.hh"
#include "src/common_cpp/Recon/SciFi/SimpleLine.hh"

namespace MAUS {

class SciFiStraightPRTrack : public SciFiBasePRTrack {
  public:
    /** Default constructor */
    SciFiStraightPRTrack();

    /** Constructor using explicit parameters */
    SciFiStraightPRTrack(int tracker, double x0, double mx, double x_chisq,
                         double y0, double my, double y_chisq,
                         const DoubleArray& covariance);

    /** Constructor using SimpleLines */
    SciFiStraightPRTrack(int tracker, SimpleLine line_x, SimpleLine line_y,
                         const DoubleArray& covariance);

    /** Copy constructor */
    SciFiStraightPRTrack(const SciFiStraightPRTrack &_strk);

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

    /** Equality operator - any pointers are deep copied */
    SciFiStraightPRTrack& operator=(const SciFiStraightPRTrack &_strk);


    /** Get the x position of the track at z=0 */
    double get_x0() const { return _x0; }

    /** Get the global x position of the track at z=0 */
    double get_global_x0() const { return _global_x0; }

    /** Set the x position of the track at z=0 */
    void set_x0(double x0) { _x0 = x0; }

    /** Set the global x position of the track at z=0 */
    void set_global_x0(double global_x0) { _global_x0 = global_x0; }

    /** Get the gradient of the line in x-z */
    double get_mx() const { return _mx; }

    /** Get the gradient of the line in global x-z */
    double get_global_mx() const { return _global_mx; }

    /** Set the gradient of the line in x-z */
    void set_mx(double mx) { _mx = mx; }

    /** Set the gradient of the line in global x-z */
    void set_global_mx(double global_mx) { _global_mx = global_mx; }

    /** Get the chi_2 of the line fit in x-z */
    double get_x_chisq() const { return _x_chisq; }

    /** Set the chi_2 of the line fit in x-z */
    void set_x_chisq(double x_chisq) { _x_chisq = x_chisq; }

    /** Get the y position of the track at z=0 */
    double get_y0() const { return _y0; }

    /** Get the global y position of the track at z=0 */
    double get_global_y0() const { return _global_y0; }

    /** Set the y position of the track at z=0 */
    void set_y0(double y0) { _y0 = y0; }

    /** Set the global y position of the track at z=0 */
    void set_global_y0(double global_y0) { _global_y0 = global_y0; }

    /** Get the gradient of the line in y-z */
    double get_my() const { return _my; }

    /** Get the gradient of the line in global y-z */
    double get_global_my() const { return _global_my; }

    /** Set the gradient of the line in y-z */
    void set_my(double my) { _my = my; }

    /** Set the gradient of the line in global y-z */
    void set_global_my(double global_my) { _global_my = global_my; }

    /** Get the chi_2 of the line fit in y-z */
    double get_y_chisq() const { return _y_chisq; }

    /** Set the chi_2 of the line fit in y-z */
    void set_y_chisq(double y_chisq) { _y_chisq = y_chisq; }

    /** Get the tracker number */
    int get_tracker() const { return _tracker; }

    /** Set the tracker number */
    void set_tracker(int tracker) { _tracker = tracker; }

    /** Get the track type, for straight, 1 for helical */
    int get_type() const { return _type; }


  private:
    int _tracker;
    static const int _type = 0; // 0 for straight, 1 for helical

    double _x0;
    double _mx;
    double _x_chisq;
    double _y0;
    double _my;
    double _y_chisq;
    double _global_x0;
    double _global_mx;
    double _global_y0;
    double _global_my;

    MAUS_VERSIONED_CLASS_DEF(SciFiStraightPRTrack)
};

typedef std::vector<SciFiStraightPRTrack*> SciFiStraightPRTrackPArray;
typedef std::vector<SciFiStraightPRTrack> SciFiStraightPRTrackArray;

} // ~namespace MAUS

#endif