LinearRegressionProcessor.h

#ifndef LINEARREGRESSIONPROCESSOR_H
#define LINEARREGRESSIONPROCESSOR 1

#include "marlin/Processor.h"
//#include "lcio.h"
//#include <string>
#include <vector>

#include <EVENT/TrackerHit.h>

namespace AIDA
{

class IHistogram1D;
class IHistogram2D;


}


namespace marlintpc
{

class LinearRegressionProcessor : public marlin::Processor
{

    public:

        LinearRegressionProcessor();


        ~LinearRegressionProcessor();


        Processor * newProcessor();


        void init();


        void processRunHeader(lcio::LCRunHeader *run);


        virtual void processEvent(lcio::LCEvent *evt);


        virtual void check(lcio::LCEvent *evt);


        virtual void end();

        /* Sort function to sort the Hits along the track
         */
        static bool compareHit(const std::pair<TrackerHit*, double> & pair1,
                               const std::pair<TrackerHit*, double> & pair2)
        {
            return pair1.second < pair2.second;
        }

    private:

        std::string _input_track_candidates_collection_name;

        std::string _output_seed_tracks_collection_name;

        int _outputIsTransient; //< if not 0 the hits collection is set transient
        //(default 0)


        std::vector<AIDA::IHistogram2D *> HistoLineVec;
        std::vector<AIDA::IHistogram2D *> HistoTrackVec;
};

}// namespace marlintoc

#endif // TRACKSEEDERPROCESSOR_H