Метод simple для решения уравнений навье стокса

Уравнение Навье-Стокса и симуляция жидкостей на CUDA

Привет, Хабр. В этой статье мы разберемся с уравнением Навье-Стокса для несжимаемой жидкости, численно его решим и сделаем красивую симуляцию, работающую за счет параллельного вычисления на CUDA. Основная цель — показать, как можно применить математику, лежащую в основе уравнения, на практике при решении задачи моделирования жидкостей и газов.

Метод simple для решения уравнений навье стокса

Видео:Уравнения Навье-Стокса - Numberphile на русском.Скачать

Уравнения Навье-Стокса - Numberphile на русском.

Уравнение Навье-Стокса для несжимаемой жидкости

Метод simple для решения уравнений навье стокса

Я думаю каждый хоть раз слышал об этом уравнении, некоторые, быть может, даже аналитически решали его частные случаи, но в общем виде эта задачи остается неразрешенной до сих пор. Само собой, мы не ставим в этой статье цель решить задачу тысячелетия, однако итеративный метод применить к ней мы все же можем. Но для начала, давайте разберемся с обозначениями в этой формуле.

Условно уравнение Навье-Стокса можно разделить на пять частей:

  • Метод simple для решения уравнений навье стокса— обозначает скорость изменения скорости жидкости в точке (его мы и будем считать для каждой частицы в нашей симуляции).
  • Метод simple для решения уравнений навье стокса— перемещение жидкости в пространстве.
  • Метод simple для решения уравнений навье стокса— давление, оказываемое на частицу (здесь Метод simple для решения уравнений навье стокса— коэффициент плотности жидкости).
  • Метод simple для решения уравнений навье стокса— вязкость среды (чем она больше, тем сильнее жидкость сопротивляется силе, применяемой к ее части), Метод simple для решения уравнений навье стокса— коэффициент вязкости).
  • Метод simple для решения уравнений навье стокса— внешние силы, которые мы применяем к жидкости (в нашем случае сила будет играть вполне конкретную роль — она будет отражать действия, совершаемые пользователем.

Также, так как мы будем рассматривать случай несжимаемой и однородной жидкости, мы имеем еще одно уравнение: Метод simple для решения уравнений навье стокса. Энергия в среде постоянна, никуда не уходит, ниоткуда не приходит.

Будет неправильно обделить всех читателей, которые не знакомы с векторным анализом, поэтому заодно и бегло пройдемся по всем операторам, присутствующим в уравнении (однако, настоятельно рекомендую вспомнить, что такое производная, дифференциал и вектор, так как они лежат в основе всего того, о чем пойдет речь ниже).

Начнем мы с с оператора набла, представляющего из себя вот такой вектор (в нашем случае он будет двухкомпонентным, так как жидкость мы будет моделировать в двумерном пространстве):

Метод simple для решения уравнений навье стокса

Оператор набла представляет из себя векторный дифференциальный оператор и может быть применен как к скалярной функции, так и к векторной. В случае скаляра мы получаем градиент функции (вектор ее частных производных), а в случае вектора — сумму частых производных по осям. Главная особенность данного оператора в том, что через него можно выразить основные операции векторного анализа — grad (градиент), div (дивергенция), rot (ротор) и Метод simple для решения уравнений навье стокса(оператор Лапласа). Стоит сразу же отметить, что выражение Метод simple для решения уравнений навье стоксане равносильно Метод simple для решения уравнений навье стокса— оператор набла не обладает коммутативностью.

Как мы увидим далее, эти выражения заметно упрощаются при переходе на дискретное пространство, в котором мы и будем проводить все вычисления, так что не пугайтесь, если на данный момент вам не очень понятно, что же со всем этим делать. Разбив задачу на несколько частей, мы последовательно решим каждую из них и представим все это в виде последовательного применения нескольких функций к нашей среде.

Видео:Science show. Выпуск 51. Уравнение Навье - СтоксаСкачать

Science show. Выпуск 51. Уравнение Навье - Стокса

Численное решение уравнения Навье-Стокса

Чтобы представить нашу жидкость в программе, нам необходимо получить математическую репрезентацию состояния каждой частицы жидкости в произвольный момент времени. Самый удобный для этого метод — создать векторное поле частиц, хранящее их состояние, в виде координатной плоскости:

Метод simple для решения уравнений навье стокса

В каждой ячейке нашего двумерного массива мы будем хранить скорость частицы в момент времени Метод simple для решения уравнений навье стокса, а расстояние между частицами обозначим за Метод simple для решения уравнений навье стоксаи Метод simple для решения уравнений навье стоксасоответственно. В коде же нам будет достаточно изменять значение скорости каждую итерацию, решая набор из нескольких уравнений.

Теперь выразим градиент, дивергенцию и оператор Лапласа с учетом нашей координатной сетки ( Метод simple для решения уравнений навье стокса— индексы в массиве, Метод simple для решения уравнений навье стокса— взятие соответствующих компонентов у вектора):

ОператорОпределениеДискретный аналог
gradМетод simple для решения уравнений навье стоксаМетод simple для решения уравнений навье стокса
divМетод simple для решения уравнений навье стокса Метод simple для решения уравнений навье стокса
Метод simple для решения уравнений навье стоксаМетод simple для решения уравнений навье стоксаМетод simple для решения уравнений навье стокса
rotМетод simple для решения уравнений навье стоксаМетод simple для решения уравнений навье стокса

Мы можем еще сильнее упростить дискретные формулы векторных операторов, если положим, что Метод simple для решения уравнений навье стокса. Данное допущение не будет сильно сказываться на точности алгоритма, однако уменьшает количество операций на каждую итерацию, да и в целом делает выражения приятней взгляду.

Перемещение частиц

Данные утверждения работают только в том случае, если мы можем найти ближайшие частицы относительно рассматриваемой на данный момент. Чтобы свести на нет все возможные издержки, связанные с поиском таковых, мы будет отслеживать не их перемещение, а то, откуда приходят частицы в начале итерации путем проекции траектории движения назад во времени (проще говоря, вычитать вектор скорости, помноженный на изменение времени, из текущей позиции). Используя этот прием для каждого элемента массива, мы будем точно уверены, что у любой частицы будут «соседи»:

Метод simple для решения уравнений навье стокса

Положив, что Метод simple для решения уравнений навье стокса— элемент массива, хранящий состояния частицы, получаем следующую формулу для вычисления ее состояния через время Метод simple для решения уравнений навье стокса(мы полагаем, что все необходимые параметры в виде ускорения и давления уже рассчитаны):

Метод simple для решения уравнений навье стокса

Заметим сразу же, что при достаточно малом Метод simple для решения уравнений навье стоксаи скорости мы можем так и не выйти за пределы ячейки, поэтому очень важно правильно подобрать ту силу импульса, которую пользователь будет придавать частицам.

Чтобы избежать потери точности в случае попадания проекции на границу клеток или в случае получения нецелых координат, мы будем проводить билинейную интерполяцию состояний четырех ближайших частиц и брать ее за истинное значение в точке. В принципе, такой метод практически не уменьшит точность симуляции, и вместе с тем он достаточно прост в реализации, так что его и будем использовать.

Вязкость

Метод simple для решения уравнений навье стокса

. В таком случае итеративное уравнение для скорости примет следующий вид:

Метод simple для решения уравнений навье стокса

Мы несколько преобразуем данное равенство, приведя его к виду Метод simple для решения уравнений навье стокса(стандартный вид системы линейных уравнений):

Метод simple для решения уравнений навье стокса

где Метод simple для решения уравнений навье стокса— единичная матрица. Такие преобразования нам необходимы, чтобы в последствии применить метод Якоби для решения нескольких схожих систем уравнений. Его мы также обсудим в дальнейшем.

Внешние силы

Метод simple для решения уравнений навье стокса

Импульс-вектор легко посчитать как разность между предыдущей позицией мыши и текущей (если такая имелась), и здесь как раз-таки можно проявить креативность. Именно в этой части алгоритма мы можем внедрить добавление цветов в жидкость, ее подсветку и т.п. К внешним силам также можно отнести гравитацию и температуру, и хоть реализовать такие параметры несложно, в данной статье рассматривать их мы не будем.

Давление

Давление в уравнении Навье-Стокса — та сила, которая препятствует частицам заполнять все доступное им пространство после применения к ним какой-либо внешней силы. Сходу его расчет весьма затруднителен, однако нашу задачу можно значительно упростить, применив теорему разложения Гельмгольца.

Назовем Метод simple для решения уравнений навье стоксавекторное поле, полученное после расчета перемещения, внешних сил и вязкости. Оно будет иметь ненулевую дивергенцию, что противоречит условию несжимаемости жидкости (Метод simple для решения уравнений навье стокса), и чтобы это исправить, необходимо рассчитать давление. Согласно теореме разложения Гельмгольца, Метод simple для решения уравнений навье стоксаможно представить как сумму двух полей:

Метод simple для решения уравнений навье стокса

где Метод simple для решения уравнений навье стокса— и есть искомое нами векторное поле с нулевой дивергенцией. Доказательство этого равенства в данной статье приводиться не будет, однако в конце вы сможете найти ссылку с подробным объяснением. Мы же можем применить оператор набла к обоим частям выражения, чтобы получить следующую формулу для расчета скалярного поля давления:

Метод simple для решения уравнений навье стокса

Записанное выше выражение представляет из себя уравнение Пуассона для давления. Его мы также можем решить вышеупомянутым методом Якоби, и тем самым найти последнюю неизвестную переменную в уравнении Навье-Стокса. В принципе, системы линейных уравнений можно решать самыми разными и изощренными способами, но мы все же остановимся на самом простом из них, чтобы еще больше не нагружать данную статью.

Граничные и начальные условия

Любое дифференциальное уравнение, моделируемое на конечной области, требует правильно заданных начальных или граничных условий, иначе мы с очень большой вероятностью получим физически неверный результат. Граничные условия устанавливаются для контролирования поведения жидкости близ краев координатной сетки, а начальные условия задают параметры, которые имеют частицы в момент запуска программы.

Начальные условия будут весьма простыми — изначально жидкость неподвижна (скорость частиц равна нулю), и давление также равно нулю. Граничные условия будут задаваться для скорости и давления приведенными формулами:

Метод simple для решения уравнений навье стокса

Метод simple для решения уравнений навье стокса

Тем самым, скорость частиц на краях будет противоположна скорости у краев (тем самым они будут отталкиваться от края), а давление равно значению непосредственно рядом с границей. Данные операции следует применить ко всем ограничивающим элементам массива (к примеру, есть размер сетки Метод simple для решения уравнений навье стокса, то алгоритм мы применим для клеток, отмеченных на рисунке синим):

Метод simple для решения уравнений навье стокса

Краситель

Метод simple для решения уравнений навье стокса

В формуле Метод simple для решения уравнений навье стоксаотвечает за пополнение красителем области (возможно, в зависимости от того, куда нажмет пользователь), Метод simple для решения уравнений навье стоксанепосредственно является количество красителя в точке, а Метод simple для решения уравнений навье стокса— коэффициент диффузии. Решить его не составляет большого труда, так как вся основная работа по выводу формул уже проведена, и достаточно лишь сделает несколько подстановок. Краску можно реализовать в коде как цвет в формате RGB, и в таком случае задача сводится к операциям с несколькими вещественными величинами.

Завихренность

Метод simple для решения уравнений навье стокса

Метод simple для решения уравнений навье стокса

Метод simple для решения уравнений навье стокса

Метод simple для решения уравнений навье стокса

Метод simple для решения уравнений навье стоксаесть результат применения ротора к вектору скорости (его определение дано в начале статьи), Метод simple для решения уравнений навье стокса— градиент скалярного поля абсолютных значений Метод simple для решения уравнений навье стокса. Метод simple для решения уравнений навье стоксапредставляет нормализованный вектор Метод simple для решения уравнений навье стокса, а Метод simple для решения уравнений навье стокса— константа, контролирующая, насколько большими будут завихренности в нашей жидкости.

Метод Якоби для решения систем линейных уравнений

Метод simple для решения уравнений навье стокса

Для нас Метод simple для решения уравнений навье стокса— элементы массива, представляющие скалярное или векторное поле. Метод simple для решения уравнений навье стокса— номер итерации, его мы можем регулировать, чтобы как увеличить точность расчета или наоборот уменьшить, и повысить производительность.

Для расчет вязкости подставляем: Метод simple для решения уравнений навье стокса, Метод simple для решения уравнений навье стокса, Метод simple для решения уравнений навье стокса, здесь параметр Метод simple для решения уравнений навье стокса— сумма весов. Таким образом, нам необходимо хранить как минимум два векторных поля скоростей, чтобы независимо считать значения одного поля и записывать их в другое. В среднем, для расчета поля скорости методом Якоби необходимо провести 20-50 итераций, что весьма много, если бы мы выполняли вычисления на CPU.

Для уравнения давления мы сделаем следующую подстановку: Метод simple для решения уравнений навье стокса, Метод simple для решения уравнений навье стокса, Метод simple для решения уравнений навье стокса, Метод simple для решения уравнений навье стокса. В результате мы получим значение Метод simple для решения уравнений навье стоксав точке. Но так как оно используется только для расчета градиента, вычитаемого из поля скорости, дополнительные преобразования можно не выполнять. Для поля давления лучше всего выполнять 40-80 итераций, потому что при меньших числах расхождение становится заметным.

Видео:Уравнение Навье-Стокса на пальцах. МЛФ#2Скачать

Уравнение Навье-Стокса на пальцах. МЛФ#2

Реализация алгоритма

Реализовывать алгоритм мы будем на C++, также нам потребуется Cuda Toolkit (как его установить вы можете прочитать на сайте Nvidia), а также SFML. CUDA нам потребуется для распараллеливания алгоритма, а SFML будет использоваться только для создания окна и отображения картинки на экране (В принципе, это вполне можно написать на OpenGL, но разница в производительности будет несущественна, а вот код увеличится еще строк на 200).

Cuda Toolkit

Сначала мы немного поговорим о том, как использовать Cuda Toolkit для распараллеливания задач. Более подробный гайд предоставляется самой Nvidia, поэтому здесь мы ограничимся только самым необходимым. Также предполагается, что вы смогли установить компилятор, и у вас получилось собрать тестовый проект без ошибок.

Чтобы создать функцию, исполняющуюся на GPU, для начала необходимо объявить, сколько ядер мы хотим использовать, и сколько блоков ядер нужно выделить. Для этого Cuda Toolkit предоставляет нам специальную структуру — dim3, по умолчанию устанавливающую все свои значения x, y, z равными 1. Указывая ее как аргумент при вызове функции, мы можем управлять количеством выделяемых ядер. Так как работаем мы с двумерным массивом, то в конструкторе необходимо установить только два поля: x и y:

где size_x и size_y — размер обрабатываемого массива. Сигнатура и вызов функции выглядят следующим образом (тройные угловые скобки обрабатываются компилятором Cuda):

В самой функции можно восстановить индексы двумерного массива через номер блока и номер ядра в этом блоке по следующей формуле:

Следует отметить, что функция, исполняемая на видеокарте, должна быть обязательно помечена тегом __global__ , а также возвращать void, поэтому чаще всего результаты вычислений записываются в передаваемый как аргумент и заранее выделенный в памяти видеокарты массив.

За освобождение и выделение памяти на видеокарте отвечают функции CudaMalloc и CudaFree. Мы можем оперировать указателями на область памяти, которые они возвращают, но получить доступ к данным из основного кода не можем. Самый простой способ вернуть результаты вычислений — воспользоваться cudaMemcpy, схожей со стандартным memcpy, но умеющей копировать данные с видеокарты в основную память и наоборот.

SFML и рендер окна

Вооружившись всеми этими знаниями, мы наконец можем перейти к непосредственному написанию кода. Для начала давайте создадим файл main.cpp и разместим туда весь вспомогательный код для рендера окна:

строка в начале функции main

создает изображение формата RGBA в виде одномерного массива с константной длиной. Его мы будем передавать вместе с другими параметрами (позиция мыши, разница между кадрами) в функцию computeField. Последняя, как и несколько других функций, объявлены в kernel.cu и вызывают код, исполняемый на GPU. Документацию по любой из функций вы можете найти на сайте SFML, в коде файла не происходит ничего сверхинтересного, поэтому мы не будем надолго на нем останавливаться.

Вычисления на GPU

Чтобы начать писать код под gpu, для начала создадим файл kernel.cu и определим в нем несколько вспомогательных классов: Color3f, Vec2, Config, SystemConfig:

Атрибут __host__ перед именем метода означает, что код может исполнятся на CPU, __device__ , наоборот, обязует компилятор собирать код под GPU. В коде объявляются примитивы для работы с двухкомпонентными векторами, цветом, конфиги с параметрами, которые можно менять в рантайме, а также несколько статических указателей на массивы, которые мы будем использовать как буферы для вычислений.

cudaInit и cudaExit также определяеются достаточно тривиально:

В функции инициализации мы выделяем память под двумерные массивы, задаем массив цветов, которые мы будем использовать для раскраски жидкости, а также устанавливаем в конфиг значения по умолчанию. В cudaExit мы просто освобождаем все буферы. Как бы это парадоксально не звучало, для хранения двумерных массивов выгоднее всего использовать одномерные, обращение к которым будет осуществляться таким выражением:

Начнем реализацию непосредственного алгоритма с функции перемещения частиц. В advect передаются поля oldField и newField (то поле, откуда берутся данные и то, куда они записываются), размер массива, а также дельта времени и коэффициент плотности (используется для того, чтобы ускорить растворение красителя в жидкости и сделать среду не сильно чувствительной к действиям пользователя). Функция билинейной интерполяции реализована классическим образом через вычисление промежуточных значений:

Функцию диффузии вязкости было решено разделить на несколько частей: из главного кода вызывается computeDiffusion, которая вызывает diffuse и computeColor заранее указанное число раз, а затем меняет местами массив, откуда мы берем данные, и тот, куда мы их записываем. Это самый простой способ реализовать параллельную обработку данных, но мы расходует в два раза больше памяти.

Обе функции вызывают вариации метода Якоби. В теле jacobiColor и jacobiVelocity сразу же идет проверка, что текущие элементы не находятся на границе — в этом случае мы должны установить их в соответствии с формулами, изложенными в разделе Граничные и начальные условия.

Применение внешней силы реализовано через единственную функцию — applyForce, принимающую как аргументы позицию мыши, цвет красителя, а также радиус действия. С ее помощью мы можем придать скорость частицам, а также красить их. братная экспонента позволяет сделать область не слишком резкой, и при этом достаточно четкой в указанном радиусе.

Расчет завихренности представляет из себя уже более сложный процесс, поэтому его мы реализуем в computeVorticity и applyVorticity, заметим также, что для них необходимо определить два таких векторных оператора, как curl (ротор) и absGradient (градиент абсолютных значений поля). Чтобы задать дополнительные эффекты вихря, мы умножаем Метод simple для решения уравнений навье стоксакомпоненту вектора градиента на Метод simple для решения уравнений навье стокса, а затем нормализируем его, разделив на длину (не забыв при этом проверить, что вектор ненулевой):

Следующим этапом алгоритма будет вычисление скалярного поля давления и его проекция на поле скорости. Для этого нам потребуется реализовать 4 функции: divergency, которая будет считать дивергенцию скорости, jacobiPressure, реализующую метод Якоби для давления, и computePressure c computePressureImpl, проводящие итеративные вычисления поля:

Проекция умещается в две небольшие функции — project и вызываемой ей gradient для давления. Это, можно сказать, последний этап нашего алгоритма симуляции:

После проекции мы смело можем перейти к отрисовке изображения в буфер и различным пост-эффектам. В функции paint выполняется копирование цветов из поля частиц в массив RGBA. Также была реализована функция applyBloom, которая подсвечивает жидкость, когда на нее наведен курсор и нажата клавиша мыши. Из опыта, такой прием делает картину более приятной и интересной для глаз пользователя, но он вовсе не обязателен.

В постобработке также можно подсвечивать места, в которых жидкость имеет наибольшую скорость, менять цвет в зависимости от вектора движения, добавлять различные эффекты и прочее, но в нашем случае мы ограничимся своеобразным минимумом, ведь даже с ним изображения получаются весьма завораживающими (особенно в динамике):

И под конец у нас осталась одна главная функция, которую мы вызываем из main.cppcomputeField. Она сцепляет воедино все кусочки алгоритма, вызывая код на видеокарте, а также копирует данные с gpu на cpu. В ней же находится и расчет вектора импульса и выбор цвета красителя, которые мы передаем в applyForce:

Видео:Программа решения уравнений Навье-Стокса. Лекция 1. Введение.Скачать

Программа решения уравнений Навье-Стокса. Лекция 1. Введение.

Заключение

В этой статье мы разобрали численный алгоритм решения уравнения Навье-Стокса и написали небольшую программу-симуляцию для несжимаемой жидкости. Быть может мы и не разобрались во всех тонкостях, но я надеюсь, что материал оказался для вас интересным и полезным, и как минимум послужил хорошим введением в область моделирования жидкостей.

Как автор данной статьи, я буду искренне признателен любым комментариям и дополнениям, и постараюсь ответить на все возникшие у вас вопросы под этим постом.

Дополнительный материал

Весь исходный код, приведенный в данной статье, вы можете найти в моем Github-репозитории. Любые предложения по улучшению приветствуются.

Оригинальный материал, послуживший основой для данной статьи, вы можете прочесть на официальном сайте Nvidia (англ). В нем также представлены примеры реализации частей алгоритма на языке шейдеров:
developer.download.nvidia.com/books/HTML/gpugems/gpugems_ch38.html

Доказательство теоремы разложения Гельмгольца и огромное количество дополнительного материала про механику жидкостей можно найти в данной книге (англ, см. раздел 1.2):
Chorin, A.J., and J.E. Marsden. 1993. A Mathematical Introduction to Fluid Mechanics. 3rd ed. Springer.

Канал одного англоязычного ютубера, делающего качественный контент, связанной с математикой, и решением дифференциальных уравнений в частности (англ). Очень наглядные ролики, помогающие понять суть многих вещей в математике и физике:
3Blue1Brown — YouTube
Differential Equations (3Blue1Brown)

Также выражаю благодарность WhiteBlackGoose за помощь в подготовке материала для статьи.

И под конец небольшой бонус — несколько красивых скриншотов, снятых в программе:

Метод simple для решения уравнений навье стокса
Прямой поток (дефолтные настройки)

Метод simple для решения уравнений навье стокса
Водоворот (большой радиус в applyForce)

Метод simple для решения уравнений навье стокса
Волна (высокая завихренность + диффузия)

Также по многочисленным просьбам добавил видео с работой симуляции:


источники:

🎥 Видео

Уравнение Навье — Стокса для чайниковСкачать

Уравнение Навье — Стокса для чайников

Гладкое решение уравнения Навье — СтоксаСкачать

Гладкое решение уравнения Навье — Стокса

Вывод уравнений Навье-Стокса - Лекция 3Скачать

Вывод уравнений Навье-Стокса - Лекция 3

Алтаев Н. К. "Решение уравнения Навье-Стокса".Скачать

Алтаев Н. К. "Решение уравнения Навье-Стокса".

В А Садовничий, О А Олейник, Гагарин и уравнение Навье–СтоксаСкачать

В А Садовничий, О А Олейник, Гагарин и уравнение Навье–Стокса

Метод Ньютона (метод касательных) Пример РешенияСкачать

Метод Ньютона (метод касательных) Пример Решения

Вычислительная гидродинамика (ВГД). Уравнение Рейнольдса и метод конечных объемовСкачать

Вычислительная гидродинамика (ВГД). Уравнение Рейнольдса и метод конечных объемов

Задача на миллион. Уравнение Навье-СтоксаСкачать

Задача на миллион. Уравнение Навье-Стокса

Visual C++. Программа решения уравнений Навье-Стокса. Урок 6. Решение модельного уравнения энергииСкачать

Visual C++. Программа решения уравнений Навье-Стокса. Урок 6. Решение модельного уравнения энергии

Уравнение навье стоксаСкачать

Уравнение навье стокса

Как распознать талантливого математикаСкачать

Как распознать талантливого математика

Метод простых итераций пример решения нелинейных уравненийСкачать

Метод простых итераций пример решения нелинейных уравнений

Алтаев Н.К. Решение уравнения Навье-Стокса на основе методов теоретической физикиСкачать

Алтаев Н.К. Решение уравнения Навье-Стокса на основе методов теоретической физики

Навье - СтоксСкачать

Навье - Стокс

Дискретизация уравнений Навье-Стокса несжимаемой жидкости методами компьютерной алгебрыСкачать

Дискретизация уравнений Навье-Стокса несжимаемой жидкости методами компьютерной алгебры

anim_3d_fps_5_all_allСкачать

anim_3d_fps_5_all_all
Поделиться или сохранить к себе: