main.cpp

#include <iostream>

#include "glframework/core.h"
#include "wrapper/checkError.h"
#include "application/Application.h"
#include "glframework/shader.h"
#include "glframework/texture.h"
#include "glframework/geometry.h"

#include "application/camera/camera.h"
#include "application/camera/orthographicCamera.h"
#include "application/camera/perspectiveCamera.h"

#include "application/camera/cameraControl.h"
#include "application/camera/gameCameraControl.h"
#include "application/camera/trackBallCameraControl.h"

#include "application/assimpLoader.h"

#include "glframework/material/material.h"
#include "glframework/material/phongMaterial.h"
#include "glframework/material/whiteMaterial.h"
#include "glframework/material/opacityMaskMaterial.h"
#include "glframework/material/screenMaterial.h"
#include "glframework/material/cubeMaterial.h"
#include "glframework/material/phongEnvMaterial.h"

#include "glframework/mesh.h"

#include "glframework/light/directionalLight.h"
#include "glframework/light/ambientLight.h"
#include "glframework/light/pointLight.h"

#include "glframework/renderer/renderer.h"

#include "glframework/frameBuffer/frameBuffer.h"

#include "imgui/imgui.h"
#include "imgui/imgui_impl_glfw.h"
#include "imgui/imgui_impl_opengl3.h"

int WIDTH = 800, HEIGHT = 600;

Renderer* renderer = nullptr;
Scene* scene = nullptr;
Camera* camera = nullptr;
CameraControl* cameraControl = nullptr;

DirectionalLight* dirLight = nullptr;
AmbientLight* ambLight = nullptr;

glm::vec3 clearColor{};

FrameBuffer* frameBuffer = nullptr;

void OnResize(int width, int height) {
    GL_CALL(glViewport(0, 0, width, height));
}

void onKey(int key, int action, int mods) {
    cameraControl->onKey(key, action, mods);
}

void onMouse(int button, int action, int mods) {
    double x = 0, y = 0;
    glApp->getCursorPosition(&x, &y);
    cameraControl->onMouse(button, action, x, y);
}

void onCursor(double xpos, double ypos) {
    cameraControl->onCursor(xpos, ypos);
}

void onScroll(double offset) {
    cameraControl->onScroll(offset);
}

void prepareCamera() {
    camera = new PerspectiveCamera(60.0f, (float)(glApp->getWidth()) / (float)(glApp->getHeight()), 0.1f, 1000.0f);
    // cameraControl = new TrackBallCameraControl();
    cameraControl = new GameCameraControl();
    cameraControl->setCamera(camera);
    cameraControl->setSensitivity(0.2f);
}

void setModelBlend(Object* obj, bool blend, float opacity) {
    if (obj->getType() == ObjectType::Mesh) {
        Mesh* mesh = (Mesh *) obj;
        Material* material = mesh->mMaterial;
        material->mBlend = blend;
        material->mOpacity = opacity;
        material->mDepthWrite = false;
    }
    auto children = obj->getChildren();
    for (auto child: children)
        setModelBlend(child, blend, opacity);
}

void prepare() {
    renderer = new Renderer();
    scene = new Scene();

    auto boxGeo = Geometry::createBox(1.0f);
    auto boxMat = new CubeMaterial();
    boxMat->mDiffuse = new Texture("assets/textures/sphericalMap.jpg", 0);
    // boxMat->mDepthWrite = false;    // 若绘制天空盒, 需关闭深度写入(或让shader使用输出深度值=1)
    auto boxMesh = new Mesh(boxGeo, boxMat);
    scene->addChild(boxMesh);

    auto sphereGeo = Geometry::createSphere(4.0f);
    auto sphereMat = new PhongMaterial();
    sphereMat->mDiffuse = new Texture("assets/textures/noir.png", 0);
    auto sphereMesh = new Mesh(sphereGeo, sphereMat);
    scene->addChild(sphereMesh);

    dirLight = new DirectionalLight();
    dirLight->mDirection = glm::vec3(-1.0f);
    dirLight->mSpecularIntensity = 0.1f;
    
    ambLight = new AmbientLight();
    ambLight->mColor = glm::vec3(0.1f);
}

void initIMGUI() {
    ImGui::CreateContext();
    ImGui::StyleColorsDark();

    ImGui_ImplGlfw_InitForOpenGL(glApp->getWindow(), true);
    ImGui_ImplOpenGL3_Init("#version 410");
}

void renderIMGUI() {
    ImGui_ImplOpenGL3_NewFrame();
    ImGui_ImplGlfw_NewFrame();
    ImGui::NewFrame();

    ImGui::Begin("Hello, world!");
    ImGui::Text("ChangeColor Demo");
    ImGui::Button("Test Button", ImVec2(40, 20));
    ImGui::ColorEdit3("Clear Color", (float *) &clearColor);
    ImGui::End();

    ImGui::Render();
    int display_w, display_h;
    glfwGetFramebufferSize(glApp->getWindow(), &display_w, &display_h);
    glViewport(0, 0, display_w, display_h);

    ImGui_ImplOpenGL3_RenderDrawData(ImGui::GetDrawData());
}

int main() {    
    if (!glApp->init(WIDTH, HEIGHT))
        return -1;

    glApp->setResizeCallback(OnResize);
    glApp->setKeyBoardCallback(onKey);
    glApp->setMouseCallback(onMouse);
    glApp->setCursorCallback(onCursor);
    glApp->setScrollCallback(onScroll);

    // 设置openGL视口并清理颜色
    glViewport(0, 0, WIDTH, HEIGHT);
    glClearColor(0.0f, 0.0f, 0.0f, 1.0f);   // 设置用于Clear时的颜色, 以便在Clear时将整个画布设置为该颜色

    prepare();
    prepareCamera();
    initIMGUI();

    // 执行窗体循环
    while (glApp->update()) {
        cameraControl->update();
        renderer->setClearColor(clearColor);
        renderer->render(scene, camera, dirLight, ambLight);
        renderIMGUI();
    }

    glApp->destroy();
    return 0;
}