CompactSecID.cc

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#include <tracking/trackFindingVXD/filterMap/map/CompactSecIDs.h>
#include "tracking/dataobjects/FullSecID.h"
 
#include <gtest/gtest.h>
#include <iostream>
#include <vector>
 
using namespace std;
using namespace Belle2;
 
 
TEST(CompactSecIDs, basicCheck)
{
  CompactSecIDs compactSecIds;
 
  vector< double > uSup = { .25, .5, .75 };
  vector< double > vSup = { .25, .5, .75, .8};
 
  vector< vector< FullSecID > > sectors;
 
  sectors.resize(uSup.size() + 1);
 
  int counter = 0;
  for (unsigned int i = 0; i < uSup.size() + 1 ; i++) {
    sectors[i].resize(vSup.size() + 1);
    for (unsigned int j = 0; j < vSup.size() + 1; j++) {
      sectors[i][j] = FullSecID(VxdID(6, 3, 2), false, 12 + counter);
      counter ++;
    }
  }
  // successfully insert
  EXPECT_EQ(sectors.size()*sectors[0].size(),
            compactSecIds.addSectors(uSup, vSup, sectors)) ;
 
  // successfully retrieve
  EXPECT_EQ(1, compactSecIds.getCompactID(sectors[0][0]));
  EXPECT_EQ(2, compactSecIds.getCompactIDFastAndDangerous(sectors[0][1]));
 
  // successfully retrieve
  EXPECT_EQ(0, compactSecIds.getCompactID(FullSecID(VxdID(6, 3, 2), false, 1)));
  EXPECT_EQ(0, compactSecIds.getCompactID(FullSecID(VxdID(7, 3, 2), false, 1)));
  EXPECT_EQ(0, compactSecIds.getCompactID(FullSecID(VxdID(5, 3, 2), false, 1)));
 
  // cannot insert twice
  EXPECT_EQ(0, compactSecIds.addSectors(uSup, vSup, sectors));
 
 
 
}
 
 
TEST(CompactSecIDs, stressTest)
{
 
  CompactSecIDs compactSecIds;
  vector<int> layers  = { 1, 2, 3, 4, 5, 6};
  vector<int> ladders = { 8, 12, 7, 10, 12, 16};
  vector<int> sensors = { 2, 2, 2, 3, 4, 5};
 
 
  vector< double > uSup = { .25, .5, .75 };
  vector< double > vSup = { .2, .4, .6, .8};
 
 
  vector< vector< FullSecID > > sectors;
 
  sectors.resize(uSup.size() + 1);
 
 
  for (auto layer : layers)
    for (int ladder = 1 ; ladder <= ladders[layer - 1] ; ladder++)
      for (int sensor = 1 ; sensor <=  sensors[layer - 1] ; sensor++) {
        int counter = 0;
        for (unsigned int i = 0; i < uSup.size() + 1; i++) {
          sectors[i].resize(vSup.size() + 1);
          for (unsigned int j = 0; j < vSup.size() + 1 ; j++) {
            sectors[i][j] = FullSecID(VxdID(layer, ladder, sensor),
                                      sensor % 2 == 0 ? false : true,  counter);
            counter ++;
          }
        }
 
        // Assert successful insertion
        EXPECT_EQ(sectors.size()*sectors[0].size(), compactSecIds.addSectors(uSup, vSup, sectors));
      }
 
  for (auto layer : layers)
    for (int ladder = 1 ; ladder <= ladders[layer - 1] ; ladder++)
      for (int sensor = 1 ; sensor <=  sensors[layer - 1] ; sensor++) {
        int counter = 0;
        for (unsigned int i = 0; i < uSup.size() + 1 ; i++) {
          sectors[i].resize(vSup.size() + 1);
          for (unsigned int j = 0; j < vSup.size() + 1; j++) {
            sectors[i][j] = FullSecID(VxdID(layer, ladder, sensor),
                                      sensor % 2 == 0 ? false : true, counter);
            counter ++;
          }
        }
 
        // unsuccessful insertion because the sectors are already in compactSecIds
        EXPECT_EQ(0, compactSecIds.addSectors(uSup, vSup, sectors));
      }
 
  int expected = 0;
 
  for (auto layer : layers)
    for (int ladder = 1 ; ladder <= ladders[layer - 1] ; ladder++)
      for (int sensor = 1 ; sensor <=  sensors[layer - 1] ; sensor++)
        for (unsigned int sector = 0 ; sector < (uSup.size() + 1) * (vSup.size() + 1) ; sector ++) {
          FullSecID aSector(VxdID(layer, ladder, sensor),
                            sensor % 2 == 0 ? false : true
                            , sector);
          // successful retrieve
          EXPECT_EQ(++ expected, compactSecIds.getCompactID(aSector));
 
        }
 
  //  list< pair< float, float> >;
 
  for (double u = 0. ; u < 1.; u += .25) {
    for (double v = 0.; v < 1.; v += .2)
      cout << compactSecIds.getFullSecID(VxdID(1, 1, 1), u, v) << "\t";
    cout << endl;
  }
 
  // the compactsecids don't throw exceptions anymore
  for (double u = .25 ; u < 1.25; u += .25) {
    for (double v = 0.2; v < 1.2; v += .2) {
      if (u < 0. or u > 1. or v < 0. or v >1.) {
        EXPECT_ANY_THROW(compactSecIds.getFullSecID(VxdID(1, 1, 1), u, v));
      } else {
        cout << compactSecIds.getFullSecID(VxdID(1, 1, 1), u, v) << "\t";
      }
    }
    cout << endl;
  }
 
}