CorrelationAverageTutorial: init.py

#from ex import Lattice,ReciprocalUnitCell,SpatialUnitCell
#from ex.gui import LatticeOverlay,UnitCellOverlay
#from alg import FFT,Conjugate,PeakSearch,HighPassFilter
#from iplt import PeakList
#from ex.alg import LatticeFilter, GaussianLatticeFilter

from ost.geom import *
from ost.io import LoadImage,SaveImage
from ost.img import *
from ost.img.alg import *
from iplt import Lattice,ReciprocalUnitCell,SpatialUnitCell
from iplt.alg import *
from iplt.gui import LatticeOverlay,UnitCellOverlay
from ost.gui import PointlistOverlay

def ShowPeaks(viewer,peaklist):
    peak_overlay=PointlistOverlay()
    peak_overlay.Add(PointList(peaklist))
    viewer.AddOverlay(peak_overlay)

def CorrelationAverage(image,peaklist,width,height):
        result=CreateImage(Size(width,height))
        result.SetSpatialSampling(image.GetSpatialSampling())
        for p in peaklist:
                itmp=image.Extract(Extent(Size(width,height),p))
                itmp.SetSpatialOrigin(Point(0,0))
                result+=itmp
        return result

def CreateLatticeFromUnitCell(uc,sampling):
        return Lattice(uc.GetVecA()/sampling[0],uc.GetVecB()/sampling[1])

def FilterPeaklist(peaklist,real_lattice,rel_dis):
        result=PeakList()
        for p in peaklist:
                comp=real_lattice.CalcComponents(p)
                dx=abs(comp[0]-round(comp[0]))
                dy=abs(comp[0]-round(comp[0]))
                if rel_dis*rel_dis>dx*dx+dy*dy:
                        result.push_back(p)
        return result
        
def CrossCorrelate(im,ref):
        imfft=im.Apply(alg.DFT())
        reffft=ref.Apply(alg.DFT())
        reffft.ApplyIP(alg.Conj())
        result=imfft*reffft
        result.ApplyIP(alg.DFT())
        return result