WireRouting/lib/Clip.cpp

#include "pch.h" // use stdafx.h in Visual Studio 2017 and earlier
#include "Clip.h"

ClipSet clipSet;

Clip::Clip()
{
	emc = 0;
	bundleNodeID = 0;
	maxDia = MAXDia;
	id = 0;
	used = 0;
}

// 清除所有通过卡箍的线和通道信息
void Clip::clear()
{
	diaset.clear();
	bundleNodeID = 0;
	used = 0;
}

// 在卡箍里加入直径为d的线缆
void Clip::addwire(double d)
{
	diaset.push_back(d);
	dia = sqrt(dia * dia + d * d);
}

// 返回通过卡箍的线束捆直径
double Clip::getDia()
{
	double dia1 = 1.154 * dia;
	if (dia1 >= 10)
		dia1 *= 1.235 / 1.154;
	return dia1;
}

// 判断直径为d的线缆能否通过
bool Clip::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;
	return false;
}

// 判断从p点到卡箍，应该从哪一面进入，正面为0，反面为1
int Clip::inDir(P p)
{
	double angle = Angle(coord - p, dir);
	if (angle > pi / 2)
		return 1;
	return 0;
}

ClipSet::ClipSet()
{
	clipnum = 0;
}

void ClipSet::clearWire()
{
	for (int i = 1; i <= clipnum; i++)
		c[i].clear();
}

// 新加入一个卡箍
void ClipSet::addclip(P p, P d, string name, int emc)
{
	clipnum++;
	c[clipnum].coord = p;
	c[clipnum].coord.ref = clipnum;
	c[clipnum].coord.type = 0;
	c[clipnum].coord.setDir(p - d);

	c[clipnum].id = clipnum;
	c[clipnum].dir = p - d;
	c[clipnum].name = name;
	c[clipnum].emc = emc;
	clipid[p] = clipnum;
}

// 添加通过卡箍的导线
void ClipSet::addwire(int id, double d)
{
	c[id].addwire(d);
}

void ClipSet::addwire(P &p, double d)
{
	int id = clipid[p];
	c[id].addwire(d);
}

// 获取指向编号为id的卡箍的指针
Clip *ClipSet::getClipPointer(int id, int bundleNodeID)
{
	if (bundleNodeID != -1)
		c[id].bundleNodeID = bundleNodeID;
	return &c[id];
}

// 判断两个卡箍是否互相为最近(已知i最近的卡箍是j)
bool ClipSet::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;
}

// 判断卡箍能否通过
bool ClipSet::transitable(int id, int emc, double d)
{
	if (c[id].emc != emc)
		return false;
	return c[id].transitable(d);
}

// 根据卡箍位置估计飞机的近似中心线
void ClipSet::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);

		MAXY = max(MAXY, tp.y);
		MINY = min(MINY, tp.y);

		MAXZ = max(MAXZ, tp.z);
		MINZ = min(MINZ, tp.z);
	}
	Ycenter /= clipnum;
	Zcenter /= clipnum;
}

// 打印卡箍距离近似中心线的距离
void ClipSet::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;
	}
}

// 根据卡箍位置先创建一个近似的通道，每个卡箍连到距离最近的两个卡箍
void ClipSet::buildChannel()
{
	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);
			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));
		}
		sort(vp.begin(), vp.end());
		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)
			{
				c[i].channelEdge.push_back(cid);
				dir = c[cid].coord - c[i].coord;
				sum++;
				sumDis += vp[j].first;
			}
			else
			{
				P dir2 = c[cid].coord - c[i].coord;
				if (Angle(dir, dir2) >= pi / 2 - minAngle)
				{
					c[i].channelEdge.push_back(cid);
					sum++;
					sumDis += vp[j].first;
					break;
				}
			}
		}
		if (c[i].channelEdge.size() < 2 && !flag)
			flag = true, i--;
		else
			flag = false;
	}
}

// 打印所有卡箍信息到clipDir文件，卡箍坐标后面会加上卡箍类型
/*
	0代表航向卡箍
	1代表垂直向卡箍
	2代表其他方向卡箍
*/
void ClipSet::print()
{
	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;
	}
}