Utils.ModelBuilder

Does anyone use these functions in Utils.ModelBuilder?

defineTwoLayers
defineBlock
layeredModel
addBlock

If not, can we remove them, or refactor them?

[jcapriot]

We’re thinking of moving this into the discretize repository as they are mesh independent and could be more useful for a more general approach.

[rowanc1]

if __name__ == '__main__':

    from discretize import TensorMesh
    from matplotlib import pyplot as plt

    # Define the mesh

    testDim = 2
    h1 = 0.3*np.ones(7)
    h1[0] = 0.5
    h1[-1] = 0.6
    h2 = .5 * np.ones(4)
    h3 = .4 * np.ones(6)
    x0 = np.zeros(3)

    if testDim == 1:
        h = [h1]
        x0 = x0[0]
    elif testDim == 2:
        h = [h1, h2]
        x0 = x0[0:2]
    else:
        h = [h1, h2, h3]

    M = TensorMesh(h, x0)

    ccMesh = M.gridCC

    # ------------------- Test conductivities! --------------------------
    print('Testing 1 block conductivity')

    p0 = np.array([0.5,0.5,0.5])[:testDim]
    p1 = np.array([1.0,1.0,1.0])[:testDim]
    vals = np.array([100,1e-6])

    sigma = defineBlockConductivity(ccMesh,p0,p1,vals)

    # Plot sigma model
    print(sigma.shape)
    M.plotImage(sigma)
    print('Done with block! :)')
    plt.show()

    # -----------------------------------------
    print('Testing the two layered model')
    vals = np.array([100,1e-5]);
    depth    = 1.0;

    sigma = defineTwoLayeredConductivity(ccMesh,depth,vals)

    M.plotImage(sigma)
    print(sigma)
    print('layer model!')
    plt.show()

    # -----------------------------------------
    print('Testing scalar conductivity')

    if testDim == 1:
        pFunction = lambda x: np.exp(x)
    elif testDim == 2:
        pFunction = lambda x,y: np.exp(x+y)
    elif testDim == 3:
        pFunction = lambda x,y,z: np.exp(x+y+z)

    sigma = scalarConductivity(ccMesh,pFunction)

    # Plot sigma model
    M.plotImage(sigma)
    print(sigma)
    print('Scalar conductivity defined!')
    plt.show()

    # -----------------------------------------

I have removed this from the bottom of the ModelBuilder code. It is not being run.