File: NavStatePath.cpp

package info (click to toggle)
vecgeom 1.2.8%2Bdfsg-2
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 24,016 kB
  • sloc: cpp: 88,803; ansic: 6,888; python: 1,035; sh: 582; sql: 538; makefile: 23
file content (190 lines) | stat: -rw-r--r-- 5,799 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
/// \file NavStatePath.cpp
/// \author Sandro Wenzel (sandro.wenzel@cern.ch)
/// \date 17.04.2014

#include "VecGeom/navigation/NavStatePath.h"

#include <iostream>
#include <list>
#include <sstream>

namespace vecgeom {
inline namespace VECGEOM_IMPL_NAMESPACE {

VECCORE_ATT_HOST_DEVICE
Vector3D<Precision> NavStatePath::GlobalToLocal(Vector3D<Precision> const &globalpoint, int tolevel) const
{
  Vector3D<Precision> tmp = globalpoint;
  Vector3D<Precision> current;
  for (int level = 0; level < tolevel; ++level) {
    Transformation3D const *m = At(level)->GetTransformation();
    current                   = m->Transform(tmp);
    tmp                       = current;
  }
  return tmp;
}

VECCORE_ATT_HOST_DEVICE
void NavStatePath::TopMatrix(int tolevel, Transformation3D &global_matrix) const
{
  for (int i = 1; i < tolevel; ++i) {
    global_matrix.MultiplyFromRight(*(At(i)->GetTransformation()));
  }
}

// returning a "delta" transformation that can transform
// coordinates given in reference frame of this->Top() to the reference frame of other->Top()
// simply with otherlocalcoordinate = delta.Transform( thislocalcoordinate )
VECCORE_ATT_HOST_DEVICE
void NavStatePath::DeltaTransformation(NavStatePath const &other, Transformation3D &delta) const
{
  Transformation3D g2;
  Transformation3D g1;
  other.TopMatrix(g2);
  this->TopMatrix(g1);
  g1.Inverse(delta);
  g2.SetProperties();
  delta.SetProperties();
  delta.FixZeroes();
  delta.MultiplyFromRight(g2);
  delta.FixZeroes();
}

/**
 * function that transforms a global point to local point in reference frame of deepest volume in current navigation
 * state
 * ( equivalent to using a global matrix )
 */
VECCORE_ATT_HOST_DEVICE
Vector3D<Precision> NavStatePath::GlobalToLocal(Vector3D<Precision> const &globalpoint) const
{
  Vector3D<Precision> tmp = globalpoint;
  Vector3D<Precision> current;
  for (int level = 0; level < fCurrentLevel; ++level) {
    Transformation3D const *m = At(level)->GetTransformation();
    current                   = m->Transform(tmp);
    tmp                       = current;
  }
  return tmp;
}

size_t FindIndexWithinMother(VPlacedVolume const *mother, VPlacedVolume const *daughter)
{
  for (size_t d = 0; d < mother->GetDaughters().size(); ++d) {
    if (mother->GetDaughters()[d] == daughter) return d;
  }
  assert(false && "did not find index of a daughter volume within mother");
  return static_cast<uint>(-1);
}

VPlacedVolume const *GetDaughterWithinMother(VPlacedVolume const *mother, uint index)
{
  if (index < (uint)mother->GetDaughters().size()) return mother->GetDaughters()[index];

  return NULL;
}

void NavStatePath::GetPathAsListOfIndices(std::list<uint> &indices) const
{
  indices.clear();
  if (IsOutside()) return;
  for (uint level = fCurrentLevel - 1; level > 0; --level) {
    indices.push_front(FindIndexWithinMother(At(level - 1), At(level)));
  }
  indices.push_front(0);
}

VECCORE_ATT_HOST_DEVICE
void NavStatePath::Print() const
{
// printf("VariableSizeObj: fPath=%p (%l bytes)\n", fPath, sizeof(fPath));
#ifndef VECCORE_CUDA
  printf("NavStatePath: level=%i/%i,  onBoundary=%s, path=<", fCurrentLevel - 1, GetMaxLevel(),
         (fOnBoundary ? "true" : "false"));
  for (int i = 0; i < fCurrentLevel; ++i)
    printf("/%s", ToPlacedVolume(fPath[i]) ? ToPlacedVolume(fPath[i])->GetLabel().c_str() : "NULL");
  printf(">\n");
#else
  printf("NavStatePath: level=%i/%i,  onBoundary=%s, topVol=<%p>, this=%p\n", fCurrentLevel - 1, GetMaxLevel(),
         (fOnBoundary ? "true" : "false"), Top(), (const void *)this);
#endif
}

void NavStatePath::ResetPathFromListOfIndices(VPlacedVolume const *world, std::list<uint> const &indices)
{
  // clear current nav state
  fCurrentLevel = indices.size();
  if (indices.size() > 0) {
    fPath[0] = ToIndex(world);
    // have to disregard first one;
    // then iterate through list
    int counter = 0;
    for (auto x : indices) {
      if (counter > 0) fPath[counter] = ToIndex(GetDaughterWithinMother(At(counter - 1), x));
      counter++;
    }
  }
}

std::string NavStatePath::RelativePath(NavStatePath const &other) const
{
  int lastcommonlevel = -1;
  int maxlevel        = Min(GetCurrentLevel(), other.GetCurrentLevel());
  std::stringstream str;
  //  algorithm: start on top and go down until paths split
  for (int i = 0; i < maxlevel; i++) {
    if (this->At(i) == other.At(i)) {
      lastcommonlevel = i;
    } else {
      break;
    }
  }

  auto filledlevel1 = GetCurrentLevel() - 1;
  auto filledlevel2 = other.GetCurrentLevel() - 1;

  // paths are the same
  if (filledlevel1 == lastcommonlevel && filledlevel2 == lastcommonlevel) {
    return std::string("");
  }

  // emit only ups
  if (filledlevel1 > lastcommonlevel && filledlevel2 == lastcommonlevel) {
    for (int i = 0; i < filledlevel1 - lastcommonlevel; ++i) {
      str << "/up";
    }
    return str.str();
  }

  // emit only downs
  if (filledlevel1 == lastcommonlevel && filledlevel2 > lastcommonlevel) {
    for (int i = lastcommonlevel + 1; i <= filledlevel2; ++i) {
      str << "/down";
      str << "/" << other.ValueAt(i);
    }
    return str.str();
  }

  // mixed case: first up; then down
  if (filledlevel1 > lastcommonlevel && filledlevel2 > lastcommonlevel) {
    // emit ups
    int level = filledlevel1;
    for (; level > lastcommonlevel + 1; --level) {
      str << "/up";
    }

    level = lastcommonlevel + 1;
    // emit horiz ( exists when there is a turning point )
    int delta = other.ValueAt(level) - this->ValueAt(level);
    if (delta != 0) str << "/horiz/" << delta;

    level++;
    // emit downs with index
    for (; level <= filledlevel2; ++level) {
      str << "/down/" << other.ValueAt(level);
    }
  }
  return str.str();
}
} // namespace VECGEOM_IMPL_NAMESPACE
} // namespace vecgeom