DataFormats/GeometrySurface/test/FrameTransformTest.cpp - master - githubmirror/cmssw

DataFormats/GeometrySurface/test/FrameTransformTest.cpp
120 строк4 KB
Cms Build
Clang-Format
11 июн 2019, 11:30
11 июн 2019, 11:30044a211
#include <cmath>
#include <iostream>

#include "DataFormats/GeometrySurface/interface/SOARotation.h"
#include "DataFormats/GeometrySurface/interface/TkRotation.h"
#include "DataFormats/GeometrySurface/interface/GloballyPositioned.h"
#include "DataFormats/GeometrySurface/interface/BoundPlane.h"
#include "DataFormats/GeometrySurface/interface/Cylinder.h"

using namespace std;

template <typename T>
void go() {
  typedef TkRotation<T> Rotation;
  typedef SOARotation<T> SRotation;
  typedef GloballyPositioned<T> Frame;
  typedef SOAFrame<T> SFrame;
  typedef typename Frame::PositionType Position;
  typedef typename Frame::GlobalVector GlobalVector;
  typedef typename Frame::GlobalPoint GlobalPoint;
  typedef typename Frame::LocalVector LocalVector;
  typedef typename Frame::LocalPoint LocalPoint;

  std::cout << "size of Rot       " << sizeof(Rotation) << std::endl;
  std::cout << "size of Pos       " << sizeof(Position) << std::endl;
  std::cout << "size of Point     " << sizeof(GlobalPoint) << std::endl;
  std::cout << "size of Frame     " << sizeof(Frame) << std::endl;
  std::cout << "size of soa Rot   " << sizeof(SRotation) << std::endl;
  std::cout << "size of soa Frame " << sizeof(SFrame) << std::endl;

  double a = 0.01;
  double ca = cos(a);
  double sa = sin(a);

  Rotation r1(ca, sa, 0, -sa, ca, 0, 0, 0, 1);
  ;
  Frame f1(Position(2, 3, 4), r1);
  cout << "f1.position() " << f1.position() << endl;
  cout << "f1.rotation() " << endl << f1.rotation() << endl;

  Rotation r2(GlobalVector(0, 1, 0), GlobalVector(0, 0, 1));
  Frame f2(Position(5, 6, 7), r2);
  cout << "f2.position() " << f2.position() << endl;
  cout << "f2.rotation() " << endl << f2.rotation() << endl;

  Rotation r3 = r2 * r1.transposed();
  SRotation sr3(r3);

  GlobalPoint pos2(f2.position());
  LocalPoint lp3 = f1.toLocal(pos2);
  Frame f3(GlobalPoint(lp3.basicVector()), r3);
  cout << "f3.position() " << f3.position() << endl;
  cout << "f3.rotation() " << endl << f3.rotation() << endl;

  SFrame sf1(f1.position().x(), f1.position().y(), f1.position().z(), f1.rotation());

  SFrame sf3(f3.position().x(), f3.position().y(), f3.position().z(), sr3);

  // test
  GlobalPoint gp(11, 22, 33);
  LocalPoint p_in1 = f1.toLocal(gp);
  typename Frame::ToLocal ff1(f1);
  LocalPoint p_in2 = f2.toLocal(gp);
  LocalPoint p_in3 = f3.toLocal(GlobalPoint(p_in1.basicVector()));
  cout << "p_in1 " << p_in1 << endl;
  cout << "p_in1 " << ff1.toLocal(gp) << endl;
  cout << "p_in2 " << p_in2 << endl;
  cout << "p_in3 " << p_in3 << endl;

  LocalPoint p_in1_from3(f3.toGlobal(p_in3).basicVector());
  cout << "p_in1_from3 " << p_in1_from3 << endl;

  T xx, yy, zz;
  sf1.toLocal(gp.x(), gp.y(), gp.z(), xx, yy, zz);
  cout << "p_in1 soa " << xx << ',' << yy << ',' << zz << endl;
  sf3.toLocal(p_in1.x(), p_in1.y(), p_in1.z(), xx, yy, zz);
  cout << "p_in3 soa " << xx << ',' << yy << ',' << zz << endl;
  sf3.toGlobal(p_in3.x(), p_in3.y(), p_in3.z(), xx, yy, zz);
  cout << "p_in1_from3 soa " << xx << ',' << yy << ',' << zz << endl;
}

void cyl() {
  using T = float;
  typedef TkRotation<T> Rotation;
  typedef GloballyPositioned<T> Frame;
  typedef typename Frame::PositionType Position;
  typedef typename Frame::GlobalVector GlobalVector;
  typedef typename Frame::GlobalPoint GlobalPoint;
  typedef typename Frame::LocalVector LocalVector;
  typedef typename Frame::LocalPoint LocalPoint;

  // cylinder
  {
    std::cout << " Trivial Cylinder" << std::endl;
    Rotation ll(GlobalVector(1, 0, 0), GlobalVector(0, 1, 0));
    Position p0(0, 0, 0);
    Cylinder cyl(5., p0, ll);
    Plane t = cyl.fastTangent(GlobalPoint(3., 4., 1.));
    std::cout << t.position() << '\n' << t.rotation() << std::endl;
    std::cout << t.rotation().x() * cyl.rotation().z().cross((t.position() - cyl.position()).basicVector()).unit()
              << std::endl;
  }

  {
    std::cout << " rotated, displaced Cylinder" << std::endl;
    Rotation ll(Basic3DVector<T>(1, 1, 1), .3);
    Cylinder cyl(5.f, Position(2, -1, 3), ll);
    Plane t = cyl.fastTangent(LocalPoint(3., 4., 1.));
    std::cout << t.position() << '\n' << t.rotation() << std::endl;
    std::cout << t.rotation().x() * cyl.rotation().z().cross((t.position() - cyl.position()).basicVector()).unit()
              << std::endl;
  }
}

int main() {
  go<float>();
  cyl();
  std::cout << std::endl;
  go<double>();
}