From e019746996c8e5e6cb5a7d09eb9fac45d6f2e043 Mon Sep 17 00:00:00 2001
From: Mikhail Chaikovskii <chaykovskiym.e@gmail.com>
Date: Mon, 10 Apr 2023 18:46:35 +0300
Subject: [PATCH] tests

---
 app/rule_extention.py                | 10 ++++------
 examples/example_simstep.py          |  5 +++--
 rostok/virtual_experiment/sensors.py |  3 +++
 3 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/app/rule_extention.py b/app/rule_extention.py
index 13515d8..64c951b 100644
--- a/app/rule_extention.py
+++ b/app/rule_extention.py
@@ -3,14 +3,12 @@ import networkx as nx
 import numpy as np
 import pychrono as chrono
 
-
-import numpy as np
-
-from rostok.block_builder_api.block_blueprints import TransformBlueprint, PrimitiveBodyBlueprint, \
-EnvironmentBodyBlueprint, RevolveJointBlueprint
+from rostok.block_builder_api.block_blueprints import (
+    EnvironmentBodyBlueprint, PrimitiveBodyBlueprint, RevolveJointBlueprint,
+    TransformBlueprint)
+from rostok.block_builder_api.block_parameters import JointInputType
 from rostok.block_builder_api.easy_body_shapes import Box
 from rostok.block_builder_chrono.blocks_utils import FrameTransform
-from rostok.block_builder_api.block_parameters import JointInputType
 from rostok.graph_grammar import node_vocabulary, rule_vocabulary
 from rostok.graph_grammar.node import ROOT
 
diff --git a/examples/example_simstep.py b/examples/example_simstep.py
index 8269923..1877911 100644
--- a/examples/example_simstep.py
+++ b/examples/example_simstep.py
@@ -19,7 +19,7 @@ mechs = [
 mechs = [get_terminal_graph_no_joints]
 for get_graph in mechs:
     # Constants
-    MAX_TIME = 0.5
+    MAX_TIME = 1000
     TIME_STEP = 1e-3
 
     graph = get_graph()
@@ -46,7 +46,8 @@ for get_graph in mechs:
                                           FrameTransform([0, 0.1, 0], [1, 0, 0, 0]))
     sim.set_flags_stop_simulation(flags)
     sim.change_config_system(config_sys)
-
+    #body: chrono.ChBody = sim.grasp_object.body
+    #body.SetPos_dt(chrono.ChVectorD(-1,0,0))
     # Start simulation
     sim_output = sim.simulate_system(TIME_STEP, True)
 
diff --git a/rostok/virtual_experiment/sensors.py b/rostok/virtual_experiment/sensors.py
index 5e36b65..b13f67e 100644
--- a/rostok/virtual_experiment/sensors.py
+++ b/rostok/virtual_experiment/sensors.py
@@ -48,6 +48,9 @@ class ContactReporter(chrono.ReportContactCallback):
         body_b = chrono.CastToChBody(contactobjB)
         idx_a = None
         idx_b = None
+        #print(plane_coord)
+        if react_forces.x>0 or react_forces.y>0 or react_forces.z>0:
+            print(react_forces.x, react_forces.z, react_forces.y)
         for body_tuple in self._body_list:
             if body_a == body_tuple[1].body:
                 idx_a = body_tuple[0]
-- 
GitLab