PendulumScene.cpp

#include "PendulumGraphicsItem.h"
#include "PendulumScene.h"
#include "ToQtCoordinates.h"

#include <iostream>
#include <QGraphicsSceneMouseEvent>

#define XPOS 0.0
#define YPOS 0.0
#define WIDTH 450
#define HEIGHT 650

PendulumScene::PendulumScene(QObject *parent)
    :
      QGraphicsScene(XPOS, YPOS, WIDTH, HEIGHT, parent)
{
    addRect(XPOS,YPOS,WIDTH,HEIGHT,QPen(Qt::black));
    toQtCoordinates = new ToQtCoordinates(
                ToQtCoordinates::Type::MidPoint,WIDTH,HEIGHT);

    int count =1;
    for(int i = 1; i <= numOfPendulums; i++){
        vectorOfOmegas.push_back(2*PI/(60.0/(maxNumOfCyclesPerMinute-i)));
        vectorOfLengths.push_back(initialLength+2*radius*(i-1));

        vectorOfPendulums.push_back(
                    std::make_shared<PendulumGraphicsItem>(
                        toQtCoordinates->convertToQtCoordinates(
                            QPointF(
                                -vectorOfLengths.back()*sin(initialTetha),
                                altura - vectorOfLengths.back()*cos(initialTetha)
                                )
                            ),
                        radius
                        )
                    );

        addItem(vectorOfPendulums.back().get());

        count++;
    }
}

void PendulumScene::mousePressEvent(QGraphicsSceneMouseEvent *mouseEvent)
{
    std::cout << "Scene position: " << mouseEvent->scenePos().x() << " , "  << mouseEvent->scenePos().y() << std::endl;
    std::cout << "Scene: " << this->width() << " , "  << this->height() << std::endl;
}

void PendulumScene::setPendulumSceneToTimeZero()
{
    currentTime = 0;
    vectorOfThetas.clear();
    setPendulumnPositionsAtTimet(currentTime);
    update();
}

void PendulumScene::onUpdate()
{
    vectorOfThetas.clear();
    setPendulumnPositionsAtTimet(currentTime);

    currentTime += timeIncrement;
    update();
}

void PendulumScene::setPendulumColors(int index)
{
    for(int i = 0; i < vectorOfPendulums.size(); i++){

        switch (index) {
        case 0:
            vectorOfPendulums.at(i)->setColor(ColorOfBall::Yellow);
            break;
        case 1:
            if(i%2 == 0){
                vectorOfPendulums.at(i)->setColor(ColorOfBall::Red);
            }
            else{
                vectorOfPendulums.at(i)->setColor(ColorOfBall::Blue);
            }
            break;
        case 2:
            if(i%3 == 0){
                vectorOfPendulums.at(i)->setColor(ColorOfBall::Red);
            }
            else if(i%3 == 1){
                vectorOfPendulums.at(i)->setColor(ColorOfBall::Blue);
            }
            else{
                vectorOfPendulums.at(i)->setColor(ColorOfBall::Green);
            }
            break;
        case 3:
            break;
        }
    }
    update();
}

void PendulumScene::setInitialTetha(double initialTetha)
{
    this->initialTetha = initialTetha*PI/180;
    vectorOfThetas.clear();

    setPendulumnPositionsAtTimet(currentTime);
    update();
}

void PendulumScene::setPendulumnPositionsAtTimet(double currentTime)
{
    for(int i = 0; i < numOfPendulums; i++){

        vectorOfThetas.push_back(initialTetha*cos(vectorOfOmegas.at(i)*currentTime + PI));

        vectorOfPendulums.at(i)->setCenterOfBall(
                    toQtCoordinates->convertToQtCoordinates(
                        QPointF(
                            vectorOfLengths.at(i)*sin(vectorOfThetas.at(i)),
                            altura - vectorOfLengths.at(i)*cos(vectorOfThetas.at(i))
                            )
                        )
                    );
    }
}

void PendulumScene::setTimeIncrement(double timeIncrement)
{
    this->timeIncrement = timeIncrement;
}