RegulatoryElement.cs

#region License
/******************************************************************************
* Copyright 2018-2021 The AutoCore Authors. All Rights Reserved.
* 
* Licensed under the GNU Lesser General Public License, Version 3.0 (the "License"); 
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* 
* https://www.gnu.org/licenses/lgpl-3.0.html
* 
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*****************************************************************************/
#endregion

using UnityEditor;
using UnityEngine;

namespace Packages.MapToolbox
{
    [ExecuteInEditMode]
    [RequireComponent(typeof(Relation))]
    public class RegulatoryElement : MonoBehaviour
    {
        public Way ref_line;
        public Way refers;
        public enum SubType
        {
            road,
            traffic_light,
            traffic_sign
        }
        public SubType subType;
        public Relation Relation => GetComponent<Relation>() ?? gameObject.AddComponent<Relation>();
        internal static RegulatoryElement AddNew(Lanelet2Map map)
        {
            var ret = map.AddChildGameObject<RegulatoryElement>(map.transform.ChildMapId());
            ret.gameObject.RecordUndoCreateGo();
            return ret;
        }
        internal bool drawGizmos = false;
        private void OnDrawGizmos()
        {
            if (drawGizmos && ref_line.Nodes.Count > 0 && refers.Nodes.Count > 0)
            {
                Gizmos.color = Color.green;
                var start = Vector3.zero;
                foreach (var item in ref_line.Nodes)
                {
                    start += item.Position;
                }
                start /= ref_line.Nodes.Count;
                var end = Vector3.zero;
                foreach (var item in refers.Nodes)
                {
                    end += item.Position;
                }
                end /= refers.Nodes.Count;
                Gizmos.DrawLine(start, end);
            }
        }
    }
    [CanEditMultipleObjects]
    [CustomEditor(typeof(RegulatoryElement))]
    class RegulatoryElementEditor : Editor
    {
        private void OnEnable()
        {
            foreach (RegulatoryElement item in targets)
            {
                item.drawGizmos = true;
            }
        }
        private void OnDisable()
        {
            foreach (RegulatoryElement item in targets)
            {
                item.drawGizmos = false;
            }
        }
    }
}