Vamos Automotive Simulator git

//  Copyright (C) 2001-2009 Sam Varner
//
//  This file is part of Vamos Automotive Simulator.
//
//  Vamos is free software: you can redistribute it and/or modify it under the terms of
//  the GNU General Public License as published by the Free Software Foundation, either
//  version 3 of the License, or (at your option) any later version.
//
//  Vamos is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
//  without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
//  PURPOSE.  See the GNU General Public License for more details.
//
//  You should have received a copy of the GNU General Public License
//  along with Vamos.  If not, see <http://www.gnu.org/licenses/>.

#ifndef VAMOS_BODY_DRIVETRAIN_HPP_INCLUDED
#define VAMOS_BODY_DRIVETRAIN_HPP_INCLUDED

#include "../geometry/Constants.hpp"
#include "Clutch.hpp"
#include "Differential.hpp"
#include "Engine.hpp"
#include "Transmission.hpp"

#include <memory>

namespace Vamos_Body
{
/// The drivetrain manages the engine, clutch, transmission and differential.
class Drivetrain
{
public:
    Drivetrain(Engine *engine, Clutch *clutch, Transmission *transmission,
               Differential *differential);

    void propagate(double time);

    /// Process the input parameters.
    void input(double gas, double clutch, double left_wheel_speed, double right_wheel_speed);
    void reset();
    std::shared_ptr<Engine> engine() { return mp_engine; }
    Transmission& transmission() { return *mp_transmission; }
    Clutch& clutch() { return *mp_clutch; }
    /// @return the torque for a driven wheel.
    double torque(Vamos_Geometry::Direction side) const;
    void set_auto_neutral(bool state) { m_auto_neutral = state; }

private:
    std::shared_ptr<Engine> mp_engine; ///< Also a particle in the body.
    std::unique_ptr<Clutch> mp_clutch;
    std::unique_ptr<Transmission> mp_transmission;
    std::unique_ptr<Differential> mp_differential;
    /// The throttle position.
    double m_gas = 0.0;
    /// If true, shift to neutral instead of stalling.
    bool m_auto_neutral = false;
};
} // namespace Vamos_Body

#endif // VAMOS_BODY_DRIVETRAIN_HPP_INCLUDED