package com.ard.tianyu.domain.dto.converter.wayline;

import com.ard.common.utils.bean.BeanUtils;
import com.ard.tianyu.domain.dto.state.PointData;
import com.ard.tianyu.domain.dto.wayline.SurroundModel;
import com.ard.tianyu.netty.protobuf.TelemetryDataPB;

import java.util.List;
import java.util.stream.Collectors;

public class SurroundModelConverter {
    public static TelemetryDataPB.SurroundModel convertEntityToProto(SurroundModel surroundModel) {
        // 使用构建器来创建 Protobuf 对象
        TelemetryDataPB.SurroundModel.Builder builder = TelemetryDataPB.SurroundModel.newBuilder();

        // 复制基本属性
        builder.setServiceId(surroundModel.getServiceId());
        builder.setCameraId(surroundModel.getCameraId());
        builder.setIsModeling(surroundModel.getIsModeling());

        // 处理 repeated fields
        if (surroundModel.getArea() != null && !surroundModel.getArea().isEmpty()) {
            List<TelemetryDataPB.PointData> area = surroundModel.getArea().stream()
                    .map(PointDataConverter::convertEntityToProto)
                    .collect(Collectors.toList());
            builder.addAllArea(area);
        }

        // 构建并返回 Protobuf 对象
        return builder.build();
    }

    public static SurroundModel convertProtoToEntity(TelemetryDataPB.SurroundModel proto) {
        SurroundModel surroundModel = new SurroundModel();
        BeanUtils.copyProperties(proto, surroundModel);

        // 处理 repeated fields
        if (proto.getAreaCount() > 0) {
            List<PointData> area = proto.getAreaList().stream()
                    .map(PointDataConverter::convertProtoToEntity)
                    .collect(Collectors.toList());
            surroundModel.setArea(area);
        }
        
        return surroundModel;
    }
}