/*
 * PndLmdDigiProducer.cxx
 *
 *  Created on: Aug 10, 2009
 *      Author: huagen
 */

//the implementation of digiproducer

#include "PndLmdDigiProducer.h"
#include "PndLmdDigiPara.h"
#include "PndLmdDigiStrips.h"
#include "PndLmdMCPoint.h"
#include "PndLmdCalStrip.h"
#include "PndLmdStrips.h"
#include "PndLmdGeoHandling.h"

#include "TGeoManager.h"
#include "FairRun.h"
#include "FairRuntimeDb.h"
#include "FairGeoNode.h"
#include "FairGeoVector.h"
#include "PndStringVector.h"
#include "FairParamList.h"
#include "FairRootManager.h"

#include "TClonesArray.h"
#include "TVector3.h"

#include <iostream>
#include <vector>
using namespace std;


PndLmdDigiProducer::PndLmdDigiProducer() : FairTask("Pnd LMD Digi Producer")
{
  fBranchName = "PndLmdMCPoint";
}

PndLmdDigiProducer::PndLmdDigiProducer(Int_t k,SensorSegment segment)
                    :FairTask("Pnd LMD Digi Producer")
{
	fBranchName = "PndLmdMCPoint";
//	fDigiParTrap = digipar;
	fSegment = segment;
	}

PndLmdDigiProducer::PndLmdDigiProducer(Double_t k,SensorSide side)
                   :FairTask("Pnd LMD Digi Producer")
 {
	fBranchName = "PndLmdMCPoint";
  //  fDigiParTrap = digipar;
    fSide = side;
  }


PndLmdDigiProducer :: ~PndLmdDigiProducer()
{
	delete fGeoH;
	}

void PndLmdDigiProducer::SetParContainers()
{
	FairRun* ana = FairRun::Instance();
	FairRuntimeDb* rtdb = ana->GetRuntimeDb();
	fDigiParTrap = (PndLmdDigiPara*)(rtdb->getContainer("LmdStripDigiParaTrap"));
	//if(!fDigiParTrap) cout<<"The digi parameters has been set to container!"<<endl;
	//fDigiParTrap->print();
	}


InitStatus PndLmdDigiProducer::ReInit()
{
   SetParContainers();
   return kSUCCESS;
}

//init method for the digiproducer, input/output array, digipar file etc.
InitStatus PndLmdDigiProducer::Init()
{
    FairRun* ana = FairRun::Instance();
    FairRootManager* ioman = FairRootManager::Instance();

    fGeoH = new PndLmdGeoHandling(gGeoManager);

    if(! ioman)
    {
      std::cout<<"-E- PndLmdDigiProducer::Init :"
      <<" The FariRootManager is not initialized!"<<std::endl;
      return kFATAL;
    }

    //input array initialization
    fPointArray = (TClonesArray*)ioman->GetObject(fBranchName);
    if(! fPointArray)
    {
    	std::cout<<"-E- PndLmdDigiProducer::Init :"
    	<<"No PndLmdMCPoint Array"<<std::endl;
     return kERROR;
    }

    //creat and register output array
    fStripArray = new TClonesArray("PndLmdDigiStrips");
    ioman->Register("LMDStripDigis","LMD",fStripArray,kTRUE);

    if(! fDigiParTrap)
    {
       std::cout<<"-E- PndLmdDigiProducer::DigiParTrap Container doesn't exist!"<<std::endl;
       return kERROR;
    }
//print the digipar parameters for check
    fDigiParTrap->print();

    fStripCalTrapLeft = new PndLmdCalStrip(fDigiParTrap,kLEFT);
    cout<<"the digi paras for left segment!"<<endl;
    fStripCalTrapLeft->print();

    fStripCalTrapRight = new PndLmdCalStrip(fDigiParTrap,kRIGHT);
    cout<<"the digi paras for right segment!"<<endl;
    fStripCalTrapRight->print();

    std::cout<<"the PndLmdDigiProcuder initialized successfully!"<<std::endl;

    return kSUCCESS;
}

