PyCTBN/main_package/classes/sample_path.py

import pandas as pd
import numpy as np
import os
import importer as imp
import trajectory as tr


class SamplePath():
    """
    Rappresenta l'aggregazione di una o più traiettorie.
    Ha il compito dato di costruire tutte gli oggetti Trajectory a partire
    dai dataset contenuti nella directory files_path.

    :importer: l'oggetto Importer che ha il compito di caricare i dataset
    :trajectories: lista contenente le traiettorie create
    """

    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)