mirror of
https://github.com/OpenCMT/uLib.git
synced 2025-12-06 07:21:31 +01:00
131 lines
3.6 KiB
C++
131 lines
3.6 KiB
C++
/*//////////////////////////////////////////////////////////////////////////////
|
|
// CMT Cosmic Muon Tomography project //////////////////////////////////////////
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
|
|
All rights reserved
|
|
|
|
Authors: Sara Vanini < sara.vanini@pd.infn.it >
|
|
|
|
------------------------------------------------------------------
|
|
This library is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU Lesser General Public
|
|
License as published by the Free Software Foundation; either
|
|
version 3.0 of the License, or (at your option) any later version.
|
|
|
|
This library is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public
|
|
License along with this library.
|
|
|
|
//////////////////////////////////////////////////////////////////////////////*/
|
|
|
|
|
|
|
|
#include "Structured2DGrid.h"
|
|
|
|
using namespace uLib;
|
|
|
|
//_____________________________________________________________
|
|
Structured2DGrid::Structured2DGrid()
|
|
{
|
|
// reset uni-dim space
|
|
m_Dims = Vector2i(0,0);
|
|
m_Origin = Vector2f(0.,0.);
|
|
m_Increments = Vector2i(0,0);
|
|
m_binW = Vector2f(1.,1.);
|
|
m_debug = true;
|
|
}
|
|
|
|
|
|
//_____________________________________________________________
|
|
void Structured2DGrid::SetDims(const Vector2i &size)
|
|
{
|
|
this->m_Dims = size;
|
|
|
|
// set increments
|
|
this->m_Increments[0] = 1;
|
|
this->m_Increments[1] = m_Dims[0];
|
|
}
|
|
|
|
//_____________________________________________________________
|
|
void Structured2DGrid::SetPhysicalSpace(const Vector2f &origin, const Vector2f &spacing)
|
|
{
|
|
m_Origin = origin;
|
|
m_binW = spacing;
|
|
|
|
m_uni2phys_T = Translation<float,2>(m_Origin) * Scaling(m_binW);
|
|
m_phys2uni_T = m_uni2phys_T.matrix().inverse();
|
|
|
|
return;
|
|
}
|
|
|
|
|
|
//_____________________________________________________________
|
|
bool Structured2DGrid::IsInsideBounds(const Vector2f &pt) const
|
|
{
|
|
Vector2i pLoc = this->PhysicsToUnitSpace(pt);
|
|
return this->IsInsideGrid(pLoc);
|
|
}
|
|
|
|
|
|
//_____________________________________________________________
|
|
Vector2i Structured2DGrid::UnMap(int index) const
|
|
{
|
|
Vector2i v( 0,0 );
|
|
Vector2i iv = m_Increments;
|
|
|
|
int id = 0;
|
|
for(int k=0; k<2; ++k) {
|
|
int inc = iv.maxCoeff(&id);
|
|
v(id) = index / inc;
|
|
index -= v(id) * inc;
|
|
iv(id) = 0;
|
|
}
|
|
return v;
|
|
}
|
|
|
|
//_____________________________________________________________
|
|
bool Structured2DGrid::IsInsideGrid(const Vector2i &v) const
|
|
{
|
|
int vok = 1;
|
|
vok *= (v(0) >= 0 && v(0) < m_Dims[0]);
|
|
vok *= (v(1) >= 0 && v(1) < m_Dims[1]);
|
|
|
|
return vok;
|
|
}
|
|
|
|
|
|
//_____________________________________________________________
|
|
Vector2i Structured2DGrid::PhysicsToUnitSpace(const Vector2f &pt) const
|
|
{
|
|
Vector2i out;
|
|
Vector2f pLoc = m_phys2uni_T * pt;
|
|
|
|
out << (int)floor(pLoc(0)),
|
|
(int)floor(pLoc(1));
|
|
|
|
return out;
|
|
}
|
|
|
|
//_____________________________________________________________
|
|
Vector2f Structured2DGrid::UnitToPhysicsSpace(const Vector2i &ind) const
|
|
{
|
|
Vector2f p;
|
|
p = m_uni2phys_T * ind.cast<float>();
|
|
|
|
return p;
|
|
}
|
|
|
|
//_____________________________________________________________
|
|
void Structured2DGrid::PrintSelf(std::ostream &o)
|
|
{
|
|
o << "uLib Structured Data 2D: \n"
|
|
<< "dim = " << this->GetDims().transpose() << "\n"
|
|
<< "spacing = " << this->GetSpacing().transpose() << "\n";
|
|
}
|
|
|