void PndLmdDigiProducer::Exec(Option_t* opt)
{
	//reset the output array
	fStripArray->Clear();

	//define variables
	PndLmdMCPoint* point = NULL;

	Int_t nPoint = fPointArray->GetEntriesFast();
	if(fVerbose>0){
		std::cout<<"the Nr of points is"<<nPoints<<std::endl;
	}

	//loop all points in an event
	Int_t iStrip = 0;
	for(Int_t iPoint = 0;iPoint<nPoints;iPoint++)
	{
		point = (PndLmdMCPoint*)fPointArray->At(iPoint);
		if(! point){
			std::cout<<"No point !"<<std::endl;
			continue;
		}

		fCurrentStripCalLeft = fStripCalTrapLeft;
		fCurrentStripCalRight = fStripCalTrapRight;
		fCurrentDigiPar = fDigiParTrap;

		//transform the point from global coordinate to local coordinate
		TVector3 posInL = fGeoH->MasterToLocalId(point->GetEntryPoint(),point->GetDetName());
		TVector3 posOutL = fGeoH->MasterToLocalId(point->GetExitPoint(),point->GetDetName());
		//std::cout<<"the point in global coordinate is "<<point->GetEntryPoint().X()<<","<<point->GetEntryPoint().Y()
		//<<","<<point->GetEntryPoint().Z()<<std::endl;
		std::cout<<"the Entry point in local coordinate is "<<posInL.X()<<","<<posInL.Y()<<","<<posInL.Z()<<std::endl;
		std::cout<<"the Exit point in local coordinate is "<<posOutL.X()<<","<<posOutL.Y()<<","<<posOutL.Z()<<std::endl;

		Double_t eLoss = point->GetEnergyLoss();
		std::cout<<"the eLoss of this point is: "<<eLoss<<endl;

		TVector3 pos,pos1 = posInL + posOutL;
		pos.SetXYZ(pos1.X()/2.,pos1.Y()/2.,pos1.Z()/2.);
		//for Left segment
		if(pos.X()>0)
		{
		//	std::cout<<"Hits on Left segment"<<std::endl;

		//calculate the cluster strips fired
		std::vector<PndLmdStrips> leftStrips =
			fCurrentStripCalLeft->DigiStripFromPoint(posInL.X(),posInL.Y(),posInL.Z(),
					                                posOutL.X(),posOutL.Y(),posOutL.Z(),
					                                eLoss);
		      if(leftStrips.size()!=0){
		    	  for(std::vector<PndLmdStrips>::const_iterator kit=leftStrips.begin();kit!=leftStrips.end();++kit)
		    	  {//register the strips fired
		    		  AddDigi(iStrip,iPoint,kLMDHitsStrip,point->GetDetName(),
					  fCurrentStripCalLeft->CalFeFromStrip(kit->GetIndex()),
					  fCurrentStripCalLeft->CalChannelFromStrip(kit->GetIndex()),
					  kit->GetCharge());
		    		  Int_t chan1 = (fCurrentStripCalLeft->CalFeFromStrip(kit->GetIndex()))*(fCurrentDigiPar->GetNrFeChannels())
				    +( fCurrentStripCalLeft->CalChannelFromStrip(kit->GetIndex()));
				  
				  cout<<"the iPoint is: "<<iPoint<<endl;
				  cout<<"the iStrip is :"<<iStrip-1<<endl;
				  cout<<"the channel is :"<<chan1<<endl;
				  cout<<"the information of digis: "<<point->GetDetName()<<","<<kit->GetCharge()<<endl;
				  cout<<endl;
				  
		    	  }
		      }else std::cout<<"the left segment is empty"<<std::endl;

		}else{// std::cout<<"Hits on Right segment"<<std::endl;
              std::vector<PndLmdStrips> rightStrips =
            	  fCurrentStripCalRight->DigiStripFromPoint(posInL.X(),posInL.Y(),posInL.Z(),
					                                posOutL.X(),posOutL.Y(),posOutL.Z(),
					                                eLoss);
        //      cout<<"Call the calStrip class to get digi successfully"<<endl;

              if(rightStrips.size()!=0){
            	  for(std::vector<PndLmdStrips>::const_iterator kit=rightStrips.begin();
            	       kit!=rightStrips.end();++kit){
            		    //register the strips fired
            		    AddDigi(iStrip,iPoint,kLMDHitsStrip,point->GetDetName(),
            		    		fCurrentStripCalRight->CalFeFromStrip(kit->GetIndex())+fCurrentDigiPar->GetNrLeftFe(),
            		    		fCurrentStripCalRight->CalChannelFromStrip(kit->GetIndex()),
            		    		kit->GetCharge());
                         Int_t chan2 = (fCurrentStripCalRight->CalFeFromStrip(kit->GetIndex())+fCurrentDigiPar->GetNrLeftFe())*(fCurrentDigiPar->GetNrFeChannels())
                                       +(fCurrentStripCalLeft->CalChannelFromStrip(kit->GetIndex()));

/*
            		    new((*fStripArray)[iStrip]) PndLmdDigiStrips(iPoint,kLMDHitsStrip,point->GetDetName(),
            		    		fCurrentStripCalRight->CalFeFromStrip(kit->GetIndex())+fCurrentDigiPar->GetNrLeftFe(),
            		    		fCurrentStripCalRight->CalChannelFromStrip(kit->GetIndex()),
            		    		kit->GetCharge());
            		        	iStrip++;
*/
            		//    cout<<endl;
            		    cout<<"the iPoint is: "<<iPoint<<endl;
            		    cout<<"the iStrip is :"<<iStrip-1<<endl;
            		    cout<<"the channel is :"<<chan2<<endl;
            		    cout<<"the information of digis: "<<point->GetDetName()<<","<<kit->GetCharge()<<endl;
            		    cout<<endl;

            	      }
              }else std::cout<<"the right segment is empty"<<std::endl;

		     }//judge the hits on left segment or right segment, then calculate strips

	}//loop all points for one event

 //   if(fVerbose>0)
    	std::cout<<"-I- PndLmdDigiProducer: "<<nPoints<<" PndLmdMCPoint, "
                    <<iStrip<<" PndLmdDigis"<<std::endl<<std::endl;

}//public method for Exec

void PndLmdDigiProducer::AddDigi(Int_t &iStrip,Int_t iPoint,Int_t detID,TString detName,
		                         Int_t fe, Int_t channel,Double_t charge)
{
    Bool_t found = kFALSE;
    PndLmdDigiStrips* aDigi= 0;
    for(Int_t skr=0;skr<iStrip && found==kFALSE;skr++){
    	aDigi = (PndLmdDigiStrips*)fStripArray->At(skr);
    	if(aDigi->GetDetID()==detID,
    	   aDigi->GetDetName()==detName,
    	   aDigi->GetFe()==fe,
    	   aDigi->GetChannel()==channel)
    	{
    	//	aDigi->AddCharge(charge);
    	//	aDigi->AddIndex(iPoint);
    		found=kTRUE;
    	}
    }

    if(found == kFALSE){
    	new((*fStripArray)[iStrip]) PndLmdDigiStrips(iPoint,detID,detName,fe,channel,charge);
    	iStrip++;
    }
}



ClassImp(PndLmdDigiProducer);