import utils.b_spline.parameter_selection as ps
import numpy as np
import utils.b_spline.bspline_curve as bc
import utils.b_spline.bspline_surface as bs
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import torch

def surface_inter(P_control):
    k = q = 2
    piece_uv = [6, 6]
    knot_uv = [[0,0,0, 0.3,0.5,0.7, 1,1,1], [0,0,0, 0.3,0.5,0.7, 1,1,1]]
    
    P_control = torch.cat((P_control.reshape(2,6,6), torch.zeros(1, 6, 6)), dim=0)
    P_control = P_control.detach().numpy().tolist()
    
    P_piece = bs.surface(P_control, k, q, piece_uv, knot_uv)
    P_piece = torch.Tensor(P_piece)[:2,:,:].permute(1,2,0).reshape(72)
    
    return P_piece