using namespace std;
#include "TRandom.h"
#include <time.h>
#include <iostream> 
#include <iomanip>
#include "hwallhitf.h" //FK//
#include "hwallhitfsim.h"
#include "walldef.h"   //FK//
#include "hdebug.h"
#include "hades.h"
#include "hiterator.h"
#include "hruntimedb.h"
#include "hspectrometer.h"
#include "hdetector.h"
#include "hwalldetector.h" //FK//
#include "hwallraw.h"   //FK// 
#include "hwallrawsim.h"   
#include "hevent.h"
#include "hpartialevent.h"
#include "hcategory.h"
#include "hmatrixcategory.h"
#include "hrecevent.h"
#include "hlocation.h"
#include "hwallcalpar.h" //FK//
#include "hwalldigipar.h"
#include "hwallhit.h"   //FK//
#include "hwallhitsim.h"  
#include "hstartdef.h"
#include "hstart2hit.h"
#include "hwallgeompar.h"        //FK//
#include "hdetgeompar.h"         //FK//
#include "hgeomvector.h"         //FK//
#include "hspecgeompar.h"        //FK//
#include "hgeomvolume.h"           //FK//
#include "hgeomcompositevolume.h"  //FK//
#include "hgeomtransform.h"        //FK//

//*-- Created : 24/06/2000 by M.Golubeva
//*-- Modified : 27/04/2001 by T.Wojcik
//*-- Modified: 11/8/2005 by F. Krizek
// Modified by M.Golubeva 01.11.2006
//_HADES_CLASS_DESCRIPTION
/////////////////////////////////////////////////////////////////////
//
//  HWallHitf calibrates raw Wall data and puts output
//  values into cal data category
//
/////////////////////////////////////////////////////////////////////

void HWallHitFSim::initParContainer() {

    pWallDigiPar  = (HWallDigiPar *)gHades->getRuntimeDb()->getContainer("WallDigiPar");
    pWallGeometry = (HWallGeomPar *)gHades->getRuntimeDb()->getContainer("WallGeomPar");
    pSpecGeometry = (HSpecGeomPar *)gHades->getRuntimeDb()->getContainer("SpecGeomPar");
}

HWallHitFSim::HWallHitFSim(void) {

    fHitCat       = NULL;
    fRawCat       = NULL;
    pWallDigiPar  = NULL;
    pWallGeometry = NULL;
    fLoc.set(1,-1);
    iter          = NULL;

}

HWallHitFSim::HWallHitFSim(const Text_t *name,const Text_t *title) :
HWallHitF(name,title) {

    fHitCat       = NULL;
    fRawCat       = NULL;
    pWallDigiPar  = NULL;
    pWallGeometry = NULL;
    pSpecGeometry = NULL;
    fLoc.set(1,-1);
    iter          = NULL;
}

HWallHitFSim::~HWallHitFSim()
{
    if(iter){
	delete iter;
	iter = NULL;
    }
}

void HWallHitFSim::fillHit(HWallHitSim *hit, HWallRawSim *raw)
{

}

void HWallHitFSim::fillHitTrack(HWallHitSim *hit, HWallRawSim *raw)
{
    HWallHitSim *hs = (HWallHitSim *)hit;
    HWallRawSim *rs = (HWallRawSim *)raw;
    hs->setNTrack1(rs->getNTrack1());
    hs->setNTrack2(rs->getNTrack2());
}

Bool_t HWallHitFSim::init(void) {

    initParContainer();

    fRawCat = gHades->getCurrentEvent()->getCategory(catWallRaw);
    if (!fRawCat) {
	HWallDetector* wall = (HWallDetector*)(gHades->getSetup()->getDetector("Wall"));
	fRawCat = wall->buildMatrixCategory("HWallRawSim",0.5F);
	if (!fRawCat) return kFALSE;
	else gHades->getCurrentEvent()->addCategory(catWallRaw,fRawCat,"Wall");
    } else {
	if (fRawCat->getClass() != HWallRawSim::Class()) {
	    Error("HWallHitFSim::init()","Misconfigured input category");
	    return kFALSE;
	}
    }
    iter = (HIterator*)fRawCat->MakeIterator();  // cannot be "native", because
    // iter->getLocation() is used

    fHitCat = gHades->getCurrentEvent()->getCategory(catWallHit);
    if (!fHitCat) {
	HWallDetector* wall = (HWallDetector*)(gHades->getSetup()->getDetector("Wall"));
	fHitCat = wall->buildMatrixCategory("HWallHitSim",0.5F);
	if (!fHitCat) return kFALSE;
	else gHades->getCurrentEvent()->addCategory(catWallHit,fHitCat,"Wall");
    } else {
	if (fHitCat->getClass() != HWallHitSim::Class()) {
	    Error("HWallHitfSim::init()","Misconfigured output category");
	    return kFALSE;
	}
    }


    // Get Start Hit category. If StartHit is not defined
    // a Warning is displayed and fStartHitCat is set to 0

    fStartHitCat = gHades->getCurrentEvent()->getCategory(catStart2Hit);
    if (!fStartHitCat) Warning("init","Start Hit level not defining: setting start time to 0");

    return kTRUE;
}

