#include <TMath.h>
#include <TVector3.h>
using namespace ROOT;

#include <RAT/MuonClassifier.hh>
using namespace RAT;
using namespace RAT::Classifiers;

#include <vector>
using namespace std;

DS::ClassifierResult MuonClassifier::GetClassification(){

  fClassifierResult.Reset();
  if( fPMTData.empty() )
    return fClassifierResult;

  double totalQ = 0.0;

  for( vector<FitterPMT>::const_iterator iPMT = fPMTData.begin(); iPMT != fPMTData.end(); ++iPMT ){
    if( (iPMT->GetQHL()) > 0.){
      totalQ += ( iPMT->GetQHL() );
    }
  }

  const size_t numHits = fPMTData.size();

  if(numHits <= 9000 && totalQ < 25000000.){
    bool medium = false;
    fClassifierResult.SetValid(true);
    fClassifierResult.SetClassification("medium", medium);
    return fClassifierResult;
  }
  else{
    bool medium = true;
    fClassifierResult.SetValid(true);
    fClassifierResult.SetClassification("medium", medium);
    return fClassifierResult;
  }
}