My Project
plotSetUp.py
Go to the documentation of this file.
1 import numpy as np
2 colors_diverging_short = ['#d7191c','#fdae61','#ffffbf','#abd9e9','#2c7bb6']
3 colors_diverging_long = ['#a50026','#d73027','#f46d43','#fdae61','#fee090','#ffffbf','#e0f3f8','#abd9e9','#74add1','#4575b4','#313695']
4 colors_qualitative_short = ['#1b9e77','#d95f02','#7570b3','#e7298a','#66a61e']
5 colors_qualitative_long = ['#1b9e77','#d95f02','#7570b3','#e7298a','#66a61e','#e6ab02','#a6761d']
6 linestyles = ['-', '--', '-.', ':']
7 
8 
10  pyplot
11 ):
12  pyplot.rcParams["svg.fonttype"] = "none"
13  pyplot.rcParams["font.size"] = 16
14  pyplot.rcParams["savefig.directory"] = "../../Figures/Master"
15  pyplot.rcParams["savefig.format"] = "svg"
16 
17 
19  yaml_file
20 ):
21  debug_continuous_slider_speed = yaml_file["Debug"]["debug_continuous_slider_speed"]
22  if(debug_continuous_slider_speed):
23  return False
24  else:
25  return True
26 
27 
29  block_data
30 ):
31  if(len(block_data.shape) == 1):
32  return False
33  else:
34  return True
35 
36 
38  block_data,
39  yaml_file
40 ):
41  multiple_blocks = checkIfMultipleBlocks(block_data)
42  start = 0
43  stop = yaml_file["Parameters"]["max_time"]
44 
45  if(yaml_file["Parameters"]["start_speed_continuous"] > yaml_file["Parameters"]["end_speed_continuous"]):
46  stop, start = start, stop
47 
48  if(multiple_blocks):
49  n_time_steps = block_data.shape[1]
50  else:
51  n_time_steps = len(block_data)
52 
53  time_steps = np.linspace(start=start, stop=stop, num=n_time_steps)
54  return time_steps
55 
56 
58  block_data,
59  yaml_file,
60  step_as_continous=False
61 ):
62  multiple_blocks = checkIfMultipleBlocks(block_data)
63  if(not step_as_continous):
64  start = yaml_file["Parameters"]["start_speed_continuous"]
65  stop = yaml_file["Parameters"]["end_speed_continuous"]
66  else:
67  start = yaml_file["Parameters"]["slider_speed"]
68  increment = yaml_file["Parameters"]["increment"]
69  if(increment < 0):
70  start = start - increment
71  # this is done to match that each velocity interval
72  # is the speed starting at it own speed and ending at the next
73  interval = yaml_file["Parameters"]["interval"]
74  max_time = yaml_file["Parameters"]["max_time"]
75 
76  stop = max_time // interval * increment + start
77 
78  if(multiple_blocks):
79  n_time_steps = block_data.shape[1]
80  else:
81  n_time_steps = len(block_data)
82 
83  time_steps = np.linspace(start=start, stop=stop, num=n_time_steps)
84  return time_steps
85 
86 
88  yaml_file
89 ):
90  if(not checkStepAsContinous(yaml_file)):
91  start_speed_continuous = yaml_file["Parameters"]["start_speed_continuous"]
92  end_speed_continuous = yaml_file["Parameters"]["end_speed_continuous"]
93  else:
94  start_speed_continuous = yaml_file["Parameters"]["slider_speed"]
95  increment = yaml_file["Parameters"]["increment"]
96  interval = yaml_file["Parameters"]["interval"]
97  max_time = yaml_file["Parameters"]["max_time"]
98 
99  end_speed_continuous = max_time // interval * increment +\
100  start_speed_continuous
101 
102  direction = end_speed_continuous - start_speed_continuous
103 
104  if(direction < 0):
105  return 3
106  else:
107  return 4
def createSliderVelocityStepValues(block_data, yaml_file, step_as_continous=False)
Definition: plotSetUp.py:61
def checkIfMultipleBlocks(block_data)
Definition: plotSetUp.py:30
def checkStepAsContinous(yaml_file)
Definition: plotSetUp.py:20
def choseColourFromDirection(yaml_file)
Definition: plotSetUp.py:89
def createTimeStepValues(block_data, yaml_file)
Definition: plotSetUp.py:40
def setSvgLatexMatplotlibSettings(pyplot)
Definition: plotSetUp.py:11