HoughTrans.cxx

Go to the documentation of this file.

//
// HoughTrans
//
// Mark's hough transform.
//
// Created:  M. Dorman -- May, 2005
//
// $Author: kordosky $ 
//
// $Revision: 1.3 $
// 
// $Name:  $
//
// $Id: HoughTrans.cxx,v 1.3 2005/08/02 11:22:56 kordosky Exp $
//

#include "HoughTrans.h"
#include <iostream>
#include <cmath>

HoughTrans::HoughTrans()//:houghSpace("houghSpace","houghSpace",80,-2.0,2.0,80,-4.0,4.0),vtxz(0)
{
  houghSpace = new TH2F("houghSpace","houghSpace",80,-2.0,2.0,80,-4.0,4.0);
  houghSpace->SetDirectory(0);
  vtxz = 0.0;
}

HoughTrans::~HoughTrans(){
  // Mark, you need to delete histograms created with new!!!
  delete houghSpace; houghSpace=0;

}

HoughTrans::HoughTrans(Int_t numSamps,Float_t lowGrad,Float_t highGrad)//:
//houghSpace("houghSpace","houghSpace",numSamps,lowGrad,highGrad,numSamps,-4.0,4.0),vtxz(0)
{
  
  houghSpace = new TH2F("houghSpace","houghSpace",numSamps,lowGrad,highGrad,numSamps,-4.0,4.0);
  houghSpace->SetDirectory(0);
  vtxz = 0.0;
}

HoughTrans::HoughTrans(Int_t numSamps,Float_t lowGrad,Float_t highGrad,Float_t lowIcept,Float_t highIcept)//:
//houghSpace("houghSpace","houghSpace",numSamps,lowGrad,highGrad,numSamps,lowIcept,highIcept),vtxz(0)
{
  houghSpace = new TH2F("houghSpace","houghSpace",numSamps,lowGrad,highGrad,numSamps,lowIcept,highIcept);
  houghSpace->SetDirectory(0);
  vtxz = 0.0;
}

void HoughTrans::SetVtxz(Float_t vertz)
{
  vtxz=vertz;
}

void HoughTrans::FillHough(NtpSRStrip* stp)
{
  FillHough(stp->z,stp->tpos);
}

void HoughTrans::FillHough(Float_t zPos, Float_t tPos)
{
  if(zPos<vtxz) return;
  for(Int_t i=1;i<houghSpace->GetNbinsX();i++){
    Float_t gradient=(houghSpace->GetXaxis())->GetBinCenter(i);
    Float_t icept=tPos-gradient*(zPos-vtxz);
    if(icept>4.0 || icept<-4.0) continue; //max MINOS transverse dimensions
    houghSpace->Fill(gradient,icept);
  }
}

void HoughTrans::ResetHough()
{
  houghSpace->Reset();
}

void HoughTrans::DeleteHough()
{
  delete houghSpace;
}

Int_t HoughTrans::GetPeakHeight()
{
  Int_t binmax = static_cast<Int_t>(houghSpace->GetBinContent(houghSpace->GetMaximumBin()));
  return binmax;
}

Int_t HoughTrans::GetPeakGradBin()
{
  Int_t x=0;
  Int_t y=0;
  Int_t z=0;
  houghSpace->GetMaximumBin(x,y,z);
  return x;
}

Int_t HoughTrans::GetPeakIceptBin()
{
  Int_t x=0;
  Int_t y=0;
  Int_t z=0;
  houghSpace->GetMaximumBin(x,y,z);
  return y;
}

Float_t HoughTrans::GetPeakGradVal()
{
  Int_t x=0;
  Int_t y=0;
  Int_t z=0;
  houghSpace->GetMaximumBin(x,y,z);
  return (houghSpace->GetXaxis())->GetBinCenter(x);
}

Float_t HoughTrans::GetPeakIceptVal()
{
  Int_t x=0;
  Int_t y=0;
  Int_t z=0;
  houghSpace->GetMaximumBin(x,y,z);
  return (houghSpace->GetYaxis())->GetBinCenter(y);
}

Float_t HoughTrans::GetRms(Float_t frac)
{
  Int_t bincount = 0;
  Int_t mRmsInput[6400];
  Int_t cRmsInput[6400];
  for(Int_t mBin=1;mBin<houghSpace->GetNbinsX();mBin++){
    for(Int_t cBin=1;cBin<houghSpace->GetNbinsY();cBin++){
      if(houghSpace->GetBinContent(mBin,cBin)>=frac*GetPeakHeight()){
        mRmsInput[bincount]=mBin;
        cRmsInput[bincount]=cBin;
        bincount++;
      }
    }
  }
  if(bincount==0) return 0;
  Float_t squaresum = 0;
  for (Int_t count=0;count<bincount;count++) {
    squaresum+=(pow(static_cast<Float_t>((mRmsInput[count]-GetPeakGradBin())),2)+pow(static_cast<Float_t>((cRmsInput[count]-GetPeakIceptBin())),2));
  }
  Float_t Rms = sqrt(squaresum/bincount);
  return Rms;
}

---