/* 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/>.
 *
 */

#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
 *  @author A. Dobbs
 *  @brief Class representing straight tracks in the tracker formed by pattern recognition
 */
class SciFiStraightPRTrack : public SciFiBasePRTrack {
  public:
    /** @brief Default constructor */
    SciFiStraightPRTrack();

    /** @brief Constructor using explicit parameters
     *  @param tracker The tracker number
     *  @param x0 Intercept of the x-z fit in local tracker coordinates
     *  @param mx Gradient of the x-z fit in local tracker coordinates
     *  @param x_chisq ChiSq of the x-z fit
     *  @param y0 Intercept of the y-z fit in local tracker coordinates
     *  @param my Gradient of the y-z fit in local tracker coordinates
     *  @param y_chisq ChiSq of the y-z fit
     *  @param covariance Covariances of the fit
     */
    SciFiStraightPRTrack(int tracker, double x0, double mx, double x_chisq,
                         double y0, double my, double y_chisq,
                         const DoubleArray& covariance);

    /** @brief Constructor using SimpleLines
     *  @param tracker The tracker number
     *  @param line_x SimpleLine used to construct track
     *  @param line_y SimpleLine used to construct track
     *  @param covariance Covariances of the fit
     */
    SciFiStraightPRTrack(int tracker, SimpleLine line_x, SimpleLine line_y,
                         const DoubleArray& covariance);

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

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

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

    /** Combine the different chi sqs into one final value */
    virtual void form_total_chi_squared();

    /** 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;                 /** Tracker number, 0 for upstream, 1 for downstream */
    static const int _type = 0;   /** Track type identifier, 0 for straight, 1 for helical */
    double _x0;                   /** Intercept of the x-z fit in local tracker coordinates */
    double _mx;                   /** Gradient of the x-z fit in local tracker coordinates */
    double _x_chisq;              /** ChiSq of the x-z fit */
    double _y0;                   /** Intercept of the y-z fit in local tracker coordinates */
    double _my;                   /** Gradient of the y-z fit in local tracker coordinates */
    double _y_chisq;              /** ChiSq of the y-z fit */
    double _global_x0;            /** Intercept of the x-z fit in global coordinates */
    double _global_mx;            /** Gradient of the x-z fit in global coordinates */
    double _global_y0;            /** Intercept of the y-z fit in global coordinates */
    double _global_my;            /** Gradient of the y-z fit in global coordinates */

    MAUS_VERSIONED_CLASS_DEF(SciFiStraightPRTrack)
};

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

} // ~namespace MAUS

#endif