#ifndef CORRECTIONS_C
#define CORRECTIONS_C

#include "../src/TCcluster.h"
#include "../src/TCtrack.h"
#include "../src/TCalign.h"
#include <algorithm>
#include <vector>
#include <set>

//v correction function from fit to profile
double f1(double x){
  return -0.0096+0.024*x;
}
double f2(double x){
  return -0.0154+0.042*x;
}
double f3(double x){
  return -0.0154+0.039*x;
}
double f4(double x){
  return -0.0597+0.152*x;
}
double f5(double x){
  if (x<0.1){
	return -0.01253;
  }
  if (x>0.1&&x<0.2){
	return -0.01069;
  }
  if (x>0.2&&x<0.3){
	return -0.006227;
  }
  if (x>0.3&&x<0.4){
	return 0.00002076;
  }
  if (x>0.4&&x<0.5){
	return 0.002216;
  }
  if (x>0.5&&x<0.6){
	return 0.00589;
  }
  if (x>0.6&&x<0.7){
	return 0.01087;
  }
  if (x>0.7){
	return 0.01572;
  }
}
double f6(double x){
  return 0.04096*x-0.01777;
}

TCtrack vertical(TCtrack t){

  //  std::vector<int> vecInt;
  std::vector<TCcluster> tpc;//will hold clusters in TPC
  for(int i=0;i<t.nCl();++i){
    TCcluster c = t.getCl(i);
    if(c.getId()>=200 && c.getId()<=299){
	  tpc.push_back(c);
	}
  }
  //std::sort(tpc.begin(),tpc.end(),TCclusterSortUVW());

  int detId;
  std::vector<TCcluster> corrClusters;
  for(unsigned int icl=0;icl<tpc.size();++icl){   

	detId=t.getCl(icl).getId();
	if(!t.getCl(icl).getFit()) continue;
	TVector3 pos = t.getCl(icl).posUVW();

	if(t.getCl(icl).nPadY()==1){
	  if(t.getCl(icl).posUVW().Y()<0.050001)continue;
	  if(t.getCl(icl).posUVW().Y()>0.7499999)continue;
	}
	if(t.getCl(icl).nPadY()==2){
	  if(t.getCl(icl).posUVW().Y()<0.1499999)continue;
	  if(t.getCl(icl).posUVW().Y()>0.6500001)continue;
	}


	if(t.getCl(icl).nPadY()==1)
	  pos.SetY(pos.Y()-f1(pos.Y()));
	if(t.getCl(icl).nPadY()==2)
	  pos.SetY(pos.Y()-f2(pos.Y()));
	if(t.getCl(icl).nPadY()==3)
	  pos.SetY(pos.Y()-f3(pos.Y()));
	if(t.getCl(icl).nPadY()>3)
	  pos.SetY(pos.Y()-f4(pos.Y()));
	  
	//pos.SetZ(pos.X());
	//pos.SetY(0.4);
	t.getCl(icl).posUVW(pos);
	  
	corrClusters.push_back(t.getCl(icl));
	
  }

  TCtrack corrTrack;
  corrTrack.addClusters(corrClusters);
  TCalign::getInstance();
  corrTrack.fit(detId);
  return corrTrack;
   
}

TCtrack mean_vertical(TCtrack t){

  std::vector<TCcluster> tpc;//will hold clusters in TPC
  for(int i=0;i<t.nCl();++i){
    TCcluster c = t.getCl(i);
    if(c.getId()>=200 && c.getId()<=299){
	  tpc.push_back(c);
	}
  }

  int detId;
  std::vector<TCcluster> corrClusters;
  for(unsigned int icl=0;icl<tpc.size();++icl){   

	detId=t.getCl(icl).getId();
	if(!t.getCl(icl).getFit()) continue;
	TVector3 pos = t.getCl(icl).posUVW();
	TVector3 res = t.getCl(icl).getRes();

	pos.SetY(pos.Y()-100*f5(pos.Y()));

	//t.getCl(icl).getRes(res);
  	t.getCl(icl).posUVW(pos);
	  
	corrClusters.push_back(t.getCl(icl));
	
  }

  TCtrack corrTrack;
  corrTrack.addClusters(corrClusters);
  TCalign::getInstance();
  corrTrack.fit(detId);
  return corrTrack;
   
}

TCtrack without_end_pads(TCtrack t){

  std::vector<TCcluster> tpc;//will hold clusters in TPC
  for(int i=0;i<t.nCl();++i){
    TCcluster c = t.getCl(i);
    if(c.getId()>=200 && c.getId()<=299){
	  tpc.push_back(c);
	}
  }

  int detId;
  std::vector<TCcluster> corrClusters;
  for(unsigned int icl=0;icl<tpc.size();++icl){   
	TCcluster c = t.getCl(icl);
	detId=c.getId();
	if(!c.getFit()) continue;
	TVector3 pos = c.posUVW();
// 	if(pos.Y()<0.1) continue;
// 	if(pos.Y()>0.7) continue;
// 	if(pos.Y()>0.30&&pos.Y()<0.5) continue;
 	if(pos.Y()<0.1) c.setFit(false);
 	if(pos.Y()>0.7) c.setFit(false);
 	if(pos.Y()>0.30&&pos.Y()<0.5) c.setFit(false);


	//pos.SetY(pos.Y());

	//c.posUVW(pos);
	  
	corrClusters.push_back(c);

	//std::cout << "@@@ " << c.getFit() << " " <<c.posUVW().Y() << " "  << corrClusters.at(corrClusters.size()-1).getFit() << " " << corrClusters.at(corrClusters.size()-1).posUVW().Y() << " " << c.getRes().Y() << std::endl;
  }

  TCtrack corrTrack;
  corrTrack.addClusters(corrClusters);
  TCalign::getInstance();
  corrTrack.fit(detId);
  for(int i=0;i<corrTrack.nCl();++i){
	std::cout << "@@@ " << corrTrack.getCl(i).getFit() << std::endl;
	std::cout << "@@@ " << corrTrack.getCl(i).posUVW().Y() << std::endl;
	corrTrack.getCl(i).getRes().Print();
  }
  return corrTrack;
   
}

#endif