from gizela.data.ObsDirection import ObsDirection
from gizela.data.ObsZAngle import ObsZAngle
from gizela.data.ObsSDistance import ObsSDistance
from gizela.data.ObsCluster import ObsCluster
from math import pi
dist = ObsDistance("AB", 100.01)
dist2 = ObsDistance("AB", 100.03)
dist3 = ObsDistance(fromid="AB", toid="CD", val=100.05)
dir = ObsDirection("AB", 90.01/200*pi)
dir2 = ObsDirection("AB", 90.03/200*pi)
zen = ObsZAngle("AB", 90.01/200*pi)
sdist = ObsSDistance("AB", 100.01)
cl = ObsCluster("CD")
cl.append_obs(dist)
#cl.append_obs(dist2)
#cl.append_obs(dist3)
cl.append_obs(dir)
cl.append_obs(dir2)
cl.append_obs(zen)
cl.append_obs(sdist)
cl3 = ObsCluster("AB")
cl3.append_obs(dist3)
lst = ObsClusterList()
lst.append_cluster(cl)
#import copy
#cl2 = copy.deepcopy(cl)
lst.append_cluster(cl3)
# obsCluster.append_obs(ObsDirection("8", 0))
# obsCluster.append_obs(ObsZAngle("999", math.pi/2))
# obsCluster.append_obs(ObsZAngle("1", -1*math.pi/4))
# obsCluster.append_obs(ObsZAngle("2", 1*math.pi/4))
# obsCluster.append_obs(ObsZAngle("3", 0))
# obsCluster.append_obs(ObsZAngle("14", 0))
# obsCluster.append_obs(ObsZAngle("15", 0))
# obsCluster.append_obs(ObsZAngle("16", 0))
# obsCluster.append_obs(ObsZAngle("17", 0))
# obsCluster.append_obs(ObsZAngle("18", 0))
# obsCluster.append_obs(ObsSDistance("91", 100))
obsCluster.append_obs(ObsHeightDiff("999","11", 0, 1))
obsCluster.append_obs(ObsHeightDiff("999","12", 0, 1))
obsCluster.append_obs(ObsHeightDiff("999","13", 0, 1))
obsCluster.append_obs(ObsHeightDiff("999","14", 0, 1))
obsCluster.append_obs(ObsHeightDiff("999","15", 0, 1))
obsCluster.append_obs(ObsHeightDiff("999","16", 0, 1))
obsCluster.append_obs(ObsHeightDiff("999","17", 0, 1))
obsCluster.append_obs(ObsHeightDiff("999","18", 0, 1))
obsClusterList.append_cluster(obsCluster)
# corr = ObsCorrSphere(ellipsoid, centralPointGeo, centralPointLoc, pointList)
corr = ObsCorrSphere(Ellipsoid(), centralPointGeo, centralPointLoc, 'sw', 'left-handed', pointList, obsClusterList)
obsClusterList.compute_corr(corr)
#obsCluster.compute_obs(corr)
