kicad/qa/pns/playground.cpp

295 lines
10 KiB
C++
Raw Normal View History

/*
* This program source code file is part of KiCad, a free EDA CAD application.
*
* Copyright (C) 2020 KiCad Developers.
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This program 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, you may find one here:
* http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
* or you may search the http://www.gnu.org website for the version 2 license,
* or you may write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
*/
// WARNING - this Tom's crappy PNS hack tool code. Please don't complain about its quality
// (unless you want to improve it).
#include <pgm_base.h>
#include <qa_utils/utility_registry.h>
#include "pns_log_viewer_frame.h"
#include "label_manager.h"
2021-06-09 13:32:18 +00:00
#include <geometry/shape_arc.h>
2021-06-09 21:28:29 +00:00
std::shared_ptr<PNS_LOG_VIEWER_OVERLAY> overlay;
2021-06-09 13:32:18 +00:00
2021-06-09 21:39:01 +00:00
bool collideArc2Arc( const SHAPE_ARC& a1, const SHAPE_ARC& a2, int clearance, SEG& minDistSeg )
2021-06-09 13:32:18 +00:00
{
2021-06-09 21:39:01 +00:00
SEG mediatrix( a1.GetCenter(), a2.GetCenter() );
2021-06-09 13:32:18 +00:00
std::vector<VECTOR2I> ips;
2021-06-09 21:28:29 +00:00
// Basic case - arcs intersect
if( a1.Intersect( a2, &ips ) > 0 )
2021-06-09 13:32:18 +00:00
{
2021-06-09 21:28:29 +00:00
minDistSeg.A = minDistSeg.B = ips[0];
return true;
2021-06-09 13:32:18 +00:00
}
// Arcs don't intersect, build a list of points to check
std::vector<VECTOR2I> ptsA;
std::vector<VECTOR2I> ptsB;
2021-06-09 21:39:01 +00:00
bool cocentered = ( mediatrix.A == mediatrix.B );
2021-06-09 21:28:29 +00:00
// 1: Interior points of both arcs, which are on the line segment between the two centres
2021-06-09 21:39:01 +00:00
if( !cocentered )
{
a1.IntersectLine( mediatrix, &ptsA );
a2.IntersectLine( mediatrix, &ptsB );
2021-06-09 21:39:01 +00:00
}
// 2: Check arc end points
ptsA.push_back( a1.GetP0() );
ptsA.push_back( a1.GetP1() );
ptsB.push_back( a2.GetP0() );
ptsB.push_back( a2.GetP1() );
2021-06-09 13:32:18 +00:00
// 3: Endpoint of one and "projected" point on the other, which is on the
// line segment through that endpoint and the centre of the other arc
a1.IntersectLine( SEG( a2.GetP0(), a1.GetCenter() ), &ptsA );
a1.IntersectLine( SEG( a2.GetP1(), a1.GetCenter() ), &ptsA );
a2.IntersectLine( SEG( a1.GetP0(), a2.GetCenter() ), &ptsB );
a2.IntersectLine( SEG( a1.GetP1(), a2.GetCenter() ), &ptsB );
double minDist = std::numeric_limits<double>::max();
2021-06-09 21:39:01 +00:00
bool minDistFound = false;
2021-06-09 13:32:18 +00:00
// @todo performance might be improved by only checking certain points (e.g only check end
// points against other end points or their corresponding "projected" points)
for( const VECTOR2I& ptA : ptsA )
{
for( const VECTOR2I& ptB : ptsB )
2021-06-09 13:32:18 +00:00
{
double dist = ( ptA - ptB ).EuclideanNorm() - a1.GetWidth() / 2.0 - a2.GetWidth() / 2.0;
2021-06-09 13:32:18 +00:00
if( dist < clearance )
{
2021-06-09 21:39:01 +00:00
if( !minDistFound || dist < minDist )
2021-06-09 13:32:18 +00:00
{
2021-06-09 21:28:29 +00:00
minDist = dist;
minDistSeg = SEG( ptA, ptB );
2021-06-09 13:32:18 +00:00
}
minDistFound = true;
}
}
}
2021-06-09 13:32:18 +00:00
return minDistFound;
}
int playground_main_func( int argc, char* argv[] )
{
auto frame = new PNS_LOG_VIEWER_FRAME( nullptr );
Pgm().App().SetTopWindow( frame ); // wxApp gets a face.
frame->Show();
2021-06-09 21:39:01 +00:00
struct ARC_DATA
{
2021-07-11 21:35:53 +00:00
double cx, cy, sx, sy, ca, w;
2021-06-09 21:39:01 +00:00
};
2021-07-11 21:35:53 +00:00
const ARC_DATA test_data [] =
2021-06-09 21:39:01 +00:00
{
{73.843527, 74.355869, 71.713528, 72.965869, -76.36664803, 0.2},
{71.236473, 74.704131, 73.366472, 76.094131, -76.36664803, 0.2},
{82.542335, 74.825975, 80.413528, 73.435869, -76.4, 0.2},
{76.491192, 73.839894, 78.619999, 75.23, -76.4, 0.2},
{89.318807, 74.810106, 87.19, 73.42, -76.4, 0.2},
{87.045667, 74.632941, 88.826472, 75.794131, -267.9, 0.2},
2021-07-11 21:35:53 +00:00
{94.665667, 73.772941, 96.446472, 74.934131, -267.9, 0.2},
{94.750009, 73.74012, 93.6551, 73.025482, -255.5, 0.2},
{72.915251, 80.493054, 73.570159, 81.257692, -260.5, 0.2}, // end points clearance false positive
2021-06-09 21:39:01 +00:00
{73.063537, 82.295989, 71.968628, 81.581351, -255.5, 0.2},
2021-07-11 21:35:53 +00:00
{79.279991, 80.67988, 80.3749, 81.394518, -255.5, 0.2},
{79.279991, 80.67988, 80.3749, 81.694518, -255.5, 0.2 },
2021-06-09 21:39:01 +00:00
{88.495265, 81.766089, 90.090174, 82.867869, -255.5, 0.2},
{86.995265, 81.387966, 89.090174, 82.876887, -255.5, 0.2},
{96.149734, 81.792126, 94.99, 83.37, -347.2, 0.2},
2021-07-11 21:35:53 +00:00
{94.857156, 81.240589, 95.91, 83.9, -288.5, 0.2},
{72.915251, 86.493054, 73.970159, 87.257692, -260.5, 0.2}, // end points clearance #1
{73.063537, 88.295989, 71.968628, 87.581351, -255.5, 0.2},
{78.915251, 86.393054, 79.970159, 87.157692, 99.5, 0.2}, // end points clearance #2 - false positive
{79.063537, 88.295989, 77.968628, 87.581351, -255.5, 0.2},
{85.915251, 86.993054, 86.970159, 87.757692, 99.5, 0.2}, // intersection - false negative
{86.063537, 88.295989, 84.968628, 87.581351, -255.5, 0.2},
{94.6551, 88.295989, 95.6551, 88.295989, 90.0, 0.2 }, // simulating diff pair
{94.6551, 88.295989, 95.8551, 88.295989, 90.0, 0.2 },
{73.77532, 93.413654, 75.70532, 93.883054, 60.0, 0.1 }, // one arc fully enclosed in other
{73.86532, 93.393054, 75.86532, 93.393054, 90.0, 0.3 },
{79.87532, 93.413654, 81.64532, 94.113054, 60.0, 0.1 }, // concentric
{79.87532, 93.413654, 81.86532, 93.393054, 90.0, 0.3 }
2021-06-09 21:39:01 +00:00
};
2021-06-09 21:28:29 +00:00
overlay = frame->GetOverlay();
2021-07-11 21:35:53 +00:00
overlay->SetIsFill(false);
overlay->SetLineWidth(10000);
2021-06-09 21:28:29 +00:00
std::vector<SHAPE_ARC> arcs;
2021-06-09 21:39:01 +00:00
int n_arcs = sizeof( test_data ) / sizeof( ARC_DATA );
2021-06-09 21:28:29 +00:00
2021-06-09 21:39:01 +00:00
BOX2I vp;
2021-06-09 21:39:01 +00:00
for( int i = 0; i < n_arcs; i++ )
2021-06-09 21:28:29 +00:00
{
2021-06-09 21:39:01 +00:00
const ARC_DATA& d = test_data[i];
2021-06-09 21:28:29 +00:00
2021-07-11 21:35:53 +00:00
SHAPE_ARC arc( VECTOR2D( Millimeter2iu( d.cx ), Millimeter2iu( d.cy ) ),
VECTOR2D( Millimeter2iu( d.sx ), Millimeter2iu( d.sy ) ), d.ca,
Millimeter2iu( d.w ) );
2021-06-09 21:28:29 +00:00
2021-06-09 21:39:01 +00:00
arcs.push_back( arc );
2021-06-09 21:28:29 +00:00
2021-06-09 21:39:01 +00:00
if( i == 0 )
vp = arc.BBox();
else
vp.Merge( arc.BBox() );
2021-06-09 21:28:29 +00:00
}
2021-12-07 22:53:56 +00:00
printf("Read %zu arcs\n", arcs.size() );
2021-06-09 21:28:29 +00:00
LABEL_MANAGER labelMgr( frame->GetPanel()->GetGAL() );
2021-06-09 21:39:01 +00:00
frame->GetPanel()->GetView()->SetViewport( BOX2D( vp.GetOrigin(), vp.GetSize() ) );
2021-06-09 21:28:29 +00:00
for(int i = 0; i < arcs.size(); i+= 2)
{
SEG closestDist;
std::vector<VECTOR2I> ips;
2021-07-11 21:35:53 +00:00
bool collides = collideArc2Arc( arcs[i], arcs[i+1], 0, closestDist );
int ni = arcs[i].Intersect( arcs[i+1], &ips );
2021-06-09 13:32:18 +00:00
2021-07-11 21:35:53 +00:00
overlay->SetLineWidth( 10000.0 );
overlay->SetStrokeColor( GREEN );
2021-06-09 13:32:18 +00:00
for( int j = 0; j < ni; j++ )
2021-07-11 21:35:53 +00:00
overlay->AnnotatedPoint( ips[j], arcs[i].GetWidth() );
2021-06-09 13:32:18 +00:00
2021-06-09 21:28:29 +00:00
if( collides )
{
overlay->SetStrokeColor( YELLOW );
overlay->Line( closestDist.A, closestDist.B );
overlay->SetLineWidth( 10000.0 );
overlay->SetGlyphSize( { 100000.0, 100000.0 } );
overlay->BitmapText( wxString::Format( "dist=%d, l=%d", closestDist.Length() ),
closestDist.A + VECTOR2I( 0, -arcs[i].GetWidth() ), 0 );
2021-06-09 21:28:29 +00:00
}
2021-06-09 13:32:18 +00:00
overlay->SetLineWidth( 10000.0 );
overlay->SetStrokeColor( CYAN );
overlay->AnnotatedPoint( arcs[i].GetP0(), arcs[i].GetWidth() / 2 );
overlay->AnnotatedPoint( arcs[i + 1].GetP0(), arcs[i + 1].GetWidth() / 2 );
overlay->AnnotatedPoint( arcs[i].GetArcMid(), arcs[i].GetWidth() / 2 );
overlay->AnnotatedPoint( arcs[i + 1].GetArcMid(), arcs[i + 1].GetWidth() / 2 );
overlay->AnnotatedPoint( arcs[i].GetP1(), arcs[i].GetWidth() / 2 );
overlay->AnnotatedPoint( arcs[i + 1].GetP1(), arcs[i + 1].GetWidth() / 2 );
2021-07-11 21:35:53 +00:00
overlay->SetStrokeColor( RED );
overlay->Arc( arcs[i] );
overlay->SetStrokeColor( MAGENTA );
overlay->Arc( arcs[i + 1] );
2021-06-09 13:32:18 +00:00
}
2021-06-09 21:28:29 +00:00
overlay = nullptr;
return 0;
}
int drawShapes( int argc, char* argv[] )
{
SHAPE_ARC arc( VECTOR2I( 206000000, 140110000 ), VECTOR2I( 201574617, 139229737 ),
VECTOR2I( 197822958, 136722959 ), 250000 );
SHAPE_LINE_CHAIN lc( { /* VECTOR2I( 159600000, 142500000 ), VECTOR2I( 159600000, 142600000 ),
VECTOR2I( 166400000, 135800000 ), VECTOR2I( 166400000, 111600000 ),
VECTOR2I( 190576804, 111600000 ), VECTOR2I( 192242284, 113265480 ),
VECTOR2I( 192255720, 113265480 ),*/
VECTOR2I( 203682188, 124691948 ), VECTOR2I( 203682188, 140332188 ),
/* VECTOR2I( 206000000, 142650000 ) */ },
false );
auto frame = new PNS_LOG_VIEWER_FRAME( nullptr );
Pgm().App().SetTopWindow( frame ); // wxApp gets a face.
frame->Show();
overlay = frame->GetOverlay();
overlay->SetIsFill( false );
overlay->SetLineWidth( arc.GetWidth() );
overlay->SetStrokeColor( RED );
overlay->Arc( arc );
overlay->SetLineWidth( arc.GetWidth() / 20 );
overlay->SetStrokeColor( GREEN );
overlay->Polyline( lc );
overlay->SetLineWidth( 80000.0 );
overlay->SetStrokeColor( CYAN );
for( int i = 0; i < lc.PointCount(); ++i )
{
int mult = ( i % 2 ) ? 1 : -1;
overlay->AnnotatedPoint( lc.GetPoint( i ), arc.GetWidth() * 2 );
overlay->SetGlyphSize( { 800000.0, 800000.0 } );
overlay->BitmapText( wxString::Format( "x=%d, y=%d", lc.GetPoint( i ).x, lc.GetPoint( i ).y ),
lc.GetPoint( i ) + VECTOR2I( 0, mult*arc.GetWidth() * 4 ), 0 );
}
arc.Collide( &lc, 100000 );
BOX2I vp = arc.BBox();
vp.Merge( lc.BBox() );
vp.Inflate( (800000 + arc.GetWidth() * 4 )*2);
frame->GetPanel()->GetView()->SetViewport( BOX2D( vp.GetOrigin(), vp.GetSize() ) );
overlay = nullptr;
return 0;
}
static bool registered = UTILITY_REGISTRY::Register( {
"playground",
"Geometry/drawing playground",
playground_main_func,
} );
static bool registered1 = UTILITY_REGISTRY::Register( {
"drawShapes",
"drawing shapes",
drawShapes,
} );