UDRIGEEKCup2024/Assets/GolfControls/BallPhysics.cs

using System;
using UnityAtoms.BaseAtoms;
using UnityEngine;

namespace GolfControls
{
    public class BallPhysics : MonoBehaviour
    {
        private float baseDrag;
        private float baseAngularDrag;

        private Rigidbody body;
        
        // Start is called before the first frame update
        void Start()
        {
            body = gameObject.GetComponent<Rigidbody>();
            baseDrag = body.drag;
            baseAngularDrag = body.angularDrag;
        }

        private void OnCollisionEnter(Collision other)
        {
            PhysicsTerrain terrain = other.gameObject.GetComponent<PhysicsTerrain>();
            if (terrain == null) return;
            body.drag = baseDrag + terrain.drag;
            body.angularDrag = baseAngularDrag + terrain.angularDrag;
        }
    }
}