Commit afeb585c authored by Domanski, Marian M.'s avatar Domanski, Marian M.

fix circular imports

parent a83d6cae
......@@ -2,8 +2,6 @@ from abc import ABC, abstractmethod
import numpy as np
from fluegg.results import FullResults, QuantileResults
class ResultsRecorder(ABC):
......@@ -106,6 +104,17 @@ class FullResultsRecorder(ResultsRecorder):
np.nan, (simclock.number_of_times(),
particles.position().shape[0], ))
def _normalize_axis(self, position_axis):
if position_axis == 0:
abs_length = self._domain_length
elif position_axis == 1:
abs_length = self._width
elif position_axis == 2:
abs_length = self._depth
positions = self._positions[:, :, position_axis] / abs_length
return positions
def depth(self):
""" returns depth array
"""
......@@ -167,6 +176,8 @@ class FullResultsRecorder(ResultsRecorder):
"""
from fluegg.results import FullResults
return FullResults(self)
def width(self):
......@@ -226,4 +237,6 @@ class QuantileResultsRecorder(ResultsRecorder):
"""
from fluegg.results import QuantileResults
return QuantileResults(self)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment