steeplechase/unity/Assets/Scripts/Wraparound.cs

using System.Collections;
using System.Collections.Generic;
using UnityEngine;

public class Wraparound : MonoBehaviour
{
    [SerializeField] bool _justVertical = false;
    // Start is called before the first frame update
    void Start()
    {
        
    }

    void ClearTrail()
    {
        DragonMovement dragonMovement = GetComponent<DragonMovement>();
        if (dragonMovement)
        {
            dragonMovement.ClearTrail();
        }
    }

    // Update is called once per frame
    void Update()
    {
        const int SCREEN_HEIGHT = 1080;
        const int SCREEN_WIDTH = 1920;
        if (this.transform.position.y < 0)
        {
            this.transform.position = new Vector3(transform.position.x, transform.position.y + SCREEN_HEIGHT, transform.position.z);
            ClearTrail();
        }

        if (this.transform.position.y > SCREEN_HEIGHT)
        {
            this.transform.position = new Vector3(transform.position.x, transform.position.y - SCREEN_HEIGHT, transform.position.z);
            ClearTrail();
        }

        if (_justVertical == false)
        {
            if (this.transform.position.x < 0)
            {
                this.transform.position = new Vector3(transform.position.x + SCREEN_WIDTH, transform.position.y, transform.position.z);
            }

            if (this.transform.position.x > SCREEN_WIDTH)
            {
                this.transform.position = new Vector3(transform.position.x - SCREEN_WIDTH, transform.position.y, transform.position.z);
            }
        }
    }
}