chocc/geo/geo.h

# pragma once
#include<iostream>
#include<ostream>
#include<Eigen/Dense>
#include<vector>
using namespace std;
using namespace Eigen;


const int acc=49;//圆一圈采样acc个点
typedef Vector3d Point3D;
typedef Vector2d Point2D;



struct Circ2D
{
    Point2D center;
    double r;
};

struct Point3DPolar 
{
    double r;
    double theta;
    double phi;
};

struct Sphere
{
    double r;
    Point3D center;
    Sphere(double r=-1,Point3D center=Vector3d(0,0,0)):r(r),center(center){}
    Vector3d getSurPoint(const Vector3d& p);
};

struct Cone
{
    Point3D top;
    Vector3d dir;
    double h1,h2;
    double r;
};

struct InfCone
{
    Point3D top;
    Vector3d dir;
    double angle;
};


struct Triangle
{
    Point3D a;
    Point3D b;
    Point3D c;
};

struct Triangle2D
{
    Point2D a;
    Point2D b;
    Point2D c;
};


struct Plane
{
    Vector3d dirc;
    Point3D point;
    Plane(Vector3d dirc=Vector3d(0,0,0),Point3D point=Vector3d(0,0,0)):dirc(dirc),point(point){}
};

struct Line2D
{
    Vector2d dirc;
    Point2D point;
};

struct Circ
{
    Point3D center;
    Vector3d dir;
    double r;
    Vector3d vertica[2];
    void getvertica(){
        Matrix<double,1,3> equation(dir);
        FullPivLU<Matrix<double,1,3>> solve(equation);
        vertica[0]=(Matrix<double,3,2>(solve.kernel()).col(0)).normalized();
        vertica[1]=dir.cross(vertica[0]).normalized();
    }
    Circ(Point3D center=Vector3d(0,0,0),Vector3d dir=Vector3d(0,0,1),double r=1):center(center),dir(dir),r(r){
        dir=dir.normalized();
        getvertica();
    }
    int geti(const Point3D& p,const int& num=acc){
        double angle=atan2((p-center).dot(vertica[0]),(p-center).dot(vertica[1]));
        if(angle<0){
            angle+=2*M_PI;
        }
        return int(round(angle/(2*M_PI)*num))%num;
    }
    double getangle(const Point3D& p){
        double angle=atan2((p-center).dot(vertica[0]),(p-center).dot(vertica[1]));
        if(angle<0){
            angle+=2*M_PI;
        }
        return angle;
    }
};


//作为map的索引
struct edgeNum:pair<int,int>
{
    //下面的定义是为了方便map做索引
    edgeNum(int i,int j){
        this->first=max(i,j);
        this->second=min(i,j);
    }
    bool operator<(edgeNum a){
        if(this->first<a.first){
            return true;
        }
        else if(this->first==a.first&&this->second<a.second){
            return true;
        }
        else{
            return false;
        }
    }
};

//作为
struct FaceNum{
    vector<int> data;
    FaceNum(int a,int b,int c,int d){
        data.push_back(a);
        data.push_back(b);
        data.push_back(c);
        data.push_back(d);
        sort(data.begin(),data.end(),greater<int>());
    }
    FaceNum(const vector<int>& x){
        data.push_back(x[0]);
        data.push_back(x[1]);
        data.push_back(x[2]);
        data.push_back(x[3]);
        sort(data.begin(),data.end(),greater<int>());
    }
    bool operator<(const FaceNum& a) const{
        if(this->data[0]<a.data[0]){
            return true;
        }
        else if(this->data[0]==a.data[0]&&
            this->data[1]<a.data[1]){
            return true;
        }
        else if(this->data[0]==a.data[0]&&
            this->data[1]==a.data[1]&&
            this->data[2]<a.data[2]){
            return true;
        }
        else if(this->data[0]==a.data[0]&&
            this->data[1]==a.data[1]&&
            this->data[2]==a.data[2]&&
            this->data[3]<a.data[3]){
            return true;
        }
        else{
            return false;
        }
    }
};

//作为方向向量的索引
int Veckey(Vector3d x){
    return int((x.x()+4*x.y()+16*x.z())*3000);
}

struct print3d
{
    vector<Point3D> p;
    vector<Vector3i> f;
};


const int MESHN=6;
const double MESHR=10;
const double LenBetweenCirc=100;