def generate_rectangle_sets(fname, r, p, q, max_side_length):

    trajectories = paths.read_dong_csv("/uusoc/scratch/raven4/dongx/trajectory_paper_code/data/samples/{}.tsv".format(fname))
    disc = utils.disc_to_func("disc")
    while True:
        red, blue, planted_reg, _ = pyscan.plant_full_square(trajectories, r, p, q, disc)
        print(planted_reg.upX() - planted_reg.lowX(), max_side_length)
        if planted_reg.upX() - planted_reg.lowX() < max_side_length:
            break
    return red, blue
def generate_disk_sets(fname, r, p, q, min_r, max_r):

    disc = utils.disc_to_func("disc")
    trajectories = paths.read_dong_csv("/uusoc/scratch/raven1/dongx/trajectory_paper_code/data/samples/{}.tsv".format(fname))

    while True:
        red, blue, planted_reg , planted_mx = pyscan.plant_full_disk(trajectories, r, p, q, disc)
        print(min_r, planted_reg.get_radius(), max_r)
        if min_r < planted_reg.get_radius() < max_r:
            break
    return red, blue
Exemple #3
0
                "s": s,
                "time": end_time - start_time,
                "m_disc_approx": mx,
                "m_disc": actual_mx
            }
            writer.writerow(row)
            f.flush()
            print(row)
            if max_time is not None and end_time - start_time > max_time:
                return


if __name__ == "__main__":

    #trajectories = paths.read_geolife_files(10000)
    trajectories = paths.read_dong_csv(
        "/data/Dong_sets/Trajectory_Sets/samples/osm_eu_sample_100k_nw.tsv")
    st_pts, end_pts = pyscan.trajectories_to_flux(trajectories)
    st_pts = [
        pyscan.WPoint(1.0, float(p[0]), float(p[1]), 1.0) for p in st_pts
    ]
    end_pts = [
        pyscan.WPoint(1.0, float(p[0]), float(p[1]), 1.0) for p in end_pts
    ]

    r = .01
    q = .8
    eps_r = .01

    for region_name in ["halfplane", "disk"]:
        scan_f = utils.range_to_func(region_name)
        print("Started planting")
        chord_l = math.sqrt(4 * alpha * i -  2 * alpha * alpha)
        sample = [pyscan.grid_direc_kernel(pyscan.dp_compress(traj, alpha), chord_l, alpha) for traj in trajectories]
        print("Grid Direc Error {} {}".format(50 * i, 50 * post_process_error(test_disk_error(trajectories, sample, alpha, max_r))))


    sample = [pyscan.grid_kernel(pyscan.dp_compress(traj, alpha), alpha) for traj in trajectories]
    print("Grid : {}".format(50 * post_process_error(test_disk_error(trajectories, sample, alpha, max_r))))

    sample = [pyscan.halfplane_kernel(pyscan.dp_compress(traj, alpha), alpha) for traj in trajectories]
    print("Halfplane : {}".format(50 * post_process_error(test_disk_error(trajectories, sample, alpha, max_r))))

    sample = [pyscan.dp_compress(traj, alpha) for traj in trajectories]
    print("DP Error: {}".format(50 * post_process_error(test_disk_error(trajectories, sample, alpha, max_r))))

#trajectories = paths.read_geolife_files(1000)
trajectories = paths.read_dong_csv("/data/Dong_sets/Trajectory_Sets/samples/osm_eu_sample_10k_nw.tsv", filter_long=True)

#trajectories = clean(trajectories)
#print(len(trajectories))
#plot_plane_full_trajectories(trajectories, .25, 1.0, 0.0)

r=.02
p =.5
q= 1.0
eps_r=.01
plot_trajectory_set(trajectories, 10000)

#plot_full_trajectories_intersection_rect(trajectories, r)
# OSM EU 1/30000, 1/300
# BJTAXI 1/500 1/5
#testing(trajectories, 1/30000, 1/300)
def generate_halfplane_sets(fname, r, p, q):
    disc = utils.disc_to_func("disc")
    trajectories = paths.read_dong_csv("/uusoc/scratch/raven1/dongx/trajectory_paper_code/data/samples/{}.tsv".format(fname))
    red, blue, planted_reg, planted_mx = pyscan.plant_full_halfplane(trajectories, r, p, q, disc)
    return red, blue
Exemple #6
0
import flux_testing
import paths
import utils
import pyscan
import file

if __name__ == "__main__":

    #trajectories = paths.read_geolife_files(100)
    trajectories = paths.read_dong_csv(file.PATH +
                                       "bjtaxi_samples_100k_nw.tsv")

    st_pts, end_pts = pyscan.trajectories_to_flux(trajectories)
    st_pts = [
        pyscan.WPoint(1.0, float(p[0]), float(p[1]), 1.0) for p in st_pts
    ]
    end_pts = [
        pyscan.WPoint(1.0, float(p[0]), float(p[1]), 1.0) for p in end_pts
    ]

    r = .01
    q = .2
    p = .5
    eps_r = .001
    disc = utils.disc_to_func("disc")
    #red, blue, _, _ = pyscan.plant_partial_disk(trajectories, r, p, q, eps_r, disc)

    for region_name in ["disk"]:  #, "rectangle", "disk"]:
        if region_name == "disk":
            method = pyscan.plant_disk
        elif region_name == "rectangle":