279 lines
5.5 KiB

10 months ago
#include "stdafx.h"
#include "KDtree.h"
#include <cmath>
#include <vector>
#include <queue>
#include <map>
#include <iostream>
#include <fstream>
#include <sstream>
10 months ago
using namespace std;
// ������Ϣ
struct Clip
{
string name; // ����
P coord, dir; // �����ͷ���,dir�ķ�����Ϊ����
int emc; // ��������,δȷ��Ϊ0
double dia; // ͨ��������������ֱ��
double maxDia; // ��������ͨ��������������ֱ��
int id; // ��������
int bundleNodeID; // ������ʱ�����ĸ�Bundle
vector<double> diaset; // �Ѿ�ͨ�����������°뾶����
vector<int> channelEdge; // �ڽ���ͨ�������ӵ��Ŀ�������
bool used; // �����Ƿ���ʹ��
10 months ago
Clip()
{
emc = 0;
bundleNodeID = 0;
maxDia = MAXDia;
id = 0;
used = 0;
10 months ago
}
// ��������ͨ���������ߺ�ͨ����Ϣ
void clear()
{
10 months ago
diaset.clear();
bundleNodeID = 0;
used = 0;
10 months ago
}
// �ڿ���������ֱ��Ϊd������
inline void addwire(double d)
{
10 months ago
diaset.push_back(d);
dia = sqrt(dia * dia + d * d);
10 months ago
}
// ����ͨ��������������ֱ��
inline double getDia()
{
double dia1 = 1.154 * dia;
if (dia1 >= 10)
dia1 *= 1.235 / 1.154;
10 months ago
return dia1;
}
// �ж�ֱ��Ϊd�������ܷ�ͨ��
inline bool transitable(double d)
{
double dia1 = sqrt(d * d + dia * dia);
dia1 = 1.154 * dia;
if (dia1 >= 10)
dia1 *= 1.235 / 1.154;
if (dia1 <= maxDia)
return true;
10 months ago
return false;
}
// �жϴ�p�㵽������Ӧ�ô���һ�����룬����Ϊ0������Ϊ1
inline int inDir(P p)
{
double angle = Angle(coord - p, dir);
if (angle > pi / 2)
return 1;
10 months ago
return 0;
}
};
// ���������������еĿ�����Ϣ
struct ClipSet
{
Clip c[N]; // �������ϣ��±���1~clipnum
int clipnum; // �����ܺ�,�������ŷ�Χ1`clipnum
map<P, int> clipid; // ���ݵ������ҵ���������
10 months ago
ClipSet()
{
clipnum = 0;
10 months ago
}
// ����ͨ��������������Ϣ
void clearWire()
{
for (int i = 1; i <= clipnum; i++)
c[i].clear();
10 months ago
}
// �¼���һ������
inline void addclip(P p, P d, string name, int emc)
{
10 months ago
clipnum++;
c[clipnum].coord = p;
c[clipnum].coord.ref = clipnum;
c[clipnum].coord.type = 0;
c[clipnum].coord.setDir(p - d);
10 months ago
c[clipnum].id = clipnum;
c[clipnum].dir = p - d;
c[clipnum].name = name;
c[clipnum].emc = emc;
clipid[p] = clipnum;
10 months ago
}
// ����ͨ�������ĵ���
inline void addwire(int id, double d)
{
10 months ago
c[id].addwire(d);
}
inline void addwire(P &p, double d)
{
int id = clipid[p];
10 months ago
c[id].addwire(d);
}
// ��ȡָ������Ϊid�Ŀ�����ָ��
Clip *getClipPointer(int id, int bundleNodeID)
{
if (bundleNodeID != -1)
c[id].bundleNodeID = bundleNodeID;
10 months ago
return &c[id];
}
// �ж����������Ƿ�����Ϊ����(��֪i�����Ŀ�����j)
inline bool bothConnect(int i, int j)
{
if (c[j].channelEdge.empty())
return false;
if (c[j].channelEdge[0] == i)
return true;
if (c[j].channelEdge.size() == 2 &&
c[j].channelEdge[1] == i)
return true;
return false;
10 months ago
}
// �жϿ����ܷ�ͨ��
inline bool transitable(int id, int emc, double d)
{
if (c[id].emc != emc)
return false;
10 months ago
return c[id].transitable(d);
}
// ���ݿ���λ�ù��Ʒɻ��Ľ���������
inline void computeCenter()
{
Ycenter = 0;
Zcenter = 0;
for (int i = 1; i <= clipnum; i++)
{
Clip clip = c[i];
P tp = c[i].coord;
Ycenter += tp.y;
Zcenter += tp.z;
MAXX = max(MAXX, tp.x);
MINX = min(MINX, tp.x);
10 months ago
MAXY = max(MAXY, tp.y);
MINY = min(MINY, tp.y);
10 months ago
MAXZ = max(MAXZ, tp.z);
MINZ = min(MINZ, tp.z);
10 months ago
}
Ycenter /= clipnum;
Zcenter /= clipnum;
10 months ago
}
// ��ӡ�����������������ߵľ���
inline void printDis()
{
for (int i = 1; i <= clipnum; i++)
{
Clip clip = c[i];
P tp = c[i].coord;
double dis = (Ycenter - tp.y) * (Ycenter - tp.y) + (Zcenter - tp.z) * (Zcenter - tp.z);
cout << sqrt(dis) << endl;
10 months ago
}
}
// ���ݿ���λ���ȴ���һ�����Ƶ�ͨ����ÿ����������������������������
inline void buildChannel()
{
10 months ago
vector<int> pointSet;
bool flag = false; // flag=trueʱ��������������Ҫ���¼���
double sumDis = 300;
int sum = 1;
for (int i = 1; i <= clipnum; i++)
{
if (pointSet.empty() || flag)
{
pointSet = kdtree.search_by_dis(c[i].coord, segmentLength);
10 months ago
c[i].channelEdge.clear();
}
vector<pair<double, int>> vp;
for (int j = 0; j < pointSet.size(); j++)
{
int cid = pointSet[j];
if (cid == i)
continue;
if (c[i].coord == c[cid].coord)
continue;
double dis = distan(c[i].coord, c[cid].coord);
vp.push_back(make_pair(dis, cid));
10 months ago
}
sort(vp.begin(), vp.end());
10 months ago
P dir;
for (int j = 0; j < vp.size(); j++)
{
int cid = vp[j].second;
if (vp[j].first > 2 * sumDis / sum)
break;
if (j == 0)
{
10 months ago
c[i].channelEdge.push_back(cid);
dir = c[cid].coord - c[i].coord;
10 months ago
sum++;
sumDis += vp[j].first;
10 months ago
}
else
{
P dir2 = c[cid].coord - c[i].coord;
if (Angle(dir, dir2) >= pi / 2 - minAngle)
{
10 months ago
c[i].channelEdge.push_back(cid);
sum++;
sumDis += vp[j].first;
10 months ago
break;
}
}
}
if (c[i].channelEdge.size() < 2 && !flag)
flag = true, i--;
else
flag = false;
10 months ago
}
}
// ��ӡ���п�����Ϣ��clipDir�ļ��������������������Ͽ�������
10 months ago
/*
0򿨹
1ֱ򿨹
2򿨹
*/
inline void print()
{
10 months ago
ofstream ofs;
ofs.open("clipDir.txt", ios::out);
for (int i = 1; i <= clipnum; i++)
{
Clip clip = c[i];
P tp = c[i].coord;
P dirtp = c[i].dir;
ofs << tp.x << " " << tp.y << " " << tp.z;
double angel = Angle(dirtp, DX);
// if(angel>pi/2)angel=pi-angel;
// angel=min(angel,pi/2-angel);
if (angel <= minAngle || angel >= pi - minAngle)
ofs << 0 << endl;
else if (angel >= pi / 2 - minAngle && angel <= pi / 2 + minAngle)
ofs << 1 << endl;
else
ofs << 2 << endl;
10 months ago
}
}
} clipSet;