colobot/src/object/implementation/programmable_impl.h

/*
 * This file is part of the Colobot: Gold Edition source code
 * Copyright (C) 2001-2015, Daniel Roux, EPSITEC SA & TerranovaTeam
 * http://epsiteс.ch; http://colobot.info; http://github.com/colobot
 *
 * This program 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.
 *
 * This program 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 this program. If not, see http://gnu.org/licenses
 */

#pragma once

#include "object/interface/interactive_object.h"
#include "object/interface/programmable_object.h"

#include "object/trace_color.h"

#include "math/vector.h"

#include <sstream>

class CObject;

enum TraceOper
{
    TO_STOP         = 0,    // stop
    TO_ADVANCE      = 1,    // advance
    TO_RECEDE       = 2,    // back
    TO_TURN         = 3,    // rotate
    TO_PEN          = 4,    // color change
};

struct TraceRecord
{
    TraceOper   oper;
    float       param;
};

class CProgrammableObjectImpl : public CProgrammableObject
{
public:
    explicit CProgrammableObjectImpl(ObjectInterfaceTypes& types, CObject* object);
    virtual ~CProgrammableObjectImpl();

    bool EventProcess(const Event& event);

    bool IsProgram() override;
    void RunProgram(Program* program) override;
    int GetProgram() override;
    void StopProgram() override;

    bool IntroduceVirus() override;
    void SetActiveVirus(bool bActive) override;
    bool GetActiveVirus() override;

    void SetScriptRun(Program* rank) override;
    Program* GetScriptRun() override;
    void SetSoluceName(const std::string& name) override;
    const std::string& GetSoluceName() override;

    bool ReadSoluce(const std::string& filename) override;
    bool ReadProgram(Program* program, const std::string& filename) override;
    bool GetCompile(Program* program) override;
    bool WriteProgram(Program* program, const std::string& filename) override;
    bool ReadStack(FILE *file) override;
    bool WriteStack(FILE *file) override;

    Program* AddProgram() override;
    void AddProgram(std::unique_ptr<Program> program) override;
    void RemoveProgram(Program* program) override;
    Program* CloneProgram(Program* program) override;

    std::vector<std::unique_ptr<Program>>& GetPrograms() override;
    int GetProgramCount() override;
    Program* GetProgram(int index) override;
    Program* GetOrAddProgram(int index) override;
    int GetProgramIndex(Program* program) override;

    void TraceRecordStart() override;
    void TraceRecordStop() override;
    bool IsTraceRecord() override;

    void SetActivity(bool bMode) override;
    bool GetActivity() override;

    void SetCmdLine(unsigned int rank, float value);
    float GetCmdLine(unsigned int rank) override;
    std::vector<float>& GetCmdLine();

private:
    //! Save current status to recording buffer
    void        TraceRecordFrame();
    //! Save this operation to recording buffer
    bool        TraceRecordOper(TraceOper oper, float param);
    //! Convert this recording operation to CBot instruction
    bool        TraceRecordPut(std::stringstream& buffer, TraceOper oper, float param);

private:
    CObject* m_object;

private:
    bool                m_activity;

    std::vector<float>  m_cmdLine;

    std::vector<std::unique_ptr<Program>> m_program;
    Program*            m_currentProgram;

    bool                m_activeVirus;

    Program*            m_scriptRun;
    std::string         m_soluceName;

    bool                m_traceRecord;
    TraceOper           m_traceOper;
    Math::Vector        m_tracePos;
    float               m_traceAngle;
    TraceColor          m_traceColor;
    int                 m_traceRecordIndex;
    std::unique_ptr<TraceRecord[]> m_traceRecordBuffer;
};