You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
This repo is archived. You can view files and clone it, but cannot push or open issues/pull-requests.
|
|
|
|
import pandas as pd
|
|
|
|
|
import numpy as np
|
|
|
|
|
import os
|
|
|
|
|
import importer as imp
|
|
|
|
|
import trajectory as tr
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class SamplePath():
|
|
|
|
|
|
|
|
|
|
def __init__(self, files_path=os.getcwd() + "/main_package/data"):
|
|
|
|
|
self.importer = imp.Importer(files_path)
|
|
|
|
|
self.trajectories = []
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def build_trajectories(self):
|
|
|
|
|
self.importer.import_data_from_csv()
|
|
|
|
|
self.importer.merge_value_columns_in_all_frames()
|
|
|
|
|
self.importer.drop_unneccessary_columns_in_all_frames()
|
|
|
|
|
for data_frame in self.importer.get_data_frames():
|
|
|
|
|
trajectory = tr.Trajectory(data_frame)
|
|
|
|
|
self.trajectories.append(trajectory)
|
|
|
|
|
self.importer.clear_data_frames()
|
|
|
|
|
|
|
|
|
|
def get_number_trajectories(self):
|
|
|
|
|
return len(self.trajectories)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
######Veloci Tests#######
|
|
|
|
|
|
|
|
|
|
s1 = SamplePath()
|
|
|
|
|
s1.build_trajectories()
|
|
|
|
|
print(s1.get_number_trajectories())
|
|
|
|
|
#print(s1.trajectories[0].get_trajectory())
|
|
|
|
|
|
|
|
|
|
|