void HWallHitFSim::fillGeometry(HWallHitSim *hit) {
    // Fills in the LAB position parameters for the given HTofHit.
    //
    // This is done by transforming the point (x,0,0) in the ROD coordinate
    // system to the LAB coordinate system. Where x is the reconstructed
    // x position inside the hit.

    HGeomVector rLocal,r;
    Float_t d,phi,theta,xx,yy,zz;
    Float_t rad2deg = 180./TMath::Pi();

    HModGeomPar *module= pWallGeometry->getModule(0);
    HGeomTransform &trans = module->getLabTransform();

    HGeomVolume *cellVol = module->getRefVolume()->getComponent(hit->getCell());
    HGeomTransform &cellTrans=cellVol->getTransform();

    // Fill r with the hit coordinates in the cell coordinate system
    // Since we do not have information about x,y,z coordinates of impact
    //  hi occures in centre thus z=0, x=0 and y=0.
    //  Note that (0,0,0) corresponds to the rod center.

    r.setX(0.);
    r.setY(0.);
    r.setZ(0.);

    rLocal = cellTrans.transFrom(r);  // transform to module coordinate system
    r      = trans.transFrom(rLocal);     // transform from module to LAB system

    HGeomVolume *tv = pSpecGeometry->getTarget(0);
    if (tv) r      -= tv->getTransform().getTransVector();   // correct for target position

    // Fill output
    d     = r.length();
    theta = (d>0.) ? (rad2deg * TMath::ACos(r.getZ() / d)) : 0.;
    phi   = rad2deg * TMath::ATan2( r.getY(), r.getX());
    if (phi < 0.) phi += 360.;
    if (tv)         r += tv->getTransform().getTransVector();   // correct for target position

    hit->setXYZLab(r.getX(), r.getY(), r.getZ());
    hit->setDistance( d );
    hit->setTheta(theta);
    hit->setPhi(phi);
    hit->getXYZLab(xx,yy,zz);

}


Int_t HWallHitFSim::execute(void) {

    HWallRawSim* raw = NULL;
    HWallHitSim* hit = NULL;
    Float_t calTime = 0, adc = 0;
    Int_t cell=-1;
    fLoc.set(1,-1);




    //------------------------------------------------------------------------
    // getting start time (some work must be done for embedding,
    // at the moment the geant tracks will get wrong subtraction!)
    //Float_t startTime         = 0.0;  // unused                                        pStartH->getTime()
    Float_t startTimeSmearing = 0.0;  // for simulation/embedding of sim tracks  pStartH->getResolution()
                                      // startTime  = startTimeSmearing  for simulation
                                      // startTime != startTimeSmearing  for embedding
                                      // startTimeSmearing == -1000      for real data
    if (fStartHitCat && fStartHitCat->getEntries()>0 ) {
	HStart2Hit* pStartH = 0;
	if((pStartH = (HStart2Hit *) fStartHitCat->getObject(0)) != NULL) {
	    //startTime = pStartH->getTime(); // unused 
            if(pStartH->getResolution() != -1000) startTimeSmearing = pStartH->getResolution();
	}
    }
    //------------------------------------------------------------------------

    Float_t startTimeLocal=startTimeSmearing;

    iter->Reset();   // this works only in split mode=2
    // (for 0 and 1 the iterator must be recreated)
    while ((raw=(HWallRawSim *)iter->Next())!=0) {

	fLoc[0] =  raw->getCell();
	if(fLoc[0] == -1) continue;

	if(raw->getTime() > 0.) {
	    fLoc[0] = raw->getCell();
	    cell=fLoc[0];

	    calTime =  pWallDigiPar->getTDC_Slope(cell)*raw->getTime()-pWallDigiPar->getTDC_Offset(cell);
	    calTime -=  startTimeLocal;

	    adc = 1.8*(raw->getCharge())/256.;
	    hit = (HWallHitSim*) fHitCat->getSlot(fLoc);
	    if(hit != NULL) {
		hit = new(hit) HWallHitSim;
		hit->setTime(calTime);
		hit->setCharge(adc);
		hit->setAddress(fLoc[0]);
		fillHit(hit,raw);
		fillGeometry(hit);
		fillHitTrack(hit,raw);
	    }
	}

    }
    return 0;

}


ClassImp(HWallHitFSim)