-
Notifications
You must be signed in to change notification settings - Fork 8
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
170 changed files
with
48,689 additions
and
0 deletions.
There are no files selected for viewing
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
|
||
Detailed instructions on how to compile and run mantaflow can be found at: | ||
http://www.mantaflow.com/install.html and | ||
http://www.mantaflow.com/quickstart.html | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,51 @@ | ||
# MLFLIP | ||
|
||
You can find the source codes for the SCA paper: [\[Um et al., 2018, "Liquid | ||
Splash Modeling with Neural | ||
Networks"\]](https://ge.in.tum.de/publications/2018-mlflip-um/). | ||
|
||
## Requirements | ||
Activate your tensorflow environment on your machine; e.g., | ||
``` | ||
source ~/tensorflow/bin/activate | ||
``` | ||
|
||
You need to compile the delivered mantaflow sources. Please refer to the | ||
general guideline to compile mantaflow: http://mantaflow.com/install.html | ||
|
||
MLFLIP requires a special cmake option for numpy support; e.g., in your build directory, | ||
``` | ||
cmake .. -DGUI=ON -DNUMPY=ON | ||
``` | ||
|
||
## Run training simulations | ||
``` | ||
for i in {00..09}; do | ||
./manta ../scenes/tsim_flip.py --nogui --seed=$i -o /tmp/tsim_flip_$i | ||
done | ||
``` | ||
|
||
|
||
## Generate training data | ||
``` | ||
for i in {00..09}; do | ||
./manta ../scenes/tdata_gen.py -o /tmp/tdata/tsim_flip_$i /tmp/tsim_flip_$i | ||
done | ||
``` | ||
|
||
|
||
## Train a model | ||
``` | ||
../scenes/tf_train.py --mve -o /tmp/tfmodel/ /tmp/tdata/ | ||
``` | ||
|
||
|
||
## Run MLFLIP | ||
``` | ||
./manta ../scenes/mlflip.py --load /tmp/tfmodel/ | ||
``` | ||
|
||
If you want to use a pre-trained model, please try: | ||
``` | ||
./manta ../scenes/mlflip.py --load ../scenes/tfmodel/ | ||
``` |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,179 @@ | ||
/* adler32.c -- compute the Adler-32 checksum of a data stream | ||
* Copyright (C) 1995-2011 Mark Adler | ||
* For conditions of distribution and use, see copyright notice in zlib.h | ||
*/ | ||
|
||
/* @(#) $Id$ */ | ||
|
||
#include "zutil.h" | ||
|
||
#define local static | ||
|
||
local uLong adler32_combine_ OF((uLong adler1, uLong adler2, z_off64_t len2)); | ||
|
||
#define BASE 65521 /* largest prime smaller than 65536 */ | ||
#define NMAX 5552 | ||
/* NMAX is the largest n such that 255n(n+1)/2 + (n+1)(BASE-1) <= 2^32-1 */ | ||
|
||
#define DO1(buf,i) {adler += (buf)[i]; sum2 += adler;} | ||
#define DO2(buf,i) DO1(buf,i); DO1(buf,i+1); | ||
#define DO4(buf,i) DO2(buf,i); DO2(buf,i+2); | ||
#define DO8(buf,i) DO4(buf,i); DO4(buf,i+4); | ||
#define DO16(buf) DO8(buf,0); DO8(buf,8); | ||
|
||
/* use NO_DIVIDE if your processor does not do division in hardware -- | ||
try it both ways to see which is faster */ | ||
#ifdef NO_DIVIDE | ||
/* note that this assumes BASE is 65521, where 65536 % 65521 == 15 | ||
(thank you to John Reiser for pointing this out) */ | ||
# define CHOP(a) \ | ||
do { \ | ||
unsigned long tmp = a >> 16; \ | ||
a &= 0xffffUL; \ | ||
a += (tmp << 4) - tmp; \ | ||
} while (0) | ||
# define MOD28(a) \ | ||
do { \ | ||
CHOP(a); \ | ||
if (a >= BASE) a -= BASE; \ | ||
} while (0) | ||
# define MOD(a) \ | ||
do { \ | ||
CHOP(a); \ | ||
MOD28(a); \ | ||
} while (0) | ||
# define MOD63(a) \ | ||
do { /* this assumes a is not negative */ \ | ||
z_off64_t tmp = a >> 32; \ | ||
a &= 0xffffffffL; \ | ||
a += (tmp << 8) - (tmp << 5) + tmp; \ | ||
tmp = a >> 16; \ | ||
a &= 0xffffL; \ | ||
a += (tmp << 4) - tmp; \ | ||
tmp = a >> 16; \ | ||
a &= 0xffffL; \ | ||
a += (tmp << 4) - tmp; \ | ||
if (a >= BASE) a -= BASE; \ | ||
} while (0) | ||
#else | ||
# define MOD(a) a %= BASE | ||
# define MOD28(a) a %= BASE | ||
# define MOD63(a) a %= BASE | ||
#endif | ||
|
||
/* ========================================================================= */ | ||
uLong ZEXPORT adler32(adler, buf, len) | ||
uLong adler; | ||
const Bytef *buf; | ||
uInt len; | ||
{ | ||
unsigned long sum2; | ||
unsigned n; | ||
|
||
/* split Adler-32 into component sums */ | ||
sum2 = (adler >> 16) & 0xffff; | ||
adler &= 0xffff; | ||
|
||
/* in case user likes doing a byte at a time, keep it fast */ | ||
if (len == 1) { | ||
adler += buf[0]; | ||
if (adler >= BASE) | ||
adler -= BASE; | ||
sum2 += adler; | ||
if (sum2 >= BASE) | ||
sum2 -= BASE; | ||
return adler | (sum2 << 16); | ||
} | ||
|
||
/* initial Adler-32 value (deferred check for len == 1 speed) */ | ||
if (buf == Z_NULL) | ||
return 1L; | ||
|
||
/* in case short lengths are provided, keep it somewhat fast */ | ||
if (len < 16) { | ||
while (len--) { | ||
adler += *buf++; | ||
sum2 += adler; | ||
} | ||
if (adler >= BASE) | ||
adler -= BASE; | ||
MOD28(sum2); /* only added so many BASE's */ | ||
return adler | (sum2 << 16); | ||
} | ||
|
||
/* do length NMAX blocks -- requires just one modulo operation */ | ||
while (len >= NMAX) { | ||
len -= NMAX; | ||
n = NMAX / 16; /* NMAX is divisible by 16 */ | ||
do { | ||
DO16(buf); /* 16 sums unrolled */ | ||
buf += 16; | ||
} while (--n); | ||
MOD(adler); | ||
MOD(sum2); | ||
} | ||
|
||
/* do remaining bytes (less than NMAX, still just one modulo) */ | ||
if (len) { /* avoid modulos if none remaining */ | ||
while (len >= 16) { | ||
len -= 16; | ||
DO16(buf); | ||
buf += 16; | ||
} | ||
while (len--) { | ||
adler += *buf++; | ||
sum2 += adler; | ||
} | ||
MOD(adler); | ||
MOD(sum2); | ||
} | ||
|
||
/* return recombined sums */ | ||
return adler | (sum2 << 16); | ||
} | ||
|
||
/* ========================================================================= */ | ||
local uLong adler32_combine_(adler1, adler2, len2) | ||
uLong adler1; | ||
uLong adler2; | ||
z_off64_t len2; | ||
{ | ||
unsigned long sum1; | ||
unsigned long sum2; | ||
unsigned rem; | ||
|
||
/* for negative len, return invalid adler32 as a clue for debugging */ | ||
if (len2 < 0) | ||
return 0xffffffffUL; | ||
|
||
/* the derivation of this formula is left as an exercise for the reader */ | ||
MOD63(len2); /* assumes len2 >= 0 */ | ||
rem = (unsigned)len2; | ||
sum1 = adler1 & 0xffff; | ||
sum2 = rem * sum1; | ||
MOD(sum2); | ||
sum1 += (adler2 & 0xffff) + BASE - 1; | ||
sum2 += ((adler1 >> 16) & 0xffff) + ((adler2 >> 16) & 0xffff) + BASE - rem; | ||
if (sum1 >= BASE) sum1 -= BASE; | ||
if (sum1 >= BASE) sum1 -= BASE; | ||
if (sum2 >= (BASE << 1)) sum2 -= (BASE << 1); | ||
if (sum2 >= BASE) sum2 -= BASE; | ||
return sum1 | (sum2 << 16); | ||
} | ||
|
||
/* ========================================================================= */ | ||
uLong ZEXPORT adler32_combine(adler1, adler2, len2) | ||
uLong adler1; | ||
uLong adler2; | ||
z_off_t len2; | ||
{ | ||
return adler32_combine_(adler1, adler2, len2); | ||
} | ||
|
||
uLong ZEXPORT adler32_combine64(adler1, adler2, len2) | ||
uLong adler1; | ||
uLong adler2; | ||
z_off64_t len2; | ||
{ | ||
return adler32_combine_(adler1, adler2, len2); | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,80 @@ | ||
/* compress.c -- compress a memory buffer | ||
* Copyright (C) 1995-2005 Jean-loup Gailly. | ||
* For conditions of distribution and use, see copyright notice in zlib.h | ||
*/ | ||
|
||
/* @(#) $Id$ */ | ||
|
||
#define ZLIB_INTERNAL | ||
#include "zlib.h" | ||
|
||
/* =========================================================================== | ||
Compresses the source buffer into the destination buffer. The level | ||
parameter has the same meaning as in deflateInit. sourceLen is the byte | ||
length of the source buffer. Upon entry, destLen is the total size of the | ||
destination buffer, which must be at least 0.1% larger than sourceLen plus | ||
12 bytes. Upon exit, destLen is the actual size of the compressed buffer. | ||
compress2 returns Z_OK if success, Z_MEM_ERROR if there was not enough | ||
memory, Z_BUF_ERROR if there was not enough room in the output buffer, | ||
Z_STREAM_ERROR if the level parameter is invalid. | ||
*/ | ||
int ZEXPORT compress2 (dest, destLen, source, sourceLen, level) | ||
Bytef *dest; | ||
uLongf *destLen; | ||
const Bytef *source; | ||
uLong sourceLen; | ||
int level; | ||
{ | ||
z_stream stream; | ||
int err; | ||
|
||
stream.next_in = (z_const Bytef *)source; | ||
stream.avail_in = (uInt)sourceLen; | ||
#ifdef MAXSEG_64K | ||
/* Check for source > 64K on 16-bit machine: */ | ||
if ((uLong)stream.avail_in != sourceLen) return Z_BUF_ERROR; | ||
#endif | ||
stream.next_out = dest; | ||
stream.avail_out = (uInt)*destLen; | ||
if ((uLong)stream.avail_out != *destLen) return Z_BUF_ERROR; | ||
|
||
stream.zalloc = (alloc_func)0; | ||
stream.zfree = (free_func)0; | ||
stream.opaque = (voidpf)0; | ||
|
||
err = deflateInit(&stream, level); | ||
if (err != Z_OK) return err; | ||
|
||
err = deflate(&stream, Z_FINISH); | ||
if (err != Z_STREAM_END) { | ||
deflateEnd(&stream); | ||
return err == Z_OK ? Z_BUF_ERROR : err; | ||
} | ||
*destLen = stream.total_out; | ||
|
||
err = deflateEnd(&stream); | ||
return err; | ||
} | ||
|
||
/* =========================================================================== | ||
*/ | ||
int ZEXPORT compress (dest, destLen, source, sourceLen) | ||
Bytef *dest; | ||
uLongf *destLen; | ||
const Bytef *source; | ||
uLong sourceLen; | ||
{ | ||
return compress2(dest, destLen, source, sourceLen, Z_DEFAULT_COMPRESSION); | ||
} | ||
|
||
/* =========================================================================== | ||
If the default memLevel or windowBits for deflateInit() is changed, then | ||
this function needs to be updated. | ||
*/ | ||
uLong ZEXPORT compressBound (sourceLen) | ||
uLong sourceLen; | ||
{ | ||
return sourceLen + (sourceLen >> 12) + (sourceLen >> 14) + | ||
(sourceLen >> 25) + 13; | ||
} |
Oops, something went wrong.