Wheat respond to irrigation application even under shallow water table conditions but regular rains at reproductive stage had very little depletion in soil moisture content.
Considerable work in California has been done with cotton and alfalfa. D.W. Grimes and Mark Grismer have published work defining shallow water table contributions.
we have developed a soil water balance and irrigation scheduling model, SIMDualKc, which is able to consider capilary rise from a shallow water table. we have applied it to maize, cotton, and grassland (see my publication list). the model is available for download at https://www.isa.ulisboa.pt/en/leaf/downloads.
Try DSSAT model. It can predict yields under irrigated and rainfed conditions. Additionally it can simulate soil water balance at any stage of the crop growth